Function TestBaseCommandSetNonParallel

cmd/river/rivercli/river_cli_test.go:239–293

Same as the above, but non-parallel so tests can use `t.Setenv`. Separated out into its own test block so that we don't have to mark the entire block above as non-parallel because a few tests can't be made parallel.

(t *testing.T)

// out into its own test block so that we don't have to mark the entire block
// above as non-parallel because a few tests can't be made parallel.
func TestBaseCommandSetNonParallel(t *testing.T) {
	ctx := context.Background()

	type testBundle struct {
		out *bytes.Buffer
	}

	setup := func(t *testing.T) (*cobra.Command, *testBundle) {
		t.Helper()

		cli := NewCLI(&Config{
			Name: "River",
		})

		var out bytes.Buffer
		cli.SetOut(&out)

		return cli.BaseCommandSet(), &testBundle{
			out: &out,
		}
	}

	t.Run("PGEnvWithoutDatabaseURL", func(t *testing.T) {
		cmd, _ := setup(t)

		testDatabaseURL := riversharedtest.TestDatabaseURL()

		config, err := pgxpool.ParseConfig(testDatabaseURL)
		require.NoError(t, err)

		dbPool, err := pgxpool.NewWithConfig(ctx, config)
		require.NoError(t, err)

		var (
			driver = riverpgxv5.New(dbPool)
			schema = riverdbtest.TestSchema(ctx, t, driver, nil)
		)

		t.Setenv("TEST_DATABASE_URL", "")

		parsedURL, err := url.Parse(testDatabaseURL)
		require.NoError(t, err)

		t.Setenv("PGDATABASE", parsedURL.Path[1:])
		t.Setenv("PGHOST", parsedURL.Hostname())
		pass, _ := parsedURL.User.Password()
		t.Setenv("PGPASSWORD", pass)
		t.Setenv("PGPORT", cmp.Or(parsedURL.Port(), "5432"))
		t.Setenv("PGSSLMODE", parsedURL.Query().Get("sslmode"))
		t.Setenv("PGUSER", parsedURL.User.Username())

		cmd.SetArgs([]string{"migrate-up", "--schema", schema})
		require.NoError(t, cmd.Execute())
	})
}

func TestBaseCommandSetPostgresTimeoutPrecedence(t *testing.T) {
	t.Parallel()