mirror of https://github.com/status-im/consul.git
117 lines
3.7 KiB
Go
117 lines
3.7 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package tenancytest
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
|
|
svctest "github.com/hashicorp/consul/agent/grpc-external/services/resource/testing"
|
|
"github.com/hashicorp/consul/internal/resource"
|
|
rtest "github.com/hashicorp/consul/internal/resource/resourcetest"
|
|
"github.com/hashicorp/consul/proto-public/pbresource"
|
|
pbtenancy "github.com/hashicorp/consul/proto-public/pbtenancy/v2beta1"
|
|
"github.com/hashicorp/consul/proto/private/prototest"
|
|
)
|
|
|
|
func TestWriteNamespace_Success(t *testing.T) {
|
|
cl := rtest.NewClient(svctest.NewResourceServiceBuilder().WithV2Tenancy(true).Run(t))
|
|
|
|
res := rtest.Resource(pbtenancy.NamespaceType, "ns1").
|
|
WithTenancy(resource.DefaultPartitionedTenancy()).
|
|
WithData(t, validNamespace()).
|
|
Build()
|
|
|
|
writeRsp, err := cl.Write(context.Background(), &pbresource.WriteRequest{Resource: res})
|
|
require.NoError(t, err)
|
|
prototest.AssertDeepEqual(t, res.Id.Type, writeRsp.Resource.Id.Type)
|
|
prototest.AssertDeepEqual(t, res.Id.Tenancy, writeRsp.Resource.Id.Tenancy)
|
|
prototest.AssertDeepEqual(t, res.Id.Name, writeRsp.Resource.Id.Name)
|
|
prototest.AssertDeepEqual(t, res.Data, writeRsp.Resource.Data)
|
|
}
|
|
|
|
func TestReadNamespace_Success(t *testing.T) {
|
|
cl := rtest.NewClient(svctest.NewResourceServiceBuilder().WithV2Tenancy(true).Run(t))
|
|
|
|
res := rtest.Resource(pbtenancy.NamespaceType, "ns1").
|
|
WithData(t, validNamespace()).
|
|
Write(t, cl)
|
|
|
|
cases := []struct {
|
|
name string
|
|
resource *pbresource.Resource
|
|
errMsg string
|
|
}{
|
|
{
|
|
name: "read namespace",
|
|
resource: rtest.Resource(pbtenancy.NamespaceType, "ns1").
|
|
WithData(t, validNamespace()).
|
|
Build(),
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
readRsp, err := cl.Read(context.Background(), &pbresource.ReadRequest{Id: tc.resource.Id})
|
|
require.NoError(t, err)
|
|
prototest.AssertDeepEqual(t, res.Id, readRsp.Resource.Id)
|
|
prototest.AssertDeepEqual(t, res.Data, readRsp.Resource.Data)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestDeleteNamespace_Success(t *testing.T) {
|
|
cl := rtest.NewClient(svctest.NewResourceServiceBuilder().WithV2Tenancy(true).Run(t))
|
|
|
|
res := rtest.Resource(pbtenancy.NamespaceType, "ns1").
|
|
WithData(t, validNamespace()).Write(t, cl)
|
|
|
|
readRsp, err := cl.Read(context.Background(), &pbresource.ReadRequest{Id: res.Id})
|
|
require.NoError(t, err)
|
|
prototest.AssertDeepEqual(t, res.Id, readRsp.Resource.Id)
|
|
|
|
_, err = cl.Delete(context.Background(), &pbresource.DeleteRequest{Id: res.Id})
|
|
require.NoError(t, err)
|
|
|
|
_, err = cl.Read(context.Background(), &pbresource.ReadRequest{Id: res.Id})
|
|
require.Error(t, err)
|
|
require.Equal(t, codes.NotFound.String(), status.Code(err).String())
|
|
|
|
}
|
|
|
|
func TestListNamespace_Success(t *testing.T) {
|
|
cl := rtest.NewClient(svctest.NewResourceServiceBuilder().WithV2Tenancy(true).Run(t))
|
|
|
|
res := rtest.Resource(pbtenancy.NamespaceType, "ns1").
|
|
WithData(t, validNamespace()).Write(t, cl)
|
|
|
|
require.NotNil(t, res)
|
|
res = rtest.Resource(pbtenancy.NamespaceType, "ns2").
|
|
WithData(t, validNamespace()).Write(t, cl)
|
|
|
|
require.NotNil(t, res)
|
|
|
|
listRsp, err := cl.List(context.Background(), &pbresource.ListRequest{Type: pbtenancy.NamespaceType, Tenancy: resource.DefaultPartitionedTenancy()})
|
|
require.NoError(t, err)
|
|
require.Len(t, listRsp.Resources, 3)
|
|
names := []string{
|
|
listRsp.Resources[0].Id.Name,
|
|
listRsp.Resources[1].Id.Name,
|
|
listRsp.Resources[2].Id.Name,
|
|
}
|
|
require.Contains(t, names, "default")
|
|
require.Contains(t, names, "ns1")
|
|
require.Contains(t, names, "ns2")
|
|
}
|
|
|
|
func validNamespace() *pbtenancy.Namespace {
|
|
return &pbtenancy.Namespace{
|
|
Description: "ns namespace",
|
|
}
|
|
}
|