140 lines
5.1 KiB
Python
140 lines
5.1 KiB
Python
from datetime import datetime
|
|
import hashlib
|
|
import os.path
|
|
from time import sleep
|
|
|
|
CHROMIUM = 'chromium'
|
|
FIREFOX = 'firefox'
|
|
SELENIUM_BROWSER = FIREFOX
|
|
|
|
from django.conf import settings
|
|
from django.contrib.staticfiles.finders import find
|
|
from django.test import LiveServerTestCase
|
|
from django.test.utils import override_settings
|
|
|
|
if SELENIUM_BROWSER == CHROMIUM:
|
|
from selenium import webdriver
|
|
elif SELENIUM_BROWSER == FIREFOX:
|
|
from selenium.webdriver.firefox.webdriver import WebDriver
|
|
else:
|
|
raise ValueError('a browser must be picked to run selenium tests')
|
|
|
|
|
|
class TestAdminPanelWidget(LiveServerTestCase):
|
|
fixtures = ['test_admin.json']
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
if SELENIUM_BROWSER == CHROMIUM:
|
|
cls.selenium = webdriver.Chrome(executable_path='/usr/lib/chromium-browser/chromedriver')
|
|
elif SELENIUM_BROWSER == FIREFOX:
|
|
cls.selenium = WebDriver()
|
|
super(TestAdminPanelWidget, cls).setUpClass()
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
cls.selenium.quit()
|
|
super(TestAdminPanelWidget, cls).tearDownClass()
|
|
|
|
def test_admin_panel_widget(self):
|
|
self._login_to_admin()
|
|
self._go_to_demo_application_in_admin()
|
|
self._assert_editor_loaded()
|
|
self._focus_cursor_in_editor()
|
|
self._enter_test_text()
|
|
self._open_image_upload_widget()
|
|
self._go_to_upload_tab()
|
|
self._switch_to_form_iframe()
|
|
self._upload_image()
|
|
self._assert_image_uploaded()
|
|
|
|
def _login_to_admin(self):
|
|
self.selenium.get('%s%s' % (self.live_server_url, '/admin/'))
|
|
username_input = self.selenium.find_element_by_name("username")
|
|
username_input.send_keys('test')
|
|
password_input = self.selenium.find_element_by_name("password")
|
|
password_input.send_keys('test')
|
|
self.selenium.find_element_by_xpath('//input[@value="Log in"]').click()
|
|
|
|
def _go_to_demo_application_in_admin(self):
|
|
self.selenium.get('%s%s' % (self.live_server_url, '/admin/demo_application/examplemodel/add/'))
|
|
|
|
def _assert_editor_loaded(self):
|
|
sleep(2)
|
|
self.selenium.find_element_by_id("cke_id_content")
|
|
|
|
def _focus_cursor_in_editor(self):
|
|
self.frame = self.selenium.find_element_by_class_name('cke_wysiwyg_frame')
|
|
self.frame.click()
|
|
|
|
def _enter_test_text(self):
|
|
self.frame.send_keys('test')
|
|
|
|
def _open_image_upload_widget(self):
|
|
self.selenium.find_element_by_css_selector("span.cke_button_icon.cke_button__image_icon").click()
|
|
sleep(1)
|
|
|
|
def _go_to_upload_tab(self):
|
|
self.selenium.find_element_by_id("cke_Upload_116").click()
|
|
sleep(1)
|
|
|
|
def _switch_to_form_iframe(self):
|
|
iframe = self.selenium.find_element_by_id('cke_111_fileInput')
|
|
self.selenium.switch_to_frame(iframe)
|
|
|
|
def _upload_image(self):
|
|
input = self.selenium.find_element_by_id("cke_111_fileInput_input")
|
|
input.send_keys(self._get_upload_file())
|
|
self.selenium.switch_to_default_content()
|
|
self.selenium.find_element_by_class_name("cke_dialog_ui_fileButton").click()
|
|
sleep(2)
|
|
|
|
def _get_upload_file(self):
|
|
return find('ckeditor/ckeditor/skins/moono/images/hidpi/close.png')
|
|
|
|
def _assert_image_uploaded(self):
|
|
upload_directory = self._get_upload_directory()
|
|
expected_image_path = os.path.join(upload_directory, 'close.png')
|
|
expected_thumbnail_path = os.path.join(upload_directory, 'close_thumb.png')
|
|
assert os.path.isfile(expected_image_path)
|
|
assert os.path.isfile(expected_thumbnail_path)
|
|
self._assert_uploaded_image_did_not_changed(expected_image_path)
|
|
self._assert_thumbnail_is_not_empty(expected_thumbnail_path)
|
|
os.remove(expected_image_path)
|
|
os.remove(expected_thumbnail_path)
|
|
|
|
def _get_upload_directory(self):
|
|
date_path = datetime.now().strftime('%Y/%m/%d')
|
|
|
|
# Complete upload path (upload_path + date_path).
|
|
upload_path = os.path.join(
|
|
settings.CKEDITOR_UPLOAD_PATH, date_path)
|
|
return os.path.join(settings.MEDIA_ROOT, upload_path)
|
|
|
|
def _assert_uploaded_image_did_not_changed(self, path):
|
|
expected_sha = self._get_sha1_for_file(self._get_upload_file())
|
|
uploaded_sha = self._get_sha1_for_file(path)
|
|
assert expected_sha == uploaded_sha
|
|
|
|
def _get_sha1_for_file(self, path):
|
|
image = open(path, 'rb')
|
|
hash = hashlib.sha1()
|
|
hash.update(image.read())
|
|
return hash.hexdigest()
|
|
|
|
def _assert_thumbnail_is_not_empty(self, path):
|
|
size = os.path.getsize(path)
|
|
assert size > 0
|
|
|
|
|
|
@override_settings(CKEDITOR_IMAGE_BACKEND=None)
|
|
class TestAdminPanelWidgetForDummyImageBackend(TestAdminPanelWidget):
|
|
def _assert_image_uploaded(self):
|
|
upload_directory = self._get_upload_directory()
|
|
expected_image_path = os.path.join(upload_directory, 'close.png')
|
|
expected_thumbnail_path = os.path.join(upload_directory, 'close_thumb.png')
|
|
assert os.path.isfile(expected_image_path)
|
|
assert not os.path.isfile(expected_thumbnail_path)
|
|
self._assert_uploaded_image_did_not_changed(expected_image_path)
|
|
os.remove(expected_image_path)
|