Docker Image (#38)

cmd/otf/login.go

@@ -3,17 +3,14 @@ package main
 import (
 	"fmt"
 
-	"github.com/leg100/otf/http"
 	"github.com/spf13/cobra"
 )
 
 const (
 	DummyToken = "dummy"
 )
 
-func LoginCommand(store KVStore) *cobra.Command {
-	var address string
-
+func LoginCommand(store KVStore, address string) *cobra.Command {
 	cmd := &cobra.Command{
 		Use:   "login",
 		Short: "Login to OTF",
@@ -28,7 +25,5 @@ func LoginCommand(store KVStore) *cobra.Command {
 		},
 	}
 
-	cmd.Flags().StringVar(&address, "address", http.DefaultAddress, "Address of OTF instance")
-
 	return cmd
 }

cmd/otf/login_test.go

@@ -1,7 +1,6 @@
 package main
 
 import (
-	"os"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -11,23 +10,9 @@ import (
 func TestLoginCommand(t *testing.T) {
 	var store KVStore = KVMap(make(map[string]string))
 
-	cmd := LoginCommand(store)
+	cmd := LoginCommand(store, "localhost:8080")
 	require.NoError(t, cmd.Execute())
 
 	token, _ := store.Load("localhost:8080")
 	assert.Equal(t, "dummy", token)
 }
-
-func TestLoginCommandWithExplicitAddress(t *testing.T) {
-	// Ensure env var doesn't interfere with test
-	os.Unsetenv("OTF_ADDRESS")
-
-	var store KVStore = KVMap(make(map[string]string))
-
-	cmd := LoginCommand(store)
-	cmd.SetArgs([]string{"--address", "otf.dev:8080"})
-	require.NoError(t, cmd.Execute())
-
-	token, _ := store.Load("otf.dev:8080")
-	assert.Equal(t, "dummy", token)
-}

cmd/otf/main.go

@@ -6,6 +6,7 @@ import (
 	"os"
 
 	cmdutil "github.com/leg100/otf/cmd"
+	"github.com/leg100/otf/http"
 	"github.com/spf13/cobra"
 )
 
@@ -21,21 +22,30 @@ func main() {
 }
 
 func Run(ctx context.Context, args []string) error {
+	cfg, err := http.NewConfig(LoadCredentials)
+	if err != nil {
+		return err
+	}
+
 	cmd := &cobra.Command{
 		Use:           "otf",
 		SilenceUsage:  true,
 		SilenceErrors: true,
+		Run:           func(cmd *cobra.Command, args []string) {},
 	}
+
+	cmd.PersistentFlags().StringVar(&cfg.Address, "address", http.DefaultAddress, "Address of OTF server")
+
 	cmd.SetArgs(args)
 
 	store, err := NewCredentialsStore()
 	if err != nil {
 		return err
 	}
 
-	cmd.AddCommand(LoginCommand(store))
-	cmd.AddCommand(OrganizationCommand())
-	cmd.AddCommand(WorkspaceCommand())
+	cmd.AddCommand(LoginCommand(store, cfg.Address))
+	cmd.AddCommand(OrganizationCommand(cfg))
+	cmd.AddCommand(WorkspaceCommand(cfg))
 
 	cmdutil.SetFlagsFromEnvVariables(cmd.Flags())
 

cmd/otf/main_test.go

@@ -26,6 +26,14 @@ func TestMain(t *testing.T) {
 			name: "login",
 			args: []string{"login", "-h"},
 		},
+		{
+			name: "login and address",
+			args: []string{"login", "-h", "--address", "test.abc:1234"},
+		},
+		{
+			name: "address",
+			args: []string{"--address", "test.abc:1234"},
+		},
 		{
 			name: "organization new",
 			args: []string{"organizations", "new", "-h"},

cmd/otf/organization.go

@@ -1,26 +1,17 @@
 package main
 
 import (
-	"os"
-
 	"github.com/leg100/otf/http"
 	"github.com/spf13/cobra"
 )
 
-func OrganizationCommand() *cobra.Command {
-	cfg, err := http.NewConfig(LoadCredentials)
-	if err != nil {
-		panic(err.Error())
-	}
-
+func OrganizationCommand(factory http.ClientFactory) *cobra.Command {
 	cmd := &cobra.Command{
 		Use:   "organizations",
 		Short: "Organization management",
 	}
-	cmd.Flags().StringVar(&cfg.Address, "address", http.DefaultAddress, "Address of OTF server")
-	cmd.Flags().StringVar(&cfg.Token, "token", os.Getenv("OTF_TOKEN"), "Authentication token")
 
-	cmd.AddCommand(OrganizationNewCommand(cfg))
+	cmd.AddCommand(OrganizationNewCommand(factory))
 
 	return cmd
 }

cmd/otf/workspace.go

@@ -1,30 +1,21 @@
 package main
 
 import (
-	"os"
-
 	"github.com/leg100/otf/http"
 	"github.com/spf13/cobra"
 )
 
-func WorkspaceCommand() *cobra.Command {
-	cfg, err := http.NewConfig(LoadCredentials)
-	if err != nil {
-		panic(err.Error())
-	}
-
+func WorkspaceCommand(factory http.ClientFactory) *cobra.Command {
 	cmd := &cobra.Command{
 		Use:   "workspaces",
 		Short: "Workspace management",
 	}
-	cmd.Flags().StringVar(&cfg.Address, "address", http.DefaultAddress, "Address of OTF server")
-	cmd.Flags().StringVar(&cfg.Token, "token", os.Getenv("OTF_TOKEN"), "Authentication token")
 
-	cmd.AddCommand(WorkspaceListCommand(cfg))
-	cmd.AddCommand(WorkspaceShowCommand(cfg))
-	cmd.AddCommand(WorkspaceEditCommand(cfg))
-	cmd.AddCommand(WorkspaceLockCommand(cfg))
-	cmd.AddCommand(WorkspaceUnlockCommand(cfg))
+	cmd.AddCommand(WorkspaceListCommand(factory))
+	cmd.AddCommand(WorkspaceShowCommand(factory))
+	cmd.AddCommand(WorkspaceEditCommand(factory))
+	cmd.AddCommand(WorkspaceLockCommand(factory))
+	cmd.AddCommand(WorkspaceUnlockCommand(factory))
 
 	return cmd
 }