fix: Fix column resolver resource set (#23)

Author: hermanschaaf

cloudquery/sdk/scheduler/table_resolver.py

@@ -1,7 +1,8 @@
-from cloudquery.sdk.schema.table import Table
-from cloudquery.sdk.schema import Resource
 from typing import Any, Generator
 
+from cloudquery.sdk.schema import Resource
+from cloudquery.sdk.schema.table import Table
+
 
 class TableResolver:
     def __init__(self, table: Table, child_resolvers=[]) -> None:
@@ -31,7 +32,7 @@ def resolve_column(self, client, resource: Resource, column_name: str):
                 resource.set(column_name, resource.item[column_name])
         else:
             if hasattr(resource.item, column_name):
-                resource.set(column_name, resource.item[column_name])
+                resource.set(column_name, getattr(resource.item, column_name))
 
     def post_resource_resolve(self, client, resource):
         return

tests/scheduler/table_resolver.py

@@ -0,0 +1,34 @@
+from dataclasses import dataclass
+
+import pyarrow as pa
+
+from cloudquery.sdk.scheduler import TableResolver
+from cloudquery.sdk.schema import Table, Resource, Column
+
+
+@dataclass
+class TestItem:
+    test_column2: int = 1
+
+
+def test_table_resolver_resolve_column():
+    test_table = Table(
+        name="test_table",
+        columns=[
+            Column(name="test_column", type=pa.int64()),
+            Column(name="test_column2", type=pa.int64()),
+        ],
+    )
+    resource_dict = Resource(table=test_table, parent=None, item={"test_column": 1})
+    resource_obj = Resource(
+        table=test_table, parent=None, item=TestItem(test_column2=2)
+    )
+    test_resolver = TableResolver(table=test_table, child_resolvers=[])
+    test_resolver.resolve_column(
+        client=None, resource=resource_dict, column_name="test_column"
+    )
+    test_resolver.resolve_column(
+        client=None, resource=resource_obj, column_name="test_column2"
+    )
+    assert resource_dict.to_list_of_arr() == [[1], [None]]
+    assert resource_obj.to_list_of_arr() == [[None], [2]]