Add main

Author: hioki-daichi

kadai4/hioki-daichi/main.go

@@ -0,0 +1,60 @@
+package main
+
+import (
+	"fmt"
+	"log"
+	"math/rand"
+	"net/http"
+	"time"
+
+	"github.com/gopherdojo/dojo3/kadai4/hioki-daichi/datehelper"
+	"github.com/gopherdojo/dojo3/kadai4/hioki-daichi/form"
+	"github.com/gopherdojo/dojo3/kadai4/hioki-daichi/fortune"
+	"github.com/gopherdojo/dojo3/kadai4/hioki-daichi/jsonhelper"
+)
+
+// for testing
+var nowFunc = time.Now
+var isDuringTheNewYearFunc = datehelper.IsDuringTheNewYear
+var toJSONFunc = jsonhelper.ToJSON
+
+func init() {
+	rand.Seed(nowFunc().UnixNano())
+}
+
+func main() {
+	http.HandleFunc("/", handler)
+	http.ListenAndServe(":8080", nil)
+}
+
+func handler(w http.ResponseWriter, r *http.Request) {
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.Header().Set("X-Content-Type-Options", "nosniff")
+
+	var ftn fortune.Fortune
+	if isDuringTheNewYearFunc() {
+		ftn = fortune.Daikichi
+	} else {
+		ftn = fortune.DrawFortune()
+	}
+
+	p := form.NewRootForm(r).NewPerson(ftn)
+
+	p.Validate()
+
+	var v interface{}
+	if len(p.Errors) > 0 {
+		w.WriteHeader(http.StatusBadRequest)
+		v = map[string][]string{"errors": p.Errors}
+	} else {
+		v = p
+	}
+
+	json, err := toJSONFunc(v)
+	if err != nil {
+		log.Println(err)
+		http.Error(w, "Internal Server Error", http.StatusInternalServerError)
+	}
+
+	fmt.Fprint(w, json)
+}

kadai4/hioki-daichi/main_test.go

@@ -0,0 +1,144 @@
+package main
+
+import (
+	"errors"
+	"io/ioutil"
+	"math/rand"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+)
+
+func TestMain_handler_StatusCode(t *testing.T) {
+	w := httptest.NewRecorder()
+	req := httptest.NewRequest("GET", "/", nil)
+	handler(w, req)
+	rw := w.Result()
+	defer rw.Body.Close()
+
+	expected := http.StatusOK
+	actual := rw.StatusCode
+	if actual != expected {
+		t.Errorf(`unexpected status code: expected: "%d" actual: "%d"`, expected, actual)
+	}
+}
+
+func TestMain_handler_ResponseBody(t *testing.T) {
+	cases := map[string]struct {
+		seed      int64
+		nameParam string
+		expected  string
+	}{
+		"KYOU":       {seed: 0, nameParam: "", expected: "{\"name\":\"Gopher\",\"fortune\":\"凶\"}\n"},
+		"DAIKYOU":    {seed: 1, nameParam: "", expected: "{\"name\":\"Gopher\",\"fortune\":\"大凶\"}\n"},
+		"SUEKICHI":   {seed: 2, nameParam: "", expected: "{\"name\":\"Gopher\",\"fortune\":\"末吉\"}\n"},
+		"KICHI":      {seed: 3, nameParam: "", expected: "{\"name\":\"Gopher\",\"fortune\":\"吉\"}\n"},
+		"CHUKICHI":   {seed: 4, nameParam: "", expected: "{\"name\":\"Gopher\",\"fortune\":\"中吉\"}\n"},
+		"SHOKICHI":   {seed: 5, nameParam: "", expected: "{\"name\":\"Gopher\",\"fortune\":\"小吉\"}\n"},
+		"DAICHIKI":   {seed: 9, nameParam: "", expected: "{\"name\":\"Gopher\",\"fortune\":\"大吉\"}\n"},
+		"name param": {seed: 9, nameParam: "hioki-daichi", expected: "{\"name\":\"hioki-daichi\",\"fortune\":\"大吉\"}\n"},
+	}
+
+	for n, c := range cases {
+		c := c
+		t.Run(n, func(t *testing.T) {
+			rand.Seed(c.seed)
+
+			w := httptest.NewRecorder()
+			req := httptest.NewRequest("GET", "/", nil)
+
+			if c.nameParam != "" {
+				q := req.URL.Query()
+				q.Add("name", c.nameParam)
+				req.URL.RawQuery = q.Encode()
+			}
+
+			handler(w, req)
+			rw := w.Result()
+			defer rw.Body.Close()
+
+			b, err := ioutil.ReadAll(rw.Body)
+			if err != nil {
+				t.Fatalf("err %s", err)
+			}
+
+			expected := c.expected
+			actual := string(b)
+			if actual != expected {
+				t.Errorf(`unexpected response body: expected: "%s" actual: "%s"`, expected, actual)
+			}
+		})
+	}
+}
+
+func TestMain_handler_DuringTheNewYear(t *testing.T) {
+	isDuringTheNewYearFunc = func() bool {
+		return true
+	}
+
+	w := httptest.NewRecorder()
+	req := httptest.NewRequest("GET", "/", nil)
+	handler(w, req)
+	rw := w.Result()
+	defer rw.Body.Close()
+
+	b, err := ioutil.ReadAll(rw.Body)
+	if err != nil {
+		t.Fatalf("err %s", err)
+	}
+
+	expected := "{\"name\":\"Gopher\",\"fortune\":\"大吉\"}\n"
+	actual := string(b)
+	if actual != expected {
+		t.Errorf(`unexpected response body: expected: "%s" actual: "%s"`, expected, actual)
+	}
+}
+
+func TestMain_handler_ValidationError(t *testing.T) {
+	w := httptest.NewRecorder()
+	req := httptest.NewRequest("GET", "/", nil)
+	q := req.URL.Query()
+	q.Add("name", "123456789012345678901234567890123")
+	req.URL.RawQuery = q.Encode()
+	handler(w, req)
+	rw := w.Result()
+	defer rw.Body.Close()
+
+	b, err := ioutil.ReadAll(rw.Body)
+	if err != nil {
+		t.Fatalf("err %s", err)
+	}
+
+	if rw.StatusCode != http.StatusBadRequest {
+		t.Errorf(`unexpected status code: expected: %d actual: %d`, http.StatusBadRequest, rw.StatusCode)
+	}
+
+	expected := "{\"errors\":[\"Name is too long (maximum is 32 characters)\"]}\n"
+	actual := string(b)
+	if actual != expected {
+		t.Errorf(`unexpected response body: expected: "%s" actual: "%s"`, expected, actual)
+	}
+}
+
+func TestMain_handler_ToJSONError(t *testing.T) {
+	toJSONFunc = func(v interface{}) (string, error) {
+		return "", errors.New("error in TestMain_handler_ToJSONError")
+	}
+
+	w := httptest.NewRecorder()
+	req := httptest.NewRequest("GET", "/", nil)
+	handler(w, req)
+	rw := w.Result()
+	defer rw.Body.Close()
+
+	b, err := ioutil.ReadAll(rw.Body)
+	if err != nil {
+		t.Fatalf("err %s", err)
+	}
+
+	expected := "Internal Server Error\n"
+	actual := string(b)
+	if actual != expected {
+		t.Errorf(`unexpected response body: expected: "%s" actual: "%s"`, expected, actual)
+	}
+}