Python check_plan_expirationの例

コード例 #1

ファイル: test_models.py プロジェクト: huguesmayolle/famille

 def test_check_plan_expiration_basic(self, send):
     self.famille.plan = "basic"
     self.famille.save()
     check_plan_expiration(None, None, self.famille.user)
     f = models.Famille.objects.get(pk=self.famille.pk)
     self.assertEquals(f.plan, "basic")
     self.assertFalse(send.called)

コード例 #2

ファイル: test_models.py プロジェクト: huguesmayolle/famille

 def test_check_plan_expiration_not_expired(self, send):
     self.famille.plan = "premium"
     self.famille.plan_expires_at = datetime(2500, 1, 1)
     self.famille.save()
     check_plan_expiration(None, None, self.famille.user)
     f = models.Famille.objects.get(pk=self.famille.pk)
     self.assertEquals(f.plan, "premium")
     self.assertEquals(f.plan_expires_at.replace(tzinfo=None), datetime(2500, 1, 1))
     self.assertFalse(send.called)

コード例 #3

ファイル: test_models.py プロジェクト: huguesmayolle/famille

 def test_check_plan_expiration_expired(self, send):
     self.famille.plan = "premium"
     self.famille.plan_expires_at = datetime(2000, 1, 1)
     self.famille.save()
     check_plan_expiration(None, None, self.famille.user)
     f = models.Famille.objects.get(pk=self.famille.pk)
     self.assertEquals(f.plan, "basic")
     self.assertIsNone(f.plan_expires_at)
     self.assertTrue(send.called)