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