snapshot of https://github.com/ory/keto.git (master, 2026-09-16; blobs > 131072B omitted)

cmd/client/grpc_client_test.go

@@ -0,0 +1,65 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package client
+
+import (
+	"bytes"
+	"os"
+	"testing"
+
+	"github.com/spf13/cobra"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestGetRemote(t *testing.T) {
+	setup := func() (cmd *cobra.Command, stdOut, stdErr *bytes.Buffer) {
+		cmd = &cobra.Command{}
+		stdOut, stdErr = &bytes.Buffer{}, &bytes.Buffer{}
+		cmd.SetOut(stdOut)
+		cmd.SetErr(stdErr)
+		RegisterRemoteURLFlags(cmd.Flags())
+		return
+	}
+
+	setEnv := func(t *testing.T, env, val string) {
+		require.NoError(t, os.Setenv(env, val))
+		t.Cleanup(func() {
+			require.NoError(t, os.Unsetenv(EnvReadRemote))
+		})
+	}
+
+	t.Run("case=prefers flag value", func(t *testing.T) {
+		cmd, stdOut, stdErr := setup()
+
+		expectedRemote := "ketotest.oryapis.com"
+		require.NoError(t, cmd.Flags().Set(FlagReadRemote, "ketotest.oryapis.com"))
+		setEnv(t, EnvReadRemote, "not"+expectedRemote)
+
+		assert.Equal(t, expectedRemote, getRemote(cmd, FlagReadRemote, EnvReadRemote))
+		assert.Equal(t, 0, stdOut.Len())
+		assert.Equal(t, 0, stdErr.Len())
+	})
+
+	t.Run("case=uses env value", func(t *testing.T) {
+		cmd, stdOut, stdErr := setup()
+
+		expectedRemote := "ketotest.oryapis.com"
+		setEnv(t, EnvReadRemote, expectedRemote)
+
+		assert.Equal(t, expectedRemote, getRemote(cmd, FlagReadRemote, EnvReadRemote))
+		assert.Equal(t, 0, stdOut.Len())
+		assert.Equal(t, 0, stdErr.Len())
+	})
+
+	t.Run("case=falls back to flag default and prints warning", func(t *testing.T) {
+		cmd, stdOut, stdErr := setup()
+
+		expectedRemote := cmd.Flags().Lookup(FlagReadRemote).DefValue
+
+		assert.Equal(t, expectedRemote, getRemote(cmd, FlagReadRemote, EnvReadRemote))
+		assert.Equal(t, 0, stdOut.Len())
+		assert.Contains(t, stdErr.String(), "falling back")
+	})
+}