Add a test case for unmarshalling IsSetStmt, thanks to @johanfylling

Signed-off-by: Kris Kennaway <[email protected]>

Author: KrisKennawayDD

v1/ir/encoding/encoding_test.go

@@ -11,62 +11,85 @@ import (
 )
 
 func TestRoundTrip(t *testing.T) {
+	tests := []struct {
+		note    string
+		modules map[string]string
+	}{
+		{
+			note: "simple",
+			modules: map[string]string{
+				"test.rego": `
+					package test
+					p if {
+						input.foo == 7
+					}
+				`,
+			},
+		},
+		{
+			note: "every",
+			modules: map[string]string{
+				"test.rego": `
+					package test
+					p if {
+						every i in input.foo { i > 0 } 
+					}
+				`,
+			},
+		},
+	}
 
-	// Note: v1 module
-	c, err := ast.CompileModules(map[string]string{
-		"test.rego": `
-			package test
+	for _, tc := range tests {
+		t.Run(tc.note, func(t *testing.T) {
+			// Note: v1 module
+			c, err := ast.CompileModules(tc.modules)
 
-			p if {
-				input.foo == 7
+			if err != nil {
+				t.Fatal(err)
 			}
-		`,
-	})
-
-	if err != nil {
-		t.Fatal(err)
-	}
 
-	modules := []*ast.Module{}
+			modules := []*ast.Module{}
 
-	for _, m := range c.Modules {
-		modules = append(modules, m)
-	}
+			for _, m := range c.Modules {
+				modules = append(modules, m)
+			}
 
-	planner := planner.New().
-		WithQueries([]planner.QuerySet{
-			{
-				Name: "main",
-				Queries: []ast.Body{
-					ast.MustParseBody("data.test.p = true"),
-				},
-			},
-		}).
-		WithModules(modules).
-		WithBuiltinDecls(ast.BuiltinMap)
+			planner := planner.New().
+				WithQueries([]planner.QuerySet{
+					{
+						Name: "main",
+						Queries: []ast.Body{
+							ast.MustParseBody("data.test.p = true"),
+						},
+					},
+				}).
+				WithModules(modules).
+				WithBuiltinDecls(ast.BuiltinMap)
 
-	plan, err := planner.Plan()
-	if err != nil {
-		t.Fatal(err)
-	}
+			plan, err := planner.Plan()
+			if err != nil {
+				t.Fatal(err)
+			}
 
-	bs, err := json.MarshalIndent(plan, "", "  ")
-	if err != nil {
-		t.Fatal(err)
-	}
+			bs, err := json.MarshalIndent(plan, "", "  ")
+			if err != nil {
+				t.Fatal(err)
+			}
 
-	var cpy ir.Policy
-	err = json.Unmarshal(bs, &cpy)
-	if err != nil {
-		t.Fatal(err)
-	}
+			var cpy ir.Policy
+			err = json.Unmarshal(bs, &cpy)
+			if err != nil {
+				t.Fatal(err)
+			}
 
-	bs2, err := json.MarshalIndent(plan, "", "  ")
-	if err != nil {
-		t.Fatal(err)
-	}
+			bs2, err := json.MarshalIndent(plan, "", "  ")
+			if err != nil {
+				t.Fatal(err)
+			}
 
-	if !bytes.Equal(bs, bs2) {
-		t.Fatal("expected bytes to be equal")
+			if !bytes.Equal(bs, bs2) {
+				t.Fatal("expected bytes to be equal")
+			}
+		})
 	}
 }