Skip to content

refactor: PTY & SSH #7100

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 22 commits into from
Apr 24, 2023
Merged

refactor: PTY & SSH #7100

Changes from 1 commit
Commits
Show all changes
22 commits
Select commit Hold shift + click to select a range
0075d7d
Add ssh tests for longoutput, orphan
spikecurtis Apr 11, 2023
a491d4f
PTY/SSH tests & improvements
spikecurtis Apr 12, 2023
2cf357a
Fix some tests
spikecurtis Apr 12, 2023
28c0646
Fix linting
spikecurtis Apr 12, 2023
872e357
fmt
spikecurtis Apr 12, 2023
3f21e30
Fix windows test
spikecurtis Apr 13, 2023
e83ff6e
Windows copy test
spikecurtis Apr 14, 2023
b610579
WIP Windows pty handling
spikecurtis Apr 14, 2023
90bfe94
Fix truncation tests
spikecurtis Apr 18, 2023
d6e131c
Appease linter/fmt
spikecurtis Apr 18, 2023
2c9c6ef
Fix typo
spikecurtis Apr 18, 2023
e39e885
Rework truncation test to not assume OS buffers
spikecurtis Apr 18, 2023
8ec3d1f
Disable orphan test on Windows --- uses sh
spikecurtis Apr 18, 2023
df424e6
agent_test running SSH in pty use ptytest.Start
spikecurtis Apr 19, 2023
50e3fec
More detail about closing pseudoconsole on windows
spikecurtis Apr 19, 2023
c09083e
Code review fixes
spikecurtis Apr 20, 2023
439107d
Rearrange ptytest method order
spikecurtis Apr 20, 2023
c6a3229
Protect pty.Resize on windows from races
spikecurtis Apr 21, 2023
aa94546
Fix windows bugs
spikecurtis Apr 21, 2023
0f07cb9
PTY doesn't extend PTYCmd
spikecurtis Apr 24, 2023
eaabb3a
merge branch 'main' of https://github.com/coder/coder into spike/pty-ssh
spikecurtis Apr 24, 2023
50060d8
Fix windows types
spikecurtis Apr 24, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Windows copy test 
Signed-off-by: Spike Curtis <spike@coder.com>
  • Loading branch information
@spikecurtis
spikecurtis committed Apr 14, 2023
commit e83ff6ef00143512e20d7f58418e3eb72d6c8c2e
57 changes: 49 additions & 8 deletions pty/start_windows_test.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,9 +4,14 @@
package pty_test

import (
"bytes"
"context"
"io"
"os/exec"
"testing"
"time"

"github.com/coder/coder/pty"
"github.com/coder/coder/pty/ptytest"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
Expand All @@ -22,31 +27,67 @@ func TestStart(t *testing.T) {
t.Parallel()
t.Run("Echo", func(t *testing.T) {
t.Parallel()
pty, ps := ptytest.Start(t, exec.Command("cmd.exe", "/c", "echo", "test"))
pty.ExpectMatch("test")
ptty, ps := ptytest.Start(t, exec.Command("cmd.exe", "/c", "echo", "test"))
ptty.ExpectMatch("test")
err := ps.Wait()
require.NoError(t, err)
err = pty.Close()
err = ptty.Close()
require.NoError(t, err)
})
t.Run("Resize", func(t *testing.T) {
t.Parallel()
pty, _ := ptytest.Start(t, exec.Command("cmd.exe"))
err := pty.Resize(100, 50)
ptty, _ := ptytest.Start(t, exec.Command("cmd.exe"))
err := ptty.Resize(100, 50)
require.NoError(t, err)
err = pty.Close()
err = ptty.Close()
require.NoError(t, err)
})
t.Run("Kill", func(t *testing.T) {
t.Parallel()
pty, ps := ptytest.Start(t, exec.Command("cmd.exe"))
ptty, ps := ptytest.Start(t, exec.Command("cmd.exe"))
err := ps.Kill()
assert.NoError(t, err)
err = ps.Wait()
var exitErr *exec.ExitError
require.True(t, xerrors.As(err, &exitErr))
assert.NotEqual(t, 0, exitErr.ExitCode())
err = pty.Close()
err = ptty.Close()
require.NoError(t, err)
})
}

func Test_Start_copy(t *testing.T) {
t.Parallel()

ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
defer cancel()

pc, cmd, err := pty.Start(exec.CommandContext(ctx, "cmd.exe", "/c", "echo", "test"))
require.NoError(t, err)
b := &bytes.Buffer{}
readDone := make(chan error)
go func() {
_, err := io.Copy(b, pc.OutputReader())
readDone <- err
}()

select {
case err := <-readDone:
require.NoError(t, err)
case <-ctx.Done():
t.Error("read timed out")
}
assert.Equal(t, "test", b.String())

cmdDone := make(chan error)
go func() {
cmdDone <- cmd.Wait()
}()

select {
case err := <-cmdDone:
require.NoError(t, err)
case <-ctx.Done():
t.Error("cmd.Wait() timed out")
}
}