def test_user_agent_header_uses_application_not_version(self):
test_class = flask_request.RequestsSession()
response = self.test_class.http_fetch('https://httpbin.org/get')
self.assertEquals(response.status_code, 200)
self.assertEquals(self.test_class.session.headers['User-Agent'],
test_class.headers['User-Agent'])
def test_user_agent_uses_application_name_and_version(self):
self.app.version = '1.0.0'
user_agent = '{}/{}'.format(self.app_name, self.app.version)
test_class = flask_request.RequestsSession(self.app)
response = test_class.http_fetch('https://httpbin.org/get')
self.assertEquals(response.status_code, 200)
self.assertEquals(test_class.session.headers['User-Agent'], user_agent)
def setUp(cls):
cls.app_name = 'fancy_app'
cls.app = Flask(cls.app_name)
cls.test_class = flask_request.RequestsSession(cls.app)