allow intercepting command output for testing

Author: notnmeyer

cmd/api_key.go

@@ -24,9 +24,9 @@ var apiKeyCmd = &cobra.Command{
 		}
 
 		if isJSONOutput() {
-			return printJSON(result)
+			return printJSON(cmd.OutOrStdout(), result)
 		}
-		fmt.Printf("Valid API key for team: %s\n", result.TeamName)
+		fmt.Fprintf(cmd.OutOrStdout(), "Valid API key for team: %s\n", result.TeamName)
 		return nil
 	},
 }

cmd/auth_login.go

@@ -38,9 +38,9 @@ var loginCmd = &cobra.Command{
 		}
 
 		if isJSONOutput() {
-			return printJSON(Result{Success: true, Message: fmt.Sprintf("Authenticated as team: %s", result.TeamName)})
+			return printJSON(cmd.OutOrStdout(), Result{Success: true, Message: fmt.Sprintf("Authenticated as team: %s", result.TeamName)})
 		}
-		fmt.Printf("API key saved. Authenticated as team: %s\n", result.TeamName)
+		fmt.Fprintf(cmd.OutOrStdout(), "API key saved. Authenticated as team: %s\n", result.TeamName)
 		return nil
 	},
 }

cmd/auth_logout.go

@@ -15,9 +15,9 @@ var logoutCmd = &cobra.Command{
 			return err
 		}
 		if isJSONOutput() {
-			return printJSON(Result{Success: true})
+			return printJSON(cmd.OutOrStdout(), Result{Success: true})
 		}
-		fmt.Println("Logged out.")
+		fmt.Fprintln(cmd.OutOrStdout(), "Logged out.")
 		return nil
 	},
 }

cmd/auth_status.go

@@ -17,11 +17,11 @@ var statusCmd = &cobra.Command{
 		}
 
 		if isJSONOutput() {
-			return printJSON(cfg)
+			return printJSON(cmd.OutOrStdout(), cfg)
 		}
 
-		fmt.Printf("API Key:  %s\n", cfg.APIKey)
-		fmt.Printf("Endpoint: %s\n", cfg.EndpointURL)
+		fmt.Fprintf(cmd.OutOrStdout(), "API Key:  %s\n", cfg.APIKey)
+		fmt.Fprintf(cmd.OutOrStdout(), "Endpoint: %s\n", cfg.EndpointURL)
 		return nil
 	},
 }

cmd/output.go

@@ -3,7 +3,7 @@ package cmd
 import (
 	"encoding/json"
 	"fmt"
-	"os"
+	"io"
 )
 
 type outputFlag string
@@ -30,8 +30,8 @@ func isJSONOutput() bool {
 	return outputFormat == "json"
 }
 
-func printJSON(v any) error {
-	enc := json.NewEncoder(os.Stdout)
+func printJSON(w io.Writer, v any) error {
+	enc := json.NewEncoder(w)
 	enc.SetIndent("", "  ")
 	return enc.Encode(v)
 }

cmd/transactional.go

@@ -78,15 +78,15 @@ var transactionalListCmd = &cobra.Command{
 			if emails == nil {
 				emails = []api.TransactionalEmail{}
 			}
-			return printJSON(emails)
+			return printJSON(cmd.OutOrStdout(), emails)
 		}
 
 		if len(emails) == 0 {
-			fmt.Println("No transactional emails found.")
+			fmt.Fprintln(cmd.OutOrStdout(), "No transactional emails found.")
 			return nil
 		}
 
-		w := tabwriter.NewWriter(os.Stdout, 0, 0, 2, ' ', 0)
+		w := tabwriter.NewWriter(cmd.OutOrStdout(), 0, 0, 2, ' ', 0)
 		fmt.Fprintln(w, "ID\tNAME\tLAST UPDATED\tVARIABLES")
 		for _, e := range emails {
 			fmt.Fprintf(w, "%s\t%s\t%s\t%s\n", e.ID, e.Name, e.LastUpdated, strings.Join(e.DataVariables, ", "))
@@ -141,9 +141,9 @@ var transactionalSendCmd = &cobra.Command{
 		}
 
 		if isJSONOutput() {
-			return printJSON(Result{Success: true})
+			return printJSON(cmd.OutOrStdout(), Result{Success: true})
 		}
-		fmt.Println("Sent.")
+		fmt.Fprintln(cmd.OutOrStdout(), "Sent.")
 		return nil
 	},
 }