diff --git a/dictionary/dictionary.go b/dictionary/dictionary.go index fc374f3..941c137 100644 --- a/dictionary/dictionary.go +++ b/dictionary/dictionary.go @@ -48,3 +48,7 @@ func (d Dictionary) Update(word, definition string) error { } return nil } + +func (d Dictionary) Delete(word string) { + delete(d, word) +} diff --git a/dictionary/dictionary_test.go b/dictionary/dictionary_test.go index c9a3968..74693c6 100644 --- a/dictionary/dictionary_test.go +++ b/dictionary/dictionary_test.go @@ -91,3 +91,18 @@ func TestUpdate(t *testing.T) { assertError(t, err, ErrWordDoesNotExist) }) } + +func TestDelete(t *testing.T) { + t.Run("delete a word", func(t *testing.T) { + word := "test" + def := "this is just a test" + dict := Dictionary{word: def} + + dict.Delete(word) + _, err := dict.Search(word) + if err == nil { + t.Fatal("expected to get an error") + } + assertError(t, err, ErrNotFound) + }) +}