releases/v1/storage/storage_test.go

package storage_test

import (
	"bytes"
	"context"
	"strings"
	"testing"

	"github.com/open-policy-agent/opa/v1/storage"
	"github.com/open-policy-agent/opa/v1/storage/inmem"
)

func TestNonEmpty(t *testing.T) {

	cases := []struct {
		content string
		path    string
		exp     bool
	}{
		{
			content: `{}`,
			path:    "a/b/c",
			exp:     false,
		},
		{
			content: `{"a": {}}`,
			path:    "a/b/c",
			exp:     false,
		},
		{
			content: `{"a": {"b": {}}}`,
			path:    "a/b/c",
			exp:     false,
		},
		{
			content: `{"a": {"b": {"c": {}}}}`,
			path:    "a/b/c",
			exp:     true,
		},
		{
			content: `{"a": {"b": "x"}}`,
			path:    "a/b/c",
			exp:     true,
		},
		{
			content: `{"a": "x"}`,
			path:    "a/b/c",
			exp:     true,
		},
	}

	ctx := t.Context()

	for _, tc := range cases {
		t.Run(tc.content, func(t *testing.T) {
			store := inmem.NewFromReader(bytes.NewBufferString(tc.content))
			err := storage.Txn(ctx, store, storage.TransactionParams{}, func(txn storage.Transaction) error {
				nonEmpty, err := storage.NonEmpty(ctx, store, txn)(strings.Split(tc.path, "/"))
				if err != nil {
					t.Fatal(err)
				}
				if nonEmpty != tc.exp {
					t.Errorf("Expected %v for %v on %v but got %v", tc.exp, tc.path, tc.content, nonEmpty)
				}
				return nil
			})
			if err != nil {
				t.Error(err)
			}
		})
	}

}

type nonEmpty struct {
	storage.Store
}

func (*nonEmpty) NonEmpty(context.Context, storage.Transaction) func([]string) (bool, error) {
	return func([]string) (bool, error) {
		return true, nil
	}
}

func TestNonEmptyer(t *testing.T) {
	ctx := t.Context()
	ne := &nonEmpty{inmem.New()}

	for _, path := range []string{"a", "a/b/c"} {
		err := storage.Txn(ctx, ne, storage.TransactionParams{}, func(txn storage.Transaction) error {
			nonEmpty, err := storage.NonEmpty(ctx, ne, txn)(strings.Split(path, "/"))
			if err != nil {
				t.Fatal(err)
			}
			if nonEmpty != true {
				t.Errorf("Expected true for %v but got false", path)
			}
			return nil
		})
		if err != nil {
			t.Error(err)
		}
	}
}