class TestPageImageTags(TestCase):
def setUp(self):
self.p = Page(title='test')
self.p.save()
def setUpTemplate(self):
self.tmpl = Template(u'{% load pageimage_tags %}This will point to image \'{% pageimage \'BACKGROUND\' %}\'.')
self.c = Context({'page': self.p})
def test_simpletemplate_empty(self):
self.setUpTemplate()
tmplout = self.tmpl.render(self.c)
self.assertTrue('\'\'' in tmplout)
def test_simpletemplate_default(self):
self.setUpTemplate()
DefaultImage.objects.create(type='BACKGROUND', file='defaultback.jpg')
self.assertTrue(
'\'/static/media/defaultback.jpg\'' in self.tmpl.render(self.c)
)
def test_simpletemplate_specific(self):
self.setUpTemplate()
PageImage.objects.create(page=self.p, type='BACKGROUND', file='back.jpg')
self.assertTrue(
'\'/static/media/back.jpg\'' in self.tmpl.render(self.c)
)
def setUp(self):
self.user = User(username="******")
self.user.set_password("asdf")
self.user.save()
p = os.path.dirname(os.path.abspath(__file__))
icon = open(os.path.join(p, "test_files/icon.psd"))
f_icon = File(icon)
self.button = HybridButton(
name="Testbutton",
user=self.user,
icon=f_icon,
)
self.button.save()
page = Page(status='2', description="buttonizer", title="Buttonizer")
page.save()
class TestGetImageForPage(TestCase):
def setUp(self):
self.p = Page(title='test')
self.p.save()
def setUpDefaultImage(self):
di = DefaultImage(file='default.jpg', type='BACKGROUND')
di.save()
def setUpPageImage(self):
pi = PageImage(page=self.p, type='BACKGROUND', file='test.jpg')
pi.save()
def test_get_empty(self):
self.assertEqual(
'',
get_image_for_page(self.p, 'BACKGROUND')
)
def test_get_default_image(self):
self.setUpDefaultImage()
self.assertEqual(
'/static/media/default.jpg',
get_image_for_page(self.p, 'BACKGROUND')
)
def test_get_specific_image(self):
self.setUpDefaultImage()
self.setUpPageImage()
self.assertEqual(
'/static/media/test.jpg',
get_image_for_page(self.p, 'BACKGROUND')
)
def test_get_child_images(self):
self.setUpDefaultImage()
self.setUpPageImage()
p1 = Page.objects.create(title='child1', parent=self.p)
p2 = Page.objects.create(title='child2', parent=self.p)
PageImage.objects.create(page=p2, type='BACKGROUND', file='child2.jpg')
self.assertEqual(
'/static/media/test.jpg',
get_image_for_page(p1, 'BACKGROUND')
)
self.assertEqual(
'/static/media/child2.jpg',
get_image_for_page(p2, 'BACKGROUND')
)
def test_get_banner_image(self):
DefaultImage.objects.create(type='BANNER', file='defaultbanner.jpg')
self.assertEqual(
'/static/media/defaultbanner.jpg',
get_image_for_page(self.p, 'BANNER')
)
