1 from django.core.files.base import ContentFile
2 from django.test import TestCase
3 from sponsors.models import Sponsor, SponsorPage
10 from StringIO import StringIO as BytesIO
13 class SponsorsTest(TestCase):
14 def test_empty_page(self):
15 page = SponsorPage(name='test')
18 def test_simple_sponsor(self):
19 s = Sponsor(name='Test Sponsor')
20 im = Image.new('1', (1,1))
23 s.logo.save('test.png', ContentFile(b.getvalue()), save=True)
28 {"name": "empty-column", "sponsors": []},
29 {"name": "test-column", "sponsors": [s.id]},
32 self.assertNotIn('empty-column', page._html)
33 self.assertIn('test-column', page._html)
34 self.assertIn('Test Sponsor', page._html)