from django.test import TestCase
from selenium import webdriver
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities


class SeleniumTest(TestCase):
    def setUp(self):
        self.chrome = webdriver.Remote(command_executor='http://selenium_hub:4444/wd/hub', desired_capabilities=DesiredCapabilities.CHROME)
        self.chrome.implicitly_wait(10)

    def test_visit_site_with_chrome(self):
        self.chrome.get('http://web/admin')
        self.assertIn(self.chrome.title, "Log in | Django site admin")