taskengine/taskstat_test.go at main · mmbros/taskengine · 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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
package taskengine

import (
	"fmt"
	"strings"
	"testing"
)

func (statmap taskStatMap) String() string {
	var b strings.Builder
	b.WriteString("[\n")
	for tid, ti := range statmap {
		fmt.Fprintf(&b, " %s: todo=%d, doing=%d, done=%d, success=%d\n",
			tid, ti.Todo, ti.Doing, ti.Done, ti.Success)
	}
	b.WriteString("]")
	return b.String()
}

type statTask string

func (t statTask) TaskID() TaskID { return TaskID(t) }

func TestPick(t *testing.T) {

	// TS tranforms a string "t1, t2, ..." to Tasks
	TS := func(s string) Tasks {
		a := strings.Split(s, ",")
		res := Tasks{}
		for _, tidx := range a {
			tid := strings.TrimSpace(tidx)
			if tid != "" {
				res = append(res, statTask(tid))
			}
		}
		return res
	}

	type testCase struct {
		statmap taskStatMap
		tasks   Tasks
		want    int
	}

	statmap123 := taskStatMap{
		"t1": &TaskStat{1, 0, 0, 0},
		"t2": &TaskStat{2, 0, 0, 0},
		"t3": &TaskStat{3, 0, 0, 0},
	}

	testCases := map[string]testCase{
		"todo 123": {
			statmap: statmap123,
			tasks:   TS("t1,t2,t3"),
			want:    0,
		},
		"todo 32 alpha": {
			statmap: statmap123,
			tasks:   TS("t3,t2"),
			want:    1,
		},
		"todo 2 alpha": {
			statmap: statmap123,
			tasks:   TS("t2"),
			want:    0,
		},
		"todo -1": {
			statmap: statmap123,
			tasks:   TS(""),
			want:    -1,
		},
		// "123_y unknown task 0": {
		// 	statmap: statmap123,
		// 	tasks:   TS("t999"),
		// 	want:    0,
		// },
		// "123_y unknown task 1": {
		// 	statmap: statmap123,
		// 	tasks:   TS("t1, t999"),
		// 	want:    0,
		// },
		"done with success": {
			statmap: taskStatMap{
				"t1": &TaskStat{1, 0, 1, 1},
				"t2": &TaskStat{2, 0, 0, 0},
				"t3": &TaskStat{3, 0, 0, 0},
			},
			tasks: TS("t1,t2,t3"),
			want:  1,
		},
		"done with error": {
			statmap: taskStatMap{
				"t1": &TaskStat{1, 0, 1, 0},
				"t2": &TaskStat{2, 0, 0, 0},
				"t3": &TaskStat{3, 0, 0, 0},
			},
			tasks: TS("t1,t2,t3"),
			want:  0,
		},
		"doing": {
			statmap: taskStatMap{
				"t1": &TaskStat{1, 1, 0, 0},
				"t2": &TaskStat{2, 1, 0, 0},
				"t3": &TaskStat{3, 0, 0, 0},
			},
			tasks: TS("t1,t2,t3"),
			want:  2,
		},
		"doing success alpha": {
			statmap: taskStatMap{
				"t1": &TaskStat{1, 1, 0, 0},
				"t2": &TaskStat{2, 1, 0, 0},
				"t3": &TaskStat{3, 0, 1, 1},
			},
			tasks: TS("t2,t1,t3"),
			want:  1,
		},
	}

	for title, tc := range testCases {
		got := tc.statmap.pick(tc.tasks)

		if got != tc.want {
			t.Logf("%s: statmap = %v", title, tc.statmap)
			t.Logf("%s: tasks = %v", title, tc.tasks)
			t.Errorf("%s: want %d, got %d", title, tc.want, got)
		}
	}
}