def test_generate_secret_key(self): """Test :attr:`googauth.generate_secret_key`.""" value = googauth.generate_secret_key() self.assertTrue(len(value) == 16) value = googauth.generate_secret_key() self.assertTrue(len(value) == 16) for i in xrange(8, 129): value = googauth.generate_secret_key(i) self.assertTrue(len(value) == i) exception = None try: value = googauth.generate_secret_key(7) except TypeError, ex: exception = ex
def test_generate_code(self): """Test :attr:`googauth.generate_code`.""" secret_key = 'GBSWKZRYGVQWKMJR' value = googauth.generate_code(secret_key, 683753) self.assertTrue(value == '700446') secret_key = 'MYYDINDBMI2DSNBY' value = googauth.generate_code(secret_key, 282653) self.assertTrue(value == '060555') secret_key = googauth.generate_secret_key(8) for i in xrange(50000): value = googauth.generate_code(secret_key, i) self.assertTrue(len(value) == 6) secret_key = googauth.generate_secret_key(128) for i in xrange(50000): value = googauth.generate_code(secret_key, i) self.assertTrue(len(value) == 6)
def test_generate_code(self): """Test :attr:`googauth.generate_code`.""" secret_key = "GBSWKZRYGVQWKMJR" value = googauth.generate_code(secret_key, 683753) self.assertTrue(value == "700446") secret_key = "MYYDINDBMI2DSNBY" value = googauth.generate_code(secret_key, 282653) self.assertTrue(value == "060555") secret_key = googauth.generate_secret_key(8) for i in xrange(50000): value = googauth.generate_code(secret_key, i) self.assertTrue(len(value) == 6) secret_key = googauth.generate_secret_key(128) for i in xrange(50000): value = googauth.generate_code(secret_key, i) self.assertTrue(len(value) == 6)
def add_user(username): secret_key = googauth.generate_secret_key() # username = request.args.get('username', 'test') db = get_db() user = db.execute('select secret_key from googleauth where username=?', [username]) if user.fetchone(): return 'Error,username exist!' db.execute('insert into googleauth (username, secret_key) values (?, ?)', [username, secret_key]) db.commit() barcode_url = googauth.get_barcode_url(username, 'domain.com', secret_key) return redirect(barcode_url)
class TestFunctions(unittest.TestCase): """Test all googauth functions.""" def test_generate_code(self): """Test :attr:`googauth.generate_code`.""" secret_key = 'GBSWKZRYGVQWKMJR' value = googauth.generate_code(secret_key, 683753) self.assertTrue(value == '700446') secret_key = 'MYYDINDBMI2DSNBY' value = googauth.generate_code(secret_key, 282653) self.assertTrue(value == '060555') secret_key = googauth.generate_secret_key(8) for i in xrange(50000): value = googauth.generate_code(secret_key, i) self.assertTrue(len(value) == 6) secret_key = googauth.generate_secret_key(128) for i in xrange(50000): value = googauth.generate_code(secret_key, i) self.assertTrue(len(value) == 6) def test_verify_counter_based(self): """Test :attr:`googauth.verify_counter_based`.""" secret_key = 'MUYDINLBG4ZWEYTF' value = googauth.verify_counter_based(secret_key, '343101', 957862) self.assertTrue(value == 957863) secret_key = 'MQZTOMLDHBRDOZDF' value = googauth.verify_counter_based(secret_key, '117316', 196831) self.assertTrue(value == 196834) def test_verify_time_based(self): """Test :attr:`googauth.verify_time_based`.""" cur_time = int(time.time() / 30) secret_key = 'MQ3GEMTCMQ3TOOLG' code = googauth.generate_code(secret_key, cur_time) value = googauth.verify_time_based(secret_key, code) self.assertTrue(value == cur_time) cur_time = int(time.time() / 30) secret_key = 'G42DCMBQGJRGMZBQ' code = googauth.generate_code(secret_key) value = googauth.verify_time_based(secret_key, code) self.assertTrue(value != None) def test_generate_secret_key(self): """Test :attr:`googauth.generate_secret_key`.""" value = googauth.generate_secret_key() self.assertTrue(len(value) == 16) value = googauth.generate_secret_key() self.assertTrue(len(value) == 16) for i in xrange(8, 129): value = googauth.generate_secret_key(i) self.assertTrue(len(value) == i) exception = None try: value = googauth.generate_secret_key(7) except TypeError, ex: exception = ex self.assertTrue(isinstance(exception, TypeError)) exception = None try: value = googauth.generate_secret_key(129) except TypeError, ex: exception = ex
def __init__(self, username, password): self.username = username self.password = bcrypt.encrypt(password) self.secret_key = googauth.generate_secret_key() # this will eventually be submitted by the admin in a config file self.domain = 'pass.run107.com'