consul/agent/peering_endpoint_ce_test.go

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

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

package agent

import (
	"bytes"
	"encoding/json"
	"io"
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/stretchr/testify/require"

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

func TestHTTP_Peering_GenerateToken_CE_Failure(t *testing.T) {
	if testing.Short() {
		t.Skip("too slow for testing.Short")
	}

	t.Parallel()

	a := NewTestAgent(t, "")
	testrpc.WaitForTestAgent(t, a.RPC, "dc1")

	t.Run("Doesn't allow partitions in CE HTTP requests", func(t *testing.T) {
		reqBody := &pbpeering.GenerateTokenRequest{
			PeerName: "peering-a",
		}
		reqBodyBytes, err := json.Marshal(reqBody)
		require.NoError(t, err)
		req, err := http.NewRequest("POST", "/v1/peering/token?partition=foo",
			bytes.NewReader(reqBodyBytes))
		require.NoError(t, err)
		resp := httptest.NewRecorder()
		a.srv.h.ServeHTTP(resp, req)
		require.Equal(t, http.StatusBadRequest, resp.Code)
		body, _ := io.ReadAll(resp.Body)
		require.Contains(t, string(body), "Partitions are a Consul Enterprise feature")
	})
}

func TestHTTP_PeeringEndpoint_CE_Failure(t *testing.T) {
	if testing.Short() {
		t.Skip("too slow for testing.Short")
	}

	t.Parallel()

	a := NewTestAgent(t, "")
	testrpc.WaitForTestAgent(t, a.RPC, "dc1")

	t.Run("Doesn't allow partitions on PeeringEndpoint in CE HTTP requests", func(t *testing.T) {
		req, err := http.NewRequest("GET", "/v1/peering/foo?partition=foo", nil)
		require.NoError(t, err)
		resp := httptest.NewRecorder()
		a.srv.h.ServeHTTP(resp, req)

		require.Equal(t, http.StatusBadRequest, resp.Code)
		body, _ := io.ReadAll(resp.Body)
		require.Contains(t, string(body), "Partitions are a Consul Enterprise feature")

		req2, err2 := http.NewRequest("DELETE", "/v1/peering/foo?partition=foo", nil)
		require.NoError(t, err2)
		resp2 := httptest.NewRecorder()
		a.srv.h.ServeHTTP(resp2, req2)

		require.Equal(t, http.StatusBadRequest, resp2.Code)
		body2, _ := io.ReadAll(resp2.Body)
		require.Contains(t, string(body2), "Partitions are a Consul Enterprise feature")
	})
}