cli/connect_test.go

Rama main · réplica de lectura del repositorio original.

package cli_test

import (
	"bytes"
	"context"
	"net"
	"testing"

	"github.com/stretchr/testify/require"
	"tailscale.com/net/tsaddr"

	"github.com/coder/coder/v2/cli"
	"github.com/coder/coder/v2/codersdk/workspacesdk"
	"github.com/coder/coder/v2/testutil"
	"github.com/coder/serpent"
)

func TestConnectExists_Running(t *testing.T) {
	t.Parallel()
	ctx := testutil.Context(t, testutil.WaitShort)

	var root cli.RootCmd
	cmd, err := root.Command(root.AGPL())
	require.NoError(t, err)

	inv := (&serpent.Invocation{
		Command: cmd,
		Args:    []string{"connect", "exists", "test.example"},
	}).WithContext(withCoderConnectRunning(ctx))
	stdout := new(bytes.Buffer)
	stderr := new(bytes.Buffer)
	inv.Stdout = stdout
	inv.Stderr = stderr
	err = inv.Run()
	require.NoError(t, err)
}

func TestConnectExists_NotRunning(t *testing.T) {
	t.Parallel()
	ctx := testutil.Context(t, testutil.WaitShort)

	var root cli.RootCmd
	cmd, err := root.Command(root.AGPL())
	require.NoError(t, err)

	inv := (&serpent.Invocation{
		Command: cmd,
		Args:    []string{"connect", "exists", "test.example"},
	}).WithContext(withCoderConnectNotRunning(ctx))
	stdout := new(bytes.Buffer)
	stderr := new(bytes.Buffer)
	inv.Stdout = stdout
	inv.Stderr = stderr
	err = inv.Run()
	require.ErrorIs(t, err, cli.ErrSilent)
}

type fakeResolver struct {
	shouldReturnSuccess bool
}

func (f *fakeResolver) LookupIP(_ context.Context, _, _ string) ([]net.IP, error) {
	if f.shouldReturnSuccess {
		return []net.IP{net.ParseIP(tsaddr.CoderServiceIPv6().String())}, nil
	}
	return nil, &net.DNSError{IsNotFound: true}
}

func withCoderConnectRunning(ctx context.Context) context.Context {
	return workspacesdk.WithTestOnlyCoderContextResolver(ctx, &fakeResolver{shouldReturnSuccess: true})
}

func withCoderConnectNotRunning(ctx context.Context) context.Context {
	return workspacesdk.WithTestOnlyCoderContextResolver(ctx, &fakeResolver{shouldReturnSuccess: false})
}