def setUp(self):
self.connection = MagicMock()
connection_manager = MagicMock(client_uuid=uuid.uuid4())
connection_manager.get_random_connection_for_sql = MagicMock(
return_value=self.connection)
serialization_service = MagicMock()
serialization_service.to_object.side_effect = lambda arg: arg
serialization_service.to_data.side_effect = lambda arg: arg
self.invocation_registry = {}
correlation_id_counter = itertools.count()
invocation_service = MagicMock()
def invoke(invocation):
self.invocation_registry[next(correlation_id_counter)] = invocation
invocation_service.invoke.side_effect = invoke
self.internal_service = _InternalSqlService(connection_manager,
serialization_service,
invocation_service)
self.service = SqlService(self.internal_service)
self.result = self.service.execute("SOME QUERY")
def __init__(self, **kwargs):
config = _Config.from_dict(kwargs)
self._config = config
self._context = _ClientContext()
client_id = HazelcastClient._CLIENT_ID.get_and_increment()
self._name = self._create_client_name(client_id)
self._reactor = AsyncoreReactor()
self._serialization_service = SerializationServiceV1(config)
self._near_cache_manager = NearCacheManager(
config, self._serialization_service)
self._internal_lifecycle_service = _InternalLifecycleService(config)
self._lifecycle_service = LifecycleService(
self._internal_lifecycle_service)
self._invocation_service = InvocationService(self, config,
self._reactor)
self._address_provider = self._create_address_provider()
self._internal_partition_service = _InternalPartitionService(self)
self._partition_service = PartitionService(
self._internal_partition_service, self._serialization_service)
self._internal_cluster_service = _InternalClusterService(self, config)
self._cluster_service = ClusterService(self._internal_cluster_service)
self._connection_manager = ConnectionManager(
self,
config,
self._reactor,
self._address_provider,
self._internal_lifecycle_service,
self._internal_partition_service,
self._internal_cluster_service,
self._invocation_service,
self._near_cache_manager,
)
self._load_balancer = self._init_load_balancer(config)
self._listener_service = ListenerService(self, config,
self._connection_manager,
self._invocation_service)
self._proxy_manager = ProxyManager(self._context)
self._cp_subsystem = CPSubsystem(self._context)
self._proxy_session_manager = ProxySessionManager(self._context)
self._transaction_manager = TransactionManager(self._context)
self._lock_reference_id_generator = AtomicInteger(1)
self._statistics = Statistics(
self,
config,
self._reactor,
self._connection_manager,
self._invocation_service,
self._near_cache_manager,
)
self._cluster_view_listener = ClusterViewListenerService(
self,
self._connection_manager,
self._internal_partition_service,
self._internal_cluster_service,
self._invocation_service,
)
self._shutdown_lock = threading.RLock()
self._invocation_service.init(self._internal_partition_service,
self._connection_manager,
self._listener_service)
self._internal_sql_service = _InternalSqlService(
self._connection_manager, self._serialization_service,
self._invocation_service)
self.sql = SqlService(self._internal_sql_service)
self._init_context()
self._start()