Beispiel #1
0
 def test_deploy_fail_process(self, popen_func):
     popen_mock = Mock()
     attrs = {'wait.return_value': 1}
     popen_mock.configure_mock(**attrs)
     popen_func.return_value = popen_mock
     try:
         path = os.getcwd()
         test_deploy = LambdaDeploy(path, "s3://pb-gravity", None)
         test_deploy.deploy()
     except Exception as e:
         popen_mock.wait.assert_any_call()
         self.assertTrue(popen_mock.wait.call_count == 1, "Popen.wait was called more than once before failing!")
Beispiel #2
0
    def test_deploy_success(self, archive_func, popen_func):
        popen_mock = Mock()

        popen_attrs = {'wait.return_value': 0}
        popen_mock.configure_mock(**popen_attrs)
        popen_func.return_value = popen_mock

        shutil_attrs = {'make_archive.return_value': True}
        archive_func.configure_mock(**shutil_attrs)

        try:
            path = os.getcwd()
            test_deploy = LambdaDeploy(path, "s3://my_bucket", None)
            test_deploy.deploy()

            popen_mock.wait.assert_any_call()
            self.assertTrue(popen_mock.wait.call_count == 4, "Popen.wait has the incorrect number of calls!")
            self.assertTrue(archive_func.call_count == 1, "Did not call make_archive!")

        except Exception as e:
            print(e)
            self.assertTrue(False, "Unable to deploy!")