mirror of
https://github.com/invariantlabs-ai/invariant-gateway.git
synced 2026-05-15 20:48:17 +02:00
Hemang
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
"""Util functions for tests"""
|
|
|
|
import os
|
|
|
|
import pytest
|
|
from playwright.async_api import async_playwright
|
|
|
|
|
|
@pytest.fixture
|
|
def gateway_url():
|
|
"""Get the gateway URL from the environment variable"""
|
|
if "INVARIANT_GATEWAY_API_URL" in os.environ:
|
|
return os.environ["INVARIANT_GATEWAY_API_URL"]
|
|
raise ValueError("Please set the INVARIANT_GATEWAY_API_URL environment variable")
|
|
|
|
|
|
@pytest.fixture
|
|
def explorer_api_url():
|
|
"""Get the explorer API URL from the environment variable"""
|
|
if "INVARIANT_API_URL" in os.environ:
|
|
return os.environ["INVARIANT_API_URL"]
|
|
raise ValueError("Please set the INVARIANT_API_URL environment variable")
|
|
|
|
|
|
@pytest.fixture
|
|
async def playwright(scope="session"):
|
|
"""Fixture to create a Playwright instance"""
|
|
async with async_playwright() as playwright_instance:
|
|
yield playwright_instance
|
|
|
|
|
|
@pytest.fixture
|
|
async def browser(playwright, scope="session"):
|
|
"""Fixture to create a browser instance"""
|
|
firefox_browser = await playwright.firefox.launch(headless=True)
|
|
yield firefox_browser
|
|
await firefox_browser.close()
|
|
|
|
|
|
@pytest.fixture
|
|
async def context(browser):
|
|
"""Fixture to create a browser context"""
|
|
browser_context = await browser.new_context(ignore_https_errors=True)
|
|
yield browser_context
|
|
await browser_context.close()
|