gitlink-cli/shortcuts/label/label_test.go

package label

import (
	"encoding/json"
	"fmt"
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/gitlink-org/gitlink-cli/internal/client"
	"github.com/gitlink-org/gitlink-cli/shortcuts/common"
)

func TestLabelList(t *testing.T) {
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.Method != "GET" || r.URL.Path != "/v1/owner/repo/issue_tags.json" {
			t.Fatalf("unexpected request: %s %s", r.Method, r.URL.Path)
		}
		writeJSON(t, w, map[string]interface{}{
			"total_count": 2,
			"issue_tags": []map[string]interface{}{
				{"id": 1, "name": "bug", "color": "#FF0000"},
				{"id": 2, "name": "feature", "color": "#00FF00"},
			},
		})
	}))
	defer server.Close()

	err := runLabelShortcut(t, server, "list", map[string]string{})
	if err != nil {
		t.Fatalf("list shortcut failed: %v", err)
	}
}

func TestLabelCreate(t *testing.T) {
	var payload map[string]interface{}
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.Method == "POST" && r.URL.Path == "/v1/owner/repo/issue_tags.json" {
			payload = decodeJSON(t, r)
			writeJSON(t, w, map[string]interface{}{
				"status":  0,
				"message": "success",
			})
			return
		}
		t.Fatalf("unexpected request: %s %s", r.Method, r.URL.Path)
	}))
	defer server.Close()

	err := runLabelShortcut(t, server, "create", map[string]string{
		"name":        "enhancement",
		"color":       "#0000FF",
		"description": "New feature",
	})
	if err != nil {
		t.Fatalf("create shortcut failed: %v", err)
	}

	assertEqual(t, payload["name"], "enhancement")
	assertEqual(t, payload["color"], "#0000FF")
	assertEqual(t, payload["description"], "New feature")
}

func TestLabelCreateWithDefaultColor(t *testing.T) {
	var payload map[string]interface{}
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.Method == "POST" && r.URL.Path == "/v1/owner/repo/issue_tags.json" {
			payload = decodeJSON(t, r)
			writeJSON(t, w, map[string]interface{}{
				"status":  0,
				"message": "success",
			})
			return
		}
		t.Fatalf("unexpected request: %s %s", r.Method, r.URL.Path)
	}))
	defer server.Close()

	err := runLabelShortcut(t, server, "create", map[string]string{
		"name":  "bug",
		"color": "#F17013",
	})
	if err != nil {
		t.Fatalf("create shortcut failed: %v", err)
	}

	assertEqual(t, payload["color"], "#F17013")
}

func TestLabelUpdate(t *testing.T) {
	var payload map[string]interface{}
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.Method == "PATCH" && r.URL.Path == "/v1/owner/repo/issue_tags/3.json" {
			payload = decodeJSON(t, r)
			writeJSON(t, w, map[string]interface{}{
				"status":  0,
				"message": "success",
			})
			return
		}
		t.Fatalf("unexpected request: %s %s", r.Method, r.URL.Path)
	}))
	defer server.Close()

	err := runLabelShortcut(t, server, "update", map[string]string{
		"id":    "3",
		"name":  "critical",
		"color": "#FF0000",
	})
	if err != nil {
		t.Fatalf("update shortcut failed: %v", err)
	}

	assertEqual(t, payload["name"], "critical")
	assertEqual(t, payload["color"], "#FF0000")
}

func TestLabelUpdateRequiresAtLeastOneField(t *testing.T) {
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		t.Fatal("no request should be made when no fields are provided")
	}))
	defer server.Close()

	err := runLabelShortcut(t, server, "update", map[string]string{
		"id": "3",
	})
	if err == nil {
		t.Fatal("expected error when no update fields provided, got nil")
	}
}

func TestLabelDelete(t *testing.T) {
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.Method == "DELETE" && r.URL.Path == "/v1/owner/repo/issue_tags/3.json" {
			writeJSON(t, w, map[string]interface{}{
				"status":  0,
				"message": "success",
			})
			return
		}
		t.Fatalf("unexpected request: %s %s", r.Method, r.URL.Path)
	}))
	defer server.Close()

	err := runLabelShortcut(t, server, "delete", map[string]string{
		"id": "3",
	})
	if err != nil {
		t.Fatalf("delete shortcut failed: %v", err)
	}
}

func runLabelShortcut(t *testing.T, server *httptest.Server, name string, args map[string]string) error {
	t.Helper()
	shortcut := findLabelShortcut(t, name)
	ctx := &common.RuntimeContext{
		Client: &client.Client{
			HTTP:    server.Client(),
			BaseURL: server.URL,
		},
		Owner:  "owner",
		Repo:   "repo",
		Format: "json",
		Args:   args,
	}
	return shortcut.Run(ctx)
}

func findLabelShortcut(t *testing.T, name string) *common.Shortcut {
	t.Helper()
	for _, shortcut := range Shortcuts() {
		if shortcut.Name == name {
			return shortcut
		}
	}
	t.Fatalf("shortcut %q not found", name)
	return nil
}

func decodeJSON(t *testing.T, r *http.Request) map[string]interface{} {
	t.Helper()
	var payload map[string]interface{}
	if err := json.NewDecoder(r.Body).Decode(&payload); err != nil {
		t.Fatalf("failed to decode request body: %v", err)
	}
	return payload
}

func writeJSON(t *testing.T, w http.ResponseWriter, payload interface{}) {
	t.Helper()
	w.Header().Set("Content-Type", "application/json")
	if err := json.NewEncoder(w).Encode(payload); err != nil {
		t.Fatalf("failed to write response: %v", err)
	}
}

func assertEqual(t *testing.T, got interface{}, want interface{}) {
	t.Helper()
	if fmt.Sprintf("%v", got) != fmt.Sprintf("%v", want) {
		t.Fatalf("got %v (%T), want %v (%T)", got, got, want, want)
	}
}