Jeff Mitchell
c05324282f
This puts it more in line with other upcoming options that use "dir" instead of "directory" (plus it's easier to type), and makes it clear that this directory and the passthrough behavior is related to the UI as opposed to other upcoming items.
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())
|
|
}
|
|
})
|
|
}
|
|
}
|