add test

pkg/listen/listen_test.go

@@ -21,13 +21,23 @@ var (
 	upgrader = websocket.Upgrader{}
 )
 
-func wsHandler(t *testing.T) http.HandlerFunc {
+func wsHandler(t *testing.T, recvBuffer *bytes.Buffer) http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {
 		c, err := upgrader.Upgrade(w, r, nil)
 		require.NoError(t, err)
 		defer c.Close()
 		i := 0
 		finish := 5
+		go func() {
+			// Read messages from websocket and write to buffer
+			for {
+				_, message, err := c.ReadMessage()
+				if err != nil {
+					return
+				}
+				recvBuffer.Write(message)
+			}
+		}()
 		for {
 			// Give the Close test a chance to close before any sent
 			time.Sleep(time.Millisecond * 10)
@@ -55,7 +65,7 @@ func wsHandler(t *testing.T) http.HandlerFunc {
 }
 
 func TestListener(t *testing.T) {
-	server := httptest.NewServer(wsHandler(t))
+	server := httptest.NewServer(wsHandler(t, nil))
 	defer server.Close()
 
 	u := "ws" + strings.TrimPrefix(server.URL, "http")
@@ -78,7 +88,7 @@ func TestListener(t *testing.T) {
 }
 
 func TestListenerWithSchemaFunc(t *testing.T) {
-	server := httptest.NewServer(wsHandler(t))
+	server := httptest.NewServer(wsHandler(t, nil))
 	defer server.Close()
 
 	u := "ws" + strings.TrimPrefix(server.URL, "http")
@@ -112,3 +122,31 @@ func TestListenerWithSchemaFunc(t *testing.T) {
 `
 	require.Equal(t, want, buffer.String())
 }
+
+func TestListenerWithInput(t *testing.T) {
+	wsInBuf := &bytes.Buffer{}
+	server := httptest.NewServer(wsHandler(t, wsInBuf))
+	defer server.Close()
+
+	u := "ws" + strings.TrimPrefix(server.URL, "http")
+	url, err := url.Parse(u)
+	require.NoError(t, err)
+
+	inputCh := make(chan []byte, 5)
+	for i := 0; i < 5; i++ {
+		inputCh <- []byte{byte('a' + i)}
+	}
+	wsOutBuf := &bytes.Buffer{}
+	listener := NewListener(url, "", nil, wsOutBuf, inputCh)
+	err = listener.Listen(context.Background())
+	require.NoError(t, err)
+
+	want := `{"message":"1\n"}
+{"message":"2\n"}
+{"message":"3\n"}
+{"message":"4\n"}
+{"message":"5\n"}
+`
+	require.Equal(t, want, wsOutBuf.String())
+	require.Equal(t, "abcde", wsInBuf.String())
+}

pkg/listen/listen_windows.go

@@ -22,11 +22,7 @@ func (l *Listener) MonitorResizeEvents(ctx context.Context, resizeEvents chan<-
 		terminalSize := TerminalSize{Width: width, Height: height}
 		if terminalSize != prevTerminalSize {
 			prevTerminalSize = terminalSize
-			select {
-			case resizeEvents <- terminalSize:
-			case <-ctx.Done():
-				return nil
-			}
+			resizeEvents <- terminalSize
 		}
 
 		// sleep to avoid hot looping