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)