def setUp(self):
self.docker = Docker()
self.deployment = Deployment(1, DATA, self.docker, WORKER_OPTIONS)
class DeploymentTests(unittest.TestCase):
def setUp(self):
self.docker = Docker()
self.deployment = Deployment(1, DATA, self.docker, WORKER_OPTIONS)
@mock.patch('docker.manager.Docker.start')
@mock.patch('docker.manager.Docker.stop')
@mock.patch('frigg_worker.deployments.Deployment.clone_repo')
@mock.patch('frigg_worker.deployments.Deployment.run_task')
@mock.patch('frigg_worker.deployments.Deployment.report_run', lambda *x: None)
@mock.patch('frigg_worker.jobs.build_settings', lambda *x: BUILD_SETTINGS_WITH_NO_SERVICES)
def test_run_deploy(self, mock_run_task, mock_clone_repo, mock_docker_stop, mock_docker_start):
self.deployment.run_deploy()
mock_run_task.assert_has_calls([
mock.call('pip install -r requirements.txt'),
mock.call('gunicorn')
])
self.assertTrue(mock_clone_repo.called)
self.assertTrue(self.deployment.succeeded)
self.assertTrue(self.deployment.finished)
@mock.patch('frigg_worker.deployments.Deployment.clone_repo')
@mock.patch('frigg_worker.deployments.Deployment.run_task', side_effect=OSError())
@mock.patch('frigg_worker.deployments.Deployment.report_run', lambda *x: None)
@mock.patch('frigg_worker.jobs.build_settings', lambda *x: BUILD_SETTINGS_WITH_NO_SERVICES)
def test_run_deploy_fail_task(self, mock_run_task, mock_clone_repo):
self.deployment.run_deploy()
self.assertTrue(mock_clone_repo.called)
mock_run_task.assert_has_calls([
mock.call('pip install -r requirements.txt'),
])
self.assertFalse(self.deployment.succeeded)
self.assertTrue(self.deployment.finished)
@mock.patch('frigg_worker.deployments.Deployment.run_task')
@mock.patch('frigg_worker.deployments.Deployment.clone_repo', side_effect=GIT_ERROR)
def test_run_deploy_fail_clone(self, mock_clone, mock_run_task):
self.deployment.run_deploy()
self.assertFalse(mock_run_task.called)
self.assertFalse(self.deployment.succeeded)
@mock.patch('frigg_worker.jobs.build_settings', lambda *x: BUILD_SETTINGS_WITH_PRESET)
@mock.patch('frigg_worker.deployments.Deployment.clone_repo', lambda x: True)
@mock.patch('frigg_worker.deployments.Deployment.report_run', lambda *x: None)
@mock.patch('frigg_worker.deployments.Deployment.load_preset')
@mock.patch('frigg_worker.deployments.Deployment.run_task')
def test_run_deploy_should_load_preset(self, mock_run_task, mock_load_preset):
self.deployment.run_deploy()
self.assertTrue(mock_load_preset.called)
@mock.patch('frigg_worker.api.APIWrapper.report_run')
@mock.patch('frigg_worker.deployments.Deployment.serializer', lambda *x: {})
@mock.patch('frigg_worker.jobs.build_settings', lambda *x: {})
def test_report_run(self, mock_report_run):
self.deployment.report_run()
mock_report_run.assert_called_once_with('Deployment', 1, '{}')
@mock.patch('frigg_worker.jobs.build_settings', lambda *x: BUILD_SETTINGS_WITH_NO_SERVICES)
def test_create_pending_tasks_splitted_into_setup_tasks_and_tasks(self):
self.assertEqual([], self.deployment.tasks)
self.assertEqual([], self.deployment.setup_tasks)
self.deployment.create_pending_tasks()
self.assertEqual(['pip install -r requirements.txt', 'gunicorn'], self.deployment.tasks)
@mock.patch('frigg_worker.jobs.build_settings', lambda *x: BUILD_SETTINGS_WITH_PRESET)
def test_load_preset(self):
self.deployment.load_preset()
self.assertIn('daemon_task', self.deployment.preset)
self.assertIn('tasks', self.deployment.preset)
self.assertEqual(
self.deployment.preset['daemon_task'],
'nohup python3 manage.py runserver 0.0.0.0:8000 &'
)
self.assertEqual(
self.deployment.preset['tasks'],
['pip3 install -U gunicorn -r requirements.txt',
'python3 manage.py migrate',
'python3 manage.py collectstatic --noinput']
)
