5/Test GrantCLRCalculation 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

@@ -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_clr_calculation_factory.py

@@ -0,0 +1,17 @@
+import factory
+import pytest
+from grants.models.grant_clr_calculation import GrantCLRCalculation
+
+from .grant_clr_factory import GrantCLRFactory
+from .grant_factory import GrantFactory
+
+
+@pytest.mark.django_db
+class GrantCLRCalculationFactory(factory.django.DjangoModelFactory):
+    """Create mock GrantCLRCalculation for testing."""
+
+    class Meta:
+        model = GrantCLRCalculation
+
+    grant = factory.SubFactory(GrantFactory)
+    grantclr = factory.SubFactory(GrantCLRFactory)

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

@@ -0,0 +1,17 @@
+from datetime import datetime, timedelta
+
+import factory
+import pytest
+from grants.models.grant import GrantCLR
+
+
+@pytest.mark.django_db
+class GrantCLRFactory(factory.django.DjangoModelFactory):
+    """Create mock GrantCLR for testing."""
+
+    class Meta:
+        model = GrantCLR
+
+    round_num = 2
+    start_date = datetime.now()
+    end_date = start_date + timedelta(weeks=2)

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

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

app/grants/tests/models/test_grant_clr_calculation.py

@@ -0,0 +1,53 @@
+import pytest
+from grants.models.grant import Grant, GrantCLR
+from grants.models.grant_clr_calculation import GrantCLRCalculation
+
+from .factories.grant_clr_calculation_factory import GrantCLRCalculationFactory
+
+
+@pytest.mark.django_db
+class TestGrantCLRCalculation:
+    """Test GrantCLRCalculation model."""
+
+    def test_creation(self):
+        """Test GrantCLRCalculation returned by factory is valid."""
+
+        grant_clr_calulation = GrantCLRCalculationFactory()
+
+        assert isinstance(grant_clr_calulation, GrantCLRCalculation)
+
+    def test_grant_clr_calculation_has_latest_attribute(self):
+        """Test 'latest' attribute is present and defaults to False."""
+
+        grant_clr_calulation = GrantCLRCalculationFactory()
+
+        assert hasattr(grant_clr_calulation, 'latest')
+        assert grant_clr_calulation.latest == False
+
+    def test_grant_clr_calculation_has_associated_grant(self):
+        """Test 'grant' attribute is present and is an instance of Grant."""
+
+        grant_clr_calculation = GrantCLRCalculationFactory()
+
+        assert hasattr(grant_clr_calculation, 'grant')
+        assert isinstance(grant_clr_calculation.grant, Grant)
+
+    def test_grant_clr_calculation_has_associated_grant_clr(self):
+        """Test 'grantclr' attribute is present and is an instance of GrantCLR."""
+
+        grant_clr_calculation = GrantCLRCalculationFactory()
+
+        assert hasattr(grant_clr_calculation, 'grantclr')
+        assert isinstance(grant_clr_calculation.grantclr, GrantCLR)
+
+    def test_grant_clr_calculation_has_clr_prediction_curve(self):
+        """Test 'clr_prediction_curve' attribute is present and defaults to an empty list."""
+
+        grant_clr_calculation = GrantCLRCalculationFactory()
+
+        assert hasattr(grant_clr_calculation, 'clr_prediction_curve')
+        assert grant_clr_calculation.clr_prediction_curve == []
+        assert len(grant_clr_calculation.clr_prediction_curve) == 0
+
+
+

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