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)