coder/cli/ping_test.go at main · bamhm182/coder

Repository navigation

package cli_test

import (

"context"

"testing"

"github.com/stretchr/testify/assert"

"github.com/coder/coder/v2/agent/agenttest"

"github.com/coder/coder/v2/cli/clitest"

"github.com/coder/coder/v2/coderd/coderdtest"

"github.com/coder/coder/v2/pty/ptytest"

"github.com/coder/coder/v2/testutil"

)

func TestPing(t *testing.T) {

t.Parallel()

t.Run("OK", func(t *testing.T) {

t.Parallel()

client, workspace, agentToken := setupWorkspaceForAgent(t)

inv, root := clitest.New(t, "ping", workspace.Name)

clitest.SetupConfig(t, client, root)

pty := ptytest.New(t)

inv.Stdin = pty.Input()

inv.Stderr = pty.Output()

inv.Stdout = pty.Output()

_ = agenttest.New(t, client.URL, agentToken)

_ = coderdtest.AwaitWorkspaceAgents(t, client, workspace.ID)

ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)

defer cancel()

cmdDone := tGo(t, func() {

err := inv.WithContext(ctx).Run()

assert.NoError(t, err)

})

pty.ExpectMatch("pong from " + workspace.Name)

cancel()

<-cmdDone

})

}