mirror of https://github.com/hashicorp/consul
80 lines
2.1 KiB
Go
80 lines
2.1 KiB
Go
|
// Copyright (c) HashiCorp, Inc.
|
||
|
// SPDX-License-Identifier: MPL-2.0
|
||
|
|
||
|
//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")
|
||
|
})
|
||
|
}
|