chore(policy): create policy folder.

Author: appleboy

.github/workflows/opa.yml

@@ -37,4 +37,4 @@ jobs:
 
       - name: Run Testing
         run: |
-          ./opa test -v *.rego
+          ./opa test -v policy/*.rego

Makefile

@@ -1,10 +1,7 @@
 opa_test:
-	opa test -v *.rego
-
-rbac_test:
-	opa test -v ./rbac/*.rego
+	opa test -v policy/*.rego
 
 go_test:
 	go test -v .
 
-test: rbac_test opa_test go_test
+test: opa_test go_test

go.mod

@@ -1,4 +1,4 @@
-module github.com/go-training/opa-demo
+module github.com/go-training/opa-embed
 
 go 1.16
 

main.go

@@ -4,6 +4,8 @@ import (
 	"context"
 	"log"
 
+	"github.com/go-training/opa-embed/policy"
+
 	"github.com/open-policy-agent/opa/rego"
 )
 
@@ -29,15 +31,15 @@ func main() {
 		"object": s.Object,
 	}
 
-	policy, err := readPolicy(policyFile)
+	p, err := policy.ReadPolicy(policyFile)
 	if err != nil {
 		log.Fatal(err)
 	}
 
 	ctx := context.TODO()
 	query, err := rego.New(
 		rego.Query(defaultQuery),
-		rego.Module(policyFile, string(policy)),
+		rego.Module(policyFile, string(p)),
 	).PrepareForEval(ctx)
 
 	if err != nil {

main_test.go

@@ -6,21 +6,23 @@ import (
 	"os"
 	"testing"
 
+	"github.com/go-training/opa-embed/policy"
+
 	"github.com/open-policy-agent/opa/rego"
 )
 
 var query rego.PreparedEvalQuery
 
 func setup() {
 	var err error
-	policy, err := readPolicy(policyFile)
+	p, err := policy.ReadPolicy(policyFile)
 	if err != nil {
 		log.Fatal(err)
 	}
 
 	query, err = rego.New(
 		rego.Query(defaultQuery),
-		rego.Module(policyFile, string(policy)),
+		rego.Module(policyFile, string(p)),
 	).PrepareForEval(context.TODO())
 
 	if err != nil {

example.rego renamed to policy/example.rego

@@ -1,13 +1,13 @@
 // +build !go1.16
 
-package main
+package policy
 
 import (
 	"io/ioutil"
 	"log"
 )
 
-func readPolicy(path string) ([]byte, error) {
+func ReadPolicy(path string) ([]byte, error) {
 	// load policy
 	policy, err := ioutil.ReadFile(policyFile)
 	if err != nil {

example_test.rego renamed to policy/example_test.rego

@@ -1,6 +1,6 @@
 // +build go1.16
 
-package main
+package policy
 
 import (
 	_ "embed"
@@ -9,6 +9,6 @@ import (
 //go:embed example.rego
 var policy []byte
 
-func readPolicy(path string) ([]byte, error) {
+func ReadPolicy(path string) ([]byte, error) {
 	return policy, nil
 }