mirror of
https://github.com/neondatabase/neon.git
synced 2026-01-15 09:22:55 +00:00
78938d1b59
This adds preliminary PG17 support to Neon, based on RC1 / 2024-09-04
07b828e9d4
NOTICE: The data produced by the included version of the PostgreSQL fork
may not be compatible with the future full release of PostgreSQL 17 due to
expected or unexpected future changes in magic numbers and internals.
DO NOT EXPECT DATA IN V17-TENANTS TO BE COMPATIBLE WITH THE 17.0
RELEASE!
Co-authored-by: Anastasia Lubennikova <anastasia@neon.tech>
Co-authored-by: Alexander Bayandin <alexander@neon.tech>
Co-authored-by: Konstantin Knizhnik <knizhnik@neon.tech>
Co-authored-by: Heikki Linnakangas <heikki@neon.tech>
76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
import enum
|
|
import os
|
|
from typing import Optional
|
|
|
|
import pytest
|
|
|
|
"""
|
|
This fixture is used to determine which version of Postgres to use for tests.
|
|
"""
|
|
|
|
|
|
# Inherit PgVersion from str rather than int to make it easier to pass as a command-line argument
|
|
# TODO: use enum.StrEnum for Python >= 3.11
|
|
@enum.unique
|
|
class PgVersion(str, enum.Enum):
|
|
V14 = "14"
|
|
V15 = "15"
|
|
V16 = "16"
|
|
V17 = "17"
|
|
# Instead of making version an optional parameter in methods, we can use this fake entry
|
|
# to explicitly rely on the default server version (could be different from pg_version fixture value)
|
|
NOT_SET = "<-POSTRGRES VERSION IS NOT SET->"
|
|
|
|
# Make it less confusing in logs
|
|
def __repr__(self) -> str:
|
|
return f"'{self.value}'"
|
|
|
|
# Make this explicit for Python 3.11 compatibility, which changes the behavior of enums
|
|
def __str__(self) -> str:
|
|
return self.value
|
|
|
|
# In GitHub workflows we use Postgres version with v-prefix (e.g. v14 instead of just 14),
|
|
# sometime we need to do so in tests.
|
|
@property
|
|
def v_prefixed(self) -> str:
|
|
return f"v{self.value}"
|
|
|
|
@classmethod
|
|
def _missing_(cls, value) -> Optional["PgVersion"]:
|
|
known_values = {v.value for _, v in cls.__members__.items()}
|
|
|
|
# Allow passing version as a string with "v" prefix (e.g. "v14")
|
|
if isinstance(value, str) and value.lower().startswith("v") and value[1:] in known_values:
|
|
return cls(value[1:])
|
|
# Allow passing version as an int (e.g. 15 or 150002, both will be converted to PgVersion.V15)
|
|
elif isinstance(value, int) and str(value)[:2] in known_values:
|
|
return cls(str(value)[:2])
|
|
|
|
# Make mypy happy
|
|
# See https://github.com/python/mypy/issues/3974
|
|
return None
|
|
|
|
|
|
DEFAULT_VERSION: PgVersion = PgVersion.V16
|
|
|
|
|
|
def skip_on_postgres(version: PgVersion, reason: str):
|
|
return pytest.mark.skipif(
|
|
PgVersion(os.environ.get("DEFAULT_PG_VERSION", DEFAULT_VERSION)) is version,
|
|
reason=reason,
|
|
)
|
|
|
|
|
|
def xfail_on_postgres(version: PgVersion, reason: str):
|
|
return pytest.mark.xfail(
|
|
PgVersion(os.environ.get("DEFAULT_PG_VERSION", DEFAULT_VERSION)) is version,
|
|
reason=reason,
|
|
)
|
|
|
|
|
|
def run_only_on_default_postgres(reason: str):
|
|
return pytest.mark.skipif(
|
|
PgVersion(os.environ.get("DEFAULT_PG_VERSION", DEFAULT_VERSION)) is not DEFAULT_VERSION,
|
|
reason=reason,
|
|
)
|