NK LiNK to NK Logbook

rowers/tests/test_unit_tests.py

@@ -296,6 +296,11 @@ class PlannedSessionTests(TestCase):
         micro2 = plannedsessions.get_todays_micro(self.plan,thedate=datetime.date.today())
         self.assertEqual(micro1.id,micro2.id)
 
+    def test_get_session_ids(self):
+        ids = plannedsessions.get_my_session_ids(self.r)
+        pss = PlannedSession.objects.filter(rower__in=[self.r])
+        self.assertEqual(len(ids),len(pss))
+
 class DataPrepTests(TestCase):
     def setUp(self):
         self.u = UserFactory()