def test_publish(): """CLI - Publish""" 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._publish( client=fake_client, function_name='test', code_sha_256='12345' ) assert_equal(result, current_version + 1)
def test_publish_error(mock_logging): """CLI - Publish, 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._publish( client=fake_client, function_name='test', code_sha_256='12345' ) assert_false(result) assert_true(mock_logging.error.called)