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)