test: throttle history verification

loadtest/im-pressure/wsclient_test.go

@@ -48,3 +48,39 @@ func TestWSClientKeepaliveSendsPing(t *testing.T) {
 		t.Fatal("websocket keepalive ping was not received")
 	}
 }
+
+func TestWSClientReadLoopClosesConnectionAfterDecodeError(t *testing.T) {
+	serverSawClose := make(chan struct{}, 1)
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		conn, err := (&websocket.Upgrader{}).Upgrade(w, r, nil)
+		if err != nil {
+			return
+		}
+		defer conn.Close()
+		if err = conn.WriteMessage(websocket.TextMessage, []byte("{")); err != nil {
+			return
+		}
+		if _, _, err = conn.ReadMessage(); err != nil {
+			serverSawClose <- struct{}{}
+		}
+	}))
+	defer server.Close()
+
+	conn, _, err := websocket.DefaultDialer.Dial("ws"+strings.TrimPrefix(server.URL, "http"), nil)
+	if err != nil {
+		t.Fatal(err)
+	}
+	client := &WSClient{conn: conn, waits: make(map[string]chan ctrlPacket), done: make(chan struct{})}
+	go client.readLoop()
+
+	select {
+	case <-client.done:
+	case <-time.After(time.Second):
+		t.Fatal("read loop did not exit after malformed frame")
+	}
+	select {
+	case <-serverSawClose:
+	case <-time.After(time.Second):
+		t.Fatal("read loop exited without closing the underlying connection")
+	}
+}