mirror of
https://github.com/neondatabase/neon.git
synced 2025-12-22 21:59:59 +00:00
30a7dd630c
## Problem `TYPE_CHECKING` is used inconsistently across Python tests. ## Summary of changes - Update `ruff`: 0.7.0 -> 0.11.2 - Enable TC (flake8-type-checking): https://docs.astral.sh/ruff/rules/#flake8-type-checking-tc - (auto)fix all new issues
49 lines
1.7 KiB
Python
49 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
import pytest
|
|
from fixtures.pg_version import PgVersion
|
|
from fixtures.utils import WITH_SANITIZERS, run_only_on_postgres
|
|
|
|
if TYPE_CHECKING:
|
|
from fixtures.neon_fixtures import NeonEnvBuilder
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"sql_func",
|
|
[
|
|
"trigger_panic",
|
|
"trigger_segfault",
|
|
"💣", # calls `trigger_segfault` internally
|
|
],
|
|
)
|
|
def test_endpoint_crash(neon_env_builder: NeonEnvBuilder, sql_func: str):
|
|
"""
|
|
Test that triggering crash from neon_test_utils crashes the endpoint
|
|
"""
|
|
env = neon_env_builder.init_start()
|
|
env.create_branch("test_endpoint_crash")
|
|
endpoint = env.endpoints.create_start("test_endpoint_crash")
|
|
|
|
endpoint.safe_psql("CREATE EXTENSION neon_test_utils;")
|
|
with pytest.raises(Exception, match="This probably means the server terminated abnormally"):
|
|
endpoint.safe_psql(f"SELECT {sql_func}();")
|
|
|
|
|
|
@run_only_on_postgres([PgVersion.V17], "Currently, build vith sanitizers is possible with v17 only")
|
|
def test_sanitizers(neon_env_builder: NeonEnvBuilder):
|
|
"""
|
|
Test that undefined behavior leads to endpoint abort with sanitizers enabled
|
|
"""
|
|
env = neon_env_builder.init_start()
|
|
env.create_branch("test_ubsan")
|
|
endpoint = env.endpoints.create_start("test_ubsan")
|
|
|
|
# Test case based on https://www.postgresql.org/message-id/17167-028026e4ca333817@postgresql.org
|
|
if not WITH_SANITIZERS:
|
|
endpoint.safe_psql("SELECT 1::int4 << 128")
|
|
else:
|
|
with pytest.raises(Exception, match="This probably means the server terminated abnormally"):
|
|
endpoint.safe_psql("SELECT 1::int4 << 128")
|