def test_commit_list(config, files):
tempdir = config.base_path
base = tempdir / 'security_groups'
client = AwsClient(config)
group = client.get('mock-group')
group2 = client.get('mock-group2')
group.revoke(ip_protocol="tcp", from_port=22, to_port=22,
cidr_ip="192.168.1.0/32")
group.revoke(ip_protocol="tcp", from_port=22, to_port=22,
src_group=group2)
gr0 = Grant(ip_protocol="tcp", from_port="22", to_port="22",
cidr_ip="192.168.10.0/32")
gr1 = Grant(ip_protocol="tcp", from_port="33", to_port="44",
group="mock-group2")
gr2 = Grant(ip_protocol="tcp", from_port="22", to_port="22",
cidr_ip="192.168.20.0/32")
FileService.write_csv(base / "mock-group.csv",
[
gr0,
gr1,
])
FileService.write_csv(base / "mock-group2.csv",
[
gr2,
])
diff_list = dict(SgService.diff_list(config, AwsClient(config), []))
diff0 = diff_list['mock-group']
assert diff0.local_only == {gr0.rule, gr1.rule}
assert diff_list['mock-group2'].local_only == {gr2.rule}
SgService.commit_list(config=config, client=AwsClient(config),
file_path_list=[],
noconfirm=True)
diff_list = dict(SgService.diff_list(config, AwsClient(config), []))
assert "mock-group" not in diff_list
assert "mock-group2" not in diff_list
def test_commit_list(config, files):
tempdir = config.base_path
base = tempdir / 'security_groups'
client = AwsClient(config)
group = client.get('mock-group')
group2 = client.get('mock-group2')
group.revoke(ip_protocol="tcp",
from_port=22,
to_port=22,
cidr_ip="192.168.1.0/32")
group.revoke(ip_protocol="tcp", from_port=22, to_port=22, src_group=group2)
gr0 = Grant(ip_protocol="tcp",
from_port="22",
to_port="22",
cidr_ip="192.168.10.0/32")
gr1 = Grant(ip_protocol="tcp",
from_port="33",
to_port="44",
group="mock-group2")
gr2 = Grant(ip_protocol="tcp",
from_port="22",
to_port="22",
cidr_ip="192.168.20.0/32")
FileService.write_csv(base / "mock-group.csv", [
gr0,
gr1,
])
FileService.write_csv(base / "mock-group2.csv", [
gr2,
])
diff_list = dict(SgService.diff_list(config, AwsClient(config), []))
diff0 = diff_list['mock-group']
assert diff0.local_only == {gr0.rule, gr1.rule}
assert diff_list['mock-group2'].local_only == {gr2.rule}
SgService.commit_list(config=config,
client=AwsClient(config),
file_path_list=[],
noconfirm=True)
diff_list = dict(SgService.diff_list(config, AwsClient(config), []))
assert "mock-group" not in diff_list
assert "mock-group2" not in diff_list
