def test_version_helper(): """CLI - Publish Helper""" package = AthenaPackage(basic_streamalert_config()) publish = LambdaVersion(config=basic_streamalert_config(), package=package) current_version = 10 fake_client = MockLambdaClient('athena', current_version=current_version) result = publish._version_helper(client=fake_client, function_name='test', code_sha_256='12345', date='2017-01-01') assert_equal(result, current_version + 1)
def test_version_helper_error(mock_logging): """CLI - Publish Helper Raises Error""" package = AthenaPackage(basic_streamalert_config()) publish = LambdaVersion(config=basic_streamalert_config(), package=package) current_version = 10 fake_client = MockLambdaClient('athena', current_version=current_version, throw_exception=True) result = publish._version_helper(client=fake_client, function_name='test', code_sha_256='12345', date='2017-01-01') assert_false(result) assert_true(mock_logging.error.called)