consul/internal/tenancy/tenancytest/namespace_test.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"

	svc "github.com/hashicorp/consul/agent/grpc-external/services/resource"
	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/internal/tenancy"
	"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) {
	v2TenancyBridge := tenancy.NewV2TenancyBridge()
	config := svc.Config{TenancyBridge: v2TenancyBridge, UseV2Tenancy: true}
	client := svctest.RunResourceServiceWithConfig(t, config, tenancy.RegisterTypes)
	cl := rtest.NewClient(client)

	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) {
	v2TenancyBridge := tenancy.NewV2TenancyBridge()
	config := svc.Config{TenancyBridge: v2TenancyBridge, UseV2Tenancy: true}
	client := svctest.RunResourceServiceWithConfig(t, config, tenancy.RegisterTypes)
	cl := rtest.NewClient(client)

	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) {
	v2TenancyBridge := tenancy.NewV2TenancyBridge()
	config := svc.Config{TenancyBridge: v2TenancyBridge, UseV2Tenancy: true}
	client := svctest.RunResourceServiceWithConfig(t, config, tenancy.RegisterTypes)
	cl := rtest.NewClient(client)

	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) {
	v2TenancyBridge := tenancy.NewV2TenancyBridge()
	config := svc.Config{TenancyBridge: v2TenancyBridge, UseV2Tenancy: true}
	client := svctest.RunResourceServiceWithConfig(t, config, tenancy.RegisterTypes)
	cl := rtest.NewClient(client)

	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",
	}
}