def setUp(self): super(LoggedInApplicationTest, self).setUp() patch_config = { 'authenticate_user.return_value': { 'users': { 'id': 1234, 'emailAddress': '*****@*****.**', 'role': 'admin', 'locked': False, 'passwordChangedAt': '2015-01-01T00:00:00Z', 'active': True, 'name': 'tester' } } } self._data_api_client = mock.patch( 'app.main.views.login.data_api_client', **patch_config) self._data_api_client.start() res = self.client.post('/admin/login', data={ 'email_address': '*****@*****.**', 'password': '******' }) self._user_callback = login_manager.user_callback login_manager.user_loader(self.user_loader)
def setUp(self): super(LoggedInApplicationTest, self).setUp() patch_config = { 'authenticate_user.return_value': { 'users': { 'id': 1234, 'emailAddress': '*****@*****.**', 'role': 'admin', 'locked': False, 'passwordChangedAt': '2015-01-01T00:00:00Z', 'active': True, 'name': 'tester', 'termsAcceptedAt': '2015-01-01T00:00:00.0Z', } } } self._data_api_client = mock.patch( 'app.main.views.login.data_api_client', **patch_config ) self._data_api_client.start() res = self.client.post('/admin/login', data={ 'email_address': '*****@*****.**', 'password': '******', 'csrf_token': FakeCsrf.valid_token, }) assert res.status_code < 400 self._user_callback = login_manager.user_callback login_manager.user_loader(self.user_loader)
def setUp(self): self.app = create_app("test") self.client = self.app.test_client() self._s3_patch = mock.patch("app.main.views.services.S3") self.s3 = self._s3_patch.start() self._default_suffix_patch = mock.patch( "dmutils.validation.default_file_suffix", return_value="2015-01-01-1200" ) self._default_suffix_patch.start() self._user_callback = login_manager.user_callback def user_loader(user_id): if user_id: return User(user_id, "*****@*****.**", None, None, False) login_manager.user_loader(user_loader)
def setUp(self): self.app = create_app('test') self.client = self.app.test_client() self._s3_patch = mock.patch('app.main.views.services.S3') self.s3 = self._s3_patch.start() self._default_suffix_patch = mock.patch( 'dmutils.validation.default_file_suffix', return_value='2015-01-01-1200') self._default_suffix_patch.start() self._user_callback = login_manager.user_callback def user_loader(user_id): if user_id: return User(user_id, '*****@*****.**', None, None, False) login_manager.user_loader(user_loader)
def setup_method(self, method): super().setup_method(method) self.app.register_blueprint(login_for_tests) with mock.patch('app.data_api_client') as login_api_client: login_api_client.authenticate_user.return_value = { 'users': { 'id': 1234, 'emailAddress': '*****@*****.**', 'role': 'admin', 'locked': False, 'passwordChangedAt': '2015-01-01T00:00:00Z', 'active': True, 'name': 'tester' } } res = self.client.get('/auto-login') assert res.status_code == 200 self._user_callback = login_manager._user_callback login_manager.user_loader(self.user_loader)
def tearDown(self): self._s3_patch.stop() self._default_suffix_patch.stop() login_manager.user_loader(self._user_callback)
def tearDown(self): self._data_api_client.stop() login_manager.user_loader(self._user_callback)
from app import login_manager from app.models import User def user_loader(user_name): return User.objects(user_name=user_name).first() login_manager.user_loader(user_loader)
def teardown_method(self, method): login_manager.user_loader(self._user_callback) super().teardown_method(method)