-
Notifications
You must be signed in to change notification settings - Fork 50
/
Copy pathnext_context_test.go
115 lines (78 loc) · 3.39 KB
/
next_context_test.go
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
package integration
import (
"os"
"testing"
"github.com/naggie/dstask"
"github.com/stretchr/testify/assert"
)
func TestSettingTagContext(t *testing.T) {
repo, cleanup := makeDstaskRepo(t)
defer cleanup()
program := testCmd(repo)
output, exiterr, success := program("add", "+one", "one")
assertProgramResult(t, output, exiterr, success)
output, exiterr, success = program("add", "+one", "one")
assertProgramResult(t, output, exiterr, success)
output, exiterr, success = program("add", "+two", "two")
assertProgramResult(t, output, exiterr, success)
output, exiterr, success = program("context", "+two")
assertProgramResult(t, output, exiterr, success)
output, exiterr, success = program("next")
assertProgramResult(t, output, exiterr, success)
var tasks []dstask.Task
tasks = unmarshalTaskArray(t, output)
assert.Equal(t, "two", tasks[0].Summary, "setting +two as a context")
output, exiterr, success = program("context", "-one")
assertProgramResult(t, output, exiterr, success)
output, exiterr, success = program("next")
assertProgramResult(t, output, exiterr, success)
tasks = unmarshalTaskArray(t, output)
assert.Equal(t, "two", tasks[0].Summary, "setting -one as a context")
}
func TestSettingTagAndProjectContext(t *testing.T) {
repo, cleanup := makeDstaskRepo(t)
defer cleanup()
program := testCmd(repo)
output, exiterr, success := program("add", "+one", "+alpha", "one")
assertProgramResult(t, output, exiterr, success)
output, exiterr, success = program("add", "project:beta", "+two", "two")
assertProgramResult(t, output, exiterr, success)
output, exiterr, success = program("context", "project:beta")
assertProgramResult(t, output, exiterr, success)
output, exiterr, success = program("next")
assertProgramResult(t, output, exiterr, success)
var tasks []dstask.Task
tasks = unmarshalTaskArray(t, output)
assert.Equal(t, "two", tasks[0].Summary, "setting project:beta as a context")
output, exiterr, success = program("context", "project:beta", "+one")
assertProgramResult(t, output, exiterr, success)
output, exiterr, success = program("next")
assertProgramResult(t, output, exiterr, success)
tasks = unmarshalTaskArray(t, output)
assert.Equal(t, 0, len(tasks), "no tasks within context project:beta +one")
}
func TestContextFromEnvVar(t *testing.T) {
repo, cleanup := makeDstaskRepo(t)
defer cleanup()
program := testCmd(repo)
output, exiterr, success := program("add", "+one", "+alpha", "one")
assertProgramResult(t, output, exiterr, success)
output, exiterr, success = program("add", "project:beta", "+two", "two")
assertProgramResult(t, output, exiterr, success)
output, exiterr, success = program("context", "project:beta")
assertProgramResult(t, output, exiterr, success)
// override context with an env var
unsetEnv := setEnv("DSTASK_CONTEXT", "+one +alpha")
t.Logf("DSTASK_CONTEXT=%s", os.Getenv("DSTASK_CONTEXT"))
output, exiterr, success = program("next")
assertProgramResult(t, output, exiterr, success)
var tasks []dstask.Task
tasks = unmarshalTaskArray(t, output)
assert.Equal(t, "one", tasks[0].Summary, "'+one +alpha' context set by DSTASK_CONTEXT ")
// unset the context override, so we expect to use the on-disk context
unsetEnv()
output, exiterr, success = program("next")
assertProgramResult(t, output, exiterr, success)
tasks = unmarshalTaskArray(t, output)
assert.Equal(t, "two", tasks[0].Summary, "project:beta is on-disk context")
}