sh_cpl/unittests/unittests_cli/install_test_case.py

94 lines
3.3 KiB
Python
Raw Normal View History

2022-06-21 08:56:26 +02:00
import json
import os
import shutil
2022-06-23 23:39:41 +02:00
import subprocess
import sys
2022-05-26 13:49:31 +02:00
import unittest
2022-06-21 08:56:26 +02:00
from cpl_core.utils import String
from unittests_cli.constants import PLAYGROUND_PATH
from unittests_shared.cli_commands import CLICommands
2022-05-26 13:49:31 +02:00
class InstallTestCase(unittest.TestCase):
2022-06-21 08:56:26 +02:00
def __init__(self, methodName: str):
unittest.TestCase.__init__(self, methodName)
self._source = 'install-test-source'
self._project_file = f'src/{String.convert_to_snake_case(self._source)}/{self._source}.json'
def _get_project_settings(self):
with open(os.path.join(os.getcwd(), self._project_file), 'r', encoding='utf-8') as cfg:
# load json
project_json = json.load(cfg)
cfg.close()
return project_json
def _save_project_settings(self, settings: dict):
with open(os.path.join(os.getcwd(), self._project_file), 'w', encoding='utf-8') as project_file:
project_file.write(json.dumps(settings, indent=2))
project_file.close()
2022-05-26 13:49:31 +02:00
def setUp(self):
2022-06-21 08:56:26 +02:00
os.chdir(os.path.abspath(PLAYGROUND_PATH))
# create projects
CLICommands.new('console', self._source, '--ab', '--s')
os.chdir(os.path.join(os.getcwd(), self._source))
def cleanUp(self):
# remove projects
if not os.path.exists(os.path.abspath(os.path.join(PLAYGROUND_PATH, self._source))):
return
shutil.rmtree(os.path.abspath(os.path.join(PLAYGROUND_PATH, self._source)))
2022-06-23 23:39:41 +02:00
def _get_installed_packages(self) -> dict:
reqs = subprocess.check_output([sys.executable, '-m', 'pip', 'freeze'])
return dict([tuple(r.decode().split('==')) for r in reqs.split()])
2022-06-21 08:56:26 +02:00
def test_install_package(self):
version = '1.7.3'
package_name = 'discord.py'
package = f'{package_name}=={version}'
CLICommands.install(package)
settings = self._get_project_settings()
self.assertNotEqual(settings, {})
self.assertIn('ProjectSettings', settings)
self.assertIn('Dependencies', settings['ProjectSettings'])
self.assertIn(
package,
settings['ProjectSettings']['Dependencies']
)
2022-06-23 23:39:41 +02:00
packages = self._get_installed_packages()
2022-06-21 08:56:26 +02:00
self.assertIn(package_name, packages)
self.assertEqual(version, packages[package_name])
2022-06-23 23:39:41 +02:00
def _test_install_all(self):
2022-06-21 08:56:26 +02:00
version = '1.7.3'
package_name = 'discord.py'
package = f'{package_name}=={version}'
settings = self._get_project_settings()
self.assertIn('ProjectSettings', settings)
self.assertIn('Dependencies', settings['ProjectSettings'])
self.assertNotIn(
package,
settings['ProjectSettings']['Dependencies']
)
settings['ProjectSettings']['Dependencies'].append(package)
self._save_project_settings(settings)
CLICommands.install()
new_settings = self._get_project_settings()
self.assertEqual(settings, new_settings)
self.assertIn('ProjectSettings', new_settings)
self.assertIn('Dependencies', new_settings['ProjectSettings'])
self.assertIn(
package,
new_settings['ProjectSettings']['Dependencies']
)
2022-06-23 23:39:41 +02:00
packages = self._get_installed_packages()
self.assertIn(package_name, packages)
self.assertEqual(version, packages[package_name])
2022-06-21 08:56:26 +02:00
2022-05-26 13:49:31 +02:00