tests: add test for cancel, test for rollback to edit from cancel

JacobCoffee · JacobCoffee · commit 2c521c47fec0 · 2025-06-26T11:42:46.000-05:00
diff --git a/sponsors/tests/test_models.py b/sponsors/tests/test_models.py
@@ -227,6 +227,7 @@ def test_rollback_sponsorship_to_edit(self):
             Sponsorship.APPLIED,
             Sponsorship.APPROVED,
             Sponsorship.REJECTED,
+            Sponsorship.CANCELLED,
         ]
         for status in can_rollback_from:
             sponsorship.status = status
@@ -301,6 +302,16 @@ def test_display_agreed_fee_for_approved_and_finalized_status(self):
 
             self.assertEqual(sponsorship.agreed_fee, 2000)
 
+    def test_cancel_sponsorship(self):
+        sponsorship = Sponsorship.new(self.sponsor, self.benefits)
+        sponsorship.status = Sponsorship.APPROVED
+        sponsorship.save()
+
+        sponsorship.cancel()
+
+        self.assertEqual(sponsorship.status, Sponsorship.CANCELLED)
+        self.assertIsNotNone(sponsorship.cancelled_on)
+
 
 class SponsorshipCurrentYearTests(TestCase):
 
