def get(name):
data = json.load(open(SECRET_FILE_NAME, 'r'))
if 'cipher_texts' not in data:
return None
if name not in data['cipher_texts']:
return None
client = Client(region=data.get('region'))
return client.decrypt(data['cipher_texts'].get(name))
def get(name):
data = json.load(open(SECRET_FILE_NAME, "r"))
if "cipher_texts" not in data:
return None
if name not in data["cipher_texts"]:
return None
client = Client(region=data.get("region"))
return client.decrypt(data["cipher_texts"].get(name))
def test_put_rule(self):
rule = {
'rule': 'foo',
'description': 'bar',
'pattern': 'baz',
'schedule': 'qux'}
ok_(self.client.put_rule(rule) != {})
client = Client(region='us-east-1', profile=None, dry_run=True)
eq_(client.put_rule({'rule': 'foo'}), {})
def test_put_rule(self):
rule = {
'rule': 'foo',
'description': 'bar',
'pattern': 'baz',
'schedule': 'qux'
}
ok_(self.client.put_rule(rule) != {})
client = Client(region='us-east-1', profile=None, dry_run=True)
eq_(client.put_rule({'rule': 'foo'}), {})
def get_client(self):
return Client(region=self._config.get_region(),
profile=self._config.get_profile(),
dry_run=self._dry_run)
def setUp(self):
self.client = Client(region='us-east-1')
self.client._lambda = Mock()
self.client._kms = Mock()
self.client._events = Mock()
class ClientTestCase(TestCase):
def setUp(self):
self.client = Client(region='us-east-1')
self.client._lambda = Mock()
self.client._kms = Mock()
self.client._events = Mock()
def test_get_function_conf(self):
self.client._lambda.get_function = Mock(
return_value={'Configuration': 'foo'})
eq_(self.client.get_function_conf('test'), 'foo')
self.client._lambda.get_function = Mock(
side_effect=botocore.exceptions.ClientError({'Error': {}}, 'bar'))
eq_(self.client.get_function_conf('test'), {})
def test_create_function(self):
self.client.create_function(Mock(), TEST_CONF, True)
def test_update_function_code(self):
self.client._lambda.update_function_code = Mock(return_value={'Version': '$LATEST'})
eq_(self.client.update_function_code(Mock(), TEST_CONF, True), '$LATEST')
self.client._dry_run = True
eq_(self.client.update_function_code(Mock(), TEST_CONF, True), None)
def test_update_function_conf(self):
self.client.update_function_conf(TEST_CONF)
def test_get_alias(self):
self.client._lambda.get_alias = Mock(return_value='foo')
eq_(self.client.get_alias('function', 'alias'), 'foo')
self.client._lambda.get_alias = Mock(
side_effect=botocore.exceptions.ClientError({'Error': {}}, 'bar'))
eq_(self.client.get_alias('function', 'alias'), {})
def test_create_alias(self):
self.client.create_alias('function', 'alias', 'version')
def test_update_alias(self):
self.client.update_alias('function', 'alias', 'version')
def test_encrypt(self):
self.client._kms.encrypt = Mock(return_value={'CiphertextBlob': 'foo'})
eq_(self.client.encrypt('key', 'val'), base64.b64encode('foo'))
def test_decrypt(self):
self.client._kms.decrypt = Mock(return_value={'Plaintext': 'bar'})
eq_(self.client.decrypt(base64.b64encode('secret')), 'bar')
def test_calculate_capacity(self):
ret1 = {
'Functions': [{'FunctionName': 'foo'}, {'FunctionName': 'bar'}], 'NextMarker': 'foo'}
ret2 = {
'Functions': [{'FunctionName': 'foo'}, {'FunctionName': 'bar'}]}
self.client._lambda.list_functions = Mock(side_effect=[ret1, ret2])
self.client._calculate_versions_capacity = Mock(return_value=10)
eq_(self.client.calculate_capacity(), 40)
def test_calculate_versions_capacity(self):
ret1 = {'Versions': [{'CodeSize': 20}, {'CodeSize': 20}], 'NextMarker': 'foo'}
ret2 = {'Versions': [{'CodeSize': 20}, {'CodeSize': 20}]}
self.client._lambda.list_versions_by_function = Mock(
side_effect=[ret1, ret2])
eq_(self.client._calculate_versions_capacity('foo'), 80)
def test_get_rules_by_target(self):
self.client._get_rule_names_by_tagert = Mock(return_value=['foo', 'bar'])
self.client._events.describe_rule = Mock(return_value={'foo': 'bar'})
eq_(self.client.get_rules_by_target('foo'), [{'foo': 'bar'}, {'foo': 'bar'}])
def test_get_rule_names_by_tagert(self):
ret1 = {'RuleNames': ['foo', 'bar'], 'NextToken': 'foo'}
ret2 = {'RuleNames': ['baz', 'qux']}
self.client._events.list_rule_names_by_target = Mock(side_effect=[ret1, ret2])
eq_(self.client._get_rule_names_by_tagert('foo'), ['foo', 'bar', 'baz', 'qux'])
self.client._events.list_rule_names_by_target = Mock(
side_effect=botocore.exceptions.ClientError({'Error': {}}, 'bar'))
eq_(self.client._get_rule_names_by_tagert('foo', 'bar'), [])
def test_put_rule(self):
rule = {
'rule': 'foo',
'description': 'bar',
'pattern': 'baz',
'schedule': 'qux'}
ok_(self.client.put_rule(rule) != {})
client = Client(region='us-east-1', profile=None, dry_run=True)
eq_(client.put_rule({'rule': 'foo'}), {})
def test_put_targets(self):
self.client.put_targets(
'foo', [{'id': 'foo', 'input': 'bar', 'input_path': 'baz'}], 'baz')
def test_get_targets_by_rule(self):
ret1 = {'Targets': [{'foo': 'bar'}], 'NextToken': 'foo'}
ret2 = {'Targets': [{'baz': 'qux'}]}
self.client._events.list_targets_by_rule = Mock(side_effect=[ret1, ret2])
eq_(self.client.get_targets_by_rule('foo'), [{'foo': 'bar'}, {'baz': 'qux'}])
self.client._events.list_targets_by_rule = Mock(
side_effect=botocore.exceptions.ClientError({'Error': {}}, 'bar'))
eq_(self.client.get_targets_by_rule('foo', 'bar'), [])
def test_remove_targets(self):
self.client.remove_targets('foo', ['bar', 'baz'])
def test_delete_rule(self):
self.client.delete_rule('foo')
def test_add_permission(self):
self.client.add_permission('foo', 'bar', 'baz')
self.client._lambda.add_permission = Mock(
side_effect=botocore.exceptions.ClientError({'Error': {}}, 'bar'))
self.client.add_permission('foo', 'bar', 'baz')
def test_remove_permission(self):
self.client.remove_permission('foo', 'bar')
def test_generate_statement_id(self):
self.client._generate_statement_id('foo', 'bar')
def test_invoke(self):
self.client.invoke('foo', 'bar', 'baz')
def test_get_previous_version(self):
self.client.get_alias = Mock(return_value={'FunctionVersion': 'foo'})
eq_(self.client.get_previous_version('foo', 'bar'), 'foo')
def setUp(self):
self.client = Client(region='us-east-1')
self.client._lambda = Mock()
self.client._kms = Mock()
self.client._events = Mock()
class ClientTestCase(TestCase):
def setUp(self):
self.client = Client(region='us-east-1')
self.client._lambda = Mock()
self.client._kms = Mock()
self.client._events = Mock()
def test_get_function_conf(self):
self.client._lambda.get_function = Mock(
return_value={'Configuration': 'foo'})
eq_(self.client.get_function_conf('test'), 'foo')
self.client._lambda.get_function = Mock(
side_effect=botocore.exceptions.ClientError({'Error': {}}, 'bar'))
eq_(self.client.get_function_conf('test'), {})
def test_create_function(self):
self.client.create_function(Mock(), TEST_CONF, True)
def test_update_function_code(self):
self.client._lambda.update_function_code = Mock(
return_value={'Version': '$LATEST'})
eq_(self.client.update_function_code(Mock(), TEST_CONF, True),
'$LATEST')
self.client._dry_run = True
eq_(self.client.update_function_code(Mock(), TEST_CONF, True), None)
def test_update_function_conf(self):
self.client.update_function_conf(TEST_CONF)
def test_get_alias(self):
self.client._lambda.get_alias = Mock(return_value='foo')
eq_(self.client.get_alias('function', 'alias'), 'foo')
self.client._lambda.get_alias = Mock(
side_effect=botocore.exceptions.ClientError({'Error': {}}, 'bar'))
eq_(self.client.get_alias('function', 'alias'), {})
def test_create_alias(self):
self.client.create_alias('function', 'alias', 'version')
def test_update_alias(self):
self.client.update_alias('function', 'alias', 'version')
def test_encrypt(self):
self.client._kms.encrypt = Mock(return_value={'CiphertextBlob': 'foo'})
eq_(self.client.encrypt('key', 'val'), base64.b64encode('foo'))
def test_decrypt(self):
self.client._kms.decrypt = Mock(return_value={'Plaintext': 'bar'})
eq_(self.client.decrypt(base64.b64encode('secret')), 'bar')
def test_calculate_capacity(self):
ret1 = {
'Functions': [{
'FunctionName': 'foo'
}, {
'FunctionName': 'bar'
}],
'NextMarker': 'foo'
}
ret2 = {
'Functions': [{
'FunctionName': 'foo'
}, {
'FunctionName': 'bar'
}]
}
self.client._lambda.list_functions = Mock(side_effect=[ret1, ret2])
self.client._calculate_versions_capacity = Mock(return_value=10)
eq_(self.client.calculate_capacity(), 40)
def test_calculate_versions_capacity(self):
ret1 = {
'Versions': [{
'CodeSize': 20
}, {
'CodeSize': 20
}],
'NextMarker': 'foo'
}
ret2 = {'Versions': [{'CodeSize': 20}, {'CodeSize': 20}]}
self.client._lambda.list_versions_by_function = Mock(
side_effect=[ret1, ret2])
eq_(self.client._calculate_versions_capacity('foo'), 80)
def test_get_rules_by_target(self):
self.client._get_rule_names_by_tagert = Mock(
return_value=['foo', 'bar'])
self.client._events.describe_rule = Mock(return_value={'foo': 'bar'})
eq_(self.client.get_rules_by_target('foo'), [{
'foo': 'bar'
}, {
'foo': 'bar'
}])
def test_get_rule_names_by_tagert(self):
ret1 = {'RuleNames': ['foo', 'bar'], 'NextToken': 'foo'}
ret2 = {'RuleNames': ['baz', 'qux']}
self.client._events.list_rule_names_by_target = Mock(
side_effect=[ret1, ret2])
eq_(self.client._get_rule_names_by_tagert('foo'),
['foo', 'bar', 'baz', 'qux'])
self.client._events.list_rule_names_by_target = Mock(
side_effect=botocore.exceptions.ClientError({'Error': {}}, 'bar'))
eq_(self.client._get_rule_names_by_tagert('foo', 'bar'), [])
def test_put_rule(self):
rule = {
'rule': 'foo',
'description': 'bar',
'pattern': 'baz',
'schedule': 'qux'
}
ok_(self.client.put_rule(rule) != {})
client = Client(region='us-east-1', profile=None, dry_run=True)
eq_(client.put_rule({'rule': 'foo'}), {})
def test_put_targets(self):
self.client.put_targets('foo', [{
'id': 'foo',
'input': 'bar',
'input_path': 'baz'
}], 'baz')
def test_get_targets_by_rule(self):
ret1 = {'Targets': [{'foo': 'bar'}], 'NextToken': 'foo'}
ret2 = {'Targets': [{'baz': 'qux'}]}
self.client._events.list_targets_by_rule = Mock(
side_effect=[ret1, ret2])
eq_(self.client.get_targets_by_rule('foo'), [{
'foo': 'bar'
}, {
'baz': 'qux'
}])
self.client._events.list_targets_by_rule = Mock(
side_effect=botocore.exceptions.ClientError({'Error': {}}, 'bar'))
eq_(self.client.get_targets_by_rule('foo', 'bar'), [])
def test_remove_targets(self):
self.client.remove_targets('foo', ['bar', 'baz'])
def test_delete_rule(self):
self.client.delete_rule('foo')
def test_add_permission(self):
self.client.add_permission('foo', 'bar', 'baz')
self.client._lambda.add_permission = Mock(
side_effect=botocore.exceptions.ClientError({'Error': {}}, 'bar'))
self.client.add_permission('foo', 'bar', 'baz')
def test_remove_permission(self):
self.client.remove_permission('foo', 'bar')
def test_generate_statement_id(self):
self.client._generate_statement_id('foo', 'bar')
def test_invoke(self):
self.client.invoke('foo', 'bar', 'baz')
def test_get_previous_version(self):
self.client.get_alias = Mock(return_value={'FunctionVersion': 'foo'})
eq_(self.client.get_previous_version('foo', 'bar'), 'foo')