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)