Fix tty test and pkg doc

client_test.go

@@ -43,8 +43,8 @@ func TestRemoteStdin(t *testing.T) {
 
 		header, body := proto.SplitMessage(msg)
 
-		assert.Equal(t, "stdin body", body, []byte(tcase), bytecmp)
-		assert.Equal(t, "stdin header", header, []byte(`{"type":"stdin"}`), bytecmp)
+		assert.Equal(t, "stdin body", []byte(tcase), body, bytecmp)
+		assert.Equal(t, "stdin header", []byte(`{"type":"stdin"}`), header, bytecmp)
 	}
 }
 

doc.go

@@ -0,0 +1,6 @@
+// Package wsep provides server and client interfaces
+// for transmitting Linux command execution over a Websocket connection.
+
+// It provides stdin, stdout, stderr, tty operations, and exit code information.
+
+package wsep

internal/proto/protocol_test.go

@@ -37,7 +37,7 @@ func TestWithHeader(t *testing.T) {
 		assert.Success(t, "read buffer", err)
 
 		header, body := SplitMessage(msg)
-		assert.Equal(t, "header is expected value", header, tcase.header, bytecmp)
-		assert.Equal(t, "body is expected value", body, tcase.body, bytecmp)
+		assert.Equal(t, "header is expected value", tcase.header, header, bytecmp)
+		assert.Equal(t, "body is expected value", tcase.body, body, bytecmp)
 	}
 }

localexec_test.go

@@ -43,7 +43,7 @@ func testExecer(ctx context.Context, t *testing.T, execer Execer) {
 
 		stderr, err := ioutil.ReadAll(process.Stderr())
 		assert.Success(t, "read stderr", err)
-		assert.Equal(t, "len stderr", 0, len(stderr))
+		assert.True(t, "len stderr", len(stderr) == 0)
 	}()
 
 	err = process.Wait()

tty_test.go

@@ -37,8 +37,11 @@ func testTTY(ctx context.Context, t *testing.T, e Execer) {
 
 		stdout, err := ioutil.ReadAll(process.Stdout())
 		assert.Success(t, "read stdout", err)
+
 		t.Logf("bash tty stdout = %s", stdout)
-		assert.True(t, `bash "$" prompt found`, strings.HasSuffix(string(stdout), "$ "))
+		prompt := string(stdout)
+		assert.True(t, `bash "$" prompt found`,
+			strings.HasSuffix(prompt, "$ ") || strings.HasSuffix(prompt, "# "))
 	}()
 	go func() {
 		wg.Add(1)