Add test to FetchGopher

Author: aperezg

pkg/server/api_test.go

@@ -2,6 +2,7 @@ package server
 
 import (
 	"encoding/json"
+	"fmt"
 	"io/ioutil"
 	"net/http"
 	"net/http/httptest"
@@ -47,3 +48,65 @@ func TestFetchGophers(t *testing.T) {
 		t.Errorf("expected %d gophers, got: %d gopher", sample.Gophers, got)
 	}
 }
+
+func TestFetchGopher(t *testing.T) {
+
+	testData := []struct {
+		name   string
+		g      *gopher.Gopher
+		status int
+		err    string
+	}{
+		{name: "gopher found", g: gopherSample(), status: http.StatusOK},
+		{name: "gopher not found", g: &gopher.Gopher{ID: "123"}, status: http.StatusNotFound, err: "Gopher Not found"},
+	}
+
+	for _, tt := range testData {
+		t.Run(tt.name, func(t *testing.T) {
+			uri := fmt.Sprintf("/gophers/%s", tt.g.ID)
+			req, err := http.NewRequest("GET", uri, nil)
+			if err != nil {
+				t.Fatalf("could not created request: %v", err)
+			}
+
+			repo := inmem.NewGopherRepository(sample.Gophers)
+			s := New(repo)
+
+			rec := httptest.NewRecorder()
+			s.Router().ServeHTTP(rec, req)
+
+			res := rec.Result()
+
+			defer res.Body.Close()
+			if tt.status != res.StatusCode {
+				t.Errorf("expected %d, got: %d", tt.status, res.StatusCode)
+			}
+			b, err := ioutil.ReadAll(res.Body)
+			if err != nil {
+				t.Fatalf("could not read response: %v", err)
+			}
+
+			if tt.err == "" {
+				var got *gopher.Gopher
+				err = json.Unmarshal(b, &got)
+				if err != nil {
+					t.Fatalf("could not unmarshall response %v", err)
+				}
+
+				if *got != *tt.g {
+					t.Fatalf("expected %v, got: %v", tt.g, got)
+				}
+			}
+		})
+	}
+
+}
+
+func gopherSample() *gopher.Gopher {
+	return &gopher.Gopher{
+		ID:    "01D3XZ3ZHCP3KG9VT4FGAD8KDR",
+		Name:  "Jenny",
+		Age:   18,
+		Image: "https://storage.googleapis.com/gopherizeme.appspot.com/gophers/0ceb2c10fc0c30575c18ff1defa1ffd41501bc62.png",
+	}
+}