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