cd75858f4a
* api: swap testify for test (acl) * api: swap testify for test (agent) Please enter the commit message for your changes. Lines starting * api: swap testify for test (allocations) * api: swap testify for test (api) * api: swap testify for test (compose) * api: swap testify for test (constraint) * api: swap testify for test (consul) * api: swap testify for test (csi) * api: swap testify for test (evaluations) * api: swap testify for test (event stream) * api: swap testify for test (fs) * api: swap testify for test (ioutil) * api: swap testify for test (jobs) * api: swap testify for test (keyring) * api: swap testify for test (operator_ent) * api: swap testify for test (operator_metrics) * api: swap testify for test (operator) * api: swap testify for test (quota) * api: swap testify for test (resources) * api: swap testify for test (fix operator_metrics) * api: swap testify for test (scaling) * api: swap testify for test (search) * api: swap testify for test (sentinel) * api: swap testify for test (services) * api: swap testify for test (status) * api: swap testify for test (system) * api: swap testify for test (tasks) * api: swap testify for test (utils) * api: swap testify for test (variables) * api: remove dependencies on testify and pretty
99 lines
2.1 KiB
Go
99 lines
2.1 KiB
Go
//go:build ent
|
|
|
|
package api
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/hashicorp/nomad/api/internal/testutil"
|
|
"github.com/shoenig/test/must"
|
|
)
|
|
|
|
func TestSentinelPolicies_ListUpsert(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
c, s, _ := makeACLClient(t, nil, nil)
|
|
defer s.Stop()
|
|
ap := c.SentinelPolicies()
|
|
|
|
// Listing when nothing exists returns empty
|
|
result, qm, err := ap.List(nil)
|
|
must.NoError(t, err)
|
|
must.Positive(t, qm.LastIndex)
|
|
must.SliceEmpty(t, result)
|
|
|
|
// Register a policy
|
|
policy := &SentinelPolicy{
|
|
Name: "test",
|
|
Description: "test",
|
|
EnforcementLevel: "advisory",
|
|
Scope: "submit-job",
|
|
Policy: "main = rule { true }",
|
|
}
|
|
wm, err := ap.Upsert(policy, nil)
|
|
must.NoError(t, err)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Check the list again
|
|
result, qm, err = ap.List(nil)
|
|
must.NoError(t, err)
|
|
assertQueryMeta(t, qm)
|
|
must.Len(t, 1, result)
|
|
}
|
|
|
|
func TestSentinelPolicies_Delete(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
|
|
c, s, _ := makeACLClient(t, nil, nil)
|
|
defer s.Stop()
|
|
ap := c.SentinelPolicies()
|
|
|
|
// Register a policy
|
|
policy := &SentinelPolicy{
|
|
Name: "test",
|
|
Description: "test",
|
|
EnforcementLevel: "advisory",
|
|
Scope: "submit-job",
|
|
Policy: "main = rule { true } ",
|
|
}
|
|
wm, err := ap.Upsert(policy, nil)
|
|
must.NoError(t, err)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Delete the policy
|
|
wm, err = ap.Delete(policy.Name, nil)
|
|
must.NoError(t, err)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Check the list again
|
|
result, qm, err := ap.List(nil)
|
|
must.NoError(t, err)
|
|
assertQueryMeta(t, qm)
|
|
must.SliceEmpty(t, result)
|
|
}
|
|
|
|
func TestSentinelPolicies_Info(t *testing.T) {
|
|
testutil.Parallel(t)
|
|
|
|
c, s, _ := makeACLClient(t, nil, nil)
|
|
defer s.Stop()
|
|
ap := c.SentinelPolicies()
|
|
|
|
// Register a policy
|
|
policy := &SentinelPolicy{
|
|
Name: "test",
|
|
Description: "test",
|
|
EnforcementLevel: "advisory",
|
|
Scope: "submit-job",
|
|
Policy: "main = rule { true }",
|
|
}
|
|
wm, err := ap.Upsert(policy, nil)
|
|
must.NoError(t, err)
|
|
assertWriteMeta(t, wm)
|
|
|
|
// Query the policy
|
|
out, qm, err := ap.Info(policy.Name, nil)
|
|
must.NoError(t, err)
|
|
assertQueryMeta(t, qm)
|
|
must.Eq(t, policy.Name, out.Name)
|
|
}
|