Fix method return value

This commit is contained in:
Artem Muterko 2017-01-14 20:12:21 +02:00
parent 669ff66b59
commit 0b76ec491d

View File

@ -110,14 +110,14 @@ class TemplateHTMLRendererTests(TestCase):
class MockView(object): class MockView(object):
def get_template_names(self): def get_template_names(self):
return 'template from get_template_names method' return ['template from get_template_names method']
class MockView2(object): class MockView2(object):
template_name = 'template from template_name attribute' template_name = 'template from template_name attribute'
template_name = renderer.get_template_names(self.mock_response, template_name = renderer.get_template_names(self.mock_response,
MockView()) MockView())
assert template_name == 'template from get_template_names method' assert template_name == ['template from get_template_names method']
template_name = renderer.get_template_names(self.mock_response, template_name = renderer.get_template_names(self.mock_response,
MockView2()) MockView2())