Merge pull request #6 from gomicro/up-coverage

Up coverage

glide.yaml

@@ -1,9 +1,6 @@
 package: github.com/gomicro/steward
 import:
 - package: github.com/gorilla/mux
-  version: v1.1
 testImport:
 - package: github.com/franela/goblin
-  version: 0.0.1
 - package: github.com/onsi/gomega
-  version: v1.0

steward_test.go

@@ -68,6 +68,25 @@ func TestStewardEndpoint(t *testing.T) {
 			Expect(bodyBytes).To(ContainSubstring(`"version": "1.0.0",`))
 			Expect(bodyBytes).To(ContainSubstring(`"buildTime": "now"`))
 		})
+
+		g.It("should return an internal error for a bad payload", func() {
+			bad := func() {}
+			SetStatusResponse(bad)
+			req, err := http.NewRequest("GET", server.URL+endpoint, nil)
+			Expect(err).NotTo(HaveOccurred())
+
+			res, err := client.Do(req)
+			Expect(err).NotTo(HaveOccurred())
+			defer res.Body.Close()
+
+			Expect(res.StatusCode).To(Equal(http.StatusInternalServerError))
+		})
+
+		g.It("should set a custom status endpoint", func() {
+			e := "/foo/bar"
+			SetStatusEndpoint(e)
+			Expect(endpoint).To(Equal(e))
+		})
 	})
 }