summaryrefslogtreecommitdiffstats
path: root/openbb_platform/extensions/tests/utils/helpers.py
diff options
context:
space:
mode:
Diffstat (limited to 'openbb_platform/extensions/tests/utils/helpers.py')
-rw-r--r--openbb_platform/extensions/tests/utils/helpers.py12
1 files changed, 12 insertions, 0 deletions
diff --git a/openbb_platform/extensions/tests/utils/helpers.py b/openbb_platform/extensions/tests/utils/helpers.py
index 14ee9126f78..ac0d7aaaa9d 100644
--- a/openbb_platform/extensions/tests/utils/helpers.py
+++ b/openbb_platform/extensions/tests/utils/helpers.py
@@ -271,3 +271,15 @@ def get_required_fields(model: str) -> List[str]:
"""Get the required fields of a model."""
fields = pi.map[model]["openbb"]["QueryParams"]["fields"]
return [field for field, info in fields.items() if info.is_required()]
+
+
+def get_all_fields(model: str) -> List[str]:
+ """Get all the fields of a model."""
+ all_fields: List[str] = []
+ info = pi.map[model]
+ # for every key, grab the fields
+ for _, provider_info in info.items():
+ for field, _ in provider_info["QueryParams"]["fields"].items():
+ all_fields.append(field)
+
+ return all_fields