261 lines
7.1 KiB
Go
261 lines
7.1 KiB
Go
package issue
|
||
|
||
import (
|
||
"os"
|
||
"path/filepath"
|
||
"reflect"
|
||
"testing"
|
||
|
||
"github.com/gitlink-org/gitlink-cli/shortcuts/common"
|
||
)
|
||
|
||
func TestParseIssueNumbers(t *testing.T) {
|
||
got, err := parseIssueNumbers("1, 2,2, 3")
|
||
if err != nil {
|
||
t.Fatalf("parseIssueNumbers returned error: %v", err)
|
||
}
|
||
want := []string{"1", "2", "3"}
|
||
if !reflect.DeepEqual(got, want) {
|
||
t.Fatalf("parseIssueNumbers() = %#v, want %#v", got, want)
|
||
}
|
||
}
|
||
|
||
func TestParseIssueNumbersRejectsInvalidNumber(t *testing.T) {
|
||
if _, err := parseIssueNumbers("1,abc"); err == nil {
|
||
t.Fatal("parseIssueNumbers() expected an error for a non-integer issue number")
|
||
}
|
||
}
|
||
|
||
func TestReadCSVWithNumberHeader(t *testing.T) {
|
||
path := writeTempCSV(t, "title,number,state\nfirst,12,open\nsecond,13,open\n")
|
||
headers, rows, err := ReadCSV(path)
|
||
if err != nil {
|
||
t.Fatalf("ReadCSV returned error: %v", err)
|
||
}
|
||
col := FindColumn(headers, "number", "issue_number", "project_issues_index")
|
||
if col == -1 {
|
||
t.Fatal("column 'number' not found")
|
||
}
|
||
numbers := make([]string, 0, len(rows))
|
||
for _, row := range rows {
|
||
numbers = append(numbers, row[col])
|
||
}
|
||
numbers, err = normalizeIssueNumbers(numbers)
|
||
if err != nil {
|
||
t.Fatalf("normalizeIssueNumbers returned error: %v", err)
|
||
}
|
||
want := []string{"12", "13"}
|
||
if !reflect.DeepEqual(numbers, want) {
|
||
t.Fatalf("got %#v, want %#v", numbers, want)
|
||
}
|
||
}
|
||
|
||
func TestReadCSVWithProjectIssuesIndexHeader(t *testing.T) {
|
||
path := writeTempCSV(t, "title,project_issues_index,state\nfirst,12,open\nsecond,13,open\n")
|
||
headers, rows, err := ReadCSV(path)
|
||
if err != nil {
|
||
t.Fatalf("ReadCSV returned error: %v", err)
|
||
}
|
||
col := FindColumn(headers, "number", "issue_number", "project_issues_index")
|
||
if col == -1 {
|
||
t.Fatal("column 'project_issues_index' not found")
|
||
}
|
||
numbers := make([]string, 0, len(rows))
|
||
for _, row := range rows {
|
||
numbers = append(numbers, row[col])
|
||
}
|
||
numbers, err = normalizeIssueNumbers(numbers)
|
||
if err != nil {
|
||
t.Fatalf("normalizeIssueNumbers returned error: %v", err)
|
||
}
|
||
want := []string{"12", "13"}
|
||
if !reflect.DeepEqual(numbers, want) {
|
||
t.Fatalf("got %#v, want %#v", numbers, want)
|
||
}
|
||
}
|
||
|
||
func TestReadCSVHeaderlessReturnsNoColumnMatch(t *testing.T) {
|
||
path := writeTempCSV(t, "21,open\n22,closed\n21,duplicate\n")
|
||
headers, rows, err := ReadCSV(path)
|
||
if err != nil {
|
||
t.Fatalf("ReadCSV returned error: %v", err)
|
||
}
|
||
// 无表头时 FindColumn 返回 -1
|
||
col := FindColumn(headers, "number", "issue_number", "project_issues_index")
|
||
if col != -1 {
|
||
t.Fatalf("expected -1 for headerless CSV, got %d", col)
|
||
}
|
||
// 退回到首列（index 0）作为 issue 编号来源
|
||
col = 0
|
||
numbers := make([]string, 0, len(rows))
|
||
for _, row := range rows {
|
||
numbers = append(numbers, row[col])
|
||
}
|
||
numbers, err = normalizeIssueNumbers(numbers)
|
||
if err != nil {
|
||
t.Fatalf("normalizeIssueNumbers returned error: %v", err)
|
||
}
|
||
want := []string{"22", "21"}
|
||
if !reflect.DeepEqual(numbers, want) {
|
||
t.Fatalf("got %#v, want %#v", numbers, want)
|
||
}
|
||
}
|
||
|
||
func TestResolveIssueNumbersMergesCLIAndCSV(t *testing.T) {
|
||
path := writeTempCSV(t, "number\n2\n3\n")
|
||
ctx := &common.RuntimeContext{
|
||
Owner: "owner",
|
||
Repo: "repo",
|
||
}
|
||
got, err := ResolveIssueNumbers(ctx, "1,2", path, "")
|
||
if err != nil {
|
||
t.Fatalf("ResolveIssueNumbers returned error: %v", err)
|
||
}
|
||
want := []string{"1", "2", "3"}
|
||
if !reflect.DeepEqual(got, want) {
|
||
t.Fatalf("ResolveIssueNumbers() = %#v, want %#v", got, want)
|
||
}
|
||
}
|
||
|
||
func TestParseBool(t *testing.T) {
|
||
if !parseBool("true") {
|
||
t.Fatal("parseBool(true) = false, want true")
|
||
}
|
||
if parseBool("") {
|
||
t.Fatal("parseBool(empty) = true, want false")
|
||
}
|
||
}
|
||
|
||
func TestParseIssueNumbersEmpty(t *testing.T) {
|
||
got, err := parseIssueNumbers("")
|
||
if err != nil {
|
||
t.Fatalf("parseIssueNumbers returned error: %v", err)
|
||
}
|
||
if got != nil {
|
||
t.Fatalf("parseIssueNumbers() = %#v, want nil", got)
|
||
}
|
||
}
|
||
|
||
func TestReadIssueNumbersFromCSVMissingFile(t *testing.T) {
|
||
_, err := readIssueNumbersFromCSV("/nonexistent/file.csv")
|
||
if err == nil {
|
||
t.Fatal("expected error for missing file")
|
||
}
|
||
}
|
||
|
||
func TestReadIssueNumbersFromCSVEmpty(t *testing.T) {
|
||
path := writeTempCSV(t, "")
|
||
got, err := readIssueNumbersFromCSV(path)
|
||
if err != nil {
|
||
t.Fatalf("readIssueNumbersFromCSV error: %v", err)
|
||
}
|
||
if got != nil {
|
||
t.Fatalf("expected nil for empty CSV, got %#v", got)
|
||
}
|
||
}
|
||
|
||
func TestNormalizeIssueNumbersEmpty(t *testing.T) {
|
||
got, err := normalizeIssueNumbers([]string{"", " ", " "})
|
||
if err != nil {
|
||
t.Fatalf("normalizeIssueNumbers error: %v", err)
|
||
}
|
||
if len(got) != 0 {
|
||
t.Fatalf("expected empty, got %#v", got)
|
||
}
|
||
}
|
||
|
||
func TestMergeIssueNumbers(t *testing.T) {
|
||
got := mergeIssueNumbers([]string{"1", "2"}, []string{"2", "3"}, nil)
|
||
want := []string{"1", "2", "3"}
|
||
if !reflect.DeepEqual(got, want) {
|
||
t.Fatalf("mergeIssueNumbers() = %#v, want %#v", got, want)
|
||
}
|
||
}
|
||
|
||
func TestCollectIssueNumbersCSVOnly(t *testing.T) {
|
||
path := writeTempCSV(t, "number\n5\n6\n")
|
||
got, err := collectIssueNumbers("", path)
|
||
if err != nil {
|
||
t.Fatalf("collectIssueNumbers error: %v", err)
|
||
}
|
||
want := []string{"5", "6"}
|
||
if !reflect.DeepEqual(got, want) {
|
||
t.Fatalf("collectIssueNumbers() = %#v, want %#v", got, want)
|
||
}
|
||
}
|
||
|
||
func TestCollectIssueNumbersCLIOnly(t *testing.T) {
|
||
got, err := collectIssueNumbers("1,2,3", "")
|
||
if err != nil {
|
||
t.Fatalf("collectIssueNumbers error: %v", err)
|
||
}
|
||
want := []string{"1", "2", "3"}
|
||
if !reflect.DeepEqual(got, want) {
|
||
t.Fatalf("collectIssueNumbers() = %#v, want %#v", got, want)
|
||
}
|
||
}
|
||
|
||
func TestNormalizeIssueNumbersRejectsNonInt(t *testing.T) {
|
||
_, err := normalizeIssueNumbers([]string{"abc"})
|
||
if err == nil {
|
||
t.Fatal("expected error for non-integer")
|
||
}
|
||
}
|
||
|
||
func TestReadIssueNumbersFromCSVShortRow(t *testing.T) {
|
||
// Number column is index 1; short row skips due to len check
|
||
path := writeTempCSV(t, "title,number\nfirst,1\nsecond,\nthird,3\n")
|
||
got, err := readIssueNumbersFromCSV(path)
|
||
if err != nil {
|
||
t.Fatalf("readIssueNumbersFromCSV error: %v", err)
|
||
}
|
||
want := []string{"1", "3"}
|
||
if !reflect.DeepEqual(got, want) {
|
||
t.Fatalf("readIssueNumbersFromCSV() = %#v, want %#v", got, want)
|
||
}
|
||
}
|
||
|
||
func TestReadIssueNumbersFromCSVIssueNumberHeader(t *testing.T) {
|
||
path := writeTempCSV(t, "issue_number,title\n42,test\n")
|
||
got, err := readIssueNumbersFromCSV(path)
|
||
if err != nil {
|
||
t.Fatalf("readIssueNumbersFromCSV error: %v", err)
|
||
}
|
||
want := []string{"42"}
|
||
if !reflect.DeepEqual(got, want) {
|
||
t.Fatalf("readIssueNumbersFromCSV() = %#v, want %#v", got, want)
|
||
}
|
||
}
|
||
|
||
func TestParseBoolFalse(t *testing.T) {
|
||
if parseBool("false") {
|
||
t.Fatal("parseBool(false) = true, want false")
|
||
}
|
||
if parseBool(" FALSE ") {
|
||
t.Fatal("parseBool(FALSE) = true, want false")
|
||
}
|
||
}
|
||
|
||
func TestCollectIssueNumbersInvalidCLI(t *testing.T) {
|
||
_, err := collectIssueNumbers("abc,def", "")
|
||
if err == nil {
|
||
t.Fatal("expected error for invalid issue numbers")
|
||
}
|
||
}
|
||
|
||
func TestCollectIssueNumbersCSVReadError(t *testing.T) {
|
||
_, err := collectIssueNumbers("", "/nonexistent/file.csv")
|
||
if err == nil {
|
||
t.Fatal("expected error for missing CSV file")
|
||
}
|
||
}
|
||
|
||
func writeTempCSV(t *testing.T, content string) string {
|
||
t.Helper()
|
||
path := filepath.Join(t.TempDir(), "issues.csv")
|
||
if err := os.WriteFile(path, []byte(content), 0o600); err != nil {
|
||
t.Fatalf("write temp csv: %v", err)
|
||
}
|
||
return path
|
||
}
|