eregs/regulations-site

# -*- coding: utf-8 -*-
import unittest

from selenium.webdriver.support.ui import WebDriverWait

from regulations.uitests.base_test import BaseTest


class NavigationTest(BaseTest, unittest.TestCase):

    job_name = 'Navigation test'

    def test_navigation(self):
        self.driver.get(self.test_url + '/1005-5/2012-12121')
        html = self.driver.find_element_by_tag_name('html')
        WebDriverWait(self.driver, 30).until(
            lambda driver: 'selenium-start' in html.get_attribute('class'))

        self.driver.execute_script(
            'poffset = document.getElementById("1005-5-b-2").offsetTop')
        self.driver.execute_script('window.scrollTo(0, poffset)')

        # wayfinding header should update
        self.driver.implicitly_wait(5)
        header = self.driver.find_element_by_css_selector('#active-title')
        self.assertIn(header.text,
                      (u'§ 1005.5(b)', u'§ 1005.5(b)(1)', u'§ 1005.5(b)(2)'))

        fwd_link = self.driver.find_element_by_css_selector('li.next a')
        fwd_link.click()
        # should navigate to next section
        WebDriverWait(self.driver, 30).until(
            lambda driver: driver.find_element_by_id('1005-6'))

        self.driver.back()

        # should go back to 1005-5
        self.driver.find_element_by_xpath('//*[@id="1005-5"]')
        back_link = self.driver.find_element_by_css_selector('li.previous a')
        back_link.click()
        # should navigate to 1005-4
        WebDriverWait(self.driver, 30).until(
            lambda driver: driver.find_element_by_id('1005-4'))