Merge pull request feast-dev#63 from lokeshrangineni/feature/fix-offl…

…inestore-test-perm-issue
lokeshrangineni · Aug 6, 2024 · 7446cdd · 7446cdd
2 parents 377fce1 + a21fe89
commit 7446cdd
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/sdk/python/tests/integration/feature_repos/repo_configuration.py b/sdk/python/tests/integration/feature_repos/repo_configuration.py
@@ -488,36 +488,36 @@ def setup(self):
             Permission(
                 name="offline_permissions_perm",
                 types=Permission,
-                policy=RoleBasedPolicy(roles=["reader"]),
-                actions=[AuthzedAction.QUERY_OFFLINE],
+                policy=RoleBasedPolicy(roles=["reader", "writer"]),
+                actions=[AuthzedAction.QUERY_OFFLINE, AuthzedAction.WRITE_OFFLINE],
             ),
             Permission(
                 name="offline_entities_perm",
                 types=Entity,
                 with_subclasses=False,
-                policy=RoleBasedPolicy(roles=["reader"]),
-                actions=[AuthzedAction.QUERY_OFFLINE],
+                policy=RoleBasedPolicy(roles=["reader", "writer"]),
+                actions=[AuthzedAction.QUERY_OFFLINE, AuthzedAction.WRITE_OFFLINE],
             ),
             Permission(
                 name="offline_fv_perm",
                 types=FeatureView,
                 with_subclasses=False,
-                policy=RoleBasedPolicy(roles=["reader"]),
-                actions=[AuthzedAction.QUERY_OFFLINE],
+                policy=RoleBasedPolicy(roles=["reader", "writer"]),
+                actions=[AuthzedAction.QUERY_OFFLINE, AuthzedAction.WRITE_OFFLINE],
             ),
             Permission(
                 name="offline_odfv_perm",
                 types=OnDemandFeatureView,
                 with_subclasses=False,
-                policy=RoleBasedPolicy(roles=["reader"]),
-                actions=[AuthzedAction.QUERY_OFFLINE],
+                policy=RoleBasedPolicy(roles=["reader", "writer"]),
+                actions=[AuthzedAction.QUERY_OFFLINE, AuthzedAction.WRITE_OFFLINE],
             ),
             Permission(
                 name="offline_sfv_perm",
                 types=StreamFeatureView,
                 with_subclasses=False,
-                policy=RoleBasedPolicy(roles=["reader"]),
-                actions=[AuthzedAction.QUERY_OFFLINE],
+                policy=RoleBasedPolicy(roles=["reader", "writer"]),
+                actions=[AuthzedAction.QUERY_OFFLINE, AuthzedAction.WRITE_OFFLINE],
             ),
         ]
         self.feature_store.apply(permissions_list)