File tree 1 file changed +7
-9
lines changed
1 file changed +7
-9
lines changed Original file line number Diff line number Diff line change @@ -66,20 +66,18 @@ def test_generic_get_methods(rester):
66
66
)
67
67
68
68
if name not in ignore_generic :
69
+ key = rester .primary_key
69
70
if name not in key_only_resters :
70
- doc = rester ._query_resource_data (
71
- {"_limit" : 1 }, fields = [rester .primary_key ]
72
- )[0 ]
71
+ if key not in rester .available_fields :
72
+ key = rester .available_fields [0 ]
73
+
74
+ doc = rester ._query_resource_data ({"_limit" : 1 }, fields = [key ])[0 ]
73
75
assert isinstance (doc , rester .document_model )
74
76
75
77
if name not in search_only_resters :
76
- doc = rester .get_data_by_id (
77
- doc .model_dump ()[rester .primary_key ], fields = [rester .primary_key ]
78
- )
78
+ doc = rester .get_data_by_id (doc .model_dump ()[key ], fields = [key ])
79
79
assert isinstance (doc , rester .document_model )
80
80
81
81
elif name not in special_resters :
82
- doc = rester .get_data_by_id (
83
- key_only_resters [name ], fields = [rester .primary_key ]
84
- )
82
+ doc = rester .get_data_by_id (key_only_resters [name ], fields = [key ])
85
83
assert isinstance (doc , rester .document_model )
You can’t perform that action at this time.
0 commit comments