Add __str__ for AffiliateOffer model (#77)

Stranger6667 · Jun 18, 2018 · 17c46cb · 17c46cb
1 parent a59393e
commit 17c46cb
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/pyoffers/models/affiliate.py b/pyoffers/models/affiliate.py
@@ -7,6 +7,8 @@ class AffiliateOffer(Model):
     """
     Model that defines relation between Offer and Affiliate
     """
+    def __str__(self):
+        return '{}: {}'.format(self.__class__.__name__, self.approval_status)
 
 
 class AffiliateOfferManager(InvisibleModelManager):

diff --git a/tests/models/test_affiliate.py b/tests/models/test_affiliate.py
@@ -1,8 +1,10 @@
 # coding: utf-8
+from unittest.mock import Mock
+
 import pytest
 
 from pyoffers.exceptions import HasOffersException
-from pyoffers.models import Affiliate, AffiliateUser, OfferFile
+from pyoffers.models import Affiliate, AffiliateOffer, AffiliateUser, OfferFile
 
 
 CASSETTE_NAME = 'affiliate'
@@ -68,3 +70,9 @@ def test_get_offer_files_with_creative_code(affiliate):
     assert len(files) == 1
     assert isinstance(files[0], OfferFile)
     assert files[0].creativecode == 'http://www.example.com'
+
+
+def test_affiliate_offer_str():
+    affiliate_offer = AffiliateOffer(manager=Mock())
+    affiliate_offer.approval_status = 'approved'
+    assert str(affiliate_offer) == 'AffiliateOffer: approved'