test for remaining license

velis74 · Oct 16, 2023 · 9fd9e8b · 9fd9e8b
1 parent ee06707
commit 9fd9e8b
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/django_project_base/notifications/tests/api/test_remainig_license.py b/django_project_base/notifications/tests/api/test_remainig_license.py
@@ -0,0 +1,21 @@
+from django.contrib.contenttypes.models import ContentType
+
+from django_project_base.licensing.logic import LogAccessService, MONTHLY_ACCESS_LIMIT_IN_CURRENCY_UNITS
+from django_project_base.licensing.models import LicenseAccessUse
+from django_project_base.notifications.tests.notifications_transaction_test_case import NotificationsTransactionTestCase
+
+
+class TestRemainingLicense(NotificationsTransactionTestCase):
+    def test_remaining_license(self):
+        used_amount = 10
+        mail = self.create_notification_email_object().send()
+        LicenseAccessUse.objects.create(
+            type=LicenseAccessUse.UseType.USE,
+            user_id=str(self.test_user.pk),
+            content_type_object_id=str(mail.pk),
+            content_type=ContentType.objects.get_for_model(mail._meta.model),
+            amount=used_amount,
+            comment=dict(comment="", count=10, item_price=1, sender=""),
+        )
+        usage = LogAccessService().report(user=self.test_user)
+        self.assertEqual(MONTHLY_ACCESS_LIMIT_IN_CURRENCY_UNITS - used_amount, usage["remaining_credit"])