from corehq.apps.hqwebapp.testcases import WebUserTestCase
from corehq.apps.hqwebapp import selenium
from selenium.common.exceptions import NoSuchElementException
import re


class ReportsTestCase(WebUserTestCase):
    
    app = selenium.get_app('reports')
    max_preload_time = app.MAX_PRELOAD_TIME
    max_load_time = app.MAX_LOAD_TIME

    @classmethod
    def setUpClass(cls):
        super(ReportsTestCase, cls).setUpClass()
        cls.driver.find_element_by_link_text('Reports').click()
        assert 'Project Reports' in cls.driver.page_source

    @classmethod
    def generate_test(cls, report_name):

        def loading_report(d):
            # todo: make sure this check doesn't need to have a wait before
            # it. It's very fast, but it might.
            if 'Fetching additional data' in d.page_source:
                return True

            try:
                elem = d.find_element_by_xpath("//div[@class='hq-loading']")
                return elem.is_displayed()
            except NoSuchElementException:
                pass

            return False

        def test(self):
            self.find_element_by_link_text(report_name).click()

            elem = self.find_element_by_xpath(
                "//*[text()='Hide Filter Options'] | //*[text()='Show Filter Options']",
                duration=self.max_preload_time
            )
            if elem.text == 'Show Filter Options':
                elem.click()

            self.find_element_by_xpath("//button[@id='apply-filters']").click()

            self.wait_until_not(loading_report, time=self.max_load_time)

        return test

report_names = (
    'Daily Form Activity',
    'Submissions By Form',
    'Case Activity',
    'Form Completion Time',
    'Form Completion vs. Submission Trends',
    'Worker A
ctivity Times',

    'Submit History',
    'Case List',

    '
Export Submissions to Excel',
    'Export Cases and Users',
    'De-Identified Export', # this requires at least one saved export to be created

    'Application Status',
    'Raw Forms, Errors & Duplicates',
    'Errors & Warnings Summary',
    'Device Log Details',

    'SMS Usage'
)

for report_name in report_names:
    name = re.sub(r"[^a-z\s]", "", report_name.lower()).replace(" ", "_")
    setattr(ReportsTestCase, "test_%s" % name,
            ReportsTestCase.generate_test(report_name))

