laya-ad v2.2: sync laya-server code (default version v2.2, build_push.sh, .env.example)

0cd599e verified 10 days ago

578 Bytes

import time
from contextlib import ExitStack

import pytest
from fastapi.testclient import TestClient


@pytest.fixture
def client_factory():
    with ExitStack() as stack:
        yield lambda *args, **kwargs: stack.enter_context(TestClient(*args, **kwargs))


def wait_for_usage(client, requests):
    deadline = time.monotonic() + 5
    while True:
        usage = client.get('/internal/usage').json()
        if usage['totals']['requests'] == requests:
            return usage
        assert time.monotonic() < deadline, usage
        time.sleep(0.01)

	import time
	from contextlib import ExitStack

	import pytest
	from fastapi.testclient import TestClient


	@pytest.fixture
	def client_factory():
	with ExitStack() as stack:
	yield lambda args, kwargs: stack.enter_context(TestClient(args, **kwargs))


	def wait_for_usage(client, requests):
	deadline = time.monotonic() + 5
	while True:
	usage = client.get('/internal/usage').json()
	if usage['totals']['requests'] == requests:
	return usage
	assert time.monotonic() < deadline, usage
	time.sleep(0.01)