add tests

Author: vgvoleg

tests/aio/test_table_client.py

@@ -0,0 +1,88 @@
+import pytest
+import ydb
+
+
+class TestTableClient:
+    @pytest.mark.asyncio
+    async def test_create_table(self, driver: ydb.aio.Driver):
+        client = driver.table_client
+        table_name = "/local/testtableclient"
+        try:
+            await client.drop_table(table_name)
+        except ydb.SchemeError:
+            pass
+
+        with pytest.raises(ydb.SchemeError):
+            await client.describe_table(table_name)
+
+        description = (
+            ydb.TableDescription()
+            .with_primary_keys("key1", "key2")
+            .with_columns(
+                ydb.Column("key1", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
+                ydb.Column("key2", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
+                ydb.Column("value", ydb.OptionalType(ydb.PrimitiveType.Utf8)),
+            )
+        )
+
+        await client.create_table(table_name, description)
+
+        actual_description = await client.describe_table(table_name)
+
+        assert actual_description.columns == description.columns
+
+    @pytest.mark.asyncio
+    async def test_alter_table(self, driver: ydb.aio.Driver):
+        client = driver.table_client
+
+        table_name = "/local/testtableclient"
+        try:
+            await client.drop_table(table_name)
+        except ydb.SchemeError:
+            pass
+
+        description = (
+            ydb.TableDescription()
+            .with_primary_keys("key1", "key2")
+            .with_columns(
+                ydb.Column("key1", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
+                ydb.Column("key2", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
+                ydb.Column("value", ydb.OptionalType(ydb.PrimitiveType.Utf8)),
+            )
+        )
+
+        await client.create_table(table_name, description)
+
+        await client.alter_table(table_name, add_columns=[
+            ydb.Column("value2", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
+        ])
+
+        description = await client.describe_table(table_name)
+        assert len(description.columns) == 4
+
+    @pytest.mark.asyncio
+    async def test_copy_table(self, driver: ydb.aio.Driver):
+        client = driver.table_client
+        table_name = "/local/testtableclient"
+        try:
+            await client.drop_table(table_name)
+        except ydb.SchemeError:
+            pass
+
+        description = (
+            ydb.TableDescription()
+            .with_primary_keys("key1", "key2")
+            .with_columns(
+                ydb.Column("key1", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
+                ydb.Column("key2", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
+                ydb.Column("value", ydb.OptionalType(ydb.PrimitiveType.Utf8)),
+            )
+        )
+
+        await client.create_table(table_name, description)
+
+        await client.copy_table(table_name, table_name + "_copy")
+
+        copied_description = await client.describe_table(table_name + "_copy")
+
+        assert description.columns == copied_description.columns

tests/table/test_table_client.py

@@ -0,0 +1,85 @@
+import pytest
+import ydb
+
+
+class TestTableClient:
+    def test_create_table(self, driver_sync: ydb.Driver):
+        client = driver_sync.table_client
+        table_name = "/local/testtableclient"
+        try:
+            client.drop_table(table_name)
+        except ydb.SchemeError:
+            pass
+
+        with pytest.raises(ydb.SchemeError):
+            client.describe_table(table_name)
+
+        description = (
+            ydb.TableDescription()
+            .with_primary_keys("key1", "key2")
+            .with_columns(
+                ydb.Column("key1", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
+                ydb.Column("key2", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
+                ydb.Column("value", ydb.OptionalType(ydb.PrimitiveType.Utf8)),
+            )
+        )
+
+        client.create_table(table_name, description)
+
+        actual_description = client.describe_table(table_name)
+
+        assert actual_description.columns == description.columns
+
+    def test_alter_table(self, driver_sync: ydb.Driver):
+        client = driver_sync.table_client
+
+        table_name = "/local/testtableclient"
+        try:
+            client.drop_table(table_name)
+        except ydb.SchemeError:
+            pass
+
+        description = (
+            ydb.TableDescription()
+            .with_primary_keys("key1", "key2")
+            .with_columns(
+                ydb.Column("key1", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
+                ydb.Column("key2", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
+                ydb.Column("value", ydb.OptionalType(ydb.PrimitiveType.Utf8)),
+            )
+        )
+
+        client.create_table(table_name, description)
+
+        client.alter_table(table_name, add_columns=[
+            ydb.Column("value2", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
+        ])
+
+        description = client.describe_table(table_name)
+        assert len(description.columns) == 4
+
+    def test_copy_table(self, driver_sync: ydb.Driver):
+        client = driver_sync.table_client
+        table_name = "/local/testtableclient"
+        try:
+            client.drop_table(table_name)
+        except ydb.SchemeError:
+            pass
+
+        description = (
+            ydb.TableDescription()
+            .with_primary_keys("key1", "key2")
+            .with_columns(
+                ydb.Column("key1", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
+                ydb.Column("key2", ydb.OptionalType(ydb.PrimitiveType.Uint64)),
+                ydb.Column("value", ydb.OptionalType(ydb.PrimitiveType.Utf8)),
+            )
+        )
+
+        client.create_table(table_name, description)
+
+        client.copy_table(table_name, table_name + "_copy")
+
+        copied_description = client.describe_table(table_name + "_copy")
+
+        assert description.columns == copied_description.columns