mirror of
https://github.com/coder/coder.git
synced 2026-09-24 15:04:27 +08:00
Kyle CarberryandColin Adler
9bd83e5ec7
* fix: Add coder user to docker group on installation This makes for a simpler setup, and reduces the likelihood a user runs into a strange issue. * Add wgnet * Add ping * Add listening * Finish refactor to make this work * Add interface for swapping * Fix conncache with interface * chore: update gvisor * fix tailscale types * linting * more linting * Add coordinator * Add coordinator tests * Fix coordination * It compiles! * Move all connection negotiation in-memory * Migrate coordinator to use net.conn * Add closed func * Fix close listener func * Make reconnecting PTY work * Fix reconnecting PTY * Update CI to Go 1.19 * Add CLI flags for DERP mapping * Fix Tailnet test * Rename ConnCoordinator to TailnetCoordinator * Remove print statement from workspace agent test * Refactor wsconncache to use tailnet * Remove STUN from unit tests * Add migrate back to dump * chore: Upgrade to Go 1.19 This is required as part of #3505. * Fix reconnecting PTY tests * fix: update wireguard-go to fix devtunnel * fix migration numbers * linting * Return early for status if endpoints are empty * Update cli/server.go Co-authored-by: Colin Adler <colin1adler@gmail.com> * Update cli/server.go Co-authored-by: Colin Adler <colin1adler@gmail.com> * Fix frontend entites * Fix agent bicopy * Fix race condition for the last node * Fix down migration * Fix connection RBAC * Fix migration numbers * Fix forwarding TCP to a local port * Implement ping for tailnet * Rename to ForceHTTP * Add external derpmapping * Expose DERP region names to the API * Add global option to enable Tailscale networking for web * Mark DERP flags hidden while testing * Update DERP map on reconnect * Add close func to workspace agents * Fix race condition in upstream dependency * Fix feature columns race condition Co-authored-by: Colin Adler <colin1adler@gmail.com>
278 lines
8.5 KiB
Go
278 lines
8.5 KiB
Go
package cliflag_test
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/spf13/pflag"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/coder/coder/cli/cliflag"
|
|
"github.com/coder/coder/cryptorand"
|
|
)
|
|
|
|
// Testcliflag cannot run in parallel because it uses t.Setenv.
|
|
//
|
|
//nolint:paralleltest
|
|
func TestCliflag(t *testing.T) {
|
|
t.Run("StringDefault", func(t *testing.T) {
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
def, _ := cryptorand.String(10)
|
|
cliflag.String(flagset, name, shorthand, env, def, usage)
|
|
got, err := flagset.GetString(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, def, got)
|
|
require.Contains(t, flagset.FlagUsages(), usage)
|
|
require.Contains(t, flagset.FlagUsages(), fmt.Sprintf("Consumes $%s", env))
|
|
})
|
|
|
|
t.Run("StringEnvVar", func(t *testing.T) {
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
envValue, _ := cryptorand.String(10)
|
|
t.Setenv(env, envValue)
|
|
def, _ := cryptorand.String(10)
|
|
cliflag.String(flagset, name, shorthand, env, def, usage)
|
|
got, err := flagset.GetString(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, envValue, got)
|
|
})
|
|
|
|
t.Run("StringVarPDefault", func(t *testing.T) {
|
|
var ptr string
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
def, _ := cryptorand.String(10)
|
|
|
|
cliflag.StringVarP(flagset, &ptr, name, shorthand, env, def, usage)
|
|
got, err := flagset.GetString(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, def, got)
|
|
require.Contains(t, flagset.FlagUsages(), usage)
|
|
require.Contains(t, flagset.FlagUsages(), fmt.Sprintf("Consumes $%s", env))
|
|
})
|
|
|
|
t.Run("StringVarPEnvVar", func(t *testing.T) {
|
|
var ptr string
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
envValue, _ := cryptorand.String(10)
|
|
t.Setenv(env, envValue)
|
|
def, _ := cryptorand.String(10)
|
|
|
|
cliflag.StringVarP(flagset, &ptr, name, shorthand, env, def, usage)
|
|
got, err := flagset.GetString(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, envValue, got)
|
|
})
|
|
|
|
t.Run("EmptyEnvVar", func(t *testing.T) {
|
|
var ptr string
|
|
flagset, name, shorthand, _, usage := randomFlag()
|
|
def, _ := cryptorand.String(10)
|
|
|
|
cliflag.StringVarP(flagset, &ptr, name, shorthand, "", def, usage)
|
|
got, err := flagset.GetString(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, def, got)
|
|
require.Contains(t, flagset.FlagUsages(), usage)
|
|
require.NotContains(t, flagset.FlagUsages(), "Consumes")
|
|
})
|
|
|
|
t.Run("StringArrayDefault", func(t *testing.T) {
|
|
var ptr []string
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
def := []string{"hello"}
|
|
cliflag.StringArrayVarP(flagset, &ptr, name, shorthand, env, def, usage)
|
|
got, err := flagset.GetStringArray(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, def, got)
|
|
})
|
|
|
|
t.Run("StringArrayEnvVar", func(t *testing.T) {
|
|
var ptr []string
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
t.Setenv(env, "wow,test")
|
|
cliflag.StringArrayVarP(flagset, &ptr, name, shorthand, env, nil, usage)
|
|
got, err := flagset.GetStringArray(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, []string{"wow", "test"}, got)
|
|
})
|
|
|
|
t.Run("StringArrayEnvVarEmpty", func(t *testing.T) {
|
|
var ptr []string
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
t.Setenv(env, "")
|
|
cliflag.StringArrayVarP(flagset, &ptr, name, shorthand, env, nil, usage)
|
|
got, err := flagset.GetStringArray(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, []string{}, got)
|
|
})
|
|
|
|
t.Run("UInt8Default", func(t *testing.T) {
|
|
var ptr uint8
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
def, _ := cryptorand.Int63n(10)
|
|
|
|
cliflag.Uint8VarP(flagset, &ptr, name, shorthand, env, uint8(def), usage)
|
|
got, err := flagset.GetUint8(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, uint8(def), got)
|
|
require.Contains(t, flagset.FlagUsages(), usage)
|
|
require.Contains(t, flagset.FlagUsages(), fmt.Sprintf("Consumes $%s", env))
|
|
})
|
|
|
|
t.Run("UInt8EnvVar", func(t *testing.T) {
|
|
var ptr uint8
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
envValue, _ := cryptorand.Int63n(10)
|
|
t.Setenv(env, strconv.FormatUint(uint64(envValue), 10))
|
|
def, _ := cryptorand.Int()
|
|
|
|
cliflag.Uint8VarP(flagset, &ptr, name, shorthand, env, uint8(def), usage)
|
|
got, err := flagset.GetUint8(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, uint8(envValue), got)
|
|
})
|
|
|
|
t.Run("UInt8FailParse", func(t *testing.T) {
|
|
var ptr uint8
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
envValue, _ := cryptorand.String(10)
|
|
t.Setenv(env, envValue)
|
|
def, _ := cryptorand.Int63n(10)
|
|
|
|
cliflag.Uint8VarP(flagset, &ptr, name, shorthand, env, uint8(def), usage)
|
|
got, err := flagset.GetUint8(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, uint8(def), got)
|
|
})
|
|
|
|
t.Run("IntDefault", func(t *testing.T) {
|
|
var ptr int
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
def, _ := cryptorand.Int63n(10)
|
|
|
|
cliflag.IntVarP(flagset, &ptr, name, shorthand, env, int(def), usage)
|
|
got, err := flagset.GetInt(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, int(def), got)
|
|
require.Contains(t, flagset.FlagUsages(), usage)
|
|
require.Contains(t, flagset.FlagUsages(), fmt.Sprintf("Consumes $%s", env))
|
|
})
|
|
|
|
t.Run("IntEnvVar", func(t *testing.T) {
|
|
var ptr int
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
envValue, _ := cryptorand.Int63n(10)
|
|
t.Setenv(env, strconv.FormatUint(uint64(envValue), 10))
|
|
def, _ := cryptorand.Int()
|
|
|
|
cliflag.IntVarP(flagset, &ptr, name, shorthand, env, def, usage)
|
|
got, err := flagset.GetInt(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, int(envValue), got)
|
|
})
|
|
|
|
t.Run("IntFailParse", func(t *testing.T) {
|
|
var ptr int
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
envValue, _ := cryptorand.String(10)
|
|
t.Setenv(env, envValue)
|
|
def, _ := cryptorand.Int63n(10)
|
|
|
|
cliflag.IntVarP(flagset, &ptr, name, shorthand, env, int(def), usage)
|
|
got, err := flagset.GetInt(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, int(def), got)
|
|
})
|
|
|
|
t.Run("BoolDefault", func(t *testing.T) {
|
|
var ptr bool
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
def, _ := cryptorand.Bool()
|
|
|
|
cliflag.BoolVarP(flagset, &ptr, name, shorthand, env, def, usage)
|
|
got, err := flagset.GetBool(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, def, got)
|
|
require.Contains(t, flagset.FlagUsages(), usage)
|
|
require.Contains(t, flagset.FlagUsages(), fmt.Sprintf("Consumes $%s", env))
|
|
})
|
|
|
|
t.Run("BoolEnvVar", func(t *testing.T) {
|
|
var ptr bool
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
envValue, _ := cryptorand.Bool()
|
|
t.Setenv(env, strconv.FormatBool(envValue))
|
|
def, _ := cryptorand.Bool()
|
|
|
|
cliflag.BoolVarP(flagset, &ptr, name, shorthand, env, def, usage)
|
|
got, err := flagset.GetBool(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, envValue, got)
|
|
})
|
|
|
|
t.Run("BoolFailParse", func(t *testing.T) {
|
|
var ptr bool
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
envValue, _ := cryptorand.String(10)
|
|
t.Setenv(env, envValue)
|
|
def, _ := cryptorand.Bool()
|
|
|
|
cliflag.BoolVarP(flagset, &ptr, name, shorthand, env, def, usage)
|
|
got, err := flagset.GetBool(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, def, got)
|
|
})
|
|
|
|
t.Run("DurationDefault", func(t *testing.T) {
|
|
var ptr time.Duration
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
def, _ := cryptorand.Duration()
|
|
|
|
cliflag.DurationVarP(flagset, &ptr, name, shorthand, env, def, usage)
|
|
got, err := flagset.GetDuration(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, def, got)
|
|
require.Contains(t, flagset.FlagUsages(), usage)
|
|
require.Contains(t, flagset.FlagUsages(), fmt.Sprintf("Consumes $%s", env))
|
|
})
|
|
|
|
t.Run("DurationEnvVar", func(t *testing.T) {
|
|
var ptr time.Duration
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
envValue, _ := cryptorand.Duration()
|
|
t.Setenv(env, envValue.String())
|
|
def, _ := cryptorand.Duration()
|
|
|
|
cliflag.DurationVarP(flagset, &ptr, name, shorthand, env, def, usage)
|
|
got, err := flagset.GetDuration(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, envValue, got)
|
|
})
|
|
|
|
t.Run("DurationFailParse", func(t *testing.T) {
|
|
var ptr time.Duration
|
|
flagset, name, shorthand, env, usage := randomFlag()
|
|
envValue, _ := cryptorand.String(10)
|
|
t.Setenv(env, envValue)
|
|
def, _ := cryptorand.Duration()
|
|
|
|
cliflag.DurationVarP(flagset, &ptr, name, shorthand, env, def, usage)
|
|
got, err := flagset.GetDuration(name)
|
|
require.NoError(t, err)
|
|
require.Equal(t, def, got)
|
|
})
|
|
}
|
|
|
|
func randomFlag() (*pflag.FlagSet, string, string, string, string) {
|
|
fsname, _ := cryptorand.String(10)
|
|
flagset := pflag.NewFlagSet(fsname, pflag.PanicOnError)
|
|
name, _ := cryptorand.String(10)
|
|
shorthand, _ := cryptorand.String(1)
|
|
env, _ := cryptorand.String(10)
|
|
usage, _ := cryptorand.String(10)
|
|
|
|
return flagset, name, shorthand, env, usage
|
|
}
|