michael/Spoofy
1
0
Fork 0
mirror of https://github.com/MattKeeley/Spoofy.git synced 2026-02-03 05:23:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

update test cases

Browse Source
This commit is contained in:
Matt Keeley
2024-08-12 11:35:37 -07:00
parent a63504f5c9
commit ce128d5e89
1 changed files with 111 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

134
test.py
View File

@@ -1,55 +1,143 @@
import unittest
from modules.spoofing import Spoofing
class TestSpoofy(unittest.TestCase):
'''
0: Indicates that spoofing is possible for the domain.
1: Indicates that subdomain spoofing is possible for the domain.
2: Indicates that organizational domain spoofing is possible for the domain.
3: Indicates that spoofing might be possible for the domain.
4: Indicates that spoofing might be possible (mailbox dependent) for the domain.
5: Indicates that organizational domain spoofing may be possible for the domain.
6: Indicates that subdomain spoofing might be possible (mailbox dependent) for the domain.
7: Indicates that subdomain spoofing is possible, and organizational domain spoofing might be possible.
8: Indicates that spoofing is not possible for the domain.
'''
"""
0: Indicates that spoofing is possible for the domain.
1: Indicates that subdomain spoofing is possible for the domain.
2: Indicates that organizational domain spoofing is possible for the domain.
3: Indicates that spoofing might be possible for the domain.
4: Indicates that spoofing might be possible (mailbox dependent) for the domain.
5: Indicates that organizational domain spoofing may be possible for the domain.
6: Indicates that subdomain spoofing might be possible (mailbox dependent) for the domain.
7: Indicates that subdomain spoofing is possible, and organizational domain spoofing might be possible.
8: Indicates that spoofing is not possible for the domain.
"""
"""
--- Format ---
domain: str,
p: str,
aspf: str,
spf_record: str,
spf_all: str,
spf_dns_queries: int,
sp: None,
pct: int
"""
def test_spoofing_is_possible(self):
spoofing = Spoofing('test_0.com', 'none', 'r', 'v=spf1 include:fake.gov', '~all', 3, None, 100)
spoofing = Spoofing(
"test_0.com", "none", "r", "v=spf1 include:fake.gov", "~all", 3, None, 100
)
self.assertEqual(spoofing.spoofable, 0)
def test_subdomain_spoofing(self):
spoofing = Spoofing("test_1.com", 'reject', None, 'v=spf1 include:fakest.domain.com', '-all', 3, 'none', None)
spoofing = Spoofing(
"test_1.com",
"reject",
None,
"v=spf1 include:fakest.domain.com",
"-all",
3,
"none",
None,
)
self.assertEqual(spoofing.spoofable, 1)
def test_organizational_domain_spoofing(self):
spoofing = Spoofing('test_2.com', 'none', 'r', 'v=spf1 include:fakest.domain.com include:faker.domain.com', '-all', 2, 'reject', 100)
spoofing = Spoofing(
"test_2.com",
"none",
"r",
"v=spf1 include:fakest.domain.com include:faker.domain.com",
"-all",
2,
"reject",
100,
)
self.assertEqual(spoofing.spoofable, 2)
def test_spoofing_might_be_possible(self):
spoofing = Spoofing('test_3.com', 'none', None, 'v=spf1 include:fakest.domain.com', '~all', 1, 'quarantine', 90)
spoofing = Spoofing(
"test_3.com",
"none",
None,
"v=spf1 include:fakest.domain.com",
"~all",
1,
"quarantine",
90,
)
self.assertEqual(spoofing.spoofable, 3)
def test_spoofing_might_be_possible_mbd(self):
spoofing = Spoofing('test_4.com', 'none', None, 'v=spf1 include:fakest.domain.com', '-all', 1, None, 100)
spoofing = Spoofing(
"test_4.com",
"none",
None,
"v=spf1 include:fakest.domain.com",
"-all",
1,
None,
100,
)
self.assertEqual(spoofing.spoofable, 4)
def test_org_domain_spoofing_might_be_possible(self):
spoofing = Spoofing('test_5.com', 'none', None, 'v=spf1 include:fakest.domain.com', '-all', 1, 'reject', 100)
spoofing = Spoofing(
"test_5.com",
"none",
None,
"v=spf1 include:fakest.domain.com",
"-all",
1,
"reject",
100,
)
self.assertEqual(spoofing.spoofable, 5)
def test_subdomain_spoofing_might_be_possible_mbd(self):
spoofing = Spoofing('test_6.com', 'reject', 'r', 'v=spf1 include:fakest.domain.com', '?all', 1, 'none', 100)
spoofing = Spoofing(
"test_6.com",
"reject",
"r",
"v=spf1 include:fakest.domain.com",
"?all",
1,
"none",
100,
)
self.assertEqual(spoofing.spoofable, 6)
def test_subdomain_spoofing_and_org_spoofing_might_be_possible(self):
spoofing = Spoofing('test_7.com', 'none', None, 'v=spf1 include:fakest.domain.com', '~all', 3, 'none', 100)
spoofing = Spoofing(
"test_7.com",
"none",
None,
"v=spf1 include:fakest.domain.com",
"~all",
3,
"none",
100,
)
self.assertEqual(spoofing.spoofable, 7)
def test_spoofing_not_possible(self):
spoofing = Spoofing('test_8.com', 'none', 's', 'v=spf1 include:fakest.domain.com', '~all', 1, 'quarantine', 100)
spoofing = Spoofing(
"test_8.com",
"none",
"s",
"v=spf1 include:domain.com",
"-all",
1,
"reject",
100,
)
self.assertEqual(spoofing.spoofable, 8)
if __name__ == '__main__':
if __name__ == "__main__":
unittest.main()