diff options
Diffstat (limited to 'todolist')
| -rw-r--r-- | todolist/app.go | 2 | ||||
| -rw-r--r-- | todolist/util.go | 7 | ||||
| -rw-r--r-- | todolist/util_test.go | 13 |
3 files changed, 21 insertions, 1 deletions
diff --git a/todolist/app.go b/todolist/app.go index d4513ac..f63c41b 100644 --- a/todolist/app.go +++ b/todolist/app.go @@ -44,7 +44,7 @@ func (a *App) DeleteTodo(input string) { } a.TodoList.Delete(ids...) a.Save() - fmt.Println("Todo deleted.") + fmt.Printf("%s deleted.\n", pluralize(len(ids), "Todo", "Todos")) } func (a *App) CompleteTodo(input string) { diff --git a/todolist/util.go b/todolist/util.go index 34f9dd1..ff3dbba 100644 --- a/todolist/util.go +++ b/todolist/util.go @@ -52,3 +52,10 @@ func getNearestMonday(t time.Time) time.Time { } } } + +func pluralize(count int, singular, plural string) string { + if count > 1 { + return plural + } + return singular +} diff --git a/todolist/util_test.go b/todolist/util_test.go new file mode 100644 index 0000000..40ecb54 --- /dev/null +++ b/todolist/util_test.go @@ -0,0 +1,13 @@ +package todolist + +import ( + "testing" + + "github.com/stretchr/testify/assert" +) + +func TestPluralize(t *testing.T) { + assert := assert.New(t) + assert.Equal("todo", pluralize(1, "todo", "todos")) + assert.Equal("todos", pluralize(2, "todo", "todos")) +} |
