class TestAccountsSessions(unittest.TestCase):
def setUp(self):
self.connection, self.db = pp_tests.connect_to_mongo()
self.account_data = {
'email': '*****@*****.**',
'password': '******',
'name': 'Daniel',
'startup': 'Payparrot',
}
self.account = Accounts(self.db, self.account_data)
self.account.insert()
self.account_session = AccountsSessions(self.db, {'account_id':self.account.id})
self.account_session.insert()
def tearDown(self):
pp_tests.tear_down(self.db)
self.connection.close()
def test_create_session_setup_default_elements(self):
self.assertTrue(self.account_session.expires)
self.assertTrue(self.account_session.account_id)
self.assertTrue(self.account_session.id)
self.assertTrue(self.account_session.session_id)
def test_get_account_from_session(self):
account = Accounts.get_from_session(self.db, id = self.account_session.session_id)
self.assertEqual(self.account.id, account.id)
def login(db):
account_by_email = Accounts.findOne(db, {
'email': request.forms.get('email'),
})
if account_by_email:
account = Accounts.findOne(db, {
'email': request.forms.get('email'),
'password': sha1(account_by_email.salt+request.forms.get('password')).hexdigest()
})
if account:
account_session = AccountsSessions(db, {'account_id': account.id})
account_session.insert()
response.set_cookie('sid', account_session.session_id, path='/', expires = account_session.expires)
redirect('/logged')
return
raise UnauthorizeException()
def setUp(self):
self.connection, self.db = pp_tests.connect_to_mongo()
self.account_data = {
'email': '*****@*****.**',
'password': '******',
'name': 'Daniel',
'startup': 'Payparrot',
}
self.account = Accounts(self.db, self.account_data)
self.account.insert()
self.account_session = AccountsSessions(self.db, {'account_id':self.account.id})
self.account_session.insert()