66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
package taskwarrior
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TaskWarriorTestSetup(dir string) {
|
|
// Create a taskrc file
|
|
taskrc := fmt.Sprintf("%s/taskrc", dir)
|
|
taskrcContents := fmt.Sprintf("data.location=%s\n", dir)
|
|
os.WriteFile(taskrc, []byte(taskrcContents), 0644)
|
|
}
|
|
|
|
func TestTaskSquire_GetContext(t *testing.T) {
|
|
dir := t.TempDir()
|
|
fmt.Printf("dir: %s", dir)
|
|
TaskWarriorTestSetup(dir)
|
|
|
|
type fields struct {
|
|
configLocation string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
fields fields
|
|
prep func()
|
|
want string
|
|
}{
|
|
{
|
|
name: "Test without context",
|
|
fields: fields{
|
|
configLocation: fmt.Sprintf("%s/taskrc", dir),
|
|
},
|
|
prep: func() {},
|
|
want: "none",
|
|
},
|
|
{
|
|
name: "Test with context",
|
|
fields: fields{
|
|
configLocation: fmt.Sprintf("%s/taskrc", dir),
|
|
},
|
|
prep: func() {
|
|
f, err := os.OpenFile(fmt.Sprintf("%s/taskrc", dir), os.O_APPEND|os.O_WRONLY, 0644)
|
|
if err != nil {
|
|
t.Errorf("Failed to open file: %s", err)
|
|
}
|
|
defer f.Close()
|
|
if _, err := f.Write([]byte("context=test\ncontext.test.read=+test\ncontext.test.write=+test")); err != nil {
|
|
t.Error("Failed to write to file")
|
|
}
|
|
},
|
|
want: "test",
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
tt.prep()
|
|
ts := NewTaskSquire(tt.fields.configLocation)
|
|
if got := ts.GetActiveContext(); got.Name != tt.want {
|
|
t.Errorf("TaskSquire.GetContext() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|