def test_strict_redis_connection_instance_info(args, kwargs, expected): r = redis.StrictRedis(*args, **kwargs) r.connection_pool.connection_class = DisabledConnection connection = r.connection_pool.get_connection('SELECT') try: conn_kwargs = _conn_attrs_to_dict(connection) assert _instance_info(conn_kwargs) == expected finally: r.connection_pool.release(connection)
def test_strict_redis_connection_from_url(args, kwargs, expected): r = aredis.StrictRedis.from_url(*args, **kwargs) if r.connection_pool.connection_class is aredis.Connection: r.connection_pool.connection_class = DisabledConnection elif r.connection_pool.connection_class is aredis.UnixDomainSocketConnection: r.connection_pool.connection_class = DisabledUnixConnection else: assert False, r.connection_pool.connection_class connection = r.connection_pool.get_connection("SELECT") try: conn_kwargs = _conn_attrs_to_dict(connection) assert _instance_info(conn_kwargs) == expected finally: r.connection_pool.release(connection)
async def _nr_Connection_send_command_wrapper_(wrapped, instance, args, kwargs): transaction = current_transaction() if not transaction: return await wrapped(*args, **kwargs) host, port_path_or_id, db = (None, None, None) try: dt = transaction.settings.datastore_tracer if dt.instance_reporting.enabled or dt.database_name_reporting.enabled: conn_kwargs = _conn_attrs_to_dict(instance) host, port_path_or_id, db = _instance_info(conn_kwargs) except Exception: pass transaction._nr_datastore_instance_info = (host, port_path_or_id, db) # Older Redis clients would when sending multi part commands pass # them in as separate arguments to send_command(). Need to therefore # detect those and grab the next argument from the set of arguments. operation = args[0].strip().lower() # If it's not a multi part command, there's no need to trace it, so # we can return early. if operation.split()[0] not in _redis_multipart_commands: return await wrapped(*args, **kwargs) # Convert multi args to single arg string if operation in _redis_multipart_commands and len(args) > 1: operation = "%s %s" % (operation, args[1].strip().lower()) operation = _redis_operation_re.sub("_", operation) with DatastoreTrace(product="Redis", target=None, operation=operation, host=host, port_path_or_id=port_path_or_id, database_name=db): return await wrapped(*args, **kwargs)
def test_strict_redis_client_instance_info(args, kwargs, expected): r = redis.StrictRedis(*args, **kwargs) conn_kwargs = r.connection_pool.connection_kwargs assert _instance_info(conn_kwargs) == expected
def test_redis_client_from_url(args, kwargs, expected): r = redis.Redis.from_url(*args, **kwargs) conn_kwargs = r.connection_pool.connection_kwargs assert _instance_info(conn_kwargs) == expected