taskgraph/util_test.go at master · thought-machine/taskgraph · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
package taskgraph_test

import (
	"errors"
	"fmt"
	"testing"

	"github.com/stretchr/testify/assert"
	tg "github.com/thought-machine/taskgraph"
)

func TestErrorsMaybeErrors(t *testing.T) {
	maybes := []tg.MaybeStatus{}
	b := tg.NewBinder()
	for i := 0; i < 3; i++ {
		key := tg.NewKey[string](fmt.Sprintf("key%d", i))
		assert.NoError(t, b.Store(key.BindError(fmt.Errorf("an error %d", i))))
		o, err := tg.Optional(key).Get(b)
		assert.NoError(t, err)
		maybes = append(maybes, o)
	}
	err := tg.ErrorsMaybe(maybes...)
	assert.Error(t, err)
	assert.True(t, errors.Is(err, errors.Unwrap(err)))
}

func TestErrorsMaybeAbsent(t *testing.T) {
	maybes := []tg.MaybeStatus{}
	b := tg.NewBinder()
	for i := 0; i < 3; i++ {
		key := tg.NewKey[string](fmt.Sprintf("key%d", i))
		assert.NoError(t, b.Store(key.BindAbsent()))
		o, err := tg.Optional(key).Get(b)
		assert.NoError(t, err)
		maybes = append(maybes, o)
	}
	err := tg.ErrorsMaybe(maybes...)
	assert.NoError(t, err)
}

func TestMissingMaybe(t *testing.T) {
	maybes := map[string]tg.MaybeStatus{}
	b := tg.NewBinder()
	nElements := 3
	for i := nElements; i > 0; i-- {
		key := tg.NewKey[string](fmt.Sprintf("key%d", i))
		assert.NoError(t, b.Store(key.BindAbsent()))
		o, err := tg.Optional(key).Get(b)
		assert.NoError(t, err)
		maybes[key.ID().String()] = o
	}
	missing := tg.MissingMaybe(maybes)
	assert.Equal(t, nElements, len(missing))
	assert.ElementsMatch(t, []string{"key1", "key2", "key3"}, missing)
}