def visit_page(self, file_name, expected_text=""):
fileExt = os.path.splitext(file_name)[1]
# Чи розширення цього файла входить до списку "previewable"?
if expected_text:
self.assertNotIn(fileExt, get_viewable_extension_list())
else:
self.assertIn(fileExt, get_viewable_extension_list())
# Створюємо запис з цим файлом:
full_path = os.path.join(os.getcwd(), file_name) # повний шлях
self.report = DummyFolder().create_dummy_report(parent=self.parent,
id=1, path=full_path)
# Користувач відкриває сторінку
self.browser.get('%s%s' % (self.server_url, self.this_url))
# Бачить у полі очікувану інформацію
div_box = self.browser.find_element_by_id('preview-box')
self.assertEqual(div_box.text, expected_text)
def get_absolute_url(self):
try:
fileExt = os.path.splitext(self.filename)[1] # [0] returns path+filename
except:
fileExt = ""
if fileExt in get_viewable_extension_list():
url_name = 'folders:report-preview'
else:
url_name = 'folders:report-detail'
return reverse(url_name, kwargs={'pk': self.pk})
def get_absolute_url(self):
try:
fileExt = os.path.splitext(
self.filename)[1] # [0] returns path+filename
except:
fileExt = ""
if fileExt in get_viewable_extension_list():
url_name = 'folders:report-preview'
else:
url_name = 'folders:report-detail'
return reverse(url_name, kwargs={'pk': self.pk})
def test_get_context_data(self):
root = DummyFolder().create_dummy_root_folder()
report = DummyFolder().create_dummy_report(root, id=1)
request = RequestFactory().get('/folders/report/1/preview/')
kwargs = {'pk': 1}
view = self.cls_view()
view = setup_view(view, request, **kwargs)
view.model = Report
view.object = report
expected = get_viewable_extension_list()
context = view.get_context_data()
self.assertEqual(context['viewable_extension_list'], expected)
def get_context_data(self, **kwargs):
context = super(ReportPreview, self).get_context_data(**kwargs)
# Значення передадуться в шаблон:
context["viewable_extension_list"] = get_viewable_extension_list()
return context