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_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'
