224 lines
7.4 KiB
Go
224 lines
7.4 KiB
Go
package integration
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/kvtools/consul"
|
|
"github.com/kvtools/valkeyrie"
|
|
"github.com/kvtools/valkeyrie/store"
|
|
"github.com/pmezard/go-difflib/difflib"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
"github.com/traefik/traefik/v3/integration/try"
|
|
"github.com/traefik/traefik/v3/pkg/api"
|
|
)
|
|
|
|
// Consul test suites.
|
|
type ConsulSuite struct {
|
|
BaseSuite
|
|
kvClient store.Store
|
|
consulURL string
|
|
}
|
|
|
|
func TestConsulSuite(t *testing.T) {
|
|
suite.Run(t, new(ConsulSuite))
|
|
}
|
|
|
|
func (s *ConsulSuite) SetupSuite() {
|
|
s.BaseSuite.SetupSuite()
|
|
s.createComposeProject("consul")
|
|
s.composeUp()
|
|
|
|
consulAddr := net.JoinHostPort(s.getComposeServiceIP("consul"), "8500")
|
|
s.consulURL = fmt.Sprintf("http://%s", consulAddr)
|
|
|
|
kv, err := valkeyrie.NewStore(
|
|
context.Background(),
|
|
consul.StoreName,
|
|
[]string{consulAddr},
|
|
&consul.Config{
|
|
ConnectionTimeout: 10 * time.Second,
|
|
},
|
|
)
|
|
require.NoError(s.T(), err, "Cannot create store consul")
|
|
s.kvClient = kv
|
|
|
|
// wait for consul
|
|
err = try.Do(60*time.Second, try.KVExists(kv, "test"))
|
|
require.NoError(s.T(), err)
|
|
}
|
|
|
|
func (s *ConsulSuite) TearDownSuite() {
|
|
s.BaseSuite.TearDownSuite()
|
|
}
|
|
|
|
func (s *ConsulSuite) TearDownTest() {
|
|
err := s.kvClient.DeleteTree(context.Background(), "traefik")
|
|
if err != nil && !errors.Is(err, store.ErrKeyNotFound) {
|
|
require.ErrorIs(s.T(), err, store.ErrKeyNotFound)
|
|
}
|
|
}
|
|
|
|
func (s *ConsulSuite) TestSimpleConfiguration() {
|
|
file := s.adaptFile("fixtures/consul/simple.toml", struct{ ConsulAddress string }{s.consulURL})
|
|
|
|
data := map[string]string{
|
|
"traefik/http/routers/Router0/entryPoints/0": "web",
|
|
"traefik/http/routers/Router0/middlewares/0": "compressor",
|
|
"traefik/http/routers/Router0/middlewares/1": "striper",
|
|
"traefik/http/routers/Router0/service": "simplesvc",
|
|
"traefik/http/routers/Router0/rule": "Host(`kv1.localhost`)",
|
|
"traefik/http/routers/Router0/priority": "42",
|
|
"traefik/http/routers/Router0/tls": "",
|
|
|
|
"traefik/http/routers/Router1/rule": "Host(`kv2.localhost`)",
|
|
"traefik/http/routers/Router1/priority": "42",
|
|
"traefik/http/routers/Router1/tls/domains/0/main": "aaa.localhost",
|
|
"traefik/http/routers/Router1/tls/domains/0/sans/0": "aaa.aaa.localhost",
|
|
"traefik/http/routers/Router1/tls/domains/0/sans/1": "bbb.aaa.localhost",
|
|
"traefik/http/routers/Router1/tls/domains/1/main": "bbb.localhost",
|
|
"traefik/http/routers/Router1/tls/domains/1/sans/0": "aaa.bbb.localhost",
|
|
"traefik/http/routers/Router1/tls/domains/1/sans/1": "bbb.bbb.localhost",
|
|
"traefik/http/routers/Router1/entryPoints/0": "web",
|
|
"traefik/http/routers/Router1/service": "simplesvc",
|
|
|
|
"traefik/http/services/simplesvc/loadBalancer/servers/0/url": "http://10.0.1.1:8888",
|
|
"traefik/http/services/simplesvc/loadBalancer/servers/1/url": "http://10.0.1.1:8889",
|
|
|
|
"traefik/http/services/srvcA/loadBalancer/servers/0/url": "http://10.0.1.2:8888",
|
|
"traefik/http/services/srvcA/loadBalancer/servers/1/url": "http://10.0.1.2:8889",
|
|
|
|
"traefik/http/services/srvcB/loadBalancer/servers/0/url": "http://10.0.1.3:8888",
|
|
"traefik/http/services/srvcB/loadBalancer/servers/1/url": "http://10.0.1.3:8889",
|
|
|
|
"traefik/http/services/mirror/mirroring/service": "simplesvc",
|
|
"traefik/http/services/mirror/mirroring/mirrors/0/name": "srvcA",
|
|
"traefik/http/services/mirror/mirroring/mirrors/0/percent": "42",
|
|
"traefik/http/services/mirror/mirroring/mirrors/1/name": "srvcB",
|
|
"traefik/http/services/mirror/mirroring/mirrors/1/percent": "42",
|
|
|
|
"traefik/http/services/Service03/weighted/services/0/name": "srvcA",
|
|
"traefik/http/services/Service03/weighted/services/0/weight": "42",
|
|
"traefik/http/services/Service03/weighted/services/1/name": "srvcB",
|
|
"traefik/http/services/Service03/weighted/services/1/weight": "42",
|
|
|
|
"traefik/http/middlewares/compressor/compress": "",
|
|
"traefik/http/middlewares/striper/stripPrefix/prefixes/0": "foo",
|
|
"traefik/http/middlewares/striper/stripPrefix/prefixes/1": "bar",
|
|
}
|
|
|
|
for k, v := range data {
|
|
err := s.kvClient.Put(context.Background(), k, []byte(v), nil)
|
|
require.NoError(s.T(), err)
|
|
}
|
|
|
|
s.traefikCmd(withConfigFile(file))
|
|
|
|
// wait for traefik
|
|
err := try.GetRequest("http://127.0.0.1:8080/api/rawdata", 2*time.Second,
|
|
try.BodyContains(`"striper@consul":`, `"compressor@consul":`, `"srvcA@consul":`, `"srvcB@consul":`),
|
|
)
|
|
require.NoError(s.T(), err)
|
|
|
|
resp, err := http.Get("http://127.0.0.1:8080/api/rawdata")
|
|
require.NoError(s.T(), err)
|
|
|
|
var obtained api.RunTimeRepresentation
|
|
err = json.NewDecoder(resp.Body).Decode(&obtained)
|
|
require.NoError(s.T(), err)
|
|
got, err := json.MarshalIndent(obtained, "", " ")
|
|
require.NoError(s.T(), err)
|
|
|
|
expectedJSON := filepath.FromSlash("testdata/rawdata-consul.json")
|
|
|
|
if *updateExpected {
|
|
err = os.WriteFile(expectedJSON, got, 0o666)
|
|
require.NoError(s.T(), err)
|
|
}
|
|
|
|
expected, err := os.ReadFile(expectedJSON)
|
|
require.NoError(s.T(), err)
|
|
|
|
if !bytes.Equal(expected, got) {
|
|
diff := difflib.UnifiedDiff{
|
|
FromFile: "Expected",
|
|
A: difflib.SplitLines(string(expected)),
|
|
ToFile: "Got",
|
|
B: difflib.SplitLines(string(got)),
|
|
Context: 3,
|
|
}
|
|
|
|
text, err := difflib.GetUnifiedDiffString(diff)
|
|
require.NoError(s.T(), err, text)
|
|
}
|
|
}
|
|
|
|
func (s *ConsulSuite) assertWhoami(host string, expectedStatusCode int) {
|
|
req, err := http.NewRequest(http.MethodGet, "http://127.0.0.1:8000", nil)
|
|
require.NoError(s.T(), err)
|
|
req.Host = host
|
|
|
|
resp, err := try.ResponseUntilStatusCode(req, 15*time.Second, expectedStatusCode)
|
|
require.NoError(s.T(), err)
|
|
resp.Body.Close()
|
|
}
|
|
|
|
func (s *ConsulSuite) TestDeleteRootKey() {
|
|
// This test case reproduce the issue: https://github.com/traefik/traefik/issues/8092
|
|
|
|
file := s.adaptFile("fixtures/consul/simple.toml", struct{ ConsulAddress string }{s.consulURL})
|
|
|
|
ctx := context.Background()
|
|
svcaddr := net.JoinHostPort(s.getComposeServiceIP("whoami"), "80")
|
|
|
|
data := map[string]string{
|
|
"traefik/http/routers/Router0/entryPoints/0": "web",
|
|
"traefik/http/routers/Router0/rule": "Host(`kv1.localhost`)",
|
|
"traefik/http/routers/Router0/service": "simplesvc0",
|
|
|
|
"traefik/http/routers/Router1/entryPoints/0": "web",
|
|
"traefik/http/routers/Router1/rule": "Host(`kv2.localhost`)",
|
|
"traefik/http/routers/Router1/service": "simplesvc1",
|
|
|
|
"traefik/http/services/simplesvc0/loadBalancer/servers/0/url": "http://" + svcaddr,
|
|
"traefik/http/services/simplesvc1/loadBalancer/servers/0/url": "http://" + svcaddr,
|
|
}
|
|
|
|
for k, v := range data {
|
|
err := s.kvClient.Put(ctx, k, []byte(v), nil)
|
|
require.NoError(s.T(), err)
|
|
}
|
|
|
|
s.traefikCmd(withConfigFile(file))
|
|
|
|
// wait for traefik
|
|
err := try.GetRequest("http://127.0.0.1:8080/api/rawdata", 2*time.Second,
|
|
try.BodyContains(`"Router0@consul":`, `"Router1@consul":`, `"simplesvc0@consul":`, `"simplesvc1@consul":`),
|
|
)
|
|
require.NoError(s.T(), err)
|
|
s.assertWhoami("kv1.localhost", http.StatusOK)
|
|
s.assertWhoami("kv2.localhost", http.StatusOK)
|
|
|
|
// delete router1
|
|
err = s.kvClient.DeleteTree(ctx, "traefik/http/routers/Router1")
|
|
require.NoError(s.T(), err)
|
|
s.assertWhoami("kv1.localhost", http.StatusOK)
|
|
s.assertWhoami("kv2.localhost", http.StatusNotFound)
|
|
|
|
// delete simple services and router0
|
|
err = s.kvClient.DeleteTree(ctx, "traefik")
|
|
require.NoError(s.T(), err)
|
|
s.assertWhoami("kv1.localhost", http.StatusNotFound)
|
|
s.assertWhoami("kv2.localhost", http.StatusNotFound)
|
|
}
|