629bf2ad创建于 2023年12月18日历史提交
package tasklog

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestListTaskCallsDoneWhenComplete(t *testing.T) {
	task := NewListTask("example")
	task.Complete()

	select {
	case update, ok := <-task.Updates():
		assert.Equal(t, "example: ...", update.S)
		assert.True(t, ok,
			"tasklog: expected Updates() to remain open")
	default:
		t.Fatal("tasklog: expected update from *ListTask")
	}

	select {
	case update, ok := <-task.Updates():
		assert.False(t, ok,
			"git/githistory.log: unexpected *ListTask.Update(): %s", update)
	default:
		t.Fatal("tasklog: expected *ListTask.Updates() to be closed")
	}
}

func TestListTaskWritesEntries(t *testing.T) {
	task := NewListTask("example")
	task.Entry("1")

	select {
	case update, ok := <-task.Updates():
		assert.True(t, ok,
			"tasklog: expected ListTask.Updates() to remain open")
		assert.Equal(t, "1\n", update.S)
	default:
		t.Fatal("tasklog: expected task.Updates() to have an update")
	}
}

func TestListTaskIsNotThrottled(t *testing.T) {
	task := NewListTask("example")

	throttled := task.Throttled()

	assert.False(t, throttled,
		"tasklog: expected *ListTask to be Throttle()-d")
}