api-key command tests

Author: notnmeyer

cmd/api_key_test.go

@@ -0,0 +1,46 @@
+package cmd
+
+import (
+	"encoding/json"
+	"net/http"
+	"strings"
+	"testing"
+
+	"github.com/loops-so/cli/internal/api"
+)
+
+func TestAPIKeyCmd(t *testing.T) {
+	t.Run("text: shows team name", func(t *testing.T) {
+		serveJSON(t, http.StatusOK, `{"teamName":"Acme Corp"}`)
+		out, err := runCmd(t, "api-key")
+		if err != nil {
+			t.Fatalf("unexpected error: %v", err)
+		}
+		if !strings.Contains(out, "Acme Corp") {
+			t.Errorf("output %q should contain team name", out)
+		}
+	})
+
+	t.Run("json: valid json with team name", func(t *testing.T) {
+		serveJSON(t, http.StatusOK, `{"teamName":"Acme Corp"}`)
+		out, err := runCmd(t, "--output=json", "api-key")
+		if err != nil {
+			t.Fatalf("unexpected error: %v", err)
+		}
+		var result api.APIKeyResponse
+		if err := json.Unmarshal([]byte(out), &result); err != nil {
+			t.Fatalf("output is not valid JSON: %v\n%s", err, out)
+		}
+		if result.TeamName != "Acme Corp" {
+			t.Errorf("TeamName = %q, want %q", result.TeamName, "Acme Corp")
+		}
+	})
+
+	t.Run("api error propagates", func(t *testing.T) {
+		serveJSON(t, http.StatusUnauthorized, `{"error":"Invalid API key"}`)
+		_, err := runCmd(t, "api-key")
+		if err == nil {
+			t.Fatal("expected error, got nil")
+		}
+	})
+}

cmd/helpers_test.go

@@ -0,0 +1,39 @@
+package cmd
+
+import (
+	"bytes"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/zalando/go-keyring"
+)
+
+// runCmd executes rootCmd with the given args and returns captured stdout.
+// outputFormat is reset to "text" before and after each call.
+func runCmd(t *testing.T, args ...string) (string, error) {
+	t.Helper()
+	outputFormat = "text"
+	t.Cleanup(func() { outputFormat = "text" })
+
+	buf := &bytes.Buffer{}
+	rootCmd.SetOut(buf)
+	rootCmd.SetErr(&bytes.Buffer{})
+	rootCmd.SetArgs(args)
+	err := rootCmd.Execute()
+	return buf.String(), err
+}
+
+// serveJSON starts a test server that responds with the given status and JSON
+// body, sets LOOPS_API_KEY and LOOPS_ENDPOINT_URL, and registers cleanup.
+func serveJSON(t *testing.T, status int, body string) {
+	t.Helper()
+	keyring.MockInit()
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(status)
+		w.Write([]byte(body))
+	}))
+	t.Cleanup(srv.Close)
+	t.Setenv("LOOPS_API_KEY", "test-key")
+	t.Setenv("LOOPS_ENDPOINT_URL", srv.URL)
+}