mirror of https://github.com/hashicorp/consul
63 lines
1.9 KiB
Go
63 lines
1.9 KiB
Go
|
package resourcetest
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/google/go-cmp/cmp"
|
||
|
|
"github.com/hashicorp/consul/proto-public/pbresource"
|
||
|
|
"github.com/hashicorp/consul/proto/private/prototest"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
"google.golang.org/protobuf/testing/protocmp"
|
||
|
|
)
|
||
|
|
|
||
|
|
func RequireVersionUnchanged(t T, res *pbresource.Resource, version string) {
|
||
|
|
t.Helper()
|
||
|
|
require.Equal(t, version, res.Version)
|
||
|
|
}
|
||
|
|
|
||
|
|
func RequireVersionChanged(t T, res *pbresource.Resource, version string) {
|
||
|
|
t.Helper()
|
||
|
|
require.NotEqual(t, version, res.Version)
|
||
|
|
}
|
||
|
|
|
||
|
|
func RequireStatusCondition(t T, res *pbresource.Resource, statusKey string, condition *pbresource.Condition) {
|
||
|
|
t.Helper()
|
||
|
|
require.NotNil(t, res.Status)
|
||
|
|
status, found := res.Status[statusKey]
|
||
|
|
require.True(t, found)
|
||
|
|
prototest.AssertContainsElement(t, status.Conditions, condition)
|
||
|
|
}
|
||
|
|
|
||
|
|
func RequireStatusConditionForCurrentGen(t T, res *pbresource.Resource, statusKey string, condition *pbresource.Condition) {
|
||
|
|
t.Helper()
|
||
|
|
require.NotNil(t, res.Status)
|
||
|
|
status, found := res.Status[statusKey]
|
||
|
|
require.True(t, found)
|
||
|
|
require.Equal(t, res.Generation, status.ObservedGeneration)
|
||
|
|
prototest.AssertContainsElement(t, status.Conditions, condition)
|
||
|
|
}
|
||
|
|
|
||
|
|
func RequireResourceMeta(t T, res *pbresource.Resource, key string, value string) {
|
||
|
|
t.Helper()
|
||
|
|
require.NotNil(t, res.Metadata)
|
||
|
|
require.Contains(t, res.Metadata, key)
|
||
|
|
require.Equal(t, value, res.Metadata[key])
|
||
|
|
}
|
||
|
|
|
||
|
|
func RequireReconciledCurrentGen(t T, res *pbresource.Resource, statusKey string) {
|
||
|
|
t.Helper()
|
||
|
|
require.NotNil(t, res.Status)
|
||
|
|
status, found := res.Status[statusKey]
|
||
|
|
require.True(t, found)
|
||
|
|
require.Equal(t, res.Generation, status.ObservedGeneration)
|
||
|
|
}
|
||
|
|
|
||
|
|
func RequireOwner(t T, res *pbresource.Resource, owner *pbresource.ID, ignoreUid bool) {
|
||
|
|
t.Helper()
|
||
|
|
|
||
|
|
var opts []cmp.Option
|
||
|
|
if ignoreUid {
|
||
|
|
opts = append(opts, protocmp.IgnoreFields(owner, "uid"))
|
||
|
|
}
|
||
|
|
|
||
|
|
prototest.AssertDeepEqual(t, res.Owner, owner, opts...)
|
||
|
|
}
|