3131 AlreadyExistsException ,
3232 EnvironmentContext ,
3333 FieldSchema ,
34+ GetTableRequest ,
35+ GetTableResult ,
36+ GetTablesRequest ,
37+ GetTablesResult ,
3438 InvalidOperationException ,
3539 LockResponse ,
3640 LockState ,
@@ -293,6 +297,7 @@ def test_create_table(
293297 catalog ._client = MagicMock ()
294298 catalog ._client .__enter__ ().create_table .return_value = None
295299 catalog ._client .__enter__ ().get_table .return_value = hive_table
300+ catalog ._client .__enter__ ().get_table_req .return_value = GetTableResult (table = hive_table )
296301 catalog ._client .__enter__ ().get_database .return_value = hive_database
297302 catalog .create_table (("default" , "table" ), schema = table_schema_with_all_types , properties = {"owner" : "javaberg" })
298303
@@ -472,6 +477,7 @@ def test_create_table_with_given_location_removes_trailing_slash(
472477 catalog ._client = MagicMock ()
473478 catalog ._client .__enter__ ().create_table .return_value = None
474479 catalog ._client .__enter__ ().get_table .return_value = hive_table
480+ catalog ._client .__enter__ ().get_table_req .return_value = GetTableResult (table = hive_table )
475481 catalog ._client .__enter__ ().get_database .return_value = hive_database
476482 catalog .create_table (
477483 ("default" , "table" ), schema = table_schema_with_all_types , properties = {"owner" : "javaberg" }, location = f"{ location } /"
@@ -645,7 +651,7 @@ def test_create_v1_table(table_schema_simple: Schema, hive_database: HiveDatabas
645651
646652 catalog ._client = MagicMock ()
647653 catalog ._client .__enter__ ().create_table .return_value = None
648- catalog ._client .__enter__ ().get_table .return_value = hive_table
654+ catalog ._client .__enter__ ().get_table_req .return_value = GetTableResult ( table = hive_table )
649655 catalog ._client .__enter__ ().get_database .return_value = hive_database
650656 catalog .create_table (
651657 ("default" , "table" ), schema = table_schema_simple , properties = {"owner" : "javaberg" , "format-version" : "1" }
@@ -696,10 +702,10 @@ def test_load_table(hive_table: HiveTable) -> None:
696702 catalog = HiveCatalog (HIVE_CATALOG_NAME , uri = HIVE_METASTORE_FAKE_URL )
697703
698704 catalog ._client = MagicMock ()
699- catalog ._client .__enter__ ().get_table .return_value = hive_table
705+ catalog ._client .__enter__ ().get_table_req .return_value = GetTableResult ( table = hive_table )
700706 table = catalog .load_table (("default" , "new_tabl2e" ))
701707
702- catalog ._client .__enter__ ().get_table .assert_called_with (dbname = "default" , tbl_name = "new_tabl2e" )
708+ catalog ._client .__enter__ ().get_table_req .assert_called_with (GetTableRequest ( dbName = "default" , tblName = "new_tabl2e" ) )
703709
704710 expected = TableMetadataV2 (
705711 location = "s3://bucket/test/location" ,
@@ -796,11 +802,11 @@ def test_load_table_from_self_identifier(hive_table: HiveTable) -> None:
796802 catalog = HiveCatalog (HIVE_CATALOG_NAME , uri = HIVE_METASTORE_FAKE_URL )
797803
798804 catalog ._client = MagicMock ()
799- catalog ._client .__enter__ ().get_table .return_value = hive_table
805+ catalog ._client .__enter__ ().get_table_req .return_value = GetTableResult ( table = hive_table )
800806 intermediate = catalog .load_table (("default" , "new_tabl2e" ))
801807 table = catalog .load_table (intermediate .name ())
802808
803- catalog ._client .__enter__ ().get_table .assert_called_with (dbname = "default" , tbl_name = "new_tabl2e" )
809+ catalog ._client .__enter__ ().get_table_req .assert_called_with (GetTableRequest ( dbName = "default" , tblName = "new_tabl2e" ) )
804810
805811 expected = TableMetadataV2 (
806812 location = "s3://bucket/test/location" ,
@@ -902,7 +908,10 @@ def test_rename_table(hive_table: HiveTable) -> None:
902908 renamed_table .tableName = "new_tabl3e"
903909
904910 catalog ._client = MagicMock ()
905- catalog ._client .__enter__ ().get_table .side_effect = [hive_table , renamed_table ]
911+ catalog ._client .__enter__ ().get_table_req .side_effect = [
912+ GetTableResult (table = hive_table ),
913+ GetTableResult (table = renamed_table ),
914+ ]
906915 catalog ._client .__enter__ ().alter_table_with_environment_context .return_value = None
907916
908917 from_identifier = ("default" , "new_tabl2e" )
@@ -911,8 +920,11 @@ def test_rename_table(hive_table: HiveTable) -> None:
911920
912921 assert table .name () == to_identifier
913922
914- calls = [call (dbname = "default" , tbl_name = "new_tabl2e" ), call (dbname = "default" , tbl_name = "new_tabl3e" )]
915- catalog ._client .__enter__ ().get_table .assert_has_calls (calls )
923+ calls = [
924+ call (GetTableRequest (dbName = "default" , tblName = "new_tabl2e" )),
925+ call (GetTableRequest (dbName = "default" , tblName = "new_tabl3e" )),
926+ ]
927+ catalog ._client .__enter__ ().get_table_req .assert_has_calls (calls )
916928 catalog ._client .__enter__ ().alter_table_with_environment_context .assert_called_with (
917929 dbname = "default" ,
918930 tbl_name = "new_tabl2e" ,
@@ -926,25 +938,31 @@ def test_rename_table_from_self_identifier(hive_table: HiveTable) -> None:
926938 catalog .table_exists = MagicMock (return_value = False ) # type: ignore[method-assign]
927939
928940 catalog ._client = MagicMock ()
929- catalog ._client .__enter__ ().get_table .return_value = hive_table
941+ catalog ._client .__enter__ ().get_table_req .return_value = GetTableResult ( table = hive_table )
930942
931943 from_identifier = ("default" , "new_tabl2e" )
932944 from_table = catalog .load_table (from_identifier )
933- catalog ._client .__enter__ ().get_table .assert_called_with (dbname = "default" , tbl_name = "new_tabl2e" )
945+ catalog ._client .__enter__ ().get_table_req .assert_called_with (GetTableRequest ( dbName = "default" , tblName = "new_tabl2e" ) )
934946
935947 renamed_table = copy .deepcopy (hive_table )
936948 renamed_table .dbName = "default"
937949 renamed_table .tableName = "new_tabl3e"
938950
939- catalog ._client .__enter__ ().get_table .side_effect = [hive_table , renamed_table ]
951+ catalog ._client .__enter__ ().get_table_req .side_effect = [
952+ GetTableResult (table = hive_table ),
953+ GetTableResult (table = renamed_table ),
954+ ]
940955 catalog ._client .__enter__ ().alter_table_with_environment_context .return_value = None
941956 to_identifier = ("default" , "new_tabl3e" )
942957 table = catalog .rename_table (from_table .name (), to_identifier )
943958
944959 assert table .name () == to_identifier
945960
946- calls = [call (dbname = "default" , tbl_name = "new_tabl2e" ), call (dbname = "default" , tbl_name = "new_tabl3e" )]
947- catalog ._client .__enter__ ().get_table .assert_has_calls (calls )
961+ calls = [
962+ call (GetTableRequest (dbName = "default" , tblName = "new_tabl2e" )),
963+ call (GetTableRequest (dbName = "default" , tblName = "new_tabl3e" )),
964+ ]
965+ catalog ._client .__enter__ ().get_table_req .assert_has_calls (calls )
948966 catalog ._client .__enter__ ().alter_table_with_environment_context .assert_called_with (
949967 dbname = "default" ,
950968 tbl_name = "new_tabl2e" ,
@@ -1042,13 +1060,13 @@ def test_list_tables(hive_table: HiveTable) -> None:
10421060
10431061 catalog ._client = MagicMock ()
10441062 catalog ._client .__enter__ ().get_all_tables .return_value = ["table1" , "table2" , "table3" , "table4" ]
1045- catalog ._client .__enter__ ().get_table_objects_by_name .return_value = [tbl1 , tbl2 , tbl3 , tbl4 ]
1063+ catalog ._client .__enter__ ().get_table_objects_by_name_req .return_value = GetTablesResult ( tables = [tbl1 , tbl2 , tbl3 , tbl4 ])
10461064
10471065 got_tables = catalog .list_tables ("database" )
10481066 assert got_tables == [("database" , "table1" ), ("database" , "table2" )]
10491067 catalog ._client .__enter__ ().get_all_tables .assert_called_with (db_name = "database" )
1050- catalog ._client .__enter__ ().get_table_objects_by_name .assert_called_with (
1051- dbname = "database" , tbl_names = ["table1" , "table2" , "table3" , "table4" ]
1068+ catalog ._client .__enter__ ().get_table_objects_by_name_req .assert_called_with (
1069+ GetTablesRequest ( dbName = "database" , tblNames = ["table1" , "table2" , "table3" , "table4" ])
10521070 )
10531071
10541072
@@ -1078,7 +1096,7 @@ def test_drop_table_from_self_identifier(hive_table: HiveTable) -> None:
10781096 catalog = HiveCatalog (HIVE_CATALOG_NAME , uri = HIVE_METASTORE_FAKE_URL )
10791097
10801098 catalog ._client = MagicMock ()
1081- catalog ._client .__enter__ ().get_table .return_value = hive_table
1099+ catalog ._client .__enter__ ().get_table_req .return_value = GetTableResult ( table = hive_table )
10821100 table = catalog .load_table (("default" , "new_tabl2e" ))
10831101
10841102 catalog ._client .__enter__ ().get_all_databases .return_value = ["namespace1" , "namespace2" ]
0 commit comments