gitlink-cli/cmd/alias/alias_test.go

package alias

import (
	"os"
	"testing"
)

func TestLoadAliasesEmpty(t *testing.T) {
	tmpDir := t.TempDir()
	t.Setenv("GITLINK_CONFIG_DIR", tmpDir)

	aliases, err := loadAliases()
	if err != nil {
		t.Fatalf("loadAliases failed: %v", err)
	}
	if len(aliases) != 0 {
		t.Fatalf("expected empty aliases, got %d", len(aliases))
	}
}

func TestSaveAndLoadAliases(t *testing.T) {
	tmpDir := t.TempDir()
	t.Setenv("GITLINK_CONFIG_DIR", tmpDir)

	original := map[string]string{
		"rl": "repo +list",
		"ri": "repo +info",
	}
	if err := saveAliases(original); err != nil {
		t.Fatalf("saveAliases failed: %v", err)
	}

	loaded, err := loadAliases()
	if err != nil {
		t.Fatalf("loadAliases failed: %v", err)
	}
	if len(loaded) != 2 {
		t.Fatalf("expected 2 aliases, got %d", len(loaded))
	}
	if loaded["rl"] != "repo +list" {
		t.Errorf("expected rl -> repo +list, got %s", loaded["rl"])
	}
	if loaded["ri"] != "repo +info" {
		t.Errorf("expected ri -> repo +info, got %s", loaded["ri"])
	}
}

func TestSaveAliasesOverwrite(t *testing.T) {
	tmpDir := t.TempDir()
	t.Setenv("GITLINK_CONFIG_DIR", tmpDir)

	saveAliases(map[string]string{"rl": "repo +list"})
	saveAliases(map[string]string{"rl": "repo +list --owner Gitlink"})

	loaded, _ := loadAliases()
	if loaded["rl"] != "repo +list --owner Gitlink" {
		t.Errorf("alias should be overwritten, got %s", loaded["rl"])
	}
}

func TestLoadAliasesInvalidYAML(t *testing.T) {
	tmpDir := t.TempDir()
	t.Setenv("GITLINK_CONFIG_DIR", tmpDir)

	os.WriteFile(tmpDir+"/aliases.yaml", []byte("{{invalid yaml}}"), 0600)

	aliases, err := loadAliases()
	if err != nil {
		t.Fatalf("should not error on invalid YAML, got: %v", err)
	}
	if len(aliases) != 0 {
		t.Fatalf("should return empty map on invalid YAML, got %d", len(aliases))
	}
}

func TestNewAliasCmdStructure(t *testing.T) {
	cmd := NewAliasCmd()
	if cmd.Use != "alias" {
		t.Errorf("expected Use 'alias', got %s", cmd.Use)
	}
	if !cmd.HasSubCommands() {
		t.Error("alias command should have subcommands")
	}

	subcmds := cmd.Commands()
	if len(subcmds) != 3 {
		t.Fatalf("expected 3 subcommands, got %d", len(subcmds))
	}

	expectedUses := map[string]bool{"+list": false, "+set <name> <command>": false, "+delete <name>": false}
	for _, sub := range subcmds {
		if _, ok := expectedUses[sub.Use]; ok {
			expectedUses[sub.Use] = true
		}
	}
	for use, found := range expectedUses {
		if !found {
			t.Errorf("subcommand %q not found", use)
		}
	}
}

func TestAliasSetAndDeleteFlow(t *testing.T) {
	tmpDir := t.TempDir()
	t.Setenv("GITLINK_CONFIG_DIR", tmpDir)

	// 模拟 +set 操作：直接调用 saveAliases
	aliases := make(map[string]string)
	aliases["rl"] = "repo +list"
	aliases["ri"] = "repo +info"
	if err := saveAliases(aliases); err != nil {
		t.Fatalf("saveAliases failed: %v", err)
	}

	// 验证保存成功
	loaded, _ := loadAliases()
	if loaded["rl"] != "repo +list" {
		t.Fatalf("alias not saved correctly: %v", loaded)
	}
	if loaded["ri"] != "repo +info" {
		t.Fatalf("alias not saved correctly: %v", loaded)
	}

	// 模拟 +delete 操作：删除别名后保存
	delete(loaded, "rl")
	if err := saveAliases(loaded); err != nil {
		t.Fatalf("saveAliases after delete failed: %v", err)
	}

	// 验证删除成功
	final, _ := loadAliases()
	if _, ok := final["rl"]; ok {
		t.Fatal("alias 'rl' should have been deleted")
	}
	if final["ri"] != "repo +info" {
		t.Fatal("alias 'ri' should still exist")
	}
}

func TestAliasDeleteNonExistent(t *testing.T) {
	tmpDir := t.TempDir()
	t.Setenv("GITLINK_CONFIG_DIR", tmpDir)

	// 空别名列表，删除不存在的别名
	aliases, _ := loadAliases()
	if _, ok := aliases["nonexistent"]; ok {
		t.Fatal("nonexistent alias should not exist")
	}
	// 验证逻辑：别名不存在时不应执行删除
	// 这对应 alias.go 中 if _, ok := aliases[args[0]]; !ok 的检查
}

func TestAliasesFilePath(t *testing.T) {
	tmpDir := t.TempDir()
	t.Setenv("GITLINK_CONFIG_DIR", tmpDir)

	expected := tmpDir + "/aliases.yaml"
	got := aliasesPath()
	if got != expected {
		t.Errorf("expected path %s, got %s", expected, got)
	}
}