From a4c5fc94b79955f45664fddd8dc7e37bd76541be Mon Sep 17 00:00:00 2001
From: Vladimir V Maksimov <v.maksimov@mediatoday.ru>
Date: Tue, 10 Mar 2026 05:26:40 +0300
Subject: [PATCH] add test for TableBlockStyle column width calculation

Verifies that columns with styled rows don't get inflated widths
from struct string representation.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---
 image_table_test.go | 65 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)

diff --git a/image_table_test.go b/image_table_test.go
index 0d5894b..f3a0dac 100644
--- a/image_table_test.go
+++ b/image_table_test.go
@@ -37,6 +37,71 @@ func TestDrawTable(t *testing.T) {
 	log.Println("Готово! Файл warm_table.png создан.")
 }
 
+func TestTableBlockStyleColumnWidths(t *testing.T) {
+	// Тест воспроизводит баг: когда все строки — *TableBlockStyle,
+	// ширина первого столбца не должна раздуваться из-за строкового
+	// представления всей структуры.
+	header := []string{"Час", "Заявка", "Ерид", "Тип отклонения", "Показы", "Клики"}
+
+	colorUp := color.RGBA{R: 200, G: 255, B: 200, A: 255}
+	colorDown := color.RGBA{R: 255, G: 200, B: 200, A: 255}
+
+	rows := []any{
+		&TableBlockStyle{
+			Cells:           []any{"10", "test", "eridRanyko", "переполнение", "41% (сегодня 100 - вчера 71)", "18% (сегодня 50 - вчера 42)"},
+			BackgroundColor: colorUp,
+		},
+		&TableBlockStyle{
+			Cells:           []any{"11", "test", "eridRanyko", "недостаток", "39% (сегодня 60 - вчера 99)", "22% (сегодня 30 - вчера 39)"},
+			BackgroundColor: colorDown,
+		},
+		&TableBlockStyle{
+			Cells:           []any{"12", "test", "eridRanyko", "переполнение", "24% (сегодня 80 - вчера 65)", "10% (сегодня 44 - вчера 40)"},
+			BackgroundColor: colorUp,
+		},
+	}
+
+	img := DrawTableWarm(header, rows)
+
+	// Ширина столбца "Час" (с коротким содержимым "10","11","12") не должна
+	// быть больше 30% от общей ширины. До фикса она занимала ~70%.
+	imgWidth := img.Bounds().Dx()
+	maxFirstColWidth := imgWidth * 30 / 100
+
+	// Проверяем через пиксели: ищем первый вертикальный разделитель в заголовке
+	firstDividerX := 0
+	for x := 1; x < imgWidth; x++ {
+		r, g, b, _ := img.At(x, 16).RGBA()
+		if r>>8 == uint32(borderCol.R) && g>>8 == uint32(borderCol.G) && b>>8 == uint32(borderCol.B) {
+			firstDividerX = x
+			break
+		}
+	}
+
+	if firstDividerX == 0 {
+		t.Fatal("не нашёл вертикальный разделитель первого столбца")
+	}
+
+	t.Logf("ширина первого столбца: %d, общая ширина: %d, лимит: %d", firstDividerX, imgWidth, maxFirstColWidth)
+
+	if firstDividerX > maxFirstColWidth {
+		t.Errorf("первый столбец слишком широкий: %d > %d (%.0f%% от общей ширины)", firstDividerX, maxFirstColWidth, float64(firstDividerX)*100/float64(imgWidth))
+	}
+
+	// Сохраняем для визуальной проверки
+	file, err := os.Create("./test-data/styled_table.png")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer file.Close()
+
+	if err := png.Encode(file, img); err != nil {
+		t.Fatal(err)
+	}
+
+	t.Log("Файл test-data/styled_table.png создан")
+}
+
 func TestDocument(t *testing.T) {
 	doc := Document{
 		Blocks: []TableBlock{