def test_enrollment_set_none(self):
config = VPCTagConfig()
config.enrollment = None
config.target = None
self.assertEqual(2, len(config.get_aws_tags()))
for tag in config.get_aws_tags():
self.assertEqual(1, len(tag))
def test_dump_garbage_tags(self):
config = VPCTagConfig({
VPCTagConfig.T_TARGET: 'arn:aws:foo/bar',
'flavor': 'development',
})
self.assertTrue(config.enabled)
self.assertEqual(1, len(config.get_aws_tags()), config.get_aws_tags())
def set_vpc_config(cls, region: str, vpc_id: str,
config: VPCTagConfig) -> None:
""" enables/disables tapping the vpc"""
client = cls._get_client(region=region)
if config.enabled:
client.create_tags(Tags=config.get_aws_tags(), Resources=[vpc_id])
else:
client.delete_tags(Tags=config.get_aws_tags(), Resources=[vpc_id])
def test_enrollment_set(self):
config = VPCTagConfig()
config.enrollment = config.V_ENROLLMENT_WHITELIST
self.assertEqual(config.V_ENROLLMENT_WHITELIST, config.enrollment)
config.target = 'arn:aws:foo/bar'
self.assertEqual(2, len(config.get_aws_tags()))
for tag in config.get_aws_tags():
self.assertEqual(2, len(tag))
def test_get_aws_tags(self):
config = VPCTagConfig()
config.target = 'arn:foo/bar'
self.assertEqual([{
'Key': 'Vectra:session_mirroring_target',
'Value': 'arn:foo/bar'
}], config.get_aws_tags())
def test_empty_config(self):
config = VPCTagConfig()
self.assertEqual([], config.get_aws_tags())