def add_to_cart(self): category_page = selenium.CategoryPage( self.browser, CategoryPage.objects.first().slug, ) category_page.load() category_page.add_to_cart()
def test_add_from_category_page(self): product = Product.objects.first() page = selenium.CategoryPage(self.browser, product.category.page.slug) page.load() card = page.find_card(product.id) page.add_to_cart([card]) self.assert_add(product, 0)
def check_purchase(self): try: with selenium.SiteDriver(site_url=settings.BASE_URL) as driver: category_page = selenium.CategoryPage( driver, CategoryPage.objects.first().slug) category_page.load() category_page.add_to_cart() order_page = selenium.OrderPage(driver) order_page.load() order_page.fill_contacts() order_page.make_order() success_page = selenium.SuccessPage(driver) assert success_page.is_success() except (WebDriverException, AssertionError) as err: if self.request.retries + 1 > self.max_retries: # report on the last attempt TelegramReport().send( f'Can\'t buy a product. Got the error: {err}') raise err