init

selenium/base/fileserver/main_test.go

@@ -0,0 +1,58 @@
+package main
+
+import (
+	. "github.com/aandryashin/matchers"
+	. "github.com/aandryashin/matchers/httpresp"
+	"io/ioutil"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"path/filepath"
+	"testing"
+)
+
+var (
+	dir string
+	srv *httptest.Server
+)
+
+func init() {
+	dir, _ = ioutil.TempDir("", "fileserver")
+	srv = httptest.NewServer(mux(dir))
+}
+
+func withUrl(path string) string {
+	return srv.URL + path
+}
+
+func TestDownloadAndRemoveFile(t *testing.T) {
+	tempFile, _ := ioutil.TempFile(dir, "fileserver")
+	ioutil.WriteFile(tempFile.Name(), []byte("test-data"), 0644)
+	tempFileName := filepath.Base(tempFile.Name())
+	resp, err := http.Get(withUrl("/" + tempFileName))
+	AssertThat(t, err, Is{nil})
+	AssertThat(t, resp, Code{200})
+	_, err = os.Stat(tempFile.Name())
+	AssertThat(t, err, Is{nil})
+
+	rsp, err := http.Get(withUrl("/?json"))
+	AssertThat(t, err, Is{nil})
+	AssertThat(t, rsp, Code{http.StatusOK})
+	var files []string
+	AssertThat(t, rsp, IsJson{&files})
+	AssertThat(t, files, EqualTo{[]string{tempFileName}})
+
+	req, _ := http.NewRequest(http.MethodDelete, withUrl("/" + tempFileName), nil)
+	resp, err = http.DefaultClient.Do(req)
+	AssertThat(t, err, Is{nil})
+	AssertThat(t, resp, Code{200})
+	_, err = os.Stat(tempFile.Name())
+	AssertThat(t, err, Not{nil})
+}
+
+func TestRemoveMissingFile(t *testing.T) {
+	req, _ := http.NewRequest(http.MethodDelete, withUrl("/missing-file"), nil)
+	resp, err := http.DefaultClient.Do(req)
+	AssertThat(t, err, Is{nil})
+	AssertThat(t, resp, Code{404})
+}