Skip to content

Commit bdef7f9

Browse files
committed
Test cloud and IP records
1 parent 6a5731e commit bdef7f9

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

caso/tests/test_ssm.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,3 +67,24 @@ def mock_push(entries_cloud, entries_ip, entries_accelerator, entries_storage):
6767

6868
m.setattr(messenger, "_push", mock_push)
6969
messenger.push(ip_record_list)
70+
71+
72+
def test_cloud_ip_records_pushed(
73+
monkeypatch,
74+
cloud_record_list,
75+
expected_entries_cloud,
76+
ip_record_list,
77+
expected_entries_ip,
78+
):
79+
"""Test that cloud and IP records are correctly rendered."""
80+
81+
def mock_push(entries_cloud, entries_ip, entries_accelerator, entries_storage):
82+
assert set(entries_cloud) == set(expected_entries_cloud)
83+
assert set(entries_ip) == set(expected_entries_ip)
84+
85+
with monkeypatch.context() as m:
86+
m.setattr("caso.utils.makedirs", lambda x: None)
87+
messenger = ssm.SSMMessenger()
88+
89+
m.setattr(messenger, "_push", mock_push)
90+
messenger.push(cloud_record_list + ip_record_list)

0 commit comments

Comments
 (0)