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

Add test for wordlist from stdin.

parent 40c60e1d
No related branches found
Tags v0.2
No related merge requests found
import os import os
import pytest import pytest
import sys import sys
from io import StringIO
from diceware import ( from diceware import (
SRC_DIR, RE_LANG_CODE, SPECIAL_CHARS, get_wordlist, SRC_DIR, RE_LANG_CODE, SPECIAL_CHARS, get_wordlist,
get_wordlist_path, insert_special_char, get_passphrase, get_wordlist_path, insert_special_char, get_passphrase,
...@@ -213,3 +214,11 @@ class TestDicewareModule(object): ...@@ -213,3 +214,11 @@ class TestDicewareModule(object):
main(['-n', '1', 'mywordlist.txt', ]) main(['-n', '1', 'mywordlist.txt', ])
out, err = capsys.readouterr() out, err = capsys.readouterr()
assert out == 'Mysingleword\n' assert out == 'Mysingleword\n'
def test_main_infile_stdin(self, argv_handler, capsys):
# main() also accepts input from stdin
sys.stdin = StringIO(b"word1\n".decode("utf-8"))
sys.argv = ['diceware', '-n', '2', '-']
main()
out, err = capsys.readouterr()
assert out == 'Word1Word1\n'
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment