def test_generate_two(wordstore):
gold = wordstore(['a', 'bb', 'ccc', 'dddd', 'eeeee'])
with tempfile.NamedTemporaryFile('w') as f:
f.write(gold.to_json())
f.seek(0)
camel = generate(f.name, 1, 1, 2, 2, 'camel')
lower = generate(f.name, 1, 1, 2, 2, 'lower')
upper = generate(f.name, 1, 1, 2, 2, 'upper')
assert camel == 'Bb'
assert lower == 'bb'
assert upper == 'BB'
def test_chars(wordstore):
gold = wordstore(['a', 'bb', 'ccc', 'dddd', 'eeeee'])
with tempfile.NamedTemporaryFile('w') as f:
f.write(gold.to_json())
f.seek(0)
passphrase = generate(f.name, 0, 6, 0, 0, 'lower')
assert len(passphrase) > 6
def test_concatenation(wordstore):
gold = wordstore(['a', 'bb', 'ccc', 'dddd', 'eeeee'])
with tempfile.NamedTemporaryFile('w') as f:
f.write(gold.to_json())
f.seek(0)
passphrase = generate(f.name, 2, 1, 3, 4, 'camel')
assert passphrase in ['CccDddd', 'DdddCcc']
def main():
args = parse_command_line()
if args.command == 'build':
result = build(**vars(args))
# Handle build result.
if args.out:
dest = os.path.abspath(os.path.expanduser(args.out))
with open(dest, 'w') as f:
f.write(result)
else:
print(result)
else:
result = generate(**vars(args))
# Handle generate result.
print(result)