pkger/open_test.go

package pkger

import (
	"io/ioutil"
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/stretchr/testify/require"
)

func Test_Open_File(t *testing.T) {
	r := require.New(t)

	f, err := Open(".")
	r.NoError(err)

	ts := httptest.NewServer(http.FileServer(f))
	defer ts.Close()

	res, err := http.Get(ts.URL + "/cmd/pkger/main.go")
	r.NoError(err)
	r.Equal(200, res.StatusCode)

	b, err := ioutil.ReadAll(res.Body)
	r.NoError(err)
	r.Contains(string(b), "does not compute")

	r.NoError(f.Close())
}

func Test_Open_Dir(t *testing.T) {
	r := require.New(t)

	f, err := Open("/")
	r.NoError(err)

	ts := httptest.NewServer(http.FileServer(f))
	defer ts.Close()

	res, err := http.Get(ts.URL + "/cmd/pkger")
	r.NoError(err)
	r.Equal(200, res.StatusCode)

	b, err := ioutil.ReadAll(res.Body)
	r.NoError(err)
	r.Contains(string(b), `<a href="/cmd/pkger/main.go">/cmd/pkger/main.go</a>`)

	r.NoError(f.Close())
}

func Test_Open_File_Memory(t *testing.T) {
	r := require.New(t)

	f, err := Create("/suit/case.txt")
	r.NoError(err)
	f.Write([]byte(radio))
	r.NoError(f.Close())

	r.Equal([]byte(radio), f.data)
	r.Contains(string(f.data), "I wanna bite the hand that feeds me")

	dir, err := Open("/")
	r.NoError(err)
	defer dir.Close()

	ts := httptest.NewServer(http.FileServer(dir))
	defer ts.Close()

	res, err := http.Get(ts.URL + "/suit/case.txt")
	r.NoError(err)
	r.Equal(200, res.StatusCode)

	b, _ := ioutil.ReadAll(res.Body)
	// r.NoError(err)
	r.Contains(string(b), "I wanna bite the hand that feeds me")

}

func Test_Open_Dir_StripPrefix(t *testing.T) {
	r := require.New(t)

	ts := httptest.NewServer(http.StripPrefix("/assets/", http.FileServer(http.Dir("./testdata/public"))))
	defer ts.Close()

	res, err := http.Get(ts.URL + "/assets/radio.radio")
	r.NoError(err)
	r.Equal(200, res.StatusCode)

	b, _ := ioutil.ReadAll(res.Body)
	// r.NoError(err)
	r.Contains(string(b), "I wanna bite the hand that feeds me")

	res, err = http.Get(ts.URL + "/assets/")
	r.NoError(err)
	r.Equal(200, res.StatusCode)

	b, _ = ioutil.ReadAll(res.Body)
	// r.NoError(err)
	r.Contains(string(b), `<a href="radio.radio">radio.radio</a>`)
}

func Test_Open_Dir_Memory_StripPrefix(t *testing.T) {
	r := require.New(t)

	err := MkdirAll("/testdata/public", 0755)
	r.NoError(err)

	dir, err := Open("/testdata/public")
	r.NoError(err)
	defer dir.Close()

	ts := httptest.NewServer(http.StripPrefix("/assets/", http.FileServer(dir)))
	defer ts.Close()

	res, err := http.Get(ts.URL + "/assets/radio.radio")
	r.NoError(err)
	r.Equal(200, res.StatusCode)

	b, _ := ioutil.ReadAll(res.Body)
	// r.NoError(err)
	r.Contains(string(b), "I wanna bite the hand that feeds me")

	res, err = http.Get(ts.URL + "/assets/")
	r.NoError(err)
	r.Equal(200, res.StatusCode)

	b, _ = ioutil.ReadAll(res.Body)
	// r.NoError(err)
	r.Contains(string(b), `<a href="/radio.radio">/radio.radio</a>`)
	r.NotContains(string(b), `/public`)
	r.NotContains(string(b), `//`)
}