def test_reset_code(self): engine = create_engine('sqlite:///:memory:') mgr = ResetCodeManager(engine, True) code = mgr.generate_reset_code(1) self.assertEqual(code, mgr.generate_reset_code(1)) code2 = mgr.generate_reset_code(1, True) self.assertNotEqual(code, code2) self.assertFalse(mgr.verify_reset_code(1, code)) self.assertTrue(mgr.verify_reset_code(1, code2)) mgr.clear_reset_code(1) self.assertFalse(mgr.verify_reset_code(1, code2))