Improved coverage

Author: David Koblas

handler_test.go

@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"encoding/json"
 	"fmt"
+	"io"
 	"io/ioutil"
 	"mime/multipart"
 	"net/http"
@@ -258,3 +259,103 @@ func TestHandler_Multipart_Basic(t *testing.T) {
 		t.Fatalf("wrong result, graphql result diff: %v", testutil.Diff(expected, result))
 	}
 }
+
+func TestHandler_Multipart_Basic_ErrNoOperation(t *testing.T) {
+	body := &bytes.Buffer{}
+	writer := multipart.NewWriter(body)
+
+	err := writer.Close()
+	if err != nil {
+		t.Fatalf("unexpected writer fail %v", err)
+	}
+
+	req, err := http.NewRequest("POST", "/graphql", body)
+	req.Header.Set("Content-Type", writer.FormDataContentType())
+
+	h := handler.New(&handler.Config{
+		Schema: &testutil.StarWarsSchema,
+	})
+	result, resp := executeTest(t, h, req)
+	if resp.Code != http.StatusOK {
+		t.Fatalf("unexpected server response %v", resp.Code)
+	}
+	if len(result.Errors) != 1 || result.Errors[0].Message != "Must provide an operation." {
+		t.Fatalf("unexpected response")
+	}
+}
+
+func TestHandler_Multipart_Basic_ErrBadMap(t *testing.T) {
+	body := &bytes.Buffer{}
+	writer := multipart.NewWriter(body)
+
+	queryString := `{"query":"query HeroNameQuery { hero { name } }"}`
+	writer.WriteField("operations", queryString)
+	writer.WriteField("map", "{")
+
+	err := writer.Close()
+	if err != nil {
+		t.Fatalf("unexpected writer fail %v", err)
+	}
+
+	req, err := http.NewRequest("POST", "/graphql", body)
+	req.Header.Set("Content-Type", writer.FormDataContentType())
+
+	h := handler.New(&handler.Config{
+		Schema: &testutil.StarWarsSchema,
+	})
+	result, resp := executeTest(t, h, req)
+	if resp.Code != http.StatusOK {
+		t.Fatalf("unexpected server response %v", resp.Code)
+	}
+	if len(result.Errors) != 1 || result.Errors[0].Message != "Must provide an operation." {
+		t.Fatalf("unexpected response")
+	}
+}
+
+func TestHandler_Multipart_Basic_Upload(t *testing.T) {
+	expected := &graphql.Result{
+		Data: map[string]interface{}{
+			"hero": map[string]interface{}{
+				"name": "R2-D2",
+			},
+		},
+	}
+
+	body := &bytes.Buffer{}
+	writer := multipart.NewWriter(body)
+
+	queryString := `{
+		"query":"query HeroNameQuery { hero { name } }",
+		"variables":{"file":null}
+	}`
+
+	writer.WriteField("operations", queryString)
+	writer.WriteField("map", `{"0":["variables.file"]}`)
+	part, _ := writer.CreateFormFile("0", "test.txt")
+
+	if _, err := io.Copy(part, strings.NewReader("How now brown cow")); err != nil {
+		t.Fatalf("unexpected copy writer fail %v", err)
+	}
+
+	err := writer.Close()
+	if err != nil {
+		t.Fatalf("unexpected writer fail %v", err)
+	}
+
+	req, err := http.NewRequest("POST", "/graphql", body)
+	req.Header.Set("Content-Type", writer.FormDataContentType())
+
+	h := handler.New(&handler.Config{
+		Schema: &testutil.StarWarsSchema,
+	})
+	result, resp := executeTest(t, h, req)
+	if resp.Code != http.StatusOK {
+		t.Fatalf("unexpected server response %v", resp.Code)
+	}
+	if resp.Code != http.StatusOK {
+		t.Fatalf("unexpected server response %v", resp.Code)
+	}
+	if !reflect.DeepEqual(result, expected) {
+		t.Fatalf("wrong result, graphql result diff: %v", testutil.Diff(expected, result))
+	}
+}