growthhack.py

from django.contrib.auth import get_user_model
from django.test import LiveServerTestCase
from selenium.webdriver.phantomjs.webdriver import WebDriver


class LiveTests(LiveServerTestCase):

    @classmethod
    def setUpClass(cls):
        super(LiveTests, cls).setUpClass()
        cls.selenium = WebDriver()

        cls.user = get_user_model().objects.create_user(username="X", password="Y")

    @classmethod
    def tearDownClass(cls):
        cls.selenium.quit()
        super(LiveTests, cls).tearDownClass()

    def login(self):
        self.selenium.get(self.live_server_url)
        self.selenium.find_element_by_link_text("Login").click()

        u_i = self.selenium.find_element_by_name("username")
        u_i.send_keys(self.user.username)
        p_i = self.selenium.find_element_by_name("password")
        p_i.send_keys(self.user.password)

        self.selenium.find_element_by_xpath("//input[@type='submit']").click()


class LoggedInLiveTests(LiveTests):

    def setUp(self):
        self.login()

    def test_pass(self):
        pass