Kadai4 ramenjuniti

kadai4/ramenjuniti/Makefile

@@ -0,0 +1,14 @@
+BIN := gokuji
+
+.PHONY: test
+test:
+	go test -cover -v ./...
+
+.PHONY: build
+build: test
+	go build -o $(BIN)
+
+.PHONY: clean
+clean:
+	rm $(BIN)
+	go clean

kadai4/ramenjuniti/README.md

@@ -0,0 +1,23 @@
+# gokuji
+
+## Description
+
+おみくじ API です。1/1 - 1/3 は大吉だけが出ます。
+
+## Usage
+
+```bash
+./gokuji -p {ポート番号（デフォルトは8080）}
+```
+
+## Build
+
+```bash
+make build
+```
+
+## Test
+
+```bash
+make test
+```

kadai4/ramenjuniti/main.go

@@ -0,0 +1,23 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"net/http"
+	"time"
+
+	"github.com/gopherdojo/dojo5/kadai4/ramenjuniti/omikuji"
+)
+
+var port int
+
+func init() {
+	flag.IntVar(&port, "p", 8080, "port number")
+}
+
+func main() {
+	flag.Parse()
+	o := omikuji.New(time.Now())
+	http.HandleFunc("/", o.Handler)
+	http.ListenAndServe(fmt.Sprintf(":%d", port), nil)
+}

kadai4/ramenjuniti/omikuji/omikuji.go

@@ -0,0 +1,55 @@
+package omikuji
+
+import (
+	"encoding/json"
+	"log"
+	"math/rand"
+	"net/http"
+	"time"
+)
+
+// Omikuji contains time
+type Omikuji struct {
+	time time.Time
+}
+
+// Result contains type of omikuji result
+type Result struct {
+	Type string `json:"type"`
+}
+
+var types = []string{
+	"大吉",
+	"吉",
+	"中吉",
+	"小吉",
+	"末吉",
+	"凶",
+	"大凶",
+}
+
+// New returns Omikuji instance
+func New(t time.Time) *Omikuji {
+	rand.Seed(t.UnixNano())
+	return &Omikuji{time: t}
+}
+
+// Handler returns omikuji result
+func (o *Omikuji) Handler(w http.ResponseWriter, r *http.Request) {
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+
+	result := draw(o.time)
+	if err := json.NewEncoder(w).Encode(result); err != nil {
+		log.Println("Error:", err)
+	}
+}
+
+func draw(t time.Time) *Result {
+	yd := t.YearDay()
+
+	if yd == 1 || yd == 2 || yd == 3 {
+		return &Result{Type: types[0]}
+	}
+
+	return &Result{Type: types[rand.Intn(len(types))]}
+}

kadai4/ramenjuniti/omikuji/omikuji_test.go

@@ -0,0 +1,77 @@
+package omikuji
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"time"
+)
+
+var cases = []struct {
+	name string
+	time time.Time
+}{
+	{
+		name: "case1",
+		time: time.Date(2019, 1, 1, 0, 0, 0, 0, time.UTC),
+	},
+	{
+		name: "case2",
+		time: time.Date(2019, 1, 1, 23, 59, 59, 0, time.UTC),
+	},
+	{
+		name: "case3",
+		time: time.Date(2019, 1, 2, 0, 0, 0, 0, time.UTC),
+	},
+	{
+		name: "case4",
+		time: time.Date(2019, 1, 2, 23, 59, 59, 0, time.UTC),
+	},
+	{
+		name: "case5",
+		time: time.Date(2019, 1, 3, 0, 0, 0, 0, time.UTC),
+	},
+	{
+		name: "case6",
+		time: time.Date(2019, 1, 3, 23, 59, 59, 0, time.UTC),
+	},
+}
+
+func getResult(time time.Time, t *testing.T) *Result {
+	t.Helper()
+	o := New(time)
+	w := httptest.NewRecorder()
+	r := httptest.NewRequest("GET", "/", nil)
+	o.Handler(w, r)
+	rw := w.Result()
+	defer rw.Body.Close()
+
+	if rw.StatusCode != http.StatusOK {
+		t.Fatal("unexpected status code")
+	}
+
+	b, err := ioutil.ReadAll(rw.Body)
+	if err != nil {
+		t.Fatal("unexpected error")
+	}
+
+	re := &Result{}
+	if err := json.Unmarshal(b, &re); err != nil {
+		t.Fatal("failed json unmarshal")
+	}
+
+	return re
+}
+
+func TestOmikuji(t *testing.T) {
+	for _, c := range cases {
+		t.Run(c.name, func(t *testing.T) {
+			re := getResult(c.time, t)
+			if re.Type != "大吉" {
+				t.Errorf("got %v, want %v", re.Type, "大吉")
+			}
+		})
+	}
+}