6/Test GrantCategory

app/grants/models/clr_match.py

@@ -53,4 +53,4 @@ class CLRMatch(SuperModel):
 
     def __str__(self):
         """Return the string representation of a Grant."""
-        return f"id: {self.pk}, grant: {self.grant.pk}, round: {self.round_number}, amount: {self.amount}"
+        return f"id: {self.pk}, grant: {self.grant.pk}, round: {self.round_number}, amount: {self.amount}"

app/grants/models/contribution.py

@@ -18,6 +18,7 @@ class Contribution(SuperModel):
     CHECKOUT_TYPES = [
         ('eth_std', 'eth_std'),
         ('eth_zksync', 'eth_zksync'),
+        ('eth_polygon', 'eth_polygon'),
         ('zcash_std', 'zcash_std'),
         ('celo_std', 'celo_std'),
         ('zil_std', 'zil_std'),

app/grants/models/grant.py

@@ -488,7 +488,7 @@ class Meta:
     )
 
     in_active_clrs = models.ManyToManyField(
-        GrantCLR,
+        "GrantCLR",
         help_text="Active Grants CLR Round"
     )
     is_clr_active = models.BooleanField(default=False, help_text=_('CLR Round active or not? (auto computed)'))
@@ -885,7 +885,6 @@ def save(self, update=True, *args, **kwargs):
 
         self.clr_prediction_curve = self.calc_clr_prediction_curve
         self.clr_round_num = self.calc_clr_round_label
-
         self.search_vector = (
             SearchVector('title', weight='A') + SearchVector('description', weight='B')
         )

app/grants/models/match_pledge.py

@@ -57,4 +57,4 @@ def data_json(self):
 
     def __str__(self):
         """Return the string representation of this object."""
-        return f"{self.profile} <> {self.amount} DAI"
+        return f"{self.profile} <> {self.amount} DAI"

app/grants/models/subscription.py

@@ -554,4 +554,4 @@ def create_contribution(self, tx_id, is_successful_contribution=True):
             successful_contribution(self.grant, self, contribution)
 
         update_grant_metadata.delay(self.pk)
-        return contribution
+        return contribution

app/grants/tests/models/factories/grant_category_factory.py

@@ -0,0 +1,9 @@
+import factory
+from grants.models.grant_category import GrantCategory
+
+
+class GrantCategoryFactory(factory.django.DjangoModelFactory):
+    """Create mock GrantCategory for testing."""
+
+    class Meta:
+        model = GrantCategory

app/grants/tests/models/test_grant_category.py

@@ -0,0 +1,23 @@
+import pytest
+from grants.models.grant_category import GrantCategory
+
+from .factories.grant_category_factory import GrantCategoryFactory
+
+
+@pytest.mark.django_db
+class TestGrantCategory:
+    """Test GrantCategory model."""
+
+    def test_creation(self):
+        """Test GrantCategory returned by factory is valid."""
+
+        grant_category = GrantCategoryFactory()
+
+        assert isinstance(grant_category, GrantCategory)
+
+    def test_grant_category_has_a_category(self):
+        """Test 'category' attribute is present."""
+
+        grant_category = GrantCategoryFactory()
+
+        assert hasattr(grant_category, 'category')

requirements/test.txt

@@ -1,6 +1,7 @@
 -r base.txt
 pytest==6.2.4
 pytest-cov==2.12.0
+pytest-factoryboy==2.1.0
 pytest-isort==2.0.0
 pytest-django==4.3.0
 pytest-sugar==0.9.4