def test_create_admin_email_missed(self):
args = {'email': '*****@*****.**'}
db_adapter = Mock()
db_adapter.find_first_object.return_value = None
am = AdminManager(db_adapter)
self.assertEqual(am.create_admin(args), not_found("email does not exist in DB"))
db_adapter.find_first_object.assert_called_once_with(UserEmail, ANY)
def test_create_admin_already_exist(self):
args = {'email': '*****@*****.**'}
user = User(id=7)
user_email = UserEmail(id=7, user=user, user_id=7)
db_adapter = Mock()
db_adapter.find_first_object.side_effect = [user_email, "test"]
am = AdminManager(db_adapter)
with app.test_request_context('/', headers=None):
g.hackathon = Hackathon(id=7)
self.assertEqual(am.create_admin(args), ok())
self.assertEqual(db_adapter.find_first_object.call_count, 2)
def test_create_admin_common_logic(self):
args = {'email': '*****@*****.**', 'role_type': 1, 'remarks': 'test'}
user = User(id=7)
user_email = UserEmail(id=7, user=user, user_id=7)
db_adapter = Mock()
db_adapter.find_first_object.side_effect = [user_email, None]
db_adapter.add_object.return_value = "success"
am = AdminManager(db_adapter)
with app.test_request_context('/', headers=None):
g.hackathon = Hackathon(id=7)
self.assertEqual(am.create_admin(args), ok())
self.assertEqual(db_adapter.find_first_object.call_count, 2)
db_adapter.add_object.assert_called_once_with(ANY)
