mirror of https://github.com/hashicorp/consul
141 lines
3.0 KiB
Go
141 lines
3.0 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package controller
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/google/go-cmp/cmp/cmpopts"
|
|
"github.com/hashicorp/consul/proto-public/pbresource"
|
|
"github.com/hashicorp/consul/proto/private/prototest"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func resourceID(group string, version string, kind string, name string) *pbresource.ID {
|
|
return &pbresource.ID{
|
|
Type: &pbresource.Type{
|
|
Group: group,
|
|
GroupVersion: version,
|
|
Kind: kind,
|
|
},
|
|
Tenancy: &pbresource.Tenancy{
|
|
Partition: "default",
|
|
Namespace: "default",
|
|
PeerName: "local",
|
|
},
|
|
Name: name,
|
|
}
|
|
}
|
|
|
|
func TestMapOwner(t *testing.T) {
|
|
owner := resourceID("foo", "v99", "bar", "object")
|
|
|
|
res := &pbresource.Resource{
|
|
Id: resourceID("something", "v1", "else", "x"),
|
|
Owner: owner,
|
|
}
|
|
|
|
reqs, err := MapOwner(context.Background(), Runtime{}, res)
|
|
require.NoError(t, err)
|
|
require.Len(t, reqs, 1)
|
|
prototest.AssertDeepEqual(t, owner, reqs[0].ID)
|
|
}
|
|
|
|
func TestMapOwnerFiltered(t *testing.T) {
|
|
mapper := MapOwnerFiltered(&pbresource.Type{
|
|
Group: "foo",
|
|
GroupVersion: "v1",
|
|
Kind: "bar",
|
|
})
|
|
|
|
type testCase struct {
|
|
owner *pbresource.ID
|
|
matches bool
|
|
}
|
|
|
|
cases := map[string]testCase{
|
|
"nil-owner": {
|
|
owner: nil,
|
|
matches: false,
|
|
},
|
|
"group-mismatch": {
|
|
owner: resourceID("other", "v1", "bar", "irrelevant"),
|
|
matches: false,
|
|
},
|
|
"group-version-mismatch": {
|
|
owner: resourceID("foo", "v2", "bar", "irrelevant"),
|
|
matches: false,
|
|
},
|
|
"kind-mismatch": {
|
|
owner: resourceID("foo", "v1", "baz", "irrelevant"),
|
|
matches: false,
|
|
},
|
|
"match": {
|
|
owner: resourceID("foo", "v1", "bar", "irrelevant"),
|
|
matches: true,
|
|
},
|
|
}
|
|
|
|
for name, tcase := range cases {
|
|
t.Run(name, func(t *testing.T) {
|
|
// the runtime is not used by the mapper so its fine to pass an empty struct
|
|
req, err := mapper(context.Background(), Runtime{}, &pbresource.Resource{
|
|
Id: resourceID("foo", "v1", "other", "x"),
|
|
Owner: tcase.owner,
|
|
})
|
|
|
|
// The mapper has no error paths at present
|
|
require.NoError(t, err)
|
|
|
|
if tcase.matches {
|
|
require.NotNil(t, req)
|
|
require.Len(t, req, 1)
|
|
prototest.AssertDeepEqual(t, req[0].ID, tcase.owner, cmpopts.EquateEmpty())
|
|
} else {
|
|
require.Nil(t, req)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestReplaceType(t *testing.T) {
|
|
rtype := &pbresource.Type{
|
|
Group: "foo",
|
|
GroupVersion: "v1",
|
|
Kind: "bar",
|
|
}
|
|
|
|
tenant := &pbresource.Tenancy{
|
|
Partition: "not",
|
|
Namespace: "using",
|
|
PeerName: "the-defaults",
|
|
}
|
|
|
|
in := &pbresource.Resource{
|
|
Id: &pbresource.ID{
|
|
Type: &pbresource.Type{
|
|
Group: "other",
|
|
GroupVersion: "v2",
|
|
Kind: "baz",
|
|
},
|
|
Tenancy: tenant,
|
|
Name: "arr-matey",
|
|
},
|
|
}
|
|
|
|
mapper := ReplaceType(rtype)
|
|
|
|
reqs, err := mapper(nil, Runtime{}, in)
|
|
require.NoError(t, err)
|
|
require.Len(t, reqs, 1)
|
|
|
|
expected := &pbresource.ID{
|
|
Type: rtype,
|
|
Tenancy: tenant,
|
|
Name: "arr-matey",
|
|
}
|
|
prototest.AssertDeepEqual(t, expected, reqs[0].ID)
|
|
}
|