from galaxy_test.driver import integration_util
from galaxy_test.selenium import (
    framework
)

selenium_test = framework.selenium_test


class SeleniumIntegrationTestCase(integration_util.IntegrationTestCase, framework.TestWithSeleniumMixin, framework.UsesLibraryAssertions):

    def setUp(self):
        super().setUp()
        self.setup_selenium()

    def tearDown(self):
        self.tear_down_selenium()
        super().tearDown()

    def restart(self, handle_reconfig=None):
        super().restart(handle_reconfig=handle_reconfig)
        self.reset_driver_and_session()


__all__ = (
    'selenium_test',
    'SeleniumIntegrationTestCase',
)