def test_cache_get_error_connect(self): store = Store( MemcacheAdapter(address=os.environ['STORE_PORT_11211_TCP_ADDR'], port=8000, ignore_exc=False)) with self.assertRaises(Exception): store.cache_get('error_get_key', )
def setUp(self): self.context = {} self.headers = {} self.store = Store(MemcacheAdapter( address=os.environ['STORE_PORT_11211_TCP_ADDR'], port=os.environ['STORE_PORT_11211_TCP_PORT'] ))
def setUp(self): self.port = self.get_port() self.handler_url = f'http://localhost:{self.port}/method/' self.server = HTTPServer(('localhost', self.port), api.MainHTTPHandler) self.server_thread = Thread(target=self.server.serve_forever) self.server_thread.setDaemon(True) self.server_thread.start() self.store = Store(MemcacheAdapter( address=os.environ['STORE_PORT_11211_TCP_ADDR'], port=os.environ['STORE_PORT_11211_TCP_PORT'] ))
class MainHTTPHandler(BaseHTTPRequestHandler): router = { 'method': method_handler } store = Store(MemcacheAdapter( address=os.environ['STORE_PORT_11211_TCP_ADDR'], port=os.environ['STORE_PORT_11211_TCP_PORT'] ) ) def get_request_id(self, headers): return headers.get('HTTP_X_REQUEST_ID', uuid.uuid4().hex) def do_POST(self): response, code = {}, OK context = {'request_id': self.get_request_id(self.headers)} request = None try: data_string = self.rfile.read(int(self.headers['Content-Length'])) request = json.loads(data_string) except: code = BAD_REQUEST if request: path = self.path.strip('/') logging.info('%s: %s %s' % (self.path, data_string, context['request_id'])) if path in self.router: try: response, code = self.router[path]({'body': request, 'headers': self.headers}, context, self.store) except Exception as e: logging.exception('Unexpected error: %s' % e) code = INTERNAL_ERROR else: code = NOT_FOUND self.send_response(code) self.send_header('Content-Type', 'application/json') self.end_headers() if code not in ERRORS: r = {'response': response, 'code': code} else: r = {'error': response or ERRORS.get(code, 'Unknown Error'), 'code': code} context.update(r) logging.info(context) self.wfile.write(json.dumps(r).encode('utf-8')) return
def test_get_empty_value(self, key): store = Store( MemcacheAdapter(address=os.environ['STORE_PORT_11211_TCP_ADDR'], port=os.environ['STORE_PORT_11211_TCP_PORT'])) self.assertEqual(store.get(key), None)
def test_get(self, value): store = Store( MemcacheAdapter(address=os.environ['STORE_PORT_11211_TCP_ADDR'], port=os.environ['STORE_PORT_11211_TCP_PORT'])) store.cache_set('test_get', value) self.assertNotEqual(store.get('test_get'), None)
def test_cache_get_error_connect_and_ignore_exc(self, key): store = Store( MemcacheAdapter(address=os.environ['STORE_PORT_11211_TCP_ADDR'], port=8000)) self.assertEqual(store.cache_get(key), None)