Refactor todolist stuff out of the file store implementation

1 files changed, 6 insertions, 4 deletions

diff --git a/todolist/grouper_test.go b/todolist/grouper_test.go
index 0b610f0..4b79703 100644
--- a/todolist/grouper_test.go
+++ b/todolist/grouper_test.go
@@ -10,10 +10,11 @@ func TestGroupByContext(t *testing.T) {
 	assert := assert.New(t)
 
 	store := &FileStore{FileLocation: "todos.json"}
-	store.Load()
+	list := &TodoList{}
+	list.Load(store.Load())
 
 	grouper := &Grouper{}
-	grouped := grouper.GroupByContext(store.Todos())
+	grouped := grouper.GroupByContext(list.Todos())
 
 	assert.Equal(2, len(grouped.Groups["root"]), "")
 	assert.Equal(1, len(grouped.Groups["more"]), "")
@@ -23,10 +24,11 @@ func TestGroupByProject(t *testing.T) {
 	assert := assert.New(t)
 
 	store := &FileStore{FileLocation: "todos.json"}
-	store.Load()
+	list := &TodoList{}
+	list.Load(store.Load())
 
 	grouper := &Grouper{}
-	grouped := grouper.GroupByProject(store.Todos())
+	grouped := grouper.GroupByProject(list.Todos())
 
 	assert.Equal(2, len(grouped.Groups["test1"]), "")
 }