test(charmbracelet#58): tested --copy flag

AlejandroSuero · May 19, 2024 · d4d556c · d4d556c
1 parent 37dafec
commit d4d556c
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/freeze_test.go b/freeze_test.go
@@ -10,6 +10,7 @@ import (
 	"testing"
 
 	"github.com/aymanbagabas/go-udiff"
+	"golang.design/x/clipboard"
 )
 
 const binary = "./test/freeze-test"
@@ -56,6 +57,31 @@ func TestFreezeOutput(t *testing.T) {
 	}
 }
 
+func TestFreezeCopy(t *testing.T) {
+	output := "artichoke-test.svg"
+	defer os.Remove(output)
+
+	cmd := exec.Command(binary, "test/input/artichoke.hs", "-o", output, "--copy")
+	err := cmd.Run()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	_, err = os.Stat(output)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = clipboard.Init()
+	if err != nil {
+		t.Fatal(err)
+	}
+	png := clipboard.Read(clipboard.FmtImage)
+	if png == nil {
+		t.Fatal("clipboard is empty")
+	}
+}
+
 func TestFreezeHelp(t *testing.T) {
 	out := bytes.Buffer{}
 	cmd := exec.Command(binary)
@@ -134,6 +160,11 @@ func TestFreezeConfigurations(t *testing.T) {
 			flags:  []string{"--language", "go", "--height", "800", "--width", "750", "--config", "full", "--window=false", "--show-line-numbers"},
 			output: "bubbletea",
 		},
+		{
+			input:  "test/input/bubbletea.model",
+			flags:  []string{"--language", "go", "--height", "800", "--width", "750", "--config", "full", "--window=false", "--show-line-numbers", "--copy"},
+			output: "bubbletea-copy",
+		},
 		// {
 		// 	flags:  []string{"--execute", "layout", "--height", "800", "--config", "full", "--margin", "50,10"},
 		// 	output: "composite-2",