Files
larksuite--cli/shortcuts/drive/drive_export_common_test.go
wehub-resource-sync bf9395e022
CI / license-header (push) Has been skipped
CI / e2e-dry-run (push) Has been skipped
CI / fast-gate (push) Failing after 0s
Test PR Label Logic / test-pr-labels (push) Failing after 1s
Skill Format Check / check-format (push) Failing after 2s
CI / security (push) Failing after 5s
CI / unit-test (push) Has been skipped
CI / lint (push) Has been skipped
CI / script-test (push) Has been skipped
CI / deterministic-gate (push) Has been skipped
CI / coverage (push) Has been skipped
CI / results (push) Has been cancelled
CI / deadcode (push) Has been cancelled
CI / e2e-live (push) Has been cancelled
chore: import upstream snapshot with attribution
2026-07-13 12:22:54 +08:00

80 lines
2.4 KiB
Go

// Copyright (c) 2026 Lark Technologies Pte. Ltd.
// SPDX-License-Identifier: MIT
package drive
import "testing"
func TestDriveExportStatusLabelCoversKnownAndUnknownCodes(t *testing.T) {
t.Parallel()
tests := []struct {
name string
status driveExportStatus
want string
}{
{
name: "size limit",
status: driveExportStatus{JobStatus: 107},
want: "export_size_limit",
},
{
name: "not exist",
status: driveExportStatus{JobStatus: 123},
want: "docs_not_exist",
},
{
name: "unknown status",
status: driveExportStatus{JobStatus: 999},
want: "status_999",
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
t.Parallel()
if got := tt.status.StatusLabel(); got != tt.want {
t.Fatalf("StatusLabel() = %q, want %q", got, tt.want)
}
})
}
}
func TestParseDriveExportStatusWithoutResultKeepsTicket(t *testing.T) {
t.Parallel()
status := parseDriveExportStatus("ticket_export_test", map[string]interface{}{})
if status.Ticket != "ticket_export_test" {
t.Fatalf("ticket = %q, want %q", status.Ticket, "ticket_export_test")
}
if status.FileToken != "" {
t.Fatalf("file token = %q, want empty", status.FileToken)
}
}
func TestSanitizeExportFileNameAndEnsureExtension(t *testing.T) {
t.Parallel()
if got := sanitizeExportFileName("../quarterly:report?.pdf", "fallback.bin"); got != "quarterly_report_.pdf" {
t.Fatalf("sanitizeExportFileName() = %q, want %q", got, "quarterly_report_.pdf")
}
if got := ensureExportFileExtension("meeting-notes", "markdown"); got != "meeting-notes.md" {
t.Fatalf("ensureExportFileExtension() = %q, want %q", got, "meeting-notes.md")
}
if got := ensureExportFileExtension("report.pdf", "pdf"); got != "report.pdf" {
t.Fatalf("ensureExportFileExtension() should preserve suffix, got %q", got)
}
if got := ensureExportFileExtension("crm", "base"); got != "crm.base" {
t.Fatalf("ensureExportFileExtension() = %q, want %q", got, "crm.base")
}
if got := exportFileSuffix("base"); got != ".base" {
t.Fatalf("exportFileSuffix(base) = %q, want %q", got, ".base")
}
if got := ensureExportFileExtension("report", "pptx"); got != "report.pptx" {
t.Fatalf("ensureExportFileExtension() = %q, want %q", got, "report.pptx")
}
if got := ensureExportFileExtension("report.pptx", "pptx"); got != "report.pptx" {
t.Fatalf("ensureExportFileExtension() should preserve suffix, got %q", got)
}
}