mirror of
https://github.com/coder/coder.git
synced 2026-06-02 20:48:20 +00:00
Spike Curtis
8dc4d76890
<!-- If you have used AI to produce some or all of this PR, please ensure you have read our [AI Contribution guidelines](https://coder.com/docs/about/contributing/AI_CONTRIBUTING) before submitting. --> relates to GRU-18 Adds basic implementation for Workspace Agent Connection Watch and tests. Missing are handling of logs.
101 lines
2.6 KiB
Go
101 lines
2.6 KiB
Go
package tunneler_test
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"golang.org/x/crypto/ssh"
|
|
|
|
"github.com/coder/coder/v2/agent/agenttest"
|
|
"github.com/coder/coder/v2/coderd/coderdtest"
|
|
"github.com/coder/coder/v2/coderd/database"
|
|
"github.com/coder/coder/v2/coderd/database/dbfake"
|
|
"github.com/coder/coder/v2/codersdk"
|
|
"github.com/coder/coder/v2/codersdk/workspacesdk"
|
|
"github.com/coder/coder/v2/codersdk/workspacesdk/tunneler"
|
|
"github.com/coder/coder/v2/testutil"
|
|
)
|
|
|
|
// TestTunneler_Integration is an integration test using coderdtest. It should be removed when we integrate the Tunneler
|
|
// into coder ssh and those integration test cover this functionality.
|
|
func TestTunneler_Integration(t *testing.T) {
|
|
t.Parallel()
|
|
ctx := testutil.Context(t, testutil.WaitLong)
|
|
client, store := coderdtest.NewWithDatabase(t, nil)
|
|
logger := testutil.Logger(t)
|
|
client.SetLogger(logger.Named("client"))
|
|
first := coderdtest.CreateFirstUser(t, client)
|
|
userClient, user := coderdtest.CreateAnotherUserMutators(t, client, first.OrganizationID, nil, func(r *codersdk.CreateUserRequestWithOrgs) {
|
|
r.Username = "myuser"
|
|
})
|
|
userClient.SetLogger(logger.Named("userclient"))
|
|
r := dbfake.WorkspaceBuild(t, store, database.WorkspaceTable{
|
|
Name: "myworkspace",
|
|
OrganizationID: first.OrganizationID,
|
|
OwnerID: user.ID,
|
|
}).WithAgent().Do()
|
|
wsSDKClient := workspacesdk.New(userClient)
|
|
logs := &bytes.Buffer{}
|
|
|
|
app := &sshApplication{
|
|
t: t,
|
|
ctx: ctx,
|
|
done: make(chan struct{}),
|
|
}
|
|
|
|
tun := tunneler.NewTunneler(wsSDKClient, tunneler.Config{
|
|
WorkspaceID: r.Workspace.ID,
|
|
App: app,
|
|
WorkspaceStarter: nil,
|
|
AgentName: "",
|
|
LogWriter: logs,
|
|
DebugLogger: logger.Named("tunneler"),
|
|
})
|
|
|
|
testAgent := agenttest.New(t, client.URL, r.AgentToken)
|
|
defer testAgent.Close()
|
|
|
|
testutil.TryReceive(ctx, t, app.done)
|
|
require.Equal(t, app.result, "foo\n")
|
|
|
|
err := tun.GracefulShutdown(ctx)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
type sshApplication struct {
|
|
t *testing.T
|
|
ctx context.Context
|
|
client *ssh.Client
|
|
done chan struct{}
|
|
result string
|
|
}
|
|
|
|
func (s *sshApplication) Close() error {
|
|
return s.client.Close()
|
|
}
|
|
|
|
func (s *sshApplication) Start(conn workspacesdk.AgentConn) error {
|
|
var err error
|
|
s.client, err = conn.SSHClient(s.ctx)
|
|
if err != nil {
|
|
s.t.Error(err)
|
|
return err
|
|
}
|
|
go func() {
|
|
defer close(s.done)
|
|
sess, err := s.client.NewSession()
|
|
if err != nil {
|
|
s.t.Error("failed to create session", err)
|
|
}
|
|
defer sess.Close()
|
|
out, err := sess.Output("echo foo")
|
|
if err != nil {
|
|
s.t.Error("failed to echo", err)
|
|
}
|
|
s.result = string(out)
|
|
}()
|
|
return nil
|
|
}
|