test serveGMVote

src/device/input/parse/v01/server_test.go

@@ -82,9 +82,13 @@ func TestServeGM(t *testing.T) {
 	ctx, can := context.WithCancel(context.Background())
 	defer can()
 
-	do := func(v01 *V01, path, body string) *httptest.ResponseRecorder {
+	do := func(v01 *V01, path, body string, method ...string) *httptest.ResponseRecorder {
+		m := http.MethodPost
+		if len(method) > 0 {
+			m = method[0]
+		}
 		w := httptest.NewRecorder()
-		r := httptest.NewRequest(http.MethodPost, path, strings.NewReader(body))
+		r := httptest.NewRequest(m, path, strings.NewReader(body))
 		v01.ServeHTTP(w, r)
 		return w
 	}
@@ -235,7 +239,55 @@ func TestServeGM(t *testing.T) {
 	})
 
 	t.Run("vote", func(t *testing.T) {
-		t.Error("not impl")
+		type result map[string]string
+
+		t.Run("get non vote", func(t *testing.T) {
+			v01 := NewV01(ctx, nil)
+			v01.cfg.Users = map[string]configUser{"bel": {}}
+			resp := do(v01, "/gm/rpc/vote", "", "GET")
+			var result result
+			if err := yaml.Unmarshal(resp.Body.Bytes(), &result); err != nil {
+				t.Error(err)
+			}
+			if len(result) != 1 {
+				t.Error(result)
+			}
+			if result["bel"] != "voting" {
+				t.Error(result)
+			}
+			t.Logf("%+v", result)
+		})
+
+		t.Run("get mid vote", func(t *testing.T) {
+			v01 := NewV01(ctx, nil)
+			v01.cfg.Users = map[string]configUser{"bel": {Alias: "//zach"}}
+			resp := do(v01, "/gm/rpc/vote", "", "GET")
+			var result result
+			if err := yaml.Unmarshal(resp.Body.Bytes(), &result); err != nil {
+				t.Error(err)
+			}
+			if len(result) != 1 {
+				t.Error(result)
+			}
+			if result["bel"] != "voted" {
+				t.Error(result)
+			}
+			t.Logf("%+v", result)
+		})
+
+		t.Run("get empty", func(t *testing.T) {
+			v01 := NewV01(ctx, nil)
+			v01.cfg.Users = nil
+			resp := do(v01, "/gm/rpc/vote", "", "GET")
+			var result result
+			if err := yaml.Unmarshal(resp.Body.Bytes(), &result); err != nil {
+				t.Error(err)
+			}
+			if len(result) != 0 {
+				t.Error(result)
+			}
+			t.Logf("%+v", result)
+		})
 	})
 
 	t.Run("elect", func(t *testing.T) {