def setUp(self):
self.stubs = stubout.StubOutForTesting()
config_dict = {
'atompub': {
'url': 'http://127.0.0.1:9000/test/test_feed',
'user': '******',
'key': 'key',
'interval': 30,
'max_wait': 600,
'retries': 1,
'failures_before_reauth': 5,
'timeout': '120',
},
'event_feed': {
'feed_title': 'feed_title',
'feed_host': 'feed_host',
'use_https': False,
'port': 'port',
'atom_categories': "DATACENTER=ORD1, REGION=PREPROD-ORD"
},
'handler_auth': {
'method': 'no_auth'
},
'cufpub': {
'url': 'http://127.0.0.1:9000/test/test_feed',
'user': '******',
'key': 'key',
'interval': 30,
'max_wait': 600,
'retries': 1,
'failures_before_reauth': 5,
'timeout': '120',
},
'nova': {
'nova_flavor_field_name': 'dummy_flavor_field_name'
}
}
self.handler = CufPub()
def get(*args, **kwargs):
val = None
for arg in args:
if val:
val = val.get(arg)
else:
val = config_dict.get(arg)
if not val:
return None or kwargs.get('default')
return val
def config_with(*args):
return functools.partial(get, args)
self.stubs.Set(yagi.config, 'config_with', config_with)
self.stubs.Set(yagi.config, 'get', get)
def setUp(self):
self.stubs = stubout.StubOutForTesting()
self.mox = mox.Mox()
config_dict = {
'atompub': {
'url': 'http://127.0.0.1:9000/test/%(event_type)s',
'user': '******',
'key': 'key',
'interval': 30,
'max_wait': 600,
'retries': 1,
'failures_before_reauth': 5,
'timeout': '120',
},
'event_feed': {
'feed_title': 'feed_title',
'feed_host': 'feed_host',
'use_https': False,
'port': 'port'
},
'handler_auth': {
'method': 'no_auth'
},
'cufpub': {
'url': 'http://127.0.0.1:9000/test/%(event_type)s',
'user': '******',
'key': 'key',
'interval': 30,
'max_wait': 600,
'retries': 1,
'failures_before_reauth': 5,
'timeout': '120',
}
}
self.handler = CufPub()
def get(*args, **kwargs):
val = None
for arg in args:
if val:
val = val.get(arg)
else:
val = config_dict.get(arg)
if not val:
return None or kwargs.get('default')
return val
def config_with(*args):
return functools.partial(get, args)
self.stubs.Set(yagi.config, 'config_with', config_with)
self.stubs.Set(yagi.config, 'get', get)
def test_send_notification_successfully(self):
self.called = False
def mock_request(*args, **kwargs):
self.called = True
return MockResponse(201), None
handler = CufPub()
http_conn = HttpConnection(handler)
endpoint = handler.config_get("url")
payload_body = {"a": "b"}
self.stubs.Set(httplib2.Http, 'request', mock_request)
http_conn.send_notification(endpoint, endpoint, payload_body)
def test_response_other_than_201_raises_exception(self):
self.called = False
def mock_request(*args, **kwargs):
self.called = True
return MockResponse(400), None
handler = CufPub()
http_conn = HttpConnection(handler)
endpoint = handler.config_get("url")
payload_body = {"a": "b"}
self.stubs.Set(httplib2.Http, 'request', mock_request)
self.assertRaises(MessageDeliveryFailed, http_conn.send_notification,
endpoint, endpoint, payload_body)
def test_response_401_raises_unauthorized_exception(self):
self.called = False
def mock_request(*args, **kwargs):
self.called = True
return MockResponse(401), None
handler = CufPub()
http_conn = HttpConnection(handler)
endpoint = handler.config_get("url")
payload_body = {"a": "b"}
self.stubs.Set(httplib2.Http, 'request', mock_request)
self.assertRaises(UnauthorizedException, http_conn.send_notification,
endpoint, endpoint, payload_body)
def test_duplicate_entry_to_atomhopper_response_409(self):
self.called = False
def mock_request(*args, **kwargs):
self.called = True
return MockResponse(409), None
handler = CufPub()
http_conn = HttpConnection(handler)
endpoint = handler.config_get("url")
payload_body = {"a": "b"}
self.stubs.Set(httplib2.Http, 'request', mock_request)
status = http_conn.send_notification(endpoint, endpoint, payload_body)
self.assertEqual(status, 409)
def test_response_too_large_error_and_status_is_201_does_not_raise_exception(
self):
self.called = False
def mock_request(*args, **kwargs):
self.called = True
raise http_util.ResponseTooLargeError("desc", MockResponse(201),
"content")
handler = CufPub()
http_conn = HttpConnection(handler)
endpoint = handler.config_get("url")
payload_body = {"a": "b"}
self.stubs.Set(httplib2.Http, 'request', mock_request)
http_conn.send_notification(endpoint, endpoint, payload_body)
def test_send_notification_successfully(self):
self.called = False
content = (
"""<atom:entry xmlns:atom="http://www.w3.org/2005/Atom">"""
"""<atom:id>urn:uuid:95347e4d-4737-4438-b774-6a9219d78d2a</atom:id>"""
"""</atom:entry>""")
def mock_request(*args, **kwargs):
self.called = True
return MockResponse(201), content
handler = CufPub()
http_conn = HttpConnection(handler)
endpoint = handler.config_get("url")
payload_body = {"a": "b"}
self.mox.ReplayAll()
self.stubs.Set(httplib2.Http, 'request', mock_request)
http_conn.send_notification(endpoint, endpoint, payload_body)
self.mox.VerifyAll()
def setUp(self):
self.mox = mox.Mox()
self.stubs = stubout.StubOutForTesting()
self.handler = CufPub()
def get(*args, **kwargs):
val = None
for arg in args:
if val:
val = val.get(arg)
else:
val = self.config_dict.get(arg)
if not val:
return None or kwargs.get('default')
return val
def config_with(*args):
return functools.partial(get, args)
self.stubs.Set(yagi.config, 'config_with', config_with)
self.stubs.Set(yagi.config, 'get', get)
