import static org.junit.Assert.*;

import org.junit.Before;
import org.junit.Test;

public class TrieTest {

    /*
     * Part 1: complete
     */
    Trie trie = new Trie();

    @Before
    public void setup() {
        trie.insertString("Hello");
        trie.insertString("Hell");
        trie.insertString("He");
    }

    @Test
    public void testEmpty() {
        trie.deleteString("Hello");
        trie.deleteString("Hell");
        trie.deleteString("He");
        assertEquals(0, trie.countAllWords());
    }

    @Test
    public void countAllWords() {
        assertEquals(3, trie.count());
    }

}