37 lines
956 B
Go
37 lines
956 B
Go
package cmd
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestThemeScreenshotCommandRegistered(t *testing.T) {
|
|
root := NewRoot()
|
|
theme, _, err := root.Find([]string{"theme", "screenshot"})
|
|
if err != nil {
|
|
t.Fatalf("find theme screenshot: %v", err)
|
|
}
|
|
if theme.Name() != "screenshot" {
|
|
t.Fatalf("resolved command = %q, want screenshot", theme.Name())
|
|
}
|
|
}
|
|
|
|
func TestThemeScreenshotHasGalleryAndMobileFlags(t *testing.T) {
|
|
root := NewRoot()
|
|
cmd, _, _ := root.Find([]string{"theme", "screenshot"})
|
|
for _, name := range []string{"gallery", "mobile", "out", "slug"} {
|
|
if cmd.Flags().Lookup(name) == nil {
|
|
t.Errorf("missing --%s flag", name)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestThemeScreenshotDefaultOutIsPreviewPng(t *testing.T) {
|
|
root := NewRoot()
|
|
cmd, _, _ := root.Find([]string{"theme", "screenshot"})
|
|
out, _ := cmd.Flags().GetString("out")
|
|
if !strings.HasSuffix(out, "preview.png") {
|
|
t.Fatalf("default --out = %q, want it to end with preview.png", out)
|
|
}
|
|
}
|