X-Git-Url: https://git.mdrn.pl/django-sponsors.git/blobdiff_plain/c47ec7131be40d01ee3a38a4b2e0715dda7aff1d..f560e8bf20e2517ebf7c8e1a3ace5c344b1a0145:/example_project/tests.py

diff --git a/example_project/tests.py b/example_project/tests.py
new file mode 100644
index 0000000..ac7ed47
--- /dev/null
+++ b/example_project/tests.py
@@ -0,0 +1,35 @@
+from django.core.files.base import ContentFile
+from django.test import TestCase
+from sponsors.models import Sponsor, SponsorPage
+from PIL import Image
+
+try:
+    from io import BytesIO
+except ImportError:
+    # Python 2
+    from StringIO import StringIO as BytesIO
+
+
+class SponsorsTest(TestCase):
+    def test_empty_page(self):
+        page = SponsorPage(name='test')
+        page.save()
+
+    def test_simple_sponsor(self):
+        s = Sponsor(name='Test Sponsor')
+        im = Image.new('1', (1,1))
+        b = BytesIO()
+        im.save(b, 'PNG')
+        s.logo.save('test.png', ContentFile(b.getvalue()), save=True)
+
+        page = SponsorPage(
+            name='Sponsor',
+            sponsors=[
+                {"name": "empty-column", "sponsors": []},
+                {"name": "test-column", "sponsors": [s.id]},
+            ])
+        page.save()
+        self.assertNotIn('empty-column', page._html)
+        self.assertIn('test-column', page._html)
+        self.assertIn('Test Sponsor', page._html)
+