Skip to content

Commit 534266c

Browse files
committed
tests: use first available field if primary key n/a
1 parent 88d6b43 commit 534266c

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

tests/test_client.py

+7-9
Original file line numberDiff line numberDiff line change
@@ -66,20 +66,18 @@ def test_generic_get_methods(rester):
6666
)
6767

6868
if name not in ignore_generic:
69+
key = rester.primary_key
6970
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]
7375
assert isinstance(doc, rester.document_model)
7476

7577
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])
7979
assert isinstance(doc, rester.document_model)
8080

8181
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])
8583
assert isinstance(doc, rester.document_model)

0 commit comments

Comments
 (0)