mirror of
https://github.com/open-policy-agent/opa.git
synced 2026-08-12 19:32:48 -06:00
Stephan Renatus
6e4377af59
Users of OPA as a library are concerned about big binary blobs in their vendor/
directories. Even more so if they don't use them. This is the case for anyone
using OPA as library, but not using the wasm-backed evaluation feature.
With this change, importers of any packages other than `server` and `cmd`
will have to explicitly opt-in to using wasm evaluation features by having an
underscore import somewhere:
import _ "github.com/open-policy-agent/opa/features/wasm"
Fixes #3545.
Signed-off-by: Stephan Renatus <stephan.renatus@gmail.com>
80 lines
1.7 KiB
Go
80 lines
1.7 KiB
Go
// Copyright 2021 The OPA Authors. All rights reserved.
|
|
// Use of this source code is governed by an Apache2
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// +build opa_wasm
|
|
|
|
package repl
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"testing"
|
|
|
|
_ "github.com/open-policy-agent/opa/features/wasm"
|
|
)
|
|
|
|
func TestReplWasmTarget(t *testing.T) {
|
|
ctx := context.Background()
|
|
store := newTestStore()
|
|
var buffer bytes.Buffer
|
|
repl := newRepl(store, &buffer)
|
|
|
|
err := repl.OneShot(ctx, "target foo bar")
|
|
|
|
expected := "code bad arguments: target <mode>: expects exactly one argument"
|
|
if err == nil || err.Error() != expected {
|
|
t.Fatalf("Expected error %s, got %s", expected, err)
|
|
}
|
|
|
|
err = repl.OneShot(ctx, "target foo")
|
|
|
|
expected = "invalid target \"foo\":must be one of {rego,wasm}"
|
|
if err == nil || err.Error() != expected {
|
|
t.Fatalf("Expected error %s, got %s", expected, err)
|
|
}
|
|
|
|
buffer.Reset()
|
|
err = repl.OneShot(ctx, "target wasm")
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error: %v", err)
|
|
}
|
|
|
|
repl.OneShot(ctx, `p = true { input.foo = "bar" }`)
|
|
buffer.Reset()
|
|
repl.OneShot(ctx, "p")
|
|
|
|
if buffer.String() != "undefined\n" {
|
|
t.Fatalf("Expected undefined but got: %v", buffer.String())
|
|
}
|
|
|
|
buffer.Reset()
|
|
repl.OneShot(ctx, `p with input as {"foo": "bar"}`)
|
|
|
|
result := buffer.String()
|
|
expected = "true\n"
|
|
|
|
if result != expected {
|
|
t.Fatalf("Expected true but got: %v", result)
|
|
}
|
|
|
|
buffer.Reset()
|
|
repl.OneShot(ctx, `p with input.foo as "bar"`)
|
|
|
|
result = buffer.String()
|
|
|
|
if result != expected {
|
|
t.Fatalf("Expected true but got: %v", result)
|
|
}
|
|
|
|
buffer.Reset()
|
|
repl.OneShot(ctx, `trace`)
|
|
|
|
result = buffer.String()
|
|
|
|
expected = "warning: trace mode \"full\" is not supported with wasm target\n"
|
|
if result != expected {
|
|
t.Fatalf("Expected true but got: %v", result)
|
|
}
|
|
}
|