How to use load_html_file method in SeleniumBase

Best Python code snippet using SeleniumBase

test_google.py

Source:test_google.py

...6from mock import Mock7import os8import vcr9BASE_DIR = os.path.dirname(__file__)10def load_html_file(path):11    """Call test with a html file of the same name.12    Args:13        path: Relative path where the html file is located."""14    def test_decorator(fn):15        base_path = os.path.join(os.path.dirname(__file__), path)16        file_name = fn.__name__ + ".html"17        file_path = os.path.join(base_path, file_name)18        html_f = open(file_path, "r")19        def test_decorated(self):20            fn(self, html_f)21        return test_decorated22    return test_decorator23# HELPERS24def get_dir_vcr(name):25    return os.path.join(BASE_DIR, "vcr_cassetes", name)26class GoogleTest(unittest.TestCase):27    @load_html_file("html_files")28    # @unittest.skip("skip")29    def test_search_images(self, html_f):30        """Test method to search images."""31        class MockBrowser(object):32            """Mock browser to replace selenium driver."""33            def __init__(self, html_f):34                self.page_source = html_f.read()35                self.page_source = self.page_source.decode('utf8') if 'decode' in dir(self.page_source) else self.page_source36            def get(self, url):37                pass38            def quit(self):39                pass40        google.images.get_browser_with_url = \41            Mock(return_value=MockBrowser(html_f))42        res = google.search_images("apple", num_images=10)43        self.assertEqual(len(res), 10)44    # @load_html_file("html_files")45    # def test_calculator(self, html_f):46    @unittest.skip("skip")47    def test_calculator(self):48        """Test method to calculate in google."""49        # replace method to get html from a test html file50        # google.calculator.get_html_from_dynamic_site = \51        #     Mock(return_value=html_f.read().decode('utf8'))52        calc = google.calculate("157.3kg in grams")53        self.assertEqual(calc.value, 157300)54    # @load_html_file("html_files")55    @vcr.use_cassette(get_dir_vcr("test_exchange_rate.yaml"))56    def test_exchange_rate(self):57        """Test method to get an exchange rate in google."""58        usd_to_eur = google.exchange_rate("USD", "EUR")59        self.assertGreater(usd_to_eur, 0.0)60    # @load_html_file("html_files")61    @vcr.use_cassette(get_dir_vcr("test_convert_currency.yaml"))62    def test_convert_currency(self):63        """Test method to convert currency in google."""64        euros = google.convert_currency(5.0, "USD", "EUR")65        self.assertGreater(euros, 0.0)66    # @load_html_file("html_files")67    @vcr.use_cassette(get_dir_vcr("test_standard_search.yaml"))68    def test_standard_search(self):69        """Test method to search in google."""70        search = google.search("github")71        self.assertNotEqual(len(search), 0)72    # @load_html_file("html_files")73    @vcr.use_cassette(get_dir_vcr("test_shopping_search.yaml"))74    @unittest.skip("skip")75    def test_shopping_search(self):76        """Test method for google shopping."""77        shop = google.shopping("Disgaea 4")78        self.assertNotEqual(len(shop), 0)79class ConvertCurrencyTest(unittest.TestCase):80    def test_get_currency_req_url(self):81        """Test method to get currency conversion request url."""82        amount = 1083        from_currency = "USD"84        to_currency = "EUR"85        req_url = currency._get_currency_req_url(amount, from_currency,86                                                 to_currency)...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.