Add list validation references

Signed-off-by: Amom Mendes <[email protected]>
feast-dev · feast-ci-bot · Jan 8, 2023 · Dec 31, 2022 · Jan 1, 2023 · Jan 7, 2023
commit ba4df5cafa904e4a6fdacb2f1db064d4e8a9d67c
@@ -115,6 +115,10 @@ def get_validation_reference(
             return ValidationReference.from_proto(validation_reference)
     raise ValidationReferenceNotFound(name, project=project)
 
+def list_validation_references(
+    registry_proto: RegistryProto
+):
+    return registry_proto.validation_references
 
 def list_feature_services(
     registry_proto: RegistryProto, project: str, allow_cache: bool = False

@@ -740,6 +740,16 @@ def get_validation_reference(
             registry_proto, name, project
         )
 
+    def list_validation_references(
+            self, project: str, allow_cache: bool = False
+    ) -> List[ValidationReference]:
+        registry_proto = self._get_registry_proto(
+            project=project, allow_cache=allow_cache
+        )
+        return proto_registry_utils.list_validation_references(
+            registry_proto
+        )
+
     def delete_validation_reference(self, name: str, project: str, commit: bool = True):
         registry_proto = self._prepare_registry_for_changes(project)
         for idx, existing_validation_reference in enumerate(

@@ -408,6 +408,22 @@ def get_validation_reference(
             not_found_exception=ValidationReferenceNotFound,
         )
 
+    def list_validation_references(
+        self, project: str, allow_cache: bool = False
+    ) -> List[ValidationReference]:
+        if allow_cache:
+            self._refresh_cached_registry_if_necessary()
+            return proto_registry_utils.list_validation_references(
+                self.cached_registry_proto
+            )
+        return self._list_objects(
+            table=validation_references,
+            project=project,
+            proto_class=ValidationReferenceProto,
+            python_class=ValidationReference,
+            proto_field_name="validation_reference_proto"
+        )
+
     def list_entities(self, project: str, allow_cache: bool = False) -> List[Entity]:
         if allow_cache:
             self._refresh_cached_registry_if_necessary()

@@ -21,7 +21,7 @@ def test_e2e_local() -> None:
     """
     Tests the end-to-end workflow of apply, materialize, and online retrieval.
 
-    This test runs against several different types of repos:
+    This test runs against several types of repos:
     1. A repo with a normal FV and an entity-less FV.
     2. A repo using the SDK from version 0.19.0.
     3. A repo with a FV with a ttl of 0.