wip: added main (test failed)

Author: isuzuki

kadai4/main.go

@@ -0,0 +1,65 @@
+package main
+
+import (
+	"encoding/json"
+	"log"
+	"net/http"
+)
+
+func main() {
+	http.HandleFunc("/", handler)
+	http.ListenAndServe(":8080", nil)
+}
+
+type Response struct {
+	Msg string `json:"msg"`
+}
+
+func handler(w http.ResponseWriter, r *http.Request) {
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+
+	res := Response{Msg: "json response"}
+
+	if err := json.NewEncoder(w).Encode(res); err != nil {
+		log.Println("Error:", err)
+	}
+}
+
+/*
+package main
+
+import (
+	"encoding/json"
+	"log"
+	"net/http"
+	"time"
+
+	"github.com/isuzuki/omikuji/omikuji"
+)
+
+func main() {
+	http.HandleFunc("/", handler)
+	http.ListenAndServe(":8080", nil)
+}
+
+type Lottery struct {
+	Msg string `json:"msg"`
+}
+
+func handler(w http.ResponseWriter, r *http.Request) {
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+
+	_, m, d := time.Now().Date()
+	ret, ok := omikuji.Lottery(int(m), d)
+	var res Lottery
+	if ok {
+		res = Lottery{Msg: ret}
+	} else {
+		res = Lottery{Msg: "エラーが発生しました。"}
+	}
+
+	if err := json.NewEncoder(w).Encode(res); err != nil {
+		log.Println("Error:", err)
+	}
+}
+*/

kadai4/main_test.go

@@ -0,0 +1,31 @@
+package main
+
+import (
+	"io/ioutil"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+)
+
+func TestHandler(t *testing.T) {
+	w := httptest.NewRecorder()
+	r := httptest.NewRequest("GET", "/", nil)
+	handler(w, r)
+	rw := w.Result()
+
+	defer rw.Body.Close()
+
+	if rw.StatusCode != http.StatusOK {
+		t.Fatalf("unexpected status code: %d", rw.StatusCode)
+	}
+
+	b, err := ioutil.ReadAll(rw.Body)
+	if err != nil {
+		t.Fatal("unexpected error")
+	}
+
+	const expected = `{"msg":"json response"}`
+	if s := string(b); s != expected {
+		t.Fatalf("unexpected response: %s", s)
+	}
+}