88 lines
3.0 KiB
Python
88 lines
3.0 KiB
Python
import json
|
|
import os
|
|
import shutil
|
|
import unittest
|
|
|
|
import pkg_resources
|
|
|
|
from cpl_core.utils import String
|
|
|
|
from unittests_cli.constants import PLAYGROUND_PATH
|
|
from unittests_shared.cli_commands import CLICommands
|
|
|
|
|
|
class InstallTestCase(unittest.TestCase):
|
|
|
|
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()
|
|
|
|
def setUp(self):
|
|
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)))
|
|
|
|
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']
|
|
)
|
|
packages = dict(tuple(str(ws).split()) for ws in pkg_resources.working_set)
|
|
self.assertIn(package_name, packages)
|
|
self.assertEqual(version, packages[package_name])
|
|
|
|
def test_install_all(self):
|
|
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']
|
|
)
|
|
|
|
|