def test_run_delete(self, delete_mock): self.test_job.model = self.BuildTaskResult("banana", "delete", "active", "ban", "ana", {}, image="image", auth="YmFuYTpuYQ==") smart_contract_model.new_from_build_task = MagicMock( return_value=self.test_job.model) contract_job.EVENT = '{"task_type": "delete"}' contract_job.main() delete_mock.assert_called()
def test_main_create(self): self.test_job.model = self.BuildTaskResult("banana", "create", "active", "ban", "ana", {}, image="image", auth="YmFuYTpuYQ==") self.test_job.model.save = MagicMock() smart_contract_model.new_from_build_task = MagicMock( return_value=self.test_job.model) contract_job.EVENT = '{"task_type": "create"}' contract_job.ContractJob.create = MagicMock() contract_job.main() contract_job.ContractJob.create.assert_called()
def test_main_pass_does_update_model(self, mock_docker, mock_login, mock_ecr, mock_secrets): contract_job.EVENT = '{"txn_type": "test", "task_type": "update", "id": "123", "start_state": "inactive", "auth": "auth", "image": "image", "cmd": "cmd", "args": "[one]", "secrets": "{}", "existing_secrets": "[]", "env": "{}", "cron": "None", "seconds": "30", "execution_order": "serial", "desired_state": "active"}' # noqa: B950 self.test_job.model = self.BuildTaskResultWithHelpers( "banana", "create", "inactive", "ban", "ana", {}, image="image", auth="YmFuYTpuYQ==") self.test_job.update_model = self.BuildTaskResultWithHelpers( "banana", "create", "active", "banana", "ana", {}, image="new/image", auth="YmFuYTpuYQ=!") self.test_job.model.update_faas_fields( update_model=self.test_job.update_model) self.test_job.update_model.update_faas_fields( update_model=self.test_job.update_model) smart_contract_dao.get_contract_by_txn_type = MagicMock( return_value=self.test_job.model) smart_contract_model.new_from_build_task = MagicMock( return_value=self.test_job.update_model) contract_job.ContractJob.create_dockerfile = MagicMock() contract_job.ContractJob.create_openfaas_secrets = MagicMock() contract_job.ContractJob.delete_contract_image = MagicMock() contract_job.ContractJob.deploy_to_openfaas = MagicMock() contract_job.ContractJob.pull_image = MagicMock() contract_job.ContractJob.schedule_contract = MagicMock() mock_docker.from_env = MagicMock() mock_docker.images.push = MagicMock() mock_docker.images.get = MagicMock() try: job = contract_job.main() self.assertEqual(job.model.image, "new/image") self.assertEqual(job.update_model.image, "new/image") self.assertEqual(job.model.cmd, "banana") self.assertEqual(job.model.status["state"], "active") job.model.save.assert_called() except Exception: self.fail("should not have thrown a generic exception!")