def test_http_publisher_config(self):
"""Test publisher config parameters."""
# invalid hostname, the given url, results in an empty hostname
parsed_url = urlparse.urlparse('http:/aaa.bb/path')
self.assertRaises(ValueError, http.HttpPublisher, parsed_url)
# invalid port
parsed_url = urlparse.urlparse('http://aaa:bb/path')
self.assertRaises(ValueError, http.HttpPublisher, parsed_url)
parsed_url = urlparse.urlparse('http://localhost:90/path1')
publisher = http.HttpPublisher(parsed_url)
# By default, timeout and retry_count should be set to 1000 and 2
# respectively
self.assertEqual(1, publisher.timeout)
self.assertEqual(2, publisher.max_retries)
parsed_url = urlparse.urlparse('http://localhost:90/path1?'
'timeout=19&max_retries=4')
publisher = http.HttpPublisher(parsed_url)
self.assertEqual(19, publisher.timeout)
self.assertEqual(4, publisher.max_retries)
parsed_url = urlparse.urlparse('http://localhost:90/path1?'
'timeout=19')
publisher = http.HttpPublisher(parsed_url)
self.assertEqual(19, publisher.timeout)
self.assertEqual(2, publisher.max_retries)
parsed_url = urlparse.urlparse('http://localhost:90/path1?'
'max_retries=6')
publisher = http.HttpPublisher(parsed_url)
self.assertEqual(1, publisher.timeout)
self.assertEqual(6, publisher.max_retries)
def test_post_verify_ssl_default(self):
parsed_url = urlparse.urlparse('http://localhost:90/path1')
publisher = http.HttpPublisher(self.CONF, parsed_url)
with mock.patch.object(requests.Session, 'post') as post:
publisher.publish_samples(self.sample_data)
self.assertTrue(post.call_args[1]['verify'])
def test_post_verify_ssl_path(self):
parsed_url = urlparse.urlparse('http://localhost:90/path1?'
'verify_ssl=/path/to/cert.crt')
publisher = http.HttpPublisher(self.CONF, parsed_url)
with mock.patch.object(requests.Session, 'post') as post:
publisher.publish_samples(self.sample_data)
self.assertEqual('/path/to/cert.crt', post.call_args[1]['verify'])
def _post_batch_control_test(self, method, data, batch):
parsed_url = urlparse.urlparse('http://localhost:90/path1?'
'batch=%s' % batch)
publisher = http.HttpPublisher(self.CONF, parsed_url)
with mock.patch.object(requests.Session, 'post') as post:
getattr(publisher, method)(data)
self.assertEqual(1 if batch else 3, post.call_count)
def test_post_raw_only(self):
parsed_url = urlparse.urlparse('http://localhost:90/path1?raw_only=1')
publisher = http.HttpPublisher(self.CONF, parsed_url)
with mock.patch.object(requests.Session, 'post') as post:
publisher.publish_events(self.event_data)
self.assertEqual(
'[{"some": "aa"}, {"some": "aa"}, {"some": "aa"}]',
post.call_args[1]['data'])
def test_post_basic_auth(self):
parsed_url = urlparse.urlparse(
'http://*****:*****@localhost:90/path1?')
publisher = http.HttpPublisher(self.CONF, parsed_url)
with mock.patch.object(requests.Session, 'post') as post:
publisher.publish_samples(self.sample_data)
self.assertEqual(('alice', 'l00kingGla$$'),
post.call_args[1]['auth'])
def test_post_client_cert_auth(self):
parsed_url = urlparse.urlparse('http://localhost:90/path1?'
'clientcert=/path/to/cert.crt&'
'clientkey=/path/to/cert.key')
publisher = http.HttpPublisher(self.CONF, parsed_url)
with mock.patch.object(requests.Session, 'post') as post:
publisher.publish_samples(self.sample_data)
self.assertEqual(('/path/to/cert.crt', '/path/to/cert.key'),
post.call_args[1]['cert'])
def test_http_post_empty_data(self, thelog):
parsed_url = urlparse.urlparse('http://localhost:90/path1')
publisher = http.HttpPublisher(parsed_url)
res = mock.Mock()
res.status_code = 200
with mock.patch.object(Session, 'post', return_value=res) as m_req:
publisher.publish_events(self.empty_event_data)
self.assertEqual(0, m_req.call_count)
self.assertTrue(thelog.debug.called)
def test_http_post_events(self, thelog):
"""Test publisher post."""
parsed_url = urlparse.urlparse('http://localhost:90/path1')
publisher = http.HttpPublisher(parsed_url)
res = mock.Mock()
res.status_code = 200
with mock.patch.object(Session, 'post', return_value=res) as m_req:
publisher.publish_events(self.event_data)
self.assertEqual(1, m_req.call_count)
self.assertFalse(thelog.error.called)
res.status_code = 401
with mock.patch.object(Session, 'post', return_value=res) as m_req:
publisher.publish_samples(self.event_data)
self.assertEqual(1, m_req.call_count)
self.assertTrue(thelog.error.called)
def test_http_post_samples(self, thelog):
"""Test publisher post."""
parsed_url = urlparse.urlparse('http://localhost:90/path1')
publisher = http.HttpPublisher(self.CONF, parsed_url)
res = requests.Response()
res.status_code = 200
with mock.patch.object(requests.Session, 'post',
return_value=res) as m_req:
publisher.publish_samples(self.sample_data)
self.assertEqual(1, m_req.call_count)
self.assertFalse(thelog.exception.called)
res = requests.Response()
res.status_code = 401
with mock.patch.object(requests.Session, 'post',
return_value=res) as m_req:
publisher.publish_samples(self.sample_data)
self.assertEqual(1, m_req.call_count)
self.assertTrue(thelog.exception.called)