X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/64def8585cd770750020addd02e959b2f9a9426f..349703205c1f87da589c9cb447070955d3aaf8bb:/tests/integration/base.py

diff --git a/tests/integration/base.py b/tests/integration/base.py
index c2774c97..cdd2f5a0 100644
--- a/tests/integration/base.py
+++ b/tests/integration/base.py
@@ -1,15 +1,14 @@
+# -*- coding: utf-8 -*-
 import os
 import inspect
-from urlparse import urlparse
 
 from django.test import LiveServerTestCase
 from django.test.client import Client
 from django.conf import settings
-from django.contrib.auth.models import User, Permission
+from django.contrib.auth.models import User
 from django.utils.translation import ugettext as _
 
-from selenium import webdriver, selenium
-from selenium.webdriver.support.wait import WebDriverWait
+from selenium import webdriver
 
 
 class SeleniumTestCase(LiveServerTestCase):
@@ -19,107 +18,104 @@ class SeleniumTestCase(LiveServerTestCase):
         LiveServerTestCase.setUpClass()
         cls.browser = getattr(webdriver, os.environ.get('TEST_BROWSER', 'Firefox'))()
         cls.browser.implicitly_wait(5)
-        
+
     @classmethod
     def tearDownClass(cls):
         LiveServerTestCase.tearDownClass()
         cls.browser.quit()
-        
+
     def setUp(self):
         self.browser.delete_all_cookies()
-    
-    def create_user(self, username = 'testuser',  passwd = 'passwd', do_login = False):
+
+    def create_user(self, username='testuser',  passwd='passwd', do_login=False):
         user = User.objects.create_user(username, '', passwd)
         user._plain_passwd = passwd
         if do_login:
             self.login_user(user)
         return user
-    
+
     def create_super_user(self, *args, **kwargs):
         user = self.create_user(*args, **kwargs)
         user.is_superuser = True
         user.save()
         return user
-        
+
     def login_user(self, user):
         client = Client()
-        client.login(username = user.username, password = user._plain_passwd)
+        client.login(username=user.username, password=user._plain_passwd)
 
         if not self.browser.current_url.startswith(self.live_server_url):
             self.browser.get(self.live_server_url+'/not_existing_url')
-            
-        self.browser.find_element_by_tag_name('body') # Make sure the page is actually loaded before setting the cookie
+
+        self.browser.find_element_by_tag_name('body')  # Make sure the page is actually loaded before setting the cookie
         self.browser.delete_cookie(settings.SESSION_COOKIE_NAME)
-        self.browser.add_cookie(dict(name = settings.SESSION_COOKIE_NAME, 
-                                     value = client.cookies[settings.SESSION_COOKIE_NAME].value,
-                                     path = '/')
-                               )
-        
+        self.browser.add_cookie(dict(name=settings.SESSION_COOKIE_NAME,
+                                     value=client.cookies[settings.SESSION_COOKIE_NAME].value,
+                                     path='/'))
+
     def get_main_page(self):
         self.browser.get(self.live_server_url)
         self.browser.find_element_by_tag_name('body')
         return MainPage(self.browser)
 
-        
+
 class Page(object):
     def __init__(self, browser):
         self.browser = browser
-    
-    
+
+
 class MainPage(Page):
 
     def __init__(self, browser):
         Page.__init__(self, browser)
         self.tab = None
-    
+
     @property
     def element(self):
         return self.browser.find_element_by_tag_name('body')
-    
+
     def select_tab(self, tab_title):
         for a in self.element.find_element_by_id('tabs-nav-left').find_elements_by_tag_name('a'):
             if a.text == tab_title:
                 a.click()
                 self.tab = find_tab_class(tab_title)(self.browser)
                 return
-        raise Exception, 'Tab not found'
-        
-                
-def find_tab_class(tab_title):       
+        raise Exception('Tab not found')
+
+
+def find_tab_class(tab_title):
     for obj in globals().values():
         if inspect.isclass(obj) and issubclass(obj, MainPageTabBase) and getattr(obj, 'tab_title', None) == tab_title:
             return obj
     raise NotImplementedError
-                
+
 
 class MainPageTabBase(Page):
     def __init__(self, browser):
         Page.__init__(self, browser)
-    
+
     @property
     def element(self):
         return self.browser.find_element_by_id('content')
-        
+
 
 class AddBookPage(MainPageTabBase):
     tab_title = _('Add')
-    
+
     def put_title(self, title):
         self.element.find_element_by_id('id_title').send_keys(title)
-        
+
     def put_text(self, text):
         self.element.find_element_by_id('id_text').send_keys(text)
-        
+
     def submit(self):
         self.browser.find_element_by_css_selector('table.editable button').click()
         return self.browser
-        
-    
+
+
 class BooksListPage(MainPageTabBase):
     tab_title = _('All')
-    
+
     @property
     def visible_books_count(self):
         return len(self.element.find_element_by_id('file-list').find_elements_by_tag_name('tr')) - 2
-        
-        
\ No newline at end of file