def test_purchasedproducts_active_expired(self):
PurchasedProductFactory.create_batch(3, product=self.product, owner=self.company)
expired_product = PurchasedProductFactory(product=self.product, owner=self.company)
expired_product.expiration_date = date.today() - timedelta(days=1)
expired_product.save()
response = self.client.post(reverse("purchasedproduct"))
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context["active_products"]), 3)
self.assertEqual(len(response.context["expired_products"]), 1)
def test_purchasedjob_add_too_many(self):
product = PurchasedProductFactory(product=self.product, owner=self.company)
product.jobs_remaining = 1
product.save()
kwargs = {"product": product.pk}
response = self.client.post(
reverse("purchasedjob_add", kwargs=kwargs), data=self.purchasedjob_form_data, follow=True
)
self.assertEqual(response.status_code, 200)
self.assertEqual(PurchasedJob.objects.all().count(), 1)
response = self.client.post(reverse("purchasedjob_add", kwargs=kwargs), data=self.purchasedjob_form_data)
self.assertEqual(response.status_code, 404)
