diff --git a/tests/test_wordlist.py b/tests/test_wordlist.py index db252d3d1b01a78d5e62e39629fcd2eb04b4508c..fc612c04e3110616dce864f7c5c7af0856b00c99 100644 --- a/tests/test_wordlist.py +++ b/tests/test_wordlist.py @@ -255,3 +255,17 @@ class TestWordList(object): assert long_list[0] == "a" assert long_list[-1] == "@" assert len(long_list) == 8192 + + def test_get_wordlist_simple(self, tmpdir): + # simple wordlists can be created + in_file = tmpdir.mkdir("work").join("mywordlist") + in_file.write("a\nb\n") + result = list(WordList(str(in_file))) + assert ['a', 'b'] == result + + def test_get_wordlist_ignore_empty_lines(self, tmpdir): + # we ignore empty lines in wordlists + in_file = tmpdir.mkdir("work").join("mywordlist") + in_file.write("\n\na\n\n") + result = list(WordList(str(in_file))) + assert ['a'] == result