consul/agent/rpc/peering/service_ce_test.go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: BUSL-1.1

//go:build !consulent
// +build !consulent

package peering_test

import (
	"context"
	"testing"
	"time"

	"github.com/stretchr/testify/require"

	"github.com/hashicorp/consul/proto/private/pbpeering"
)

func TestPeeringService_RejectsPartition(t *testing.T) {
	s := newTestServer(t, nil)
	client := pbpeering.NewPeeringServiceClient(s.ClientConn(t))

	t.Run("read", func(t *testing.T) {
		ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
		t.Cleanup(cancel)

		req := &pbpeering.PeeringReadRequest{Name: "foo", Partition: "test"}
		resp, err := client.PeeringRead(ctx, req)
		require.Contains(t, err.Error(), "Partitions are a Consul Enterprise feature")
		require.Nil(t, resp)
	})

	t.Run("list", func(t *testing.T) {
		ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
		t.Cleanup(cancel)

		req := &pbpeering.PeeringListRequest{Partition: "test"}
		resp, err := client.PeeringList(ctx, req)
		require.Contains(t, err.Error(), "Partitions are a Consul Enterprise feature")
		require.Nil(t, resp)
	})
}

func TestPeeringService_IgnoresDefaultPartition(t *testing.T) {
	s := newTestServer(t, nil)
	client := pbpeering.NewPeeringServiceClient(s.ClientConn(t))

	t.Run("read", func(t *testing.T) {
		ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
		t.Cleanup(cancel)

		req := &pbpeering.PeeringReadRequest{Name: "foo", Partition: "DeFaUlT"}
		_, err := client.PeeringRead(ctx, req)
		require.NoError(t, err)
	})

	t.Run("list", func(t *testing.T) {
		ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
		t.Cleanup(cancel)

		req := &pbpeering.PeeringListRequest{Partition: "DeFaUlT"}
		_, err := client.PeeringList(ctx, req)
		require.NoError(t, err)
	})
}