test_selenium.py

import socket

from django.conf import settings
from django.urls import reverse
from django.contrib.staticfiles.testing import StaticLiveServerTestCase
from django.test import override_settings
from selenium import webdriver
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities

from exports.test.factories import ExportInMemoryFactory
from exports.models import User, Export


@override_settings(ALLOWED_HOSTS=['*'])  # Disable ALLOW_HOSTS
@override_settings(DEBUG=True)
class SeleniumStaticLiveServerTestCase(StaticLiveServerTestCase):
    """
    Provides base test class which connects to the Docker
    container running Selenium.
    You can use a VNC client on localhost:5900 (password: secret)
    to get a view of the process
    """
    host = '0.0.0.0'  # Bind to 0.0.0.0 to allow external access

    @classmethod
    def setUpClass(cls):
        super().setUpClass()
        # Set host to externally accessible web server address
        cls.host = socket.gethostbyname(socket.gethostname())

        cls.remote_selenium_address = getattr(settings, 'REMOTE_SELENIUM_SERVER', False)

        # Instantiate the remote WebDriver
        cls.selenium = webdriver.Remote(
            #  Set to: htttp://{selenium-container-name}:port/wd/hub
            #  In our example, the container is named `selenium`
            #  and runs on port 4444
            command_executor=cls.remote_selenium_address,
            # Set to CHROME since we are using the Chrome container
            desired_capabilities=DesiredCapabilities.CHROME,
        )
        cls.selenium.implicitly_wait(5)

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

    def setUp(self):
        """Open a new browser for each test."""
        super(SeleniumStaticLiveServerTestCase, self).setUp()

        test_user = User.objects.get_or_create(username='test',
                                               first_name='test',
                                               last_name='test',
                                               email='test@localhost',
                                               is_staff=True,
                                               is_active=True)[0]

        # # generate a cookie place, or get the cookie setting error from Chrome
        self.selenium.get('%s%s' % (self.live_server_url,
                                    reverse('not_allowed')))

        # bypass external Tequila auth
        self.client.force_login(test_user)

        # feed the cookie to selenium
        sessionid = self.client.cookies['sessionid']
        # add_cookie will be relative to this domain
        self.selenium.add_cookie({
            'name': 'sessionid',
            'value': sessionid.value,
            'secure': False,
            'path': '/'
        })

    def test_create(self):
        before_count = Export.objects.count()
        mpl = ExportInMemoryFactory()

        self.selenium.get('%s%s' % (self.live_server_url,
                                    reverse('crud:export-create')))
        name_input = self.selenium.find_element_by_id("id_name")
        name_input.send_keys(mpl.name)
        url_input = self.selenium.find_element_by_id("id_url")
        url_input.send_keys(mpl.url)

        self.selenium.find_element_by_id("btn-submit").click()
        self.assertGreater(Export.objects.count(), before_count)