update test.py

Added additional test, made syntax consistent.
This commit is contained in:
eman-ekaf
2023-04-05 12:23:14 -05:00
committed by GitHub
parent b44f888766
commit 2703a73714

View File

@@ -19,7 +19,7 @@ class TestSpoofy(unittest.TestCase):
unittest.TestCase().assertEqual(logic.is_spoofable('test_0.com', 'none', 'r', 'v=spf1 include:fake.gov', '~all', 3, None, 100), 0)
def test_subdomain_spoofing(self):
unittest.TestCase().assertEqual(logic.is_spoofable("test_1.com", "reject", None, "v=spf1 include:fakest.domain.com", "-all", 3, "none", None), 1)
unittest.TestCase().assertEqual(logic.is_spoofable("test_1.com", 'reject', None, 'v=spf1 include:fakest.domain.com', '-all', 3, 'none', None), 1)
def test_organizational_domain_spoofing(self):
unittest.TestCase().assertEqual(logic.is_spoofable('test_2.com', 'none', 'r', 'v=spf1 include:fakest.domain.com include:faker.domain.com', '-all', 2, 'reject', 100), 2)
@@ -37,10 +37,13 @@ class TestSpoofy(unittest.TestCase):
unittest.TestCase().assertEqual(logic.is_spoofable('test_6.com', 'reject', 'r', 'v=spf1 include:fakest.domain.com', '?all', 1, 'none', 100), 6)
def test_subdomain_spoofing_and_org_spoofing_might_be_possible(self):
unittest.TestCase().assertEqual(logic.is_spoofable('test_7.com', 'none', 'none', 'v=spf1 include:fakest.domain.com', '~all', 3, 'none', 100), 7)
unittest.TestCase().assertEqual(logic.is_spoofable('test_7.com', 'none', none, 'v=spf1 include:fakest.domain.com', '~all', 3, 'none', 100), 7)
def test_spoofing_not_possible(self):
unittest.TestCase().assertEqual(logic.is_spoofable('test_8.com', 'none', 's', 'v=spf1 include:fakest.domain.com', '~all', 1, 'quarantine', 100), 8)
def test_possible_bug_fix1(self):
unittest.TestCase().assertEqual(logic.is_spoofable('sub.test_9.com', 'none', None, None, None, None, None, None), 0)
if __name__ == '__main__':
unittest.main()
unittest.main()