[Projects] Support owner filter in follower mode (#1348)

mlrun · Sep 24, 2021 · 536ecd4 · 536ecd4
1 parent d51ab29
commit 536ecd4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/mlrun/api/utils/projects/follower.py b/mlrun/api/utils/projects/follower.py
@@ -356,8 +356,8 @@ def _filter_projects(
                 project for project in projects if project.metadata.name in names
             ]
         if owner:
-            raise NotImplementedError(
-                "Filtering projects by owner is currently not supported in follower mode"
+            projects = list(
+                filter(lambda project: project.spec.owner == owner, projects)
             )
         if state:
             projects = list(

diff --git a/tests/api/utils/projects/test_follower_member.py b/tests/api/utils/projects/test_follower_member.py
@@ -192,11 +192,13 @@ def test_list_project(
     projects_follower: mlrun.api.utils.projects.follower.Member,
     nop_leader: mlrun.api.utils.projects.remotes.leader.Member,
 ):
-    project = _generate_project(name="name-1")
+    owner = "project-owner"
+    project = _generate_project(name="name-1", owner=owner)
     archived_project = _generate_project(
         name="name-2",
         desired_state=mlrun.api.schemas.ProjectDesiredState.archived,
         state=mlrun.api.schemas.ProjectState.archived,
+        owner=owner,
     )
     label_key = "key"
     label_value = "value"
@@ -230,6 +232,11 @@ def test_list_project(
         state=mlrun.api.schemas.ProjectState.archived,
     )
 
+    # list by owner
+    _assert_list_projects(
+        projects_follower, [project, archived_project], owner=owner,
+    )
+
     # list specific names only
     _assert_list_projects(
         projects_follower,