added omikuji

Author: isuzuki

kadai4/main.go

@@ -3,63 +3,32 @@ 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"
+	"math/rand"
 	"net/http"
 	"time"
 
-	"github.com/isuzuki/omikuji/omikuji"
+	"github.com/gopherdojo/dojo6/kadai4/omikuji"
 )
 
 func main() {
+	rand.Seed(time.Now().UnixNano())
+
 	http.HandleFunc("/", handler)
 	http.ListenAndServe(":8080", nil)
 }
 
-type Lottery struct {
+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")
 
 	_, 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: "エラーが発生しました。"}
-	}
+	ret := omikuji.Do(int(m), d)
+	res := Response{Msg: ret}
 
 	if err := json.NewEncoder(w).Encode(res); err != nil {
 		log.Println("Error:", err)
 	}
 }
-*/

kadai4/main_test.go

@@ -24,7 +24,7 @@ func TestHandler(t *testing.T) {
 		t.Fatal("unexpected error")
 	}
 
-	const expected = `{"msg":"json response"}`
+	const expected = `{"msg":"大吉"}`
 	if s := string(b); s != expected {
 		t.Fatalf("unexpected response: %s", s)
 	}

kadai4/omikuji/omikuji.go

@@ -0,0 +1,31 @@
+package omikuji
+
+import (
+	"math/rand"
+)
+
+var omikuji = map[int]string{
+	0: "凶",
+	1: "末吉",
+	2: "小吉",
+	3: "中吉",
+	4: "吉",
+	5: "大吉",
+}
+
+func Do(month, day int) string {
+	var i int
+	// 1/1 ~ 1/3のみ大吉を出す
+	if month == 1 && day >= 1 && day <= 3 {
+		i = 5
+	} else {
+		i = rand.Intn(len(omikuji))
+	}
+
+	s, ok := omikuji[i]
+	if !ok {
+		panic("omikuji panic.")
+	}
+
+	return s
+}

kadai4/omikuji/omikuji_test.go

@@ -0,0 +1,26 @@
+package omikuji_test
+
+import (
+	"testing"
+
+	"github.com/gopherdojo/dojo6/kadai4/omikuji"
+)
+
+func Test_SpecificPeriodLottery(t *testing.T) {
+	periods := map[int][]int{
+		1: {1, 2, 3},
+	}
+
+	expect := "大吉"
+
+	for m, days := range periods {
+		for _, d := range days {
+			for i := 0; i < 20; i++ {
+				actual := omikuji.Do(m, d)
+				if expect != actual {
+					t.Errorf(`Omikuji error: expect="%s" actual="%s"`, expect, actual)
+				}
+			}
+		}
+	}
+}