From a3d31e3dcf7662415614dd5e35854f8e433d613b Mon Sep 17 00:00:00 2001
From: Erin Mullaney <erin.mullaney@torchbox.com>
Date: Tue, 22 Jan 2019 10:15:18 -0500
Subject: [PATCH] #810 adding a test for the new by_lead filter method
---
.../apply/funds/tests/models/test_roundsandlabs.py | 11 +++++++++++
1 file changed, 11 insertions(+)
diff --git a/opentech/apply/funds/tests/models/test_roundsandlabs.py b/opentech/apply/funds/tests/models/test_roundsandlabs.py
index 5eb8709d7..4d78ed346 100644
--- a/opentech/apply/funds/tests/models/test_roundsandlabs.py
+++ b/opentech/apply/funds/tests/models/test_roundsandlabs.py
@@ -67,6 +67,17 @@ class BaseRoundsAndLabTestCase:
self.assertEqual(fetched_obj, obj)
self.assertFalse(base_qs.active().exists())
+ def test_by_lead(self):
+ obj = self.base_factory()
+ # Create an additional round which will create a new staff lead
+ round_other_lead = RoundFactory()
+ qs_all = RoundsAndLabs.objects.with_progress()
+ qs_by_lead = qs_all.by_lead(obj.lead)
+ fetched_obj = qs_by_lead.first()
+ self.assertEqual(qs_all.count(), 2)
+ self.assertEqual(qs_by_lead.count(), 1)
+ self.assertEqual(fetched_obj.lead, obj.lead.full_name)
+
class TestForLab(BaseRoundsAndLabTestCase, TestCase):
base_factory = LabFactory
--
GitLab