runtime/runtime_test.go at main · aserto-dev/runtime · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
package runtime_test

import (
	"testing"
	"time"

	runtime "github.com/aserto-dev/runtime"
	"github.com/aserto-dev/runtime/testutil"
	"github.com/open-policy-agent/opa/v1/ast"
	"github.com/open-policy-agent/opa/v1/plugins/bundle"
	"github.com/stretchr/testify/require"
)

func TestEmptyRuntime(t *testing.T) {
	// Arrange
	assert := require.New(t)
	ctx := t.Context()

	r, err := runtime.New(ctx, &runtime.Config{})
	assert.NoError(err)

	// Act
	s := r.Status()

	// Assert
	assert.True(s.Ready)
}

func TestLocalBundle(t *testing.T) {
	// Arrange
	assert := require.New(t)
	ctx := t.Context()

	r, err := runtime.New(ctx, &runtime.Config{
		LocalBundles: runtime.LocalBundlesConfig{
			Paths: []string{testutil.AssetSimpleBundle()},
		},
	})
	assert.NoError(err)

	s := r.Status()

	// Assert
	assert.True(s.Ready)
	assert.Empty(s.Errors)
	assert.Len(s.Bundles, 1)
}

func TestFailingLocalBundle(t *testing.T) {
	// Arrange
	assert := require.New(t)

	// Act
	_, err := runtime.New(t.Context(), &runtime.Config{
		LocalBundles: runtime.LocalBundlesConfig{
			Paths: []string{testutil.AssetBuiltinsBundle()},
		},
	})

	// Assert
	assert.Error(err)
}

func TestRemoteBundleV0(t *testing.T) {
	// Arrange
	assert := require.New(t)
	ctx := t.Context()

	r, err := runtime.New(ctx, &runtime.Config{
		Config: runtime.OPAConfig{
			Services: map[string]any{
				"acmecorp": map[string]any{
					"url":                             "https://ghcr.io",
					"response_header_timeout_seconds": 5,
					"type":                            "oci",
				},
			},
			Bundles: map[string]*bundle.Source{
				"testbundle": {
					Service:  "acmecorp",
					Resource: "ghcr.io/aserto-policies/policy-peoplefinder-rbac:2",
				},
			},
		},
	},
		runtime.WithRegoVersion(ast.RegoV0),
	)

	assert.NoError(err)

	// Act
	assert.NoError(
		r.Start(ctx),
	)
	t.Cleanup(func() { r.Stop(ctx) })

	assert.NoError(
		r.WaitForPlugins(ctx, time.Second*5),
	)

	s := r.Status()

	// Assert
	assert.True(s.Ready)
	assert.Empty(s.Errors)
	assert.Len(s.Bundles, 1)
}