From 1e5d4a531a24801498f491bbc209ccff8d574e72 Mon Sep 17 00:00:00 2001
From: eman-ekaf <97634535+eman-ekaf@users.noreply.github.com>
Date: Tue, 21 Mar 2023 23:25:30 -0500
Subject: [PATCH] added new tests

added new tests to test.py
---
 test.py | 42 +++++++++++++++++++++++++++++++++++++++---
 1 file changed, 39 insertions(+), 3 deletions(-)

diff --git a/test.py b/test.py
index 4d506dc..a81eeeb 100644
--- a/test.py
+++ b/test.py
@@ -2,9 +2,45 @@ import unittest
 from libs import logic
 
 class TestSpoofy(unittest.TestCase):
-    def test_if_spoofable(self):
-        unittest.TestCase().assertEqual(logic.is_spoofable("fireeye.com", "reject", None, "v=spf1 include:%{i}._ip.%{h}._ehlo.%{d}._spf.vali.email include:mktomail.com include:stspg-customer.com -all", "-all", 3, "none", None),1)
-        unittest.TestCase().assertEqual(logic.is_spoofable('google.com', 'reject', None, 'v=spf1 include:_spf.google.com ~all', '~all', 4, None, None), 7)
+
+    '''
+      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.
+    '''
+
+    def test_spoofing_is_possible(self):
+        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)
+ 
+    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)
+    
+    def test_spoofing_might_be_possible(self):
+        unittest.TestCase().assertEqual(logic.is_spoofable('test_3.com', 'none', None, 'v=spf1 include:fakest.domain.com', '~all', 1, 'quarantine', 90), 3)
+
+    def test_spoofing_might_be_possible_mbd(self):
+        unittest.TestCase().assertEqual(logic.is_spoofable('test_4.com', 'none', None, 'v=spf1 include:fakest.domain.com', '-all', 1, None, 100), 4)
+
+    def test_org_domain_spoofing_might_be_possible(self):
+        unittest.TestCase().assertEqual(logic.is_spoofable('test_5.com', 'none', None, 'v=spf1 include:fakest.domain.com', '-all', 1, 'reject', 100), 5)
+
+    def test_subdomain_spoofing_might_be_possible_mbd(self):
+        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)
+
+    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)
 
 if __name__ == '__main__':
     unittest.main()
\ No newline at end of file