more tests

Author: yevgenypats

cloudquery/sdk/internal/servers/plugin_v3/plugin.py

@@ -3,7 +3,7 @@
 
 from cloudquery.plugin_v3 import plugin_pb2, plugin_pb2_grpc
 from cloudquery.sdk.message import SyncInsertMessage, SyncMigrateTableMessage
-from cloudquery.sdk.plugin.plugin import Plugin, SyncOptions
+from cloudquery.sdk.plugin.plugin import Plugin, SyncOptions, TableOptions
 from cloudquery.sdk.schema import tables_to_arrow_schemas
 
 
@@ -23,7 +23,7 @@ def Init(self, request: plugin_pb2.Init.Request, context):
         return plugin_pb2.Init.Response()
 
     def GetTables(self, request: plugin_pb2.GetTables.Request, context):
-        tables = self._plugin.get_tables(request.tables, request.skip_tables)
+        tables = self._plugin.get_tables(TableOptions(tables=request.tables, skip_tables=request.skip_tables))
         schema = tables_to_arrow_schemas(tables)
         tablesBytes = []
         for s in schema:

tests/serve/plugin.py

@@ -6,10 +6,11 @@
 from cloudquery.sdk import serve
 from cloudquery.sdk import plugin
 from cloudquery.plugin_v3 import plugin_pb2_grpc, plugin_pb2
+from cloudquery.sdk.internal.memdb import MemDB
 
 
 def test_plugin_serve():
-    p = plugin.Plugin("test", "v1.0.0")
+    p = MemDB()
     cmd = serve.PluginCommand(p)
     port = random.randint(5000, 50000)
     pool = futures.ThreadPoolExecutor(max_workers=1)
@@ -19,10 +20,17 @@ def test_plugin_serve():
       with grpc.insecure_channel(f'localhost:{port}') as channel:
           stub = plugin_pb2_grpc.PluginStub(channel)
           response = stub.GetName(plugin_pb2.GetName.Request())
-          assert response.name == "test"
+          assert response.name == "memdb"
 
           response = stub.GetVersion(plugin_pb2.GetVersion.Request())
-          assert response.version == "v1.0.0"
+          assert response.version == "development"
+
+          response = stub.Init(plugin_pb2.Init.Request(spec=b""))
+          assert response is not None
+
+          response = stub.GetTables(plugin_pb2.GetTables.Request())
+          print(response.tables)
+          assert response.tables is not None
     finally:
       cmd.stop()
       pool.shutdown()