mirror of
https://github.com/PostHog/posthog.git
synced 2024-11-24 09:14:46 +01:00
304540f0ce
* Add autoretry for too many simultaneous async queries Fixes POSTHOG-DNR * Use for update_cache Fixes POSTHOG-DNR * Use for export_asset Fixes POSTHOG-TMV
51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
import pytest
|
|
from clickhouse_driver.errors import ServerException
|
|
|
|
from posthog.errors import wrap_query_error
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"error,expected_type,expected_message,expected_code",
|
|
[
|
|
(AttributeError("Foobar"), "AttributeError", "Foobar", None),
|
|
(
|
|
ServerException("Estimated query execution time (34.5 seconds) is too long. Aborting query"),
|
|
"EstimatedQueryExecutionTimeTooLong",
|
|
"Estimated query execution time (34.5 seconds) is too long. Try reducing its scope by changing the time range.",
|
|
None,
|
|
),
|
|
(
|
|
ServerException("Syntax error", code=62),
|
|
"CHQueryErrorSyntaxError",
|
|
"Code: 62.\nSyntax error",
|
|
62,
|
|
),
|
|
(
|
|
ServerException("Syntax error", code=9999),
|
|
"CHQueryErrorUnknownException",
|
|
"Code: 9999.\nSyntax error",
|
|
9999,
|
|
),
|
|
(
|
|
ServerException(
|
|
"Memory limit (for query) exceeded: would use 42.00 GiB (attempt to allocate chunk of 16757643 bytes), maximum: 42.00 GiB.",
|
|
code=241,
|
|
),
|
|
"CHQueryErrorMemoryLimitExceeded",
|
|
"Query exceeds memory limits. Try reducing its scope by changing the time range.",
|
|
241,
|
|
),
|
|
(
|
|
ServerException("Too many simultaneous queries. Maximum: 100.", code=202),
|
|
"CHQueryErrorTooManySimultaneousQueries",
|
|
"Code: 202.\nToo many simultaneous queries. Try again later.",
|
|
202,
|
|
),
|
|
],
|
|
)
|
|
def test_wrap_query_error(error, expected_type, expected_message, expected_code):
|
|
new_error = wrap_query_error(error)
|
|
assert type(new_error).__name__ == expected_type
|
|
assert str(new_error) == expected_message
|
|
assert getattr(new_error, "code", None) == expected_code
|