class AccountServiceDBTest(unittest.TestCase):
def setUp(self):
self.redis = fakeredis.FakeStrictRedis()
self.service = AccountService(self.redis)
def test_store_and_load(self):
user = User(UserId('stored_id'), 'stored_email', 'stored_hash')
self.service.store(user)
user_by_email = self.service.load('stored_email')
self.assertEqual('stored_id', str(user_by_email.id))
self.assertEqual('stored_email', user_by_email.email)
self.assertEqual('stored_hash', user_by_email.pass_hash)
user_by_id = self.service.load(UserId('stored_id'))
self.assertEqual('stored_id', str(user_by_email.id))
self.assertEqual('stored_email', user_by_email.email)
self.assertEqual('stored_hash', user_by_email.pass_hash)
def test_load_missing(self):
user = self.service.load('missing_email')
self.assertIsNone(user)
class AccountServiceTest(unittest.TestCase):
def setUp(self):
self.redis = unittest.mock.Mock()
self.service = AccountService(self.redis)
self.test_user = User(UserId('u_id'), 'u_email', 'u_hash')
def test_load_by_id(self):
self.redis.get.return_value = self.test_user.to_json().encode('utf-8')
user = self.service.load(UserId('user_id'))
self.assertEqual(UserId('u_id'), user.id)
self.assertEqual('u_email', user.email)
self.assertEqual('u_hash', user.pass_hash)
self.redis.get.assert_called_with('user|by_id|user_id')
def test_load_by_id_missing(self):
self.redis.get.return_value = None
user = self.service.load(UserId('user_id'))
self.assertIsNone(user)
def test_load_by_email(self):
self.redis.get.return_value = self.test_user.to_json().encode('utf-8')
user = self.service.load('user_email')
self.assertEqual(UserId('u_id'), user.id)
self.assertEqual('u_email', user.email)
self.assertEqual('u_hash', user.pass_hash)
self.redis.get.assert_called_with('user|by_email|user_email')
def test_load_by_email_missing(self):
self.redis.get.return_value = None
user = self.service.load('user_email')
self.assertIsNone(user)
def test_store(self):
pipe_mock = self.redis.pipeline()
pipe_mock.set.return_value = pipe_mock
user = User('user_id', 'user_email', 'user_hash')
self.service.store(user)
js = user.to_json().encode('utf-8')
pipe_mock.set.assert_any_call('user|by_id|user_id', js)
pipe_mock.set.assert_any_call('user|by_email|user_email', js)
pipe_mock.execute.assert_any_call()
def setUp(self):
self.redis = unittest.mock.Mock()
self.service = AccountService(self.redis)
self.test_user = User(UserId('u_id'), 'u_email', 'u_hash')
def setUp(self):
self.redis = fakeredis.FakeStrictRedis()
self.service = AccountService(self.redis)
