add response struct

Author: NOTYWTR

request.go

@@ -14,9 +14,18 @@ import (
 	"strings"
 )
 
+//UploadResponse response struct
+type UploadResponse struct {
+	FileID     string `json:"fileId"`
+	UploadedAt string `json:"uploadedAt"`
+	Status     string `json:"status"`
+	Message    string `json:"message"`
+	Code       string `json:"code"`
+}
+
 // Upload uploads files with fileName
 // options are list of fieldSets constant
-func (s *Client) Upload(fileName string, options []string, workflowID string) (resp map[string]interface{}, err error) {
+func (s *Client) Upload(fileName string, options []string, workflowID string) (resp UploadResponse, err error) {
 	payload := &bytes.Buffer{}
 	writer := multipart.NewWriter(payload)
 	file, err := os.Open(fileName)
@@ -64,7 +73,8 @@ func (s *Client) Upload(fileName string, options []string, workflowID string) (r
 	if err != nil {
 		return
 	}
-	json.Unmarshal(body, &resp)
+	err = json.Unmarshal(body, &resp)
+
 	return
 }
 

request_test.go

@@ -32,7 +32,7 @@ func TestSypht_UploadFile(t *testing.T) {
 	if err != nil {
 		t.Error(err)
 	}
-	if resp["fileId"].(string) == "" {
+	if resp.FileID == "" {
 		t.Error("Empty fileID")
 	}
 }
@@ -49,11 +49,11 @@ func TestSypht_PredictionWithCustomFieldSet(t *testing.T) {
 	if err != nil {
 		t.Error(err)
 	}
-	if resp["fileId"].(string) == "" {
+	if resp.FileID == "" {
 		t.Error("Empty fileID")
 	}
 	time.Sleep(time.Second * 10)
-	_, err = client.Results(resp["fileId"].(string))
+	_, err = client.Results(resp.FileID)
 	if err != nil {
 		t.Error(err)
 	}