Merge pull request #28 from ydb-platform/snapshot_rw_support

SnapshotRW support

Author: vgvoleg

poetry.lock

@@ -8,7 +8,7 @@ repository = "https://github.com/ydb-platform/ydb-python-dbapi/"
 
 [tool.poetry.dependencies]
 python = "^3.8"
-ydb = "^3.21.6"
+ydb = "^3.23.2"
 
 [tool.poetry.group.dev.dependencies]
 pre-commit = "^3.5.0"

pyproject.toml

@@ -381,6 +381,7 @@ def connection(
             (dbapi.IsolationLevel.ONLINE_READONLY_INCONSISTENT, True),
             (dbapi.IsolationLevel.STALE_READONLY, True),
             (dbapi.IsolationLevel.SNAPSHOT_READONLY, True),
+            (dbapi.IsolationLevel.SNAPSHOT_READWRITE, False),
         ],
     )
     def test_isolation_level_read_only(
@@ -402,6 +403,7 @@ def test_isolation_level_read_only(
             (dbapi.IsolationLevel.ONLINE_READONLY_INCONSISTENT),
             (dbapi.IsolationLevel.STALE_READONLY),
             (dbapi.IsolationLevel.SNAPSHOT_READONLY),
+            (dbapi.IsolationLevel.SNAPSHOT_READWRITE),
         ],
     )
     def test_commit_rollback_after_begin(
@@ -469,6 +471,7 @@ def close() -> None:
             (dbapi.IsolationLevel.ONLINE_READONLY_INCONSISTENT, True),
             (dbapi.IsolationLevel.STALE_READONLY, True),
             (dbapi.IsolationLevel.SNAPSHOT_READONLY, True),
+            (dbapi.IsolationLevel.SNAPSHOT_READWRITE, False),
         ],
     )
     async def test_isolation_level_read_only(
@@ -494,6 +497,7 @@ async def test_isolation_level_read_only(
             (dbapi.IsolationLevel.ONLINE_READONLY_INCONSISTENT),
             (dbapi.IsolationLevel.STALE_READONLY),
             (dbapi.IsolationLevel.SNAPSHOT_READONLY),
+            (dbapi.IsolationLevel.SNAPSHOT_READWRITE),
         ],
     )
     async def test_commit_rollback_after_begin(

tests/test_connections.py

@@ -29,6 +29,7 @@ class IsolationLevel(str, Enum):
     ONLINE_READONLY_INCONSISTENT = "ONLINE READONLY INCONSISTENT"
     STALE_READONLY = "STALE READONLY"
     SNAPSHOT_READONLY = "SNAPSHOT READONLY"
+    SNAPSHOT_READWRITE = "SNAPSHOT READWRITE"
     AUTOCOMMIT = "AUTOCOMMIT"
 
 
@@ -57,6 +58,9 @@ class _IsolationSettings(NamedTuple):
     IsolationLevel.SNAPSHOT_READONLY: _IsolationSettings(
         ydb.QuerySnapshotReadOnly(), interactive=True
     ),
+    IsolationLevel.SNAPSHOT_READWRITE: _IsolationSettings(
+        ydb.QuerySnapshotReadWrite(), interactive=True
+    ),
 }