def test_write_failure_event_to_db():
ServiceMonitor.create_table(wait=True)
monitor = WriteToDynamoMonitor()
request = PreparedRequest()
request.body = "The body of request"
request.headers = {"bill": "Some bill"}
response = Response()
response.status_code = 500
response.headers['ben'] = "Some Ben"
response.data = "The response text"
result = monitor.failure("some-service", request, response, 1108)
assert result.id is not None
fetched_result = ServiceMonitor.query(result.service_name, ServiceMonitor.id == result.id).next()
assert fetched_result.id == result.id
assert fetched_result.event_type == "FAILED"
assert fetched_result.timestamp == result.timestamp
assert fetched_result.ttl == result.ttl
def mocked_get_ok(*args, **kwargs):
if len(args) > 0:
url = args[0]
else:
url = kwargs['url']
if url.find(ENDPOINT_USER) != -1:
jpayload = json.dumps(user1_json).encode('utf8')
elif url.find(ENDPOINT_STORIES) != -1:
jpayload = json.dumps(story_block1_json).encode('utf8')
else:
raise Exception('endpoint unknown: ne user, ne story: ' + url)
res = Response()
res.status_code = 200
res.data = jpayload
return res
