From 95a2f46d28bffe1421b193ebc7520166de50f487 Mon Sep 17 00:00:00 2001
From: Noah Metz <noah@metznet.ca>
Date: Thu, 27 Jul 2023 10:04:39 -0600
Subject: [PATCH] Fixed policies in existing test

---
 gql_test.go | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/gql_test.go b/gql_test.go
index ad60782..d9b862b 100644
--- a/gql_test.go
+++ b/gql_test.go
@@ -16,7 +16,7 @@ func TestGQL(t *testing.T) {
 }
 
 func TestGQLDB(t * testing.T) {
-  ctx := logTestContext(t, []string{"test", "signal"})
+  ctx := logTestContext(t, []string{"test", "signal", "policy"})
 
   TestUserNodeType := NodeType("TEST_USER")
   err := ctx.RegisterNodeType(TestUserNodeType, []ExtType{ACLExtType})
@@ -40,9 +40,13 @@ func TestGQLDB(t * testing.T) {
   fatalErr(t, err)
 
   t1 := NewNode(ctx, RandID(), TestThreadNodeType)
-  t1_policy := NewParentOfPolicy(Actions{"signal.abort", "state.write"})
+  t1_policy_1 := NewParentOfPolicy(Actions{"signal.abort", "state.write"})
+  t1_policy_2 := NewPerNodePolicy(NodeActions{
+    u1.ID: Actions{"parent.write"},
+  })
   t1.Extensions[ACLExtType] = NewACLExt(map[PolicyType]Policy{
-    ParentOfPolicyType: &t1_policy,
+    ParentOfPolicyType: &t1_policy_1,
+    PerNodePolicyType: &t1_policy_2,
   })
   t1.Extensions[ThreadExtType], err = NewThreadExt(ctx, BaseThreadType, nil, nil, "init", nil)
   fatalErr(t, err)
@@ -55,9 +59,13 @@ func TestGQLDB(t * testing.T) {
 
   gql, err := NewGQLNode(ctx, NewGQLExt(":0", ecdh.P256(), key, nil, nil))
   fatalErr(t, err)
-  gql_policy := NewChildOfPolicy(Actions{"signal.status"})
+  gql_policy_1 := NewChildOfPolicy(Actions{"signal.status"})
+  gql_policy_2 := NewPerNodePolicy(NodeActions{
+    u1.ID: Actions{"parent.write", "children.write", "dependencies.write"},
+  })
   gql.Extensions[ACLExtType] = NewACLExt(map[PolicyType]Policy{
-    ChildOfPolicyType: &gql_policy,
+    ChildOfPolicyType: &gql_policy_1,
+    PerNodePolicyType: &gql_policy_2,
   })
 
   ctx.Log.Logf("test", "GQL_ID: %s", gql.ID)