Skip to content
Snippets Groups Projects
Commit fe0d9a43 authored by ulif's avatar ulif
Browse files

Start a WordList class to avoid func scatter.

parent b50c7e0b
No related branches found
No related tags found
No related merge requests found
......@@ -132,3 +132,10 @@ def get_wordlist_path(name):
raise ValueError("Not a valid wordlist name: %s" % name)
basename = 'wordlist_%s.txt' % name
return os.path.join(WORDLISTS_DIR, basename)
class WordList(object):
"""A word list contains words for building passphrases.
"""
def __init__(self, opened_file=None, path=None):
pass
......@@ -3,7 +3,7 @@ import pytest
from diceware.wordlist import (
WORDLISTS_DIR, RE_WORDLIST_NAME, RE_NUMBERED_WORDLIST_ENTRY, get_wordlist,
get_signed_wordlist, get_wordlist_path, get_wordlist_names,
is_signed_wordlist, refine_wordlist_entry,
is_signed_wordlist, refine_wordlist_entry, WordList,
)
......@@ -189,3 +189,13 @@ class TestWordlistModule(object):
def test_refine_wordlist_strips_also_numbered(self):
# also numbered entries are stripped
assert refine_wordlist_entry("11111 \t foo\n") == "foo"
class TestWordList(object):
def test_create_wordlist(self, tmpdir):
# we can create `WordList` objects.
in_file = tmpdir.mkdir("work").join("mywordlist")
in_file.write("foo\n")
wlist = WordList(str(in_file))
assert wlist is not None
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment