def test_generate_key(self):
stdout = io.StringIO()
try:
keygen.main(stdout=stdout, argv=[])
except SystemExit as exc:
self.assertEqual(exc.code, 0)
key = stdout.getvalue()
f = Fernet(key)
# Make sure this doesn't raise an error about a bad key.
f.decrypt(f.encrypt('whatever'.encode('utf-8')))
def test_generate_key(self):
stdout = cStringIO.StringIO()
try:
keygen.main(stdout=stdout, argv=[])
except SystemExit, exc:
self.assertEqual(exc.code, 0)
