| package actions |
|
|
| import ( |
| "os" |
| "path/filepath" |
| "strings" |
| "testing" |
|
|
| "github.com/spf13/cobra" |
| ) |
|
|
| func newPDFCmd() *cobra.Command { |
| cmd := &cobra.Command{} |
| cmd.Flags().String("output", "", "") |
| cmd.Flags().String("tab", "", "") |
| cmd.Flags().Bool("landscape", false, "") |
| cmd.Flags().String("scale", "", "") |
| cmd.Flags().String("paper-width", "", "") |
| cmd.Flags().String("paper-height", "", "") |
| cmd.Flags().String("margin-top", "", "") |
| cmd.Flags().String("margin-bottom", "", "") |
| cmd.Flags().String("margin-left", "", "") |
| cmd.Flags().String("margin-right", "", "") |
| cmd.Flags().String("page-ranges", "", "") |
| cmd.Flags().Bool("prefer-css-page-size", false, "") |
| cmd.Flags().Bool("display-header-footer", false, "") |
| cmd.Flags().String("header-template", "", "") |
| cmd.Flags().String("footer-template", "", "") |
| cmd.Flags().Bool("generate-tagged-pdf", false, "") |
| cmd.Flags().Bool("generate-document-outline", false, "") |
| cmd.Flags().Bool("file-output", false, "") |
| cmd.Flags().String("path", "", "") |
| return cmd |
| } |
|
|
| func TestPDF(t *testing.T) { |
| m := newMockServer() |
| m.response = "FAKEPDFDATA" |
| defer m.close() |
| client := m.server.Client() |
|
|
| outFile := filepath.Join(t.TempDir(), "test.pdf") |
| cmd := newPDFCmd() |
| _ = cmd.Flags().Set("output", outFile) |
| _ = cmd.Flags().Set("tab", "tab-abc") |
| _ = cmd.Flags().Set("landscape", "true") |
| _ = cmd.Flags().Set("scale", "0.8") |
| PDF(client, m.base(), "", cmd) |
| if m.lastPath != "/tabs/tab-abc/pdf" { |
| t.Errorf("expected /tabs/tab-abc/pdf, got %s", m.lastPath) |
| } |
| if !strings.Contains(m.lastQuery, "landscape=true") { |
| t.Errorf("expected landscape=true, got %s", m.lastQuery) |
| } |
| if !strings.Contains(m.lastQuery, "scale=0.8") { |
| t.Errorf("expected scale=0.8, got %s", m.lastQuery) |
| } |
| data, err := os.ReadFile(outFile) |
| if err != nil { |
| t.Fatalf("file not written: %v", err) |
| } |
| if string(data) != "FAKEPDFDATA" { |
| t.Errorf("unexpected content: %s", string(data)) |
| } |
| } |
|
|
| func TestPDFAllOptions(t *testing.T) { |
| m := newMockServer() |
| m.response = "FAKEPDFDATA" |
| defer m.close() |
| client := m.server.Client() |
|
|
| outFile := filepath.Join(t.TempDir(), "test.pdf") |
| cmd := newPDFCmd() |
| _ = cmd.Flags().Set("output", outFile) |
| _ = cmd.Flags().Set("tab", "tab-123") |
| _ = cmd.Flags().Set("landscape", "true") |
| _ = cmd.Flags().Set("scale", "1.5") |
| _ = cmd.Flags().Set("paper-width", "11") |
| _ = cmd.Flags().Set("paper-height", "8.5") |
| _ = cmd.Flags().Set("margin-top", "1") |
| _ = cmd.Flags().Set("margin-bottom", "1") |
| _ = cmd.Flags().Set("margin-left", "0.5") |
| _ = cmd.Flags().Set("margin-right", "0.5") |
| _ = cmd.Flags().Set("page-ranges", "1-3,5") |
| _ = cmd.Flags().Set("prefer-css-page-size", "true") |
| _ = cmd.Flags().Set("display-header-footer", "true") |
| _ = cmd.Flags().Set("header-template", "<span class='title'></span>") |
| _ = cmd.Flags().Set("footer-template", "<span class='pageNumber'></span>") |
| _ = cmd.Flags().Set("generate-tagged-pdf", "true") |
| _ = cmd.Flags().Set("generate-document-outline", "true") |
|
|
| PDF(client, m.base(), "", cmd) |
| if m.lastPath != "/tabs/tab-123/pdf" { |
| t.Errorf("expected /tabs/tab-123/pdf, got %s", m.lastPath) |
| } |
|
|
| expectedParams := []string{ |
| "landscape=true", |
| "scale=1.5", |
| "paperWidth=11", |
| "paperHeight=8.5", |
| "marginTop=1", |
| "marginBottom=1", |
| "marginLeft=0.5", |
| "marginRight=0.5", |
| "preferCSSPageSize=true", |
| "displayHeaderFooter=true", |
| "generateTaggedPDF=true", |
| "generateDocumentOutline=true", |
| "raw=true", |
| } |
|
|
| for _, expected := range expectedParams { |
| if !strings.Contains(m.lastQuery, expected) { |
| t.Errorf("expected %s in query, got %s", expected, m.lastQuery) |
| } |
| } |
| } |
|
|
