test: add end-to-end integration test for HTML filter to image extraction

2026-04-07 20:14:27 -04:00
parent b13bd1d71c
commit abf8feb229
1 changed files with 72 additions and 0 deletions
@@ -0,0 +1,72 @@
+// lib/parse/integration_test.go
+package parse
+
+import (
+	"io"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"strings"
+	"testing"
+)
+
+func TestIntegration_HTMLFilterToExtractImages(t *testing.T) {
+	// Skip if html-kitty filter or w3m not available
+	filterPath := filepath.Join(os.Getenv("HOME"),
+		".local/libexec/aerc/filters/html-kitty")
+	if _, err := os.Stat(filterPath); err != nil {
+		t.Skip("html-kitty filter not found:", err)
+	}
+	if _, err := exec.LookPath("w3m"); err != nil {
+		t.Skip("w3m not found:", err)
+	}
+
+	// HTML with a data: URI image
+	html := `<html><body>
+<p>Hello World</p>
+<img src="data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNk+M9QDwADhgGAWjR9awAAAABJRU5ErkJggg==" alt="tiny red pixel">
+<p>Goodbye</p>
+</body></html>`
+
+	cmd := exec.Command("python3", filterPath)
+	cmd.Stdin = strings.NewReader(html)
+	cmd.Env = append(os.Environ(), "COLUMNS=80")
+	out, err := cmd.Output()
+	if err != nil {
+		t.Fatalf("filter failed: %v", err)
+	}
+
+	// Extract images from filter output
+	cleaned, images := ExtractImages(strings.NewReader(string(out)))
+	if len(images) == 0 {
+		t.Fatal("expected at least 1 image from filter output")
+	}
+
+	// Verify image file exists on disk
+	if _, err := os.Stat(images[0].Path); err != nil {
+		t.Errorf("image file not found: %s", images[0].Path)
+	}
+
+	// Verify alt text
+	if images[0].Alt != "tiny red pixel" {
+		t.Errorf("alt = %q, want 'tiny red pixel'", images[0].Alt)
+	}
+
+	// Verify cleaned output has placeholder
+	cleanedBytes, _ := io.ReadAll(cleaned)
+	cleanedStr := string(cleanedBytes)
+	if !strings.Contains(cleanedStr, "\x00IMG:0\x00") {
+		t.Error("cleaned output missing placeholder")
+	}
+
+	// Verify text content preserved
+	if !strings.Contains(cleanedStr, "Hello World") {
+		t.Error("text content 'Hello World' lost")
+	}
+
+	// Clean up temp files
+	for _, img := range images {
+		dir := filepath.Dir(img.Path)
+		os.RemoveAll(dir)
+	}
+}