360 lines
8.7 KiB
Go
360 lines
8.7 KiB
Go
package file
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/containous/traefik/pkg/config"
|
|
"github.com/containous/traefik/pkg/safe"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type ProvideTestCase struct {
|
|
desc string
|
|
directoryContent []string
|
|
fileContent string
|
|
traefikFileContent string
|
|
expectedNumRouter int
|
|
expectedNumService int
|
|
expectedNumTLSConf int
|
|
}
|
|
|
|
func TestProvideWithoutWatch(t *testing.T) {
|
|
for _, test := range getTestCases() {
|
|
t.Run(test.desc+" without watch", func(t *testing.T) {
|
|
provider, clean := createProvider(t, test, false)
|
|
defer clean()
|
|
configChan := make(chan config.Message)
|
|
|
|
provider.DebugLogGeneratedTemplate = true
|
|
|
|
go func() {
|
|
err := provider.Provide(configChan, safe.NewPool(context.Background()))
|
|
assert.NoError(t, err)
|
|
}()
|
|
|
|
timeout := time.After(time.Second)
|
|
select {
|
|
case conf := <-configChan:
|
|
assert.Len(t, conf.Configuration.HTTP.Services, test.expectedNumService)
|
|
assert.Len(t, conf.Configuration.HTTP.Routers, test.expectedNumRouter)
|
|
assert.Len(t, conf.Configuration.TLS, test.expectedNumTLSConf)
|
|
case <-timeout:
|
|
t.Errorf("timeout while waiting for config")
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestProvideWithWatch(t *testing.T) {
|
|
for _, test := range getTestCases() {
|
|
t.Run(test.desc+" with watch", func(t *testing.T) {
|
|
provider, clean := createProvider(t, test, true)
|
|
defer clean()
|
|
configChan := make(chan config.Message)
|
|
|
|
go func() {
|
|
err := provider.Provide(configChan, safe.NewPool(context.Background()))
|
|
assert.NoError(t, err)
|
|
}()
|
|
|
|
timeout := time.After(time.Second)
|
|
select {
|
|
case conf := <-configChan:
|
|
assert.Len(t, conf.Configuration.HTTP.Services, 0)
|
|
assert.Len(t, conf.Configuration.HTTP.Routers, 0)
|
|
assert.Len(t, conf.Configuration.TLS, 0)
|
|
case <-timeout:
|
|
t.Errorf("timeout while waiting for config")
|
|
}
|
|
|
|
if len(test.fileContent) > 0 {
|
|
if err := ioutil.WriteFile(provider.Filename, []byte(test.fileContent), 0755); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
if len(test.traefikFileContent) > 0 {
|
|
if err := ioutil.WriteFile(provider.TraefikFile, []byte(test.traefikFileContent), 0755); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
if len(test.directoryContent) > 0 {
|
|
for _, fileContent := range test.directoryContent {
|
|
createRandomFile(t, provider.Directory, fileContent)
|
|
}
|
|
}
|
|
|
|
timeout = time.After(time.Second * 1)
|
|
var numUpdates, numServices, numRouters, numTLSConfs int
|
|
for {
|
|
select {
|
|
case conf := <-configChan:
|
|
numUpdates++
|
|
numServices = len(conf.Configuration.HTTP.Services)
|
|
numRouters = len(conf.Configuration.HTTP.Routers)
|
|
numTLSConfs = len(conf.Configuration.TLS)
|
|
t.Logf("received update #%d: services %d/%d, routers %d/%d, TLS configs %d/%d", numUpdates, numServices, test.expectedNumService, numRouters, test.expectedNumRouter, numTLSConfs, test.expectedNumTLSConf)
|
|
|
|
if numServices == test.expectedNumService && numRouters == test.expectedNumRouter && numTLSConfs == test.expectedNumTLSConf {
|
|
return
|
|
}
|
|
case <-timeout:
|
|
t.Fatal("timeout while waiting for config")
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestErrorWhenEmptyConfig(t *testing.T) {
|
|
provider := &Provider{}
|
|
configChan := make(chan config.Message)
|
|
errorChan := make(chan struct{})
|
|
go func() {
|
|
err := provider.Provide(configChan, safe.NewPool(context.Background()))
|
|
assert.Error(t, err)
|
|
close(errorChan)
|
|
}()
|
|
|
|
timeout := time.After(time.Second)
|
|
select {
|
|
case <-configChan:
|
|
t.Fatal("We should not receive config message")
|
|
case <-timeout:
|
|
t.Fatal("timeout while waiting for config")
|
|
case <-errorChan:
|
|
}
|
|
}
|
|
|
|
func getTestCases() []ProvideTestCase {
|
|
return []ProvideTestCase{
|
|
{
|
|
desc: "simple file",
|
|
fileContent: createRoutersConfiguration(3) + createServicesConfiguration(6) + createTLS(5),
|
|
expectedNumRouter: 3,
|
|
expectedNumService: 6,
|
|
expectedNumTLSConf: 5,
|
|
},
|
|
{
|
|
desc: "simple file and a traefik file",
|
|
fileContent: createRoutersConfiguration(4) + createServicesConfiguration(8) + createTLS(4),
|
|
traefikFileContent: `
|
|
debug=true
|
|
`,
|
|
expectedNumRouter: 4,
|
|
expectedNumService: 8,
|
|
expectedNumTLSConf: 4,
|
|
},
|
|
{
|
|
desc: "template file",
|
|
fileContent: `
|
|
[http.routers]
|
|
{{ range $i, $e := until 20 }}
|
|
[http.routers.router{{ $e }}]
|
|
service = "application"
|
|
{{ end }}
|
|
`,
|
|
expectedNumRouter: 20,
|
|
},
|
|
{
|
|
desc: "simple directory",
|
|
directoryContent: []string{
|
|
createRoutersConfiguration(2),
|
|
createServicesConfiguration(3),
|
|
createTLS(4),
|
|
},
|
|
expectedNumRouter: 2,
|
|
expectedNumService: 3,
|
|
expectedNumTLSConf: 4,
|
|
},
|
|
{
|
|
desc: "template in directory",
|
|
directoryContent: []string{
|
|
`
|
|
[http.routers]
|
|
{{ range $i, $e := until 20 }}
|
|
[http.routers.router{{ $e }}]
|
|
service = "application"
|
|
{{ end }}
|
|
`,
|
|
`
|
|
[http.services]
|
|
{{ range $i, $e := until 20 }}
|
|
[http.services.application-{{ $e }}]
|
|
[[http.services.application-{{ $e }}.servers]]
|
|
url="http://127.0.0.1"
|
|
weight = 1
|
|
{{ end }}
|
|
`,
|
|
},
|
|
expectedNumRouter: 20,
|
|
expectedNumService: 20,
|
|
},
|
|
{
|
|
desc: "simple traefik file",
|
|
traefikFileContent: `
|
|
debug=true
|
|
[providers.file]
|
|
` + createRoutersConfiguration(2) + createServicesConfiguration(3) + createTLS(4),
|
|
expectedNumRouter: 2,
|
|
expectedNumService: 3,
|
|
expectedNumTLSConf: 4,
|
|
},
|
|
{
|
|
desc: "simple traefik file with templating",
|
|
traefikFileContent: `
|
|
temp="{{ getTag \"test\" }}"
|
|
[providers.file]
|
|
` + createRoutersConfiguration(2) + createServicesConfiguration(3) + createTLS(4),
|
|
expectedNumRouter: 2,
|
|
expectedNumService: 3,
|
|
expectedNumTLSConf: 4,
|
|
},
|
|
}
|
|
}
|
|
|
|
func createProvider(t *testing.T, test ProvideTestCase, watch bool) (*Provider, func()) {
|
|
tempDir := createTempDir(t, "testdir")
|
|
|
|
provider := &Provider{}
|
|
provider.Watch = watch
|
|
|
|
if len(test.directoryContent) > 0 {
|
|
if !watch {
|
|
for _, fileContent := range test.directoryContent {
|
|
createRandomFile(t, tempDir, fileContent)
|
|
}
|
|
}
|
|
provider.Directory = tempDir
|
|
}
|
|
|
|
if len(test.fileContent) > 0 {
|
|
if watch {
|
|
test.fileContent = ""
|
|
}
|
|
filename := createRandomFile(t, tempDir, test.fileContent)
|
|
provider.Filename = filename.Name()
|
|
|
|
}
|
|
|
|
if len(test.traefikFileContent) > 0 {
|
|
if watch {
|
|
test.traefikFileContent = ""
|
|
}
|
|
filename := createRandomFile(t, tempDir, test.traefikFileContent)
|
|
provider.TraefikFile = filename.Name()
|
|
}
|
|
|
|
return provider, func() {
|
|
os.RemoveAll(tempDir)
|
|
}
|
|
}
|
|
|
|
// createRandomFile Helper
|
|
func createRandomFile(t *testing.T, tempDir string, contents ...string) *os.File {
|
|
return createFile(t, tempDir, fmt.Sprintf("temp%d.toml", time.Now().UnixNano()), contents...)
|
|
}
|
|
|
|
// createFile Helper
|
|
func createFile(t *testing.T, tempDir string, name string, contents ...string) *os.File {
|
|
t.Helper()
|
|
fileName := path.Join(tempDir, name)
|
|
|
|
tempFile, err := os.Create(fileName)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
for _, content := range contents {
|
|
_, err = tempFile.WriteString(content)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
err = tempFile.Close()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
return tempFile
|
|
}
|
|
|
|
// createTempDir Helper
|
|
func createTempDir(t *testing.T, dir string) string {
|
|
t.Helper()
|
|
d, err := ioutil.TempDir("", dir)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return d
|
|
}
|
|
|
|
// createRoutersConfiguration Helper
|
|
func createRoutersConfiguration(n int) string {
|
|
conf := "[http.routers]\n"
|
|
for i := 1; i <= n; i++ {
|
|
conf += fmt.Sprintf(`
|
|
[http.routers."router%[1]d"]
|
|
service = "application-%[1]d"
|
|
`, i)
|
|
}
|
|
return conf
|
|
}
|
|
|
|
// createServicesConfiguration Helper
|
|
func createServicesConfiguration(n int) string {
|
|
conf := "[http.services]\n"
|
|
for i := 1; i <= n; i++ {
|
|
conf += fmt.Sprintf(`
|
|
[http.services.application-%[1]d.loadbalancer]
|
|
[[http.services.application-%[1]d.loadbalancer.servers]]
|
|
url = "http://172.17.0.%[1]d:80"
|
|
weight = 1
|
|
`, i)
|
|
}
|
|
return conf
|
|
}
|
|
|
|
// createTLS Helper
|
|
func createTLS(n int) string {
|
|
var conf string
|
|
for i := 1; i <= n; i++ {
|
|
conf += fmt.Sprintf(`[[TLS]]
|
|
EntryPoints = ["https"]
|
|
[TLS.Certificate]
|
|
CertFile = "integration/fixtures/https/snitest%[1]d.com.cert"
|
|
KeyFile = "integration/fixtures/https/snitest%[1]d.com.key"
|
|
`, i)
|
|
}
|
|
return conf
|
|
}
|
|
|
|
func TestTLSContent(t *testing.T) {
|
|
tempDir := createTempDir(t, "testdir")
|
|
defer os.RemoveAll(tempDir)
|
|
|
|
fileTLS := createRandomFile(t, tempDir, "CONTENT")
|
|
fileConfig := createRandomFile(t, tempDir, `
|
|
[[tls]]
|
|
entryPoints = ["https"]
|
|
[tls.certificate]
|
|
certFile = "`+fileTLS.Name()+`"
|
|
keyFile = "`+fileTLS.Name()+`"
|
|
`)
|
|
|
|
provider := &Provider{}
|
|
configuration, err := provider.loadFileConfig(fileConfig.Name(), true)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, "CONTENT", configuration.TLS[0].Certificate.CertFile.String())
|
|
require.Equal(t, "CONTENT", configuration.TLS[0].Certificate.KeyFile.String())
|
|
}
|