painter_test.go

package tui

import (
	"image"
	"testing"
)

func TestMask_Full(t *testing.T) {
	surface := NewTestSurface(10, 10)

	p := NewPainter(surface, NewTheme())
	p.WithMask(image.Rect(0, 0, 10, 10), func(p *Painter) {
		p.WithMask(image.Rect(0, 0, 10, 10), func(p *Painter) {
			sz := p.surface.Size()
			for x := 0; x < sz.X; x++ {
				for y := 0; y < sz.Y; y++ {
					p.DrawRune(x, y, '█')
				}
			}
		})
	})

	want := `
██████████
██████████
██████████
██████████
██████████
██████████
██████████
██████████
██████████
██████████
`
	if diff := surfaceEquals(surface, want); diff != "" {
		t.Error(diff)
	}
}

func TestMask_Inset(t *testing.T) {
	surface := NewTestSurface(10, 10)

	p := NewPainter(surface, NewTheme())
	p.WithMask(image.Rect(0, 0, 10, 10), func(p *Painter) {
		p.WithMask(image.Rect(1, 1, 9, 9), func(p *Painter) {
			sz := p.surface.Size()
			for x := 0; x < sz.X; x++ {
				for y := 0; y < sz.Y; y++ {
					p.DrawRune(x, y, '█')
				}
			}
		})
	})

	want := `
..........
.████████.
.████████.
.████████.
.████████.
.████████.
.████████.
.████████.
.████████.
..........
`
	if diff := surfaceEquals(surface, want); diff != "" {
		t.Error(diff)
	}
}

func TestMask_FirstCell(t *testing.T) {
	surface := NewTestSurface(10, 10)

	p := NewPainter(surface, NewTheme())
	p.WithMask(image.Rect(0, 0, 10, 10), func(p *Painter) {
		p.WithMask(image.Rect(0, 0, 1, 1), func(p *Painter) {
			sz := p.surface.Size()
			for x := 0; x < sz.X; x++ {
				for y := 0; y < sz.Y; y++ {
					p.DrawRune(x, y, '█')
				}
			}
		})
	})

	want := `
█.........
..........
..........
..........
..........
..........
..........
..........
..........
..........
`
	if diff := surfaceEquals(surface, want); diff != "" {
		t.Error(diff)
	}
}

func TestMask_LastCell(t *testing.T) {
	surface := NewTestSurface(10, 10)

	p := NewPainter(surface, NewTheme())
	p.WithMask(image.Rect(0, 0, 10, 10), func(p *Painter) {
		p.WithMask(image.Rect(9, 9, 10, 10), func(p *Painter) {
			sz := p.surface.Size()
			for x := 0; x < sz.X; x++ {
				for y := 0; y < sz.Y; y++ {
					p.DrawRune(x, y, '█')
				}
			}
		})
	})

	want := `
..........
..........
..........
..........
..........
..........
..........
..........
..........
.........█
`
	if diff := surfaceEquals(surface, want); diff != "" {
		t.Error(diff)
	}
}

func TestMask_MaskWithinEmptyMaskIsHidden(t *testing.T) {
	surface := NewTestSurface(10, 10)

	p := NewPainter(surface, NewTheme())
	p.WithMask(image.Rect(0, 0, 0, 0), func(p *Painter) {
		p.WithMask(image.Rect(1, 1, 9, 9), func(p *Painter) {
			sz := p.surface.Size()
			for x := 0; x < sz.X; x++ {
				for y := 0; y < sz.Y; y++ {
					p.DrawRune(x, y, '█')
				}
			}
		})
	})

	want := `
..........
..........
..........
..........
..........
..........
..........
..........
..........
..........
`
	if diff := surfaceEquals(surface, want); diff != "" {
		t.Error(diff)
	}
}

func TestWithStyle_ApplyStyle(t *testing.T) {
	surface := NewTestSurface(5, 5)

	theme := NewTheme()
	theme.SetStyle("explicit", Style{Fg: ColorWhite, Bg: ColorBlack})

	p := NewPainter(surface, theme)
	p.WithMask(image.Rect(0, 0, 5, 5), func(p *Painter) {
		p.WithMask(image.Rect(1, 1, 4, 4), func(p *Painter) {
			sz := p.surface.Size()
			for x := 0; x < sz.X; x++ {
				for y := 0; y < sz.Y; y++ {
					p.DrawRune(x, y, ' ')
				}
			}

			p.WithMask(image.Rect(2, 2, 4, 4), func(p *Painter) {
				p.WithStyle("explicit", func(p *Painter) {
					sz := p.surface.Size()
					for x := 0; x < sz.X; x++ {
						for y := 0; y < sz.Y; y++ {
							p.DrawRune(x, y, '!')
						}
					}

				})
			})
		})
	})

	wantFg := `
.....
.000.
.022.
.022.
.....
`

	wantBg := `
.....
.000.
.011.
.011.
.....
`

	if surface.FgColors() != wantFg {
		t.Errorf("got = \n%s\n\nwant = \n%s", surface.FgColors(), wantFg)
	}
	if surface.BgColors() != wantBg {
		t.Errorf("got = \n%s\n\nwant = \n%s", surface.BgColors(), wantBg)
	}

}

func TestWithStyle_Stacks(t *testing.T) {
	surface := NewTestSurface(10, 10)

	theme := NewTheme()
	theme.SetStyle("explicit", Style{Fg: Color(3)})
	theme.SetStyle("auxiliary", Style{Fg: Color(2)})

	p := NewPainter(surface, theme)
	p.WithMask(image.Rect(0, 0, 10, 10), func(p *Painter) {

		// Set "explicit" and draw upper-left and upper-right.
		p.WithStyle("explicit", func(p *Painter) {
			p.WithMask(image.Rect(1, 1, 4, 4), func(p *Painter) {
				sz := p.surface.Size()
				for x := 0; x < sz.X; x++ {
					for y := 0; y < sz.Y; y++ {
						p.DrawRune(x, y, ' ')
					}
				}
			})
			// set "auxiliary" before drawing upper-right.
			p.WithStyle("auxiliary", func(p *Painter) {
				p.WithMask(image.Rect(7, 1, 9, 3), func(p *Painter) {
					sz := p.surface.Size()
					for x := 0; x < sz.X; x++ {
						for y := 0; y < sz.Y; y++ {
							p.DrawRune(x, y, ' ')
						}
					}
				})
			})
			// Then draw bottom-right, falling back to "explicit".
			p.WithMask(image.Rect(1, 6, 4, 9), func(p *Painter) {
				sz := p.surface.Size()
				for x := 0; x < sz.X; x++ {
					for y := 0; y < sz.Y; y++ {
						p.DrawRune(x, y, ' ')
					}
				}
			})
		})

		// Use global default for bottom-right.
		p.WithMask(image.Rect(6, 6, 9, 9), func(p *Painter) {
			sz := p.surface.Size()
			for x := 0; x < sz.X; x++ {
				for y := 0; y < sz.Y; y++ {
					p.DrawRune(x, y, ' ')
				}
			}
		})
	})

	wantFg := `
..........
.333...22.
.333...22.
.333......
..........
..........
.333..000.
.333..000.
.333..000.
..........
`

	if surface.FgColors() != wantFg {
		t.Errorf("got = \n%s\n\nwant = \n%s", surface.FgColors(), wantFg)
	}
}

var styleInheritTests = []struct {
	test            string
	theme           func() *Theme
	wantFg          string
	wantDecorations string
}{
	{
		test: "no second style, keeps first",
		theme: func() *Theme {
			r := NewTheme()
			r.SetStyle("first", Style{Fg: Color(3), Bold: DecorationOn})
			return r
		},
		wantFg: `
333..333..
333..333..
333..333..
`,
		wantDecorations: `
222..222..
222..222..
222..222..
`,
	},
	{
		test: "empty second style, inherits from first",
		theme: func() *Theme {
			r := NewTheme()
			r.SetStyle("first", Style{Fg: Color(3), Bold: DecorationOn})
			r.SetStyle("second", Style{})
			return r
		},
		wantFg: `
333..333..
333..333..
333..333..
`,
		wantDecorations: `
222..222..
222..222..
222..222..
`,
	},
	{
		test: "second style overrides only explicit properties",
		theme: func() *Theme {
			r := NewTheme()
			r.SetStyle("first", Style{Fg: Color(3), Bold: DecorationOn})
			r.SetStyle("second", Style{Underline: DecorationOn})
			return r
		},
		wantFg: `
333..333..
333..333..
333..333..
`,
		wantDecorations: `
222..666..
222..666..
222..666..
`},
	{
		test: "second style resets a previously set property",
		theme: func() *Theme {
			r := NewTheme()
			r.SetStyle("first", Style{Bold: DecorationOn})
			r.SetStyle("second", Style{Bold: DecorationOff})
			return r
		},
		wantDecorations: `
222..000..
222..000..
222..000..
`},
}

func TestWithStyle_Inherit(t *testing.T) {
	for _, tt := range styleInheritTests {
		tt := tt
		t.Run(tt.test, func(t *testing.T) {

			surface := NewTestSurface(10, 3)
			theme := tt.theme()

			fill := func(p *Painter) {
				sz := p.surface.Size()
				for x := 0; x < sz.X; x++ {
					for y := 0; y < sz.Y; y++ {
						p.DrawRune(x, y, ' ')
					}
				}
			}

			p := NewPainter(surface, theme)
			p.WithMask(image.Rect(0, 0, 10, 3), func(p *Painter) {
				p.WithStyle("first", func(p *Painter) {
					p.WithMask(image.Rect(0, 0, 3, 3), fill)

					p.WithStyle("second", func(p *Painter) {
						p.WithMask(image.Rect(5, 0, 8, 3), fill)
					})
				})
			})

			gotColors := surface.FgColors()
			gotDecorations := surface.Decorations()

			if tt.wantFg != "" && gotColors != tt.wantFg {
				t.Errorf("unexpected colors: got = \n%s\nwant = \n%s", gotColors, tt.wantFg)
			}
			if gotDecorations != tt.wantDecorations {
				t.Errorf("unexpected decorations: got = \n%s\nwant = \n%s", gotDecorations, tt.wantDecorations)
			}
		})
	}
}