Skip to content

Commit

Permalink
added tests for input of no on delete
Browse files Browse the repository at this point in the history
  • Loading branch information
Michael-Hilborn committed Jul 26, 2021
1 parent 3457944 commit 9bf54a0
Show file tree
Hide file tree
Showing 2 changed files with 40 additions and 0 deletions.
16 changes: 16 additions & 0 deletions tests/test_delete_dns.py
Original file line number Diff line number Diff line change
Expand Up @@ -154,6 +154,22 @@ def mock_dns(*args, **kwargs):
successful_delete_root = "Delete DNS Records? Input 'y' or 'yes' to execute: " + Color.RED + "ERROR: No '@' DNS record found" + Color.END
successful_delete_www = Color.RED + "ERROR: No 'www' DNS record found" + Color.END
expected_out = successful_delete_root + '\n' + successful_delete_www + '\n'
with patch('sys.stdout', new = StringIO()) as fake_out:
delete.delete_dns()
assert fake_out.getvalue() == expected_out

def test_no_dns_records(monkeypatch):
sample_inputs = StringIO('n\n')
monkeypatch.setattr('sys.stdin', sample_inputs)

def mock_dns(*args, **kwargs):
return MockNeitherDNSRecordsV1()

monkeypatch.setattr(DnsRecordsV1, "new_instance", mock_dns)

delete = dns_delete()
no_delete = "Delete DNS Records? Input 'y' or 'yes' to execute: "
expected_out = no_delete
with patch('sys.stdout', new = StringIO()) as fake_out:
delete.delete_dns()
assert fake_out.getvalue() == expected_out
24 changes: 24 additions & 0 deletions tests/test_delete_edge.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,6 +83,30 @@ def mock_delete_action(*args, **kwargs):
delete = DeleteEdge(crn="testString", zone_id="testString", cis_domain="test-domain.com", apikey="testString", token="test-token")
successful_delete_action = "Delete edge function? Input 'y' or 'yes' to execute: " + Color.GREEN + "SUCCESS: Deleted edge function action with id testID" + Color.END
expected_out = successful_delete_action + '\n'
with patch('sys.stdout', new = StringIO()) as fake_out:
delete.delete_edge()
assert fake_out.getvalue() == expected_out

def test_no_delete_action(monkeypatch):
sample_inputs = StringIO('n\n')
monkeypatch.setattr('sys.stdin', sample_inputs)

def mock_get_trigg(*args, **kwargs):
return MockGetTriggers().list_triggers()

def mock_delete_trigg(*args, **kwargs):
return MockDeleteTriggers()

def mock_delete_action(*args, **kwargs):
return MockDeleteActionSuccess().response()

monkeypatch.setattr(DeleteEdge, "get_triggers", mock_get_trigg)
monkeypatch.setattr(DeleteEdge, "delete_trigger", mock_delete_trigg)
monkeypatch.setattr(DeleteEdge, "delete_action", mock_delete_action)

delete = DeleteEdge(crn="testString", zone_id="testString", cis_domain="test-domain.com", apikey="testString", token="test-token")
no_delete = "Delete edge function? Input 'y' or 'yes' to execute: "
expected_out = no_delete
with patch('sys.stdout', new = StringIO()) as fake_out:
delete.delete_edge()
assert fake_out.getvalue() == expected_out

0 comments on commit 9bf54a0

Please sign in to comment.