def test_validate_request_token_not_found(self):
token_value = "token_value"
mock_db = Mock()
mock_db.find_first_object_by.return_value = None
am = AdminManager(mock_db)
with app.test_request_context('/api', headers={"token": token_value}):
self.assertTrue("token" in request.headers)
self.assertFalse(am.validate_request())
mock_db.find_first_object_by.assert_called_once_with(AdminToken, token=token_value)
def test_validate_request_if_token_missing(self):
am = AdminManager(None)
'''more args for app.text_request_context:
path='/', base_url=None, query_string=None,
method='GET', input_stream=None, content_type=None,
content_length=None, errors_stream=None, multithread=False,
multiprocess=False, run_once=False, headers=None, data=None,
environ_base=None, environ_overrides=None, charset='utf-8'
'''
with app.test_request_context('/', headers=None):
self.assertFalse("token" in request.headers)
self.assertFalse(am.validate_request())
def test_validate_request_token_not_found(self):
token_value = "token_value"
mock_db = Mock()
mock_db.find_first_object_by.return_value = None
am = AdminManager(mock_db)
with app.test_request_context('/api', headers={"token": token_value}):
self.assertTrue("token" in request.headers)
self.assertFalse(am.validate_request())
mock_db.find_first_object_by.assert_called_once_with(
AdminToken, token=token_value)
def test_validate_request_if_token_missing(self):
am = AdminManager(None)
'''more args for app.text_request_context:
path='/', base_url=None, query_string=None,
method='GET', input_stream=None, content_type=None,
content_length=None, errors_stream=None, multithread=False,
multiprocess=False, run_once=False, headers=None, data=None,
environ_base=None, environ_overrides=None, charset='utf-8'
'''
with app.test_request_context('/', headers=None):
self.assertFalse("token" in request.headers)
self.assertFalse(am.validate_request())
def test_validate_request_token_expired(self):
token_value = "token_value"
token = AdminToken(token=token_value, admin=None, expire_date=datetime.utcnow() - timedelta(seconds=30))
mock_db = Mock()
mock_db.find_first_object_by.return_value = token
am = AdminManager(mock_db)
with app.test_request_context('/', headers={"token": token_value}):
self.assertTrue("token" in request.headers)
self.assertFalse(am.validate_request())
mock_db.find_first_object_by.assert_called_once_with(AdminToken, token=token_value)
def test_validate_request_token_expired(self):
token_value = "token_value"
token = AdminToken(token=token_value,
admin=None,
expire_date=datetime.utcnow() -
timedelta(seconds=30))
mock_db = Mock()
mock_db.find_first_object_by.return_value = token
am = AdminManager(mock_db)
with app.test_request_context('/', headers={"token": token_value}):
self.assertTrue("token" in request.headers)
self.assertFalse(am.validate_request())
mock_db.find_first_object_by.assert_called_once_with(
AdminToken, token=token_value)
