class Test(unittest.TestCase):
def setUp(self):
self.crypt = Crypt()
pass
def tearDown(self):
pass
def testSalt(self):
done = []
for i in range(1, 100):
salt = self.crypt.salt()
self.assertFalse(salt in done)
done.append(salt)
pass
def testCryptPassword(self):
username = '******'
password = '******'
hashed_val = self.crypt.crypt_password(username, password)
isValid = self.crypt.is_password_valid(username, password, hashed_val)
self.assertTrue(isValid)
def testCryptPassword_Fails(self):
username = '******'
password = '******'
hashed_val = self.crypt.crypt_password(username, password)
isValid = self.crypt.is_password_valid(username + 'foo', password,
hashed_val)
self.assertFalse(isValid)
class Test(unittest.TestCase):
def setUp(self):
self.crypt = Crypt()
pass
def tearDown(self):
pass
def testSalt(self):
done = []
for i in range(1,100):
salt = self.crypt.salt()
self.assertFalse(salt in done)
done.append(salt)
pass
def testCryptPassword(self):
username = '******'
password = '******'
hashed_val = self.crypt.crypt_password(username, password)
isValid = self.crypt.is_password_valid(username, password, hashed_val)
self.assertTrue(isValid)
def testCryptPassword_Fails(self):
username = '******'
password = '******'
hashed_val = self.crypt.crypt_password(username, password)
isValid = self.crypt.is_password_valid(username + 'foo', password, hashed_val)
self.assertFalse(isValid)
def __init__(self):
'''
Constructor
'''
self.storage = UserDataStoreFactory().get_storage()
self.signup_validation = UserSignupValidation()
self.login_validation = LoginValidation()
self.crypt = Crypt()
class UserService(object):
'''
User service
'''
def __init__(self):
'''
Constructor
'''
self.storage = UserDataStoreFactory().get_storage()
self.signup_validation = UserSignupValidation()
self.login_validation = LoginValidation()
self.crypt = Crypt()
def register(self, username, password, verify, email=None):
validation_msgs, isValid = self.signup_validation.validate(
username=username, password=password, verify=verify, email=email)
if not isValid:
raise UserRegistrationError(validation_msgs)
else:
crypt_password = self._crypt_password(username, password)
user = UserData(username=username,
password=crypt_password,
email=email)
saved_user = self.storage.save(user)
user_id = saved_user.user_id
return self.crypt.make_cookie(user_id)
def login(self, username, password, cookie):
self.check_cookie(cookie)
validation_msgs, login_ok = self.login_validation.validate(
username, password)
if not login_ok:
raise LoginError(validation_msgs)
user = self.storage.fetchByUsername(username)
user_id = user.user_id
return self.crypt.make_cookie(user_id)
def welcome(self, cookie):
self.check_cookie(cookie)
user_id = cookie.split('|')[0]
user = self.storage.fetch(int(user_id))
return user
def check_cookie(self, cookie):
if not cookie:
raise InvalidCookieError()
else:
cookie_ok = self.crypt.is_valid_cookie(cookie)
if not cookie_ok:
raise InvalidCookieError()
def _crypt_password(self, username, password):
return self.crypt.crypt_password(username, password)
def _hash_user_id(self, user_id):
return self.crypt.crypt_user_id(user_id)
class UserService(object):
'''
User service
'''
def __init__(self):
'''
Constructor
'''
self.storage = UserDataStoreFactory().get_storage()
self.signup_validation = UserSignupValidation();
self.login_validation = LoginValidation();
self.crypt = Crypt()
def register(self, username,password,verify, email=None):
validation_msgs, isValid = self.signup_validation.validate(username=username, password=password, verify=verify, email=email)
if not isValid:
raise UserRegistrationError(validation_msgs)
else:
crypt_password = self._crypt_password(username, password)
user = UserData(username=username,password=crypt_password,email=email)
saved_user = self.storage.save(user)
user_id = saved_user.user_id
return self.crypt.make_cookie(user_id)
def login(self,username,password, cookie):
self.check_cookie(cookie)
validation_msgs, login_ok = self.login_validation.validate(username, password)
if not login_ok:
raise LoginError(validation_msgs)
user = self.storage.fetchByUsername(username)
user_id = user.user_id
return self.crypt.make_cookie(user_id)
def welcome(self, cookie):
self.check_cookie(cookie)
user_id = cookie.split('|')[0]
user = self.storage.fetch(int(user_id))
return user
def check_cookie(self, cookie):
if not cookie:
raise InvalidCookieError()
else:
cookie_ok = self.crypt.is_valid_cookie(cookie)
if not cookie_ok:
raise InvalidCookieError()
def _crypt_password(self, username, password):
return self.crypt.crypt_password(username, password)
def _hash_user_id(self, user_id):
return self.crypt.crypt_user_id(user_id)
def setUp(self):
# Setup mock GAE Data Store
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_memcache_stub()
self.service = UserService()
factory = UserDataStoreFactory()
self.storage = factory.get_storage()
self.crypt = Crypt()
def setUp(self):
# Setup mock GAE Data Store
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_memcache_stub()
#instantiate classes under test
self.validation = UserSignupValidation()
self.validation_login = LoginValidation()
self.user_datastore = UserAppengineDataStore()
self.crypt = Crypt()
class LoginValidation():
def __init__(self):
factory = UserDataStoreFactory()
self.user_datastore = factory.get_storage()
self.crypt = Crypt()
pass
def validate(self, username, password):
validation_messages = {}
is_valid = False
if not self.is_valid_login(username, password):
is_valid = False
validation_messages[VERIFICATION_MESSAGES_KEYS[
5]] = VERIFICATION_MESSAGES[VERIFICATION_MESSAGES_KEYS[5]]
else:
is_valid = True
return validation_messages, is_valid
def is_valid_login(self, username, password):
is_valid = False
user = self.user_datastore.fetchByUsername(username)
if user:
user_pwd_hashed = user.password
if self.crypt.is_password_valid(username, password,
user_pwd_hashed):
is_valid = True
return is_valid
class LoginValidation():
def __init__(self):
factory = UserDataStoreFactory()
self.user_datastore = factory.get_storage()
self.crypt = Crypt()
pass
def validate(self, username, password):
validation_messages = {}
is_valid = False
if not self.is_valid_login(username, password):
is_valid = False
validation_messages[VERIFICATION_MESSAGES_KEYS[5]] = VERIFICATION_MESSAGES[VERIFICATION_MESSAGES_KEYS[5]]
else:
is_valid = True
return validation_messages, is_valid
def is_valid_login(self, username, password):
is_valid = False
user = self.user_datastore.fetchByUsername(username)
if user:
user_pwd_hashed = user.password
if self.crypt.is_password_valid(username, password, user_pwd_hashed):
is_valid = True
return is_valid
def __init__(self):
'''
Constructor
'''
self.storage = UserDataStoreFactory().get_storage()
self.signup_validation = UserSignupValidation();
self.login_validation = LoginValidation();
self.crypt = Crypt()
def setUp(self):
# Setup mock GAE Data Store
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_memcache_stub()
#instantiate classes under test
self.validation = UserSignupValidation()
self.validation_login = LoginValidation()
self.user_datastore = UserAppengineDataStore()
self.crypt = Crypt()
def setUp(self):
self.crypt = Crypt()
pass
def setUp(self):
self.crypt = Crypt()
pass
class Test(unittest.TestCase):
def setUp(self):
# Setup mock GAE Data Store
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_memcache_stub()
self.service = UserService()
factory = UserDataStoreFactory()
self.storage = factory.get_storage()
self.crypt = Crypt()
def tearDown(self):
self.testbed.deactivate()
def testLogin(self):
username = '******'
password = '******'
hashed_password = self.crypt.crypt_password(username, password)
user = UserData(username=username, password=hashed_password)
self.storage.save(user)
user_id = user.key().id()
cookie = self.crypt.make_cookie(user_id)
new_cookie = self.service.login(username, password, cookie)
self.assertFalse(new_cookie == None)
self.assertEquals(user.key().id(), 1)
def testLogin_Failed(self):
username = '******'
password = '******'
hashed_password = self.crypt.crypt_password(username, password)
user = UserData(username=username, password=hashed_password)
self.storage.save(user)
user_id = user.key().id()
cookie = self.crypt.make_cookie(user_id)
try:
self.service.login(username + 'foo', password, cookie)
self.fail('Should not get here because LoginError should be raised due to bad usernames')
except LoginError as error:
self.assertEquals(error.validation_messages['login_error'], 'Invalid login')
print error
def testRegister(self):
username = '******'
password = '******'
verify = password
cookie = self.service.register(username=username, password=password, verify=verify)
print cookie
user = self.storage.fetchByUsername(username)
self.assertFalse(cookie == None)
user_id = user.key().id()
self.assertEquals(user_id, 1)
cookie_split = cookie.split('|')
self.assertTrue(cookie_split[0] == str(user_id))
self.assertFalse(cookie_split[1] == None)
self.assertFalse(cookie_split[1] == '' )
def testRegister_failed(self):
username = '******'
password = '******'
verify = password
user = UserData(username=username, password=password)
self.storage.save(user)
try:
self.service.register(username=username, password=password, verify=verify)
self.fail('Should not get here because UserRegistrationError should be raised due to the fact that user already exists')
except UserRegistrationError as error:
self.assertEquals(error.validation_messages['user_exists_error'], 'That user already exists.')
def __init__(self):
factory = UserDataStoreFactory()
self.user_datastore = factory.get_storage()
self.crypt = Crypt()
pass
class Test(unittest.TestCase):
def setUp(self):
# Setup mock GAE Data Store
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_memcache_stub()
#instantiate classes under test
self.validation = UserSignupValidation()
self.validation_login = LoginValidation()
self.user_datastore = UserAppengineDataStore()
self.crypt = Crypt()
def tearDown(self):
self.testbed.deactivate()
def testIsValidUsername(self):
self.assertTrue(self.validation.is_valid_username("Crag"))
self.assertTrue(self.validation.is_valid_username("Crag-Doremus"))
self.assertTrue(self.validation.is_valid_username("Crag_Doremus"))
self.assertTrue(self.validation.is_valid_username("Cra"))
self.assertFalse(self.validation.is_valid_username("ca"))
self.assertFalse(self.validation.is_valid_username("cat!"))
self.assertTrue(self.validation.is_valid_username("abcdefghijklmnopqrst"))
self.assertFalse(self.validation.is_valid_username("abcdefghijklmnopqrstu"))
pass
def testIsValidPassword(self):
self.assertTrue(self.validation.is_valid_password("Craig"))
self.assertTrue(self.validation.is_valid_password("abcdefghijklmnopqrst"))
self.assertFalse(self.validation.is_valid_password("abcdefghijklmnopqrstu"))
pass
def testIsValidEmail(self):
self.assertTrue(self.validation.is_valid_email("*****@*****.**"))
self.assertTrue(self.validation.is_valid_email("*****@*****.**"))
self.assertFalse(self.validation.is_valid_email("Craigfoocom"))
pass
def testValid(self):
username = "******"
password = "******"
verify = "craig1"
email = "*****@*****.**"
# user = UserData(username=username, password=password, email=email)
# self.user_datastore.save(user)
validMsgs, isValid = self.validation.validate(username, password, verify, email)
self.assertTrue(isValid)
self.assertEmptyMessage([VERIFICATION_MESSAGES_KEYS[0],VERIFICATION_MESSAGES_KEYS[1],VERIFICATION_MESSAGES_KEYS[2],VERIFICATION_MESSAGES_KEYS[3]], validMsgs)
def testValid_BadUsername(self):
username = "******"
password = "******"
verify = "craig1"
email = "*****@*****.**"
validMsgs, isValid = self.validation.validate(username, password, verify, email)
self.assertFalse(isValid)
self.assertEquals(VERIFICATION_MESSAGES[VERIFICATION_MESSAGES_KEYS[0]], validMsgs[VERIFICATION_MESSAGES_KEYS[0]])
self.assertEmptyMessage([VERIFICATION_MESSAGES_KEYS[1],VERIFICATION_MESSAGES_KEYS[2],VERIFICATION_MESSAGES_KEYS[3]], validMsgs)
def testUserExists_True(self):
username = '******'
#add a record to the data store
userData = UserData(username=username, password='******')
self.user_datastore.save(userData)
self.assertTrue(self.validation.user_exists(username), "User " + username + " does not exist")
def testUserExists_False(self):
username = '******'
testVal ='foo'
#add a record to the data store
userData = UserData(username=username, password='******')
self.user_datastore.save(userData)
self.assertFalse(self.validation.user_exists(testVal), "User " + testVal + " exists. User added: " + userData.username)
def testValidate_UserExists(self):
'''
Test validate() with a user that already exists
'''
username = '******'
password = '******'
#add a record to the data store
userData = UserData(username=username, password=password)
self.user_datastore.save(userData)
#verify in database
user = self.user_datastore.fetchByUsername(username)
print user.username
self.assertTrue(self.validation.validate(username=username, password=password, verify=password, email=None), "User " + username + " exists")
def testValid_BadPassword(self):
username = "******"
password = "******"
verify = "c1"
email = "*****@*****.**"
validMsgs, isValid = self.validation.validate(username, password, verify, email)
self.assertFalse(isValid)
self.assertEquals(VERIFICATION_MESSAGES[VERIFICATION_MESSAGES_KEYS[1]], validMsgs[VERIFICATION_MESSAGES_KEYS[1]])
self.assertEquals(VERIFICATION_MESSAGES[VERIFICATION_MESSAGES_KEYS[2]], validMsgs[VERIFICATION_MESSAGES_KEYS[2]])
self.assertEmptyMessage([VERIFICATION_MESSAGES_KEYS[0],VERIFICATION_MESSAGES_KEYS[3]], validMsgs)
def testValid_BadEmail(self):
username = "******"
password = "******"
verify = "craig1"
email = "craigfoo.com"
validMsgs, isValid = self.validation.validate(username, password, verify, email)
self.assertFalse(isValid)
self.assertEquals(VERIFICATION_MESSAGES[VERIFICATION_MESSAGES_KEYS[3]], validMsgs[VERIFICATION_MESSAGES_KEYS[3]])
self.assertEmptyMessage([VERIFICATION_MESSAGES_KEYS[0],VERIFICATION_MESSAGES_KEYS[2],VERIFICATION_MESSAGES_KEYS[1]], validMsgs)
def testValid_PasswordsDontMatch(self):
username = "******"
password = "******"
verify = "craig"
email = "*****@*****.**"
validMsgs, isValid = self.validation.validate(username, password, verify, email)
self.assertFalse(isValid)
self.assertEquals(MISMATCHED_PASSWORDS_MESSAGE, validMsgs[VERIFICATION_MESSAGES_KEYS[1]])
self.assertEquals(MISMATCHED_PASSWORDS_MESSAGE, validMsgs[VERIFICATION_MESSAGES_KEYS[2]])
self.assertEmptyMessage([VERIFICATION_MESSAGES_KEYS[0],VERIFICATION_MESSAGES_KEYS[3]], validMsgs)
def test_is_password_and_verify_equals(self):
self.assertTrue(self.validation.is_password_and_verify_equals("craig", "craig"))
self.assertFalse(self.validation.is_password_and_verify_equals("craig", "craig1"))
def test_validate_login(self):
username = '******'
password = '******'
hashed_password = self.crypt.crypt_password(username, password)
user = UserData(username=username, password=hashed_password)
self.user_datastore.save(user)
msgs, isValid = self.validation_login.validate(username, password)
self.assertTrue(isValid)
self.assertTrue(len(msgs) == 0)
def test_validate_login_username_notfound(self):
username = '******'
password = '******'
user = UserData(username=username,password=password)
self.user_datastore.save(user)
msgs, isValid = self.validation_login.validate(username + 'foo', password)
self.assertFalse(isValid)
self.assertFalse(len(msgs) == 0)
def test_validate_login_invalid_password(self):
username = '******'
password = '******'
hashed_password = self.crypt.crypt_password(username, password)
user = UserData(username=username, password=hashed_password)
self.user_datastore.save(user)
msgs, isValid = self.validation_login.validate(username, password + 'bar')
self.assertFalse(isValid)
self.assertFalse(len(msgs) == 0)
def assertEmptyMessage(self, key_list, messages):
for key in key_list:
self.assertEquals('', messages[key], "Message with key " + key + " is not empty")
def __init__(self):
factory = UserDataStoreFactory()
self.user_datastore = factory.get_storage()
self.crypt = Crypt()
pass
class Test(unittest.TestCase):
def setUp(self):
# Setup mock GAE Data Store
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_memcache_stub()
#instantiate classes under test
self.validation = UserSignupValidation()
self.validation_login = LoginValidation()
self.user_datastore = UserAppengineDataStore()
self.crypt = Crypt()
def tearDown(self):
self.testbed.deactivate()
def testIsValidUsername(self):
self.assertTrue(self.validation.is_valid_username("Crag"))
self.assertTrue(self.validation.is_valid_username("Crag-Doremus"))
self.assertTrue(self.validation.is_valid_username("Crag_Doremus"))
self.assertTrue(self.validation.is_valid_username("Cra"))
self.assertFalse(self.validation.is_valid_username("ca"))
self.assertFalse(self.validation.is_valid_username("cat!"))
self.assertTrue(
self.validation.is_valid_username("abcdefghijklmnopqrst"))
self.assertFalse(
self.validation.is_valid_username("abcdefghijklmnopqrstu"))
pass
def testIsValidPassword(self):
self.assertTrue(self.validation.is_valid_password("Craig"))
self.assertTrue(
self.validation.is_valid_password("abcdefghijklmnopqrst"))
self.assertFalse(
self.validation.is_valid_password("abcdefghijklmnopqrstu"))
pass
def testIsValidEmail(self):
self.assertTrue(self.validation.is_valid_email("*****@*****.**"))
self.assertTrue(self.validation.is_valid_email("*****@*****.**"))
self.assertFalse(self.validation.is_valid_email("Craigfoocom"))
pass
def testValid(self):
username = "******"
password = "******"
verify = "craig1"
email = "*****@*****.**"
# user = UserData(username=username, password=password, email=email)
# self.user_datastore.save(user)
validMsgs, isValid = self.validation.validate(username, password,
verify, email)
self.assertTrue(isValid)
self.assertEmptyMessage([
VERIFICATION_MESSAGES_KEYS[0], VERIFICATION_MESSAGES_KEYS[1],
VERIFICATION_MESSAGES_KEYS[2], VERIFICATION_MESSAGES_KEYS[3]
], validMsgs)
def testValid_BadUsername(self):
username = "******"
password = "******"
verify = "craig1"
email = "*****@*****.**"
validMsgs, isValid = self.validation.validate(username, password,
verify, email)
self.assertFalse(isValid)
self.assertEquals(VERIFICATION_MESSAGES[VERIFICATION_MESSAGES_KEYS[0]],
validMsgs[VERIFICATION_MESSAGES_KEYS[0]])
self.assertEmptyMessage([
VERIFICATION_MESSAGES_KEYS[1], VERIFICATION_MESSAGES_KEYS[2],
VERIFICATION_MESSAGES_KEYS[3]
], validMsgs)
def testUserExists_True(self):
username = '******'
#add a record to the data store
userData = UserData(username=username, password='******')
self.user_datastore.save(userData)
self.assertTrue(self.validation.user_exists(username),
"User " + username + " does not exist")
def testUserExists_False(self):
username = '******'
testVal = 'foo'
#add a record to the data store
userData = UserData(username=username, password='******')
self.user_datastore.save(userData)
self.assertFalse(
self.validation.user_exists(testVal),
"User " + testVal + " exists. User added: " + userData.username)
def testValidate_UserExists(self):
'''
Test validate() with a user that already exists
'''
username = '******'
password = '******'
#add a record to the data store
userData = UserData(username=username, password=password)
self.user_datastore.save(userData)
#verify in database
user = self.user_datastore.fetchByUsername(username)
print user.username
self.assertTrue(
self.validation.validate(username=username,
password=password,
verify=password,
email=None),
"User " + username + " exists")
def testValid_BadPassword(self):
username = "******"
password = "******"
verify = "c1"
email = "*****@*****.**"
validMsgs, isValid = self.validation.validate(username, password,
verify, email)
self.assertFalse(isValid)
self.assertEquals(VERIFICATION_MESSAGES[VERIFICATION_MESSAGES_KEYS[1]],
validMsgs[VERIFICATION_MESSAGES_KEYS[1]])
self.assertEquals(VERIFICATION_MESSAGES[VERIFICATION_MESSAGES_KEYS[2]],
validMsgs[VERIFICATION_MESSAGES_KEYS[2]])
self.assertEmptyMessage(
[VERIFICATION_MESSAGES_KEYS[0], VERIFICATION_MESSAGES_KEYS[3]],
validMsgs)
def testValid_BadEmail(self):
username = "******"
password = "******"
verify = "craig1"
email = "craigfoo.com"
validMsgs, isValid = self.validation.validate(username, password,
verify, email)
self.assertFalse(isValid)
self.assertEquals(VERIFICATION_MESSAGES[VERIFICATION_MESSAGES_KEYS[3]],
validMsgs[VERIFICATION_MESSAGES_KEYS[3]])
self.assertEmptyMessage([
VERIFICATION_MESSAGES_KEYS[0], VERIFICATION_MESSAGES_KEYS[2],
VERIFICATION_MESSAGES_KEYS[1]
], validMsgs)
def testValid_PasswordsDontMatch(self):
username = "******"
password = "******"
verify = "craig"
email = "*****@*****.**"
validMsgs, isValid = self.validation.validate(username, password,
verify, email)
self.assertFalse(isValid)
self.assertEquals(MISMATCHED_PASSWORDS_MESSAGE,
validMsgs[VERIFICATION_MESSAGES_KEYS[1]])
self.assertEquals(MISMATCHED_PASSWORDS_MESSAGE,
validMsgs[VERIFICATION_MESSAGES_KEYS[2]])
self.assertEmptyMessage(
[VERIFICATION_MESSAGES_KEYS[0], VERIFICATION_MESSAGES_KEYS[3]],
validMsgs)
def test_is_password_and_verify_equals(self):
self.assertTrue(
self.validation.is_password_and_verify_equals("craig", "craig"))
self.assertFalse(
self.validation.is_password_and_verify_equals("craig", "craig1"))
def test_validate_login(self):
username = '******'
password = '******'
hashed_password = self.crypt.crypt_password(username, password)
user = UserData(username=username, password=hashed_password)
self.user_datastore.save(user)
msgs, isValid = self.validation_login.validate(username, password)
self.assertTrue(isValid)
self.assertTrue(len(msgs) == 0)
def test_validate_login_username_notfound(self):
username = '******'
password = '******'
user = UserData(username=username, password=password)
self.user_datastore.save(user)
msgs, isValid = self.validation_login.validate(username + 'foo',
password)
self.assertFalse(isValid)
self.assertFalse(len(msgs) == 0)
def test_validate_login_invalid_password(self):
username = '******'
password = '******'
hashed_password = self.crypt.crypt_password(username, password)
user = UserData(username=username, password=hashed_password)
self.user_datastore.save(user)
msgs, isValid = self.validation_login.validate(username,
password + 'bar')
self.assertFalse(isValid)
self.assertFalse(len(msgs) == 0)
def assertEmptyMessage(self, key_list, messages):
for key in key_list:
self.assertEquals('', messages[key],
"Message with key " + key + " is not empty")
