tea_test.go

package masc

import (
	"context"
	"sync/atomic"
	"testing"
	"time"
)

type incrementMsg struct{}

type testModel struct {
	Core
	executed  atomic.Value
	counter   atomic.Value
	testSuite *testSuiteT
}

func (m testModel) Init() Cmd {
	return nil
}

func (m *testModel) Update(msg Msg) (Model, Cmd) {
	switch msg.(type) {
	case incrementMsg:
		i := m.counter.Load()
		if i == nil {
			m.counter.Store(1)
		} else {
			m.counter.Store(i.(int) + 1)
		}
	}

	return m, nil
}

func (m *testModel) Render(send func(Msg)) ComponentOrHTML {
	m.executed.Store(true)
	go func() {
		time.Sleep(20 * time.Millisecond)
		for {
			if len(m.testSuite.callbacks) == 0 {
				time.Sleep(200 * time.Millisecond)
			} else {
				m.testSuite.invokeCallbackRequestAnimationFrame(0)
				return
			}
		}
	}()
	return Tag("body")
}

func TestTeaQuit(t *testing.T) {
	ts := testSuite(t)
	defer ts.done()

	ts.isUndefined.mock(`global.Call("requestAnimationFrame", func)`, 0)
	ts.strings.mock(`global.Get("document").Get("readyState")`, "complete")
	ts.strings.mock(`global.Get("document").Call("querySelector", "body").Get("nodeName")`, "BODY")
	ts.truthies.mock(`global.Get("document").Call("querySelector", "body")`, true)

	m := &testModel{}
	m.testSuite = ts
	p := NewProgram(m)
	go func() {
		for {
			time.Sleep(time.Millisecond)
			if m.executed.Load() != nil {
				p.Quit()
				return
			}
		}
	}()

	if _, err := p.Run(); err != nil {
		t.Fatal(err)
	}
}

func TestTeaWithFilter(t *testing.T) {
	testTeaWithFilter(t, 0)
	testTeaWithFilter(t, 1)
	testTeaWithFilter(t, 2)
}

func testTeaWithFilter(t *testing.T, preventCount uint32) {
	ts := testSuite(t)
	defer ts.done()

	ts.isUndefined.mock(`global.Call("requestAnimationFrame", func)`, 0)
	ts.strings.mock(`global.Get("document").Get("readyState")`, "complete")
	ts.strings.mock(`global.Get("document").Call("querySelector", "body").Get("nodeName")`, "BODY")
	ts.truthies.mock(`global.Get("document").Call("querySelector", "body")`, true)

	m := &testModel{}
	m.testSuite = ts
	shutdowns := uint32(0)
	p := NewProgram(m,
		WithFilter(func(_ Model, msg Msg) Msg {
			if _, ok := msg.(QuitMsg); !ok {
				return msg
			}
			if shutdowns < preventCount {
				atomic.AddUint32(&shutdowns, 1)
				return nil
			}
			return msg
		}))

	go func() {
		for atomic.LoadUint32(&shutdowns) <= preventCount {
			time.Sleep(time.Millisecond)
			p.Quit()
		}
	}()

	if err := p.Start(); err != nil {
		t.Fatal(err)
	}
	if shutdowns != preventCount {
		t.Errorf("Expected %d prevented shutdowns, got %d", preventCount, shutdowns)
	}
}

func TestTeaKill(t *testing.T) {
	ts := testSuite(t)
	defer ts.done()

	ts.isUndefined.mock(`global.Call("requestAnimationFrame", func)`, 0)
	ts.strings.mock(`global.Get("document").Get("readyState")`, "complete")
	ts.strings.mock(`global.Get("document").Call("querySelector", "body").Get("nodeName")`, "BODY")
	ts.truthies.mock(`global.Get("document").Call("querySelector", "body")`, true)

	m := &testModel{}
	m.testSuite = ts
	p := NewProgram(m)
	go func() {
		for {
			time.Sleep(time.Millisecond)
			if m.executed.Load() != nil {
				p.Kill()
				return
			}
		}
	}()

	if _, err := p.Run(); err != ErrProgramKilled {
		t.Fatalf("Expected %v, got %v", ErrProgramKilled, err)
	}
}

func TestTeaContext(t *testing.T) {
	ctx, cancel := context.WithCancel(context.Background())

	ts := testSuite(t)
	defer ts.done()

	ts.isUndefined.mock(`global.Call("requestAnimationFrame", func)`, 0)
	ts.strings.mock(`global.Get("document").Get("readyState")`, "complete")
	ts.strings.mock(`global.Get("document").Call("querySelector", "body").Get("nodeName")`, "BODY")
	ts.truthies.mock(`global.Get("document").Call("querySelector", "body")`, true)

	m := &testModel{}
	m.testSuite = ts
	p := NewProgram(m, WithContext(ctx))
	go func() {
		for {
			time.Sleep(time.Millisecond)
			if m.executed.Load() != nil {
				cancel()
				return
			}
		}
	}()

	if _, err := p.Run(); err != ErrProgramKilled {
		t.Fatalf("Expected %v, got %v", ErrProgramKilled, err)
	}
}

func TestTeaBatchMsg(t *testing.T) {
	inc := func() Msg {
		return incrementMsg{}
	}

	ts := testSuite(t)
	defer ts.done()

	ts.isUndefined.mock(`global.Call("requestAnimationFrame", func)`, 0)
	ts.strings.mock(`global.Get("document").Get("readyState")`, "complete")
	ts.strings.mock(`global.Get("document").Call("querySelector", "body").Get("nodeName")`, "BODY")
	ts.truthies.mock(`global.Get("document").Call("querySelector", "body")`, true)

	m := &testModel{}
	m.testSuite = ts
	p := NewProgram(m)
	go func() {
		p.Send(BatchMsg{inc, inc})

		for {
			time.Sleep(time.Millisecond)
			i := m.counter.Load()
			if i != nil && i.(int) >= 2 {
				p.Quit()
				return
			}
		}
	}()

	if _, err := p.Run(); err != nil {
		t.Fatal(err)
	}

	if m.counter.Load() != 2 {
		t.Fatalf("counter should be 2, got %d", m.counter.Load())
	}
}

func TestTeaSequenceMsg(t *testing.T) {
	inc := func() Msg {
		return incrementMsg{}
	}

	ts := testSuite(t)
	defer ts.done()

	ts.isUndefined.mock(`global.Call("requestAnimationFrame", func)`, 0)
	ts.strings.mock(`global.Get("document").Get("readyState")`, "complete")
	ts.strings.mock(`global.Get("document").Call("querySelector", "body").Get("nodeName")`, "BODY")
	ts.truthies.mock(`global.Get("document").Call("querySelector", "body")`, true)

	m := &testModel{}
	m.testSuite = ts
	p := NewProgram(m)
	go p.Send(sequenceMsg{inc, inc, Quit})

	if _, err := p.Run(); err != nil {
		t.Fatal(err)
	}

	if m.counter.Load() != 2 {
		t.Fatalf("counter should be 2, got %d", m.counter.Load())
	}
}

func TestTeaSequenceMsgWithBatchMsg(t *testing.T) {
	inc := func() Msg {
		return incrementMsg{}
	}
	batch := func() Msg {
		return BatchMsg{inc, inc}
	}

	ts := testSuite(t)
	defer ts.done()

	ts.isUndefined.mock(`global.Call("requestAnimationFrame", func)`, 0)
	ts.strings.mock(`global.Get("document").Get("readyState")`, "complete")
	ts.strings.mock(`global.Get("document").Call("querySelector", "body").Get("nodeName")`, "BODY")
	ts.truthies.mock(`global.Get("document").Call("querySelector", "body")`, true)

	m := &testModel{}
	m.testSuite = ts
	p := NewProgram(m)
	go p.Send(sequenceMsg{batch, inc, Quit})

	if _, err := p.Run(); err != nil {
		t.Fatal(err)
	}

	if m.counter.Load() != 3 {
		t.Fatalf("counter should be 3, got %d", m.counter.Load())
	}
}

func TestTeaSend(t *testing.T) {
	ts := testSuite(t)
	defer ts.done()

	ts.isUndefined.mock(`global.Call("requestAnimationFrame", func)`, 0)
	ts.strings.mock(`global.Get("document").Get("readyState")`, "complete")
	ts.strings.mock(`global.Get("document").Call("querySelector", "body").Get("nodeName")`, "BODY")
	ts.truthies.mock(`global.Get("document").Call("querySelector", "body")`, true)

	m := &testModel{}
	m.testSuite = ts
	p := NewProgram(m)

	// sending before the program is started is a blocking operation
	go p.Send(Quit())

	if _, err := p.Run(); err != nil {
		t.Fatal(err)
	}

	// sending a message after program has quit is a no-op
	p.Send(Quit())
}

func TestTeaNoRun(t *testing.T) {
	m := &testModel{}
	NewProgram(m)
}