10/Test GrantAPIKey model

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/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_api_key_factory.py

@@ -0,0 +1,13 @@
+import factory
+from grants.models.grant_api_key import GrantAPIKey
+
+from .profile_factory import ProfileFactory
+
+
+class GrantAPIKeyFactory(factory.django.DjangoModelFactory):
+    """Create mock GrantAPIKey for testing."""
+
+    class Meta:
+        model = GrantAPIKey
+
+    profile = factory.SubFactory(ProfileFactory)

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

@@ -0,0 +1,12 @@
+import factory
+from dashboard.models import Profile
+
+
+class ProfileFactory(factory.django.DjangoModelFactory):
+    """Create mock Profile for testing."""
+
+    class Meta:
+        model = Profile
+
+    handle = factory.Sequence(lambda n: "Contributor_%03d" % n)
+    data = {} 

app/grants/tests/models/test_grant_api_key.py

@@ -0,0 +1,39 @@
+import pytest
+from dashboard.models import Profile
+from grants.models.grant_api_key import GrantAPIKey
+
+from .factories.grant_api_key_factory import GrantAPIKeyFactory
+
+
+@pytest.mark.django_db
+class TestGrantAPIKey:
+    """Test GrantAPIKey model."""
+
+    def test_creation(self):
+        """Test GrantAPIKey returned by factory is valid."""
+
+        grant_api_key = GrantAPIKeyFactory()
+
+        assert isinstance(grant_api_key, GrantAPIKey)
+
+    def test_grant_api_key_has_a_key(self):
+        """Test 'key' attribute is present."""
+
+        grant_api_key = GrantAPIKeyFactory()
+
+        assert hasattr(grant_api_key, 'key')
+
+    def test_grant_api_key_has_a_secret(self):
+        """Test secret attribute is present."""
+
+        grant_api_key = GrantAPIKeyFactory()
+
+        assert hasattr(grant_api_key, 'secret')
+
+    def test_grant_api_key_has_associated_profile(self):
+        """Test profile attribute is present and is an instance of Profile."""
+
+        grant_api_key = GrantAPIKeyFactory()
+
+        assert hasattr(grant_api_key, 'profile')
+        assert isinstance(grant_api_key.profile, Profile)

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