From 52a5ed0ad48b379e66ceef8133202fa68ac6e96d Mon Sep 17 00:00:00 2001
From: Hajime Hoshi <hajimehoshi@gmail.com>
Date: Sat, 11 Jan 2020 02:38:03 +0900
Subject: [PATCH] buffered: Add tests for #1050

---
 internal/buffered/image_test.go | 132 ++++++++++++++++++++++++++++----
 1 file changed, 116 insertions(+), 16 deletions(-)

diff --git a/internal/buffered/image_test.go b/internal/buffered/image_test.go
index e8bec3718..43af344b2 100644
--- a/internal/buffered/image_test.go
+++ b/internal/buffered/image_test.go
@@ -15,7 +15,6 @@
 package buffered_test
 
 import (
-	"errors"
 	"image/color"
 	"os"
 	"testing"
@@ -40,31 +39,132 @@ func TestMain(m *testing.M) {
 	}()
 
 	for {
-		select {
-		case f := <-mainCh:
+		if err := ebiten.Run(func(*ebiten.Image) error {
+			f := <-mainCh
 			f()
+			return nil
+		}, 320, 240, 1, ""); err != nil {
+			panic(err)
 		}
 	}
 }
 
-func TestSetBeforeRun(t *testing.T) {
-	clr := color.RGBA{1, 2, 3, 4}
+type testResult struct {
+	want color.RGBA
+	got  <-chan color.RGBA
+}
 
+var testSetBeforeMainResult = func() testResult {
+	clr := color.RGBA{1, 2, 3, 4}
 	img, _ := ebiten.NewImage(16, 16, ebiten.FilterDefault)
 	img.Set(0, 0, clr)
 
-	want := clr
-	var got color.RGBA
+	ch := make(chan color.RGBA, 1)
+	go func() {
+		runOnMainThread(func() {
+			ch <- img.At(0, 0).(color.RGBA)
+		})
+	}()
 
-	runOnMainThread(func() {
-		quit := errors.New("quit")
-		if err := ebiten.Run(func(*ebiten.Image) error {
-			got = img.At(0, 0).(color.RGBA)
-			return quit
-		}, 320, 240, 1, ""); err != nil && err != quit {
-			t.Fatal(err)
-		}
-	})
+	return testResult{
+		want: clr,
+		got:  ch,
+	}
+}()
+
+func TestSetBeforeMain(t *testing.T) {
+	got := <-testSetBeforeMainResult.got
+	want := testSetBeforeMainResult.want
+
+	if got != want {
+		t.Errorf("got: %v, want: %v", got, want)
+	}
+}
+
+var testDrawImageBeforeMainResult = func() testResult {
+	const w, h = 16, 16
+	src, _ := ebiten.NewImage(w, h, ebiten.FilterDefault)
+	dst, _ := ebiten.NewImage(w, h, ebiten.FilterDefault)
+	src.Set(0, 0, color.White)
+	dst.DrawImage(src, nil)
+
+	ch := make(chan color.RGBA, 1)
+	go func() {
+		runOnMainThread(func() {
+			ch <- dst.At(0, 0).(color.RGBA)
+		})
+	}()
+
+	return testResult{
+		want: color.RGBA{0xff, 0xff, 0xff, 0xff},
+		got:  ch,
+	}
+}()
+
+func TestDrawImageBeforeMain(t *testing.T) {
+	got := <-testDrawImageBeforeMainResult.got
+	want := testDrawImageBeforeMainResult.want
+
+	if got != want {
+		t.Errorf("got: %v, want: %v", got, want)
+	}
+}
+
+var testDrawTrianglesBeforeMainResult = func() testResult {
+	const w, h = 16, 16
+	src, _ := ebiten.NewImage(w, h, ebiten.FilterDefault)
+	dst, _ := ebiten.NewImage(w, h, ebiten.FilterDefault)
+	src.Set(0, 0, color.White)
+	vs := []ebiten.Vertex{
+		{
+			DstX:   0,
+			DstY:   0,
+			SrcX:   0,
+			SrcY:   0,
+			ColorR: 1,
+			ColorG: 1,
+			ColorB: 1,
+			ColorA: 1,
+		},
+		{
+			DstX:   w,
+			DstY:   0,
+			SrcX:   w,
+			SrcY:   0,
+			ColorR: 1,
+			ColorG: 1,
+			ColorB: 1,
+			ColorA: 1,
+		},
+		{
+			DstX:   0,
+			DstY:   h,
+			SrcX:   0,
+			SrcY:   h,
+			ColorR: 1,
+			ColorG: 1,
+			ColorB: 1,
+			ColorA: 1,
+		},
+	}
+	dst.DrawTriangles(vs, []uint16{0, 1, 2}, src, nil)
+
+	ch := make(chan color.RGBA, 1)
+	go func() {
+		runOnMainThread(func() {
+			ch <- dst.At(0, 0).(color.RGBA)
+		})
+	}()
+
+	return testResult{
+		want: color.RGBA{0xff, 0xff, 0xff, 0xff},
+		got:  ch,
+	}
+}()
+
+func TestDrawTrianglesBeforeMain(t *testing.T) {
+	got := <-testDrawTrianglesBeforeMainResult.got
+	want := testDrawTrianglesBeforeMainResult.want
 
 	if got != want {
 		t.Errorf("got: %v, want: %v", got, want)