more upload tests

1 files changed, 72 insertions, 1 deletions

diff --git a/service/http/uploads_test.go b/service/http/uploads_test.go
index 80599884..deb5fa57 100644
--- a/service/http/uploads_test.go
+++ b/service/http/uploads_test.go
@@ -78,6 +78,72 @@ func TestServer_Upload_File(t *testing.T) {
 	assert.Equal(t, `{"upload":`+fs+`}`, string(b))
 }
 
+func TestServer_Upload_File_NoTmpDir(t *testing.T) {
+	st := &Handler{
+		cfg: &Config{
+			MaxRequest: 1024,
+			Uploads: &UploadsConfig{
+				Dir:    "-----",
+				Forbid: []string{},
+			},
+		},
+		rr: roadrunner.NewServer(&roadrunner.ServerConfig{
+			Command: "php ../../php-src/tests/http/client.php upload pipes",
+			Relay:   "pipes",
+			Pool: &roadrunner.Config{
+				NumWorkers:      1,
+				AllocateTimeout: 10000000,
+				DestroyTimeout:  10000000,
+			},
+		}),
+	}
+
+	assert.NoError(t, st.rr.Start())
+	defer st.rr.Stop()
+
+	hs := &http.Server{Addr: ":8021", Handler: st,}
+	defer hs.Shutdown(context.Background())
+
+	go func() { hs.ListenAndServe() }()
+	time.Sleep(time.Millisecond * 10)
+
+	var mb bytes.Buffer
+	w := multipart.NewWriter(&mb)
+
+	f := mustOpen("uploads_test.go")
+	defer f.Close()
+	fw, err := w.CreateFormFile("upload", f.Name())
+	assert.NotNil(t, fw)
+	assert.NoError(t, err)
+	io.Copy(fw, f)
+
+	w.Close()
+
+	req, err := http.NewRequest("POST", "http://localhost"+hs.Addr, &mb)
+	assert.NoError(t, err)
+
+	req.Header.Set("Content-Type", w.FormDataContentType())
+
+	r, err := http.DefaultClient.Do(req)
+	assert.NoError(t, err)
+	defer r.Body.Close()
+
+	b, err := ioutil.ReadAll(r.Body)
+	assert.NoError(t, err)
+
+	assert.NoError(t, err)
+	assert.Equal(t, 200, r.StatusCode)
+
+	fs := fileString("uploads_test.go", 5, "application/octet-stream")
+
+	assert.Equal(t, `{"upload":`+fs+`}`, string(b))
+}
+
+func Test_FileExists(t *testing.T) {
+	assert.True(t, exists("uploads_test.go"))
+	assert.False(t, exists("uploads_test."))
+}
+
 func mustOpen(f string) *os.File {
 	r, err := os.Open(f)
 	if err != nil {
@@ -91,7 +157,7 @@ type fInfo struct {
 	Size  int64  `json:"size"`
 	Mime  string `json:"mime"`
 	Error int    `json:"error"`
-	MD5   string `json:"md5"`
+	MD5   string `json:"md5,omitempty"`
 }
 
 func fileString(f string, err int, mime string) string {
@@ -110,6 +176,11 @@ func fileString(f string, err int, mime string) string {
 		MD5:   hex.EncodeToString(h.Sum(nil)),
 	}
 
+	if err != 0 {
+		v.MD5 = ""
+		v.Size = 0
+	}
+
 	r, _ := json.Marshal(v)
 	return string(r)