mirror of
https://github.com/coder/coder.git
synced 2025-07-09 11:45:56 +00:00
9f6edab53b
The VS Code extension has been refactored to use VS Code Remote SSH instead of using the private API. This changes the structure to continue using SSH, but output network information periodically to a file.
74 lines
1.9 KiB
Go
74 lines
1.9 KiB
Go
package cli_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/spf13/afero"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"cdr.dev/slog/sloggers/slogtest"
|
|
|
|
"github.com/coder/coder/agent"
|
|
"github.com/coder/coder/cli/clitest"
|
|
"github.com/coder/coder/coderd/coderdtest"
|
|
"github.com/coder/coder/codersdk"
|
|
"github.com/coder/coder/testutil"
|
|
)
|
|
|
|
// TestVSCodeSSH ensures the agent connects properly with SSH
|
|
// and that network information is properly written to the FS.
|
|
func TestVSCodeSSH(t *testing.T) {
|
|
t.Parallel()
|
|
ctx, cancel := testutil.Context(t)
|
|
defer cancel()
|
|
client, workspace, agentToken := setupWorkspaceForAgent(t, nil)
|
|
user, err := client.User(ctx, codersdk.Me)
|
|
require.NoError(t, err)
|
|
|
|
agentClient := codersdk.New(client.URL)
|
|
agentClient.SetSessionToken(agentToken)
|
|
agentCloser := agent.New(agent.Options{
|
|
Client: agentClient,
|
|
Logger: slogtest.Make(t, nil).Named("agent"),
|
|
})
|
|
defer func() {
|
|
_ = agentCloser.Close()
|
|
}()
|
|
coderdtest.AwaitWorkspaceAgents(t, client, workspace.ID)
|
|
|
|
fs := afero.NewMemMapFs()
|
|
err = afero.WriteFile(fs, "/url", []byte(client.URL.String()), 0600)
|
|
require.NoError(t, err)
|
|
err = afero.WriteFile(fs, "/token", []byte(client.SessionToken()), 0600)
|
|
require.NoError(t, err)
|
|
|
|
cmd, _ := clitest.New(t,
|
|
"vscodessh",
|
|
"--url-file", "/url",
|
|
"--session-token-file", "/token",
|
|
"--network-info-dir", "/net",
|
|
"--network-info-interval", "25ms",
|
|
fmt.Sprintf("coder-vscode--%s--%s", user.Username, workspace.Name))
|
|
done := make(chan struct{})
|
|
go func() {
|
|
//nolint // The above seems reasonable for a one-off test.
|
|
err := cmd.ExecuteContext(context.WithValue(ctx, "fs", fs))
|
|
if err != nil {
|
|
assert.ErrorIs(t, err, context.Canceled)
|
|
}
|
|
close(done)
|
|
}()
|
|
require.Eventually(t, func() bool {
|
|
entries, err := afero.ReadDir(fs, "/net")
|
|
if err != nil {
|
|
return false
|
|
}
|
|
return len(entries) > 0
|
|
}, testutil.WaitLong, testutil.IntervalFast)
|
|
cancel()
|
|
<-done
|
|
}
|