style: Format

Author: erezrokah

cloudquery/sdk/internal/memdb/memdb.py

@@ -211,7 +211,9 @@ def write(self, writer: Generator[message.WriteMessage, None, None]) -> None:
 
     def read(self, table: Table) -> Generator[message.ReadMessage, None, None]:
         for record in self._db:
-            recordMetadata = record.schema.metadata.get(METADATA_TABLE_NAME).decode("utf-8")
+            recordMetadata = record.schema.metadata.get(METADATA_TABLE_NAME).decode(
+                "utf-8"
+            )
             if recordMetadata == table.name:
                 yield message.ReadMessage(record)
 

tests/internal/memdb/memdb.py

@@ -10,7 +10,7 @@ def test_memdb():
     p.set_logger(structlog.get_logger())
     p.init(plugin.sanitize_spec(b"null"))
     msgs = []
-    for msg in p.sync(SyncOptions(tables=["*"],skip_tables=[])):
+    for msg in p.sync(SyncOptions(tables=["*"], skip_tables=[])):
         msgs.append(msg)
     assert len(msgs) == 18
 
@@ -21,5 +21,3 @@ def test_memdb():
     # other messages should be inserts
     for msg in msgs[3:]:
         assert isinstance(msg, SyncInsertMessage)
-
-

tests/serve/plugin.py

@@ -14,9 +14,15 @@
 from cloudquery.sdk.types.uuid import UUIDType
 
 test_table = Table(
-        "test_table",
-        [Column("id", pa.int64()), Column("name", pa.string()), Column("json", JSONType()), Column("uuid", UUIDType())],
-    )
+    "test_table",
+    [
+        Column("id", pa.int64()),
+        Column("name", pa.string()),
+        Column("json", JSONType()),
+        Column("uuid", UUIDType()),
+    ],
+)
+
 
 def test_plugin_serve():
     p = MemDB()
@@ -66,11 +72,13 @@ def writer_iterator():
 
             stub.Write(writer_iterator())
 
-            response = stub.GetTables(plugin_pb2.GetTables.Request(tables=["*"],skip_tables=[]))
+            response = stub.GetTables(
+                plugin_pb2.GetTables.Request(tables=["*"], skip_tables=[])
+            )
             schemas = arrow.new_schemas_from_bytes(response.tables)
             assert len(schemas) == 3
 
-            response = stub.Sync(plugin_pb2.Sync.Request(tables=["*"],skip_tables=[]))
+            response = stub.Sync(plugin_pb2.Sync.Request(tables=["*"], skip_tables=[]))
             total_migrate_tables = 0
             total_records = 0
             total_errors = 0