Commit d3bac22a authored by Cao Shufeng's avatar Cao Shufeng

set right Content-Type for configz

parent 3cb73605
...@@ -19,7 +19,6 @@ package configz ...@@ -19,7 +19,6 @@ package configz
import ( import (
"encoding/json" "encoding/json"
"fmt" "fmt"
"io"
"net/http" "net/http"
"sync" "sync"
) )
...@@ -74,7 +73,7 @@ func handle(w http.ResponseWriter, r *http.Request) { ...@@ -74,7 +73,7 @@ func handle(w http.ResponseWriter, r *http.Request) {
} }
} }
func write(w io.Writer) error { func write(w http.ResponseWriter) error {
var b []byte var b []byte
var err error var err error
func() { func() {
...@@ -85,6 +84,7 @@ func write(w io.Writer) error { ...@@ -85,6 +84,7 @@ func write(w io.Writer) error {
if err != nil { if err != nil {
return fmt.Errorf("error marshaling json: %v", err) return fmt.Errorf("error marshaling json: %v", err)
} }
w.Header().Set("Content-Type", "application/json")
_, err = w.Write(b) _, err = w.Write(b)
return err return err
} }
...@@ -44,7 +44,7 @@ func TestConfigz(t *testing.T) { ...@@ -44,7 +44,7 @@ func TestConfigz(t *testing.T) {
t.Fatalf("err: %v", err) t.Fatalf("err: %v", err)
} }
if string(body) != `{"testing":"blah"}` { if string(body) != `{"testing":"blah"}` {
t.Fatalf("unexpected output: %v", err) t.Fatalf("unexpected output: %s", body)
} }
v.Set("bing") v.Set("bing")
...@@ -58,7 +58,7 @@ func TestConfigz(t *testing.T) { ...@@ -58,7 +58,7 @@ func TestConfigz(t *testing.T) {
t.Fatalf("err: %v", err) t.Fatalf("err: %v", err)
} }
if string(body) != `{"testing":"bing"}` { if string(body) != `{"testing":"bing"}` {
t.Fatalf("unexpected output: %v", err) t.Fatalf("unexpected output: %s", body)
} }
Delete("testing") Delete("testing")
...@@ -72,6 +72,9 @@ func TestConfigz(t *testing.T) { ...@@ -72,6 +72,9 @@ func TestConfigz(t *testing.T) {
t.Fatalf("err: %v", err) t.Fatalf("err: %v", err)
} }
if string(body) != `{}` { if string(body) != `{}` {
t.Fatalf("unexpected output: %v", err) t.Fatalf("unexpected output: %s", body)
}
if resp.Header.Get("Content-Type") != "application/json" {
t.Fatalf("unexpected Content-Type: %s", resp.Header.Get("Content-Type"))
} }
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment