akvorado/console/docs_test.go

package console

import (
	"fmt"
	"io/ioutil"
	netHTTP "net/http"
	"strings"
	"testing"

	"akvorado/common/daemon"
	"akvorado/common/helpers"
	"akvorado/common/http"
	"akvorado/common/reporter"
)

func TestServeDocs(t *testing.T) {
	for _, live := range []bool{false, true} {
		name := "livefs"
		if !live {
			name = "embeddedfs"
		}
		cases := []struct {
			Path   string
			Expect string
		}{
			{"usage", `\u003ca href=\"configuration\"\u003econfiguration section\u003c/a\u003e`},
			{"intro", `data:image/svg`},
		}
		for _, tc := range cases {
			t.Run(fmt.Sprintf("%s-%s", name, tc.Path), func(t *testing.T) {
				r := reporter.NewMock(t)
				h := http.NewMock(t, r)
				c, err := New(r, Configuration{
					ServeLiveFS: live,
				}, Dependencies{
					Daemon: daemon.NewMock(t),
					HTTP:   h,
				})
				if err != nil {
					t.Fatalf("New() error:\n%+v", err)
				}
				helpers.StartStop(t, c)

				resp, err := netHTTP.Get(fmt.Sprintf("http://%s/api/v0/console/docs/%s",
					h.Address, tc.Path))
				if err != nil {
					t.Fatalf("GET /api/v0/console/docs/%s:\n%+v", tc.Path, err)
				}
				defer resp.Body.Close()
				if resp.StatusCode != 200 {
					t.Errorf("GET /api/v0/console/docs/%s: got status code %d, not 200",
						tc.Path, resp.StatusCode)
				}
				body, _ := ioutil.ReadAll(resp.Body)
				if !strings.Contains(string(body), tc.Expect) {
					t.Logf("Body:\n%s", string(body))
					t.Errorf("GET /api/v0/console/docs/%s: does not contain %q",
						tc.Path, tc.Expect)
				}
			})
		}
	}
}