Todd
b58dada40d
* Move the server daemons into their own package With this change the internal/server package now is dedicated to the server domain (workers and controllers) and not to the server daemons. The hope is that this will make it easier to keep the daemons decoupled from the representation of the servers in the database.
136 lines
2.6 KiB
Go
136 lines
2.6 KiB
Go
//go:build ui
|
|
// +build ui
|
|
|
|
package controller
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestUiRouting(t *testing.T) {
|
|
// Create a temporary directory
|
|
tempDir, err := ioutil.TempDir("", "boundary-test-")
|
|
require.NoError(t, err)
|
|
defer func() {
|
|
assert.NoError(t, os.RemoveAll(tempDir))
|
|
}()
|
|
|
|
nameContentsMap := map[string]string{
|
|
"index.html": `index`,
|
|
"favicon.png": `favicon`,
|
|
"/assets/styles.css": `css`,
|
|
"index.htm": `badindex`,
|
|
}
|
|
|
|
for k, v := range nameContentsMap {
|
|
dir := filepath.Dir(k)
|
|
if dir != "/" {
|
|
require.NoError(t, os.MkdirAll(filepath.Join(tempDir, dir), 0o755))
|
|
}
|
|
require.NoError(t, ioutil.WriteFile(filepath.Join(tempDir, k), []byte(v), 0o644))
|
|
}
|
|
|
|
c := NewTestController(t, &TestControllerOpts{DisableAutoStart: true})
|
|
|
|
c.c.conf.RawConfig.DevUiPassthroughDir = tempDir
|
|
require.NoError(t, c.c.Start())
|
|
defer c.Shutdown()
|
|
|
|
cases := []struct {
|
|
name string
|
|
path string
|
|
contentsKey string
|
|
code int
|
|
mimeType string
|
|
}{
|
|
{
|
|
"direct index",
|
|
"index.html",
|
|
"index.html",
|
|
http.StatusOK,
|
|
"text/html; charset=utf-8",
|
|
},
|
|
{
|
|
"base slash",
|
|
"",
|
|
"index.html",
|
|
http.StatusOK,
|
|
"text/html; charset=utf-8",
|
|
},
|
|
{
|
|
"no extension",
|
|
"orgs",
|
|
"index.html",
|
|
http.StatusOK,
|
|
"text/html; charset=utf-8",
|
|
},
|
|
{
|
|
"favicon",
|
|
"favicon.png",
|
|
"favicon.png",
|
|
http.StatusOK,
|
|
"image/png",
|
|
},
|
|
{
|
|
"bad index",
|
|
"index.htm",
|
|
"index.htm",
|
|
http.StatusOK,
|
|
"text/html; charset=utf-8",
|
|
},
|
|
{
|
|
"bad path",
|
|
"index.ht",
|
|
"index.ht",
|
|
http.StatusNotFound,
|
|
"text/plain; charset=utf-8",
|
|
},
|
|
{
|
|
"css",
|
|
"assets/styles.css",
|
|
"assets/styles.css",
|
|
http.StatusOK,
|
|
"text/css; charset=utf-8",
|
|
},
|
|
{
|
|
"invalid extension",
|
|
"foo.bāb",
|
|
"index.html",
|
|
http.StatusOK,
|
|
"text/html; charset=utf-8",
|
|
},
|
|
}
|
|
for _, tc := range cases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
url := fmt.Sprintf("%s/%s", c.ApiAddrs()[0], tc.path)
|
|
resp, err := http.Post(url, "", nil)
|
|
assert.NoError(err)
|
|
assert.Equal(http.StatusMethodNotAllowed, resp.StatusCode)
|
|
|
|
resp, err = http.Get(url)
|
|
assert.NoError(err)
|
|
assert.Equal(tc.code, resp.StatusCode)
|
|
assert.Equal(tc.mimeType, resp.Header.Get("content-type"))
|
|
|
|
contents, ok := nameContentsMap[tc.contentsKey]
|
|
if ok {
|
|
reader := new(bytes.Buffer)
|
|
_, err = reader.ReadFrom(resp.Body)
|
|
assert.NoError(err)
|
|
assert.Equal(contents, reader.String())
|
|
}
|
|
})
|
|
}
|
|
}
|