test(policy): update to handel statement returned from Match

ucan-wg · Nov 7, 2024 · 11bc085 · 11bc085
1 parent a4a8634
commit 11bc085
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pkg/policy/match_test.go b/pkg/policy/match_test.go
@@ -932,7 +932,9 @@ func TestInvocationValidationSpecExamples(t *testing.T) {
 		})
 		require.NoError(t, err)
 
-		assert.True(t, pol.Match(argsNode))
+		exec, stmt := pol.Match(argsNode)
+		assert.True(t, exec)
+		assert.Nil(t, stmt)
 	})
 
 	t.Run("fails on recipients (second statement)", func(t *testing.T) {
@@ -949,6 +951,8 @@ func TestInvocationValidationSpecExamples(t *testing.T) {
 		})
 		require.NoError(t, err)
 
-		assert.False(t, pol.Match(argsNode))
+		exec, stmt := pol.Match(argsNode)
+		assert.False(t, exec)
+		assert.NotNil(t, stmt)
 	})
 }