more tests

Author: yevgenypats

cloudquery/sdk/internal/memdb/memdb.py

@@ -11,7 +11,9 @@
 class MemDB(plugin.Plugin):
     def __init__(self) -> None:
         super().__init__(NAME, VERSION)
-        self._tables: List[schema.Table] = []
+        self._tables: List[schema.Table] = [
+            schema.Table("test_table", [schema.Column("test_column", pa.int64())])
+        ]
         self._memory_db: Dict[str, pa.record] = {
             "test_table": pa.record_batch([pa.array([1, 2, 3])], names=["test_column"])
         }

cloudquery/sdk/py.typed

@@ -1,7 +1,8 @@
 grpcio==1.56.2
 grpcio-tools==1.56.2
-plugin-pb-python==0.0.12
+plugin-pb-python==0.0.13
 protobuf==4.23.4
 pyarrow==12.0.1
 pytest==7.4.0
-Jinja2~=3.1.2
+Jinja2~=3.1.2
+structlog==23.1.0

requirements.txt

@@ -10,9 +10,10 @@
 description = "CloudQuery Plugin SDK for Python"
 
 dependencies = [
-    "plugin-pb-python==0.0.12",
+    "plugin-pb-python==0.0.13",
     "pyarrow==12.0.1",
     "Jinja2==3.1.2",
+    "structlog==23.1.0"
 ]
 url = "https://github.com/cloudquery/plugin-sdk-python"
 
@@ -59,10 +60,9 @@
     platforms="Posix; MacOS X; Windows",
     packages=packages,
     python_requires=">=3.7",
-    # namespace_packages=["cloudquery/sdk"],
-    # namespace_packages=namespaces,
+    namespace_packages=["cloudquery"],
     install_requires=dependencies,
     include_package_data=True,
-    package_data={"cloudquery": ["plugin_v3/py.typed", "plugin_v3/*.pyi"]},
+    package_data={"cloudquery": ["sdk/py.typed"]},
     zip_safe=False,
 )

setup.py

@@ -3,8 +3,7 @@
 import time
 from concurrent import futures
 from cloudquery.sdk import serve
-from cloudquery.sdk import plugin
-from cloudquery.plugin_v3 import plugin_pb2_grpc, plugin_pb2
+from cloudquery.plugin_v3 import plugin_pb2_grpc, plugin_pb2, arrow
 from cloudquery.sdk.internal.memdb import MemDB
 
 
@@ -28,8 +27,8 @@ def test_plugin_serve():
             assert response is not None
 
             response = stub.GetTables(plugin_pb2.GetTables.Request())
-            print(response.tables)
-            assert response.tables is not None
+            schemas = arrow.new_schemas_from_bytes(response.tables)
+            assert len(schemas) == 1
     finally:
         cmd.stop()
         pool.shutdown()