prime8dev

markdown_images_test.go (1871B)

---

 package tui
 
 import (
 	"bytes"
 	"image"
 	"image/color"
 	"image/png"
 	"net/http"
 	"net/http/httptest"
 	"strings"
 	"testing"
 
 	"github.com/charmbracelet/x/ansi"
 )
 
 func TestReadmeImagesAreCapped(t *testing.T) {
 	for _, size := range [][2]int{{2000, 400}, {400, 2000}, {1200, 1200}} {
 		img := image.NewRGBA(image.Rect(0, 0, size[0], size[1]))
 		for y := 0; y < size[1]; y++ {
 			for x := 0; x < size[0]; x++ {
 				img.Set(x, y, color.RGBA{200, 100, 50, 255})
 			}
 		}
 		var buf bytes.Buffer
 		if err := png.Encode(&buf, img); err != nil {
 			t.Fatal(err)
 		}
 		srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
 			w.Header().Set("Content-Type", "image/png")
 			_, _ = w.Write(buf.Bytes())
 		}))
 		md := "# demo\n\nsome text\n\n![shot](" + srv.URL + "/a.png)\n\n<p align=\"center\"><img src=\"" + srv.URL + "/b.png\" width=\"900\"></p>\n\nmore text\n"
 		out := renderMarkdown(md, 200)
 		_, maxRows := imageBox(200)
 		srv.Close()
 
 		rows := 0
 		for _, line := range strings.Split(out, "\n") {
 			if strings.Contains(line, "▀") || strings.Contains(line, "▄") {
 				rows++
 				if w := ansi.StringWidth(line); w > imageMaxCols {
 					t.Errorf("%v: image row %d cells wide", size, w)
 				}
 			}
 		}
 		if rows == 0 {
 			t.Fatalf("%v: no image rendered:\n%s", size, out)
 		}
 		if rows > 2*maxRows {
 			t.Errorf("%v: %d image rows for two images, cap is %d each", size, rows, maxRows)
 		}
 		if !strings.Contains(ansi.Strip(out), "more text") {
 			t.Errorf("text after images lost")
 		}
 	}
 }
 
 func TestImageBoxScalesWithPane(t *testing.T) {
 	for _, c := range []struct{ width, cols, rows int }{
 		{40, 40, 13}, {60, 60, 20}, {90, 90, 24}, {200, 96, 24}, {10, 10, 4},
 	} {
 		if cols, rows := imageBox(c.width); cols != c.cols || rows != c.rows {
 			t.Errorf("imageBox(%d) = %dx%d, want %dx%d", c.width, cols, rows, c.cols, c.rows)
 		}
 	}
 }