fix other tests

Author: BewareMyPower

include/pulsar/ServiceInfoProvider.h

@@ -33,6 +33,13 @@ class PULSAR_PUBLIC ServiceInfoProvider {
      */
     virtual ~ServiceInfoProvider() = default;
 
+    /**
+     * Get the current `ServiceInfo` connection for the client.
+     * This method is called **only once** internally when the `Client` is being initialized, and the client
+     * will use the returned `ServiceInfo` to establish the initial connection to the Pulsar cluster.
+     */
+    virtual ServiceInfo getServiceInfo() const = 0;
+
     /**
      * Initialize the ServiceInfoProvider.
      *

lib/ClientImpl.cc

@@ -129,13 +129,20 @@ ClientImpl::ClientImpl(std::unique_ptr<ServiceInfoProvider> serviceInfoProvider,
     if (loggerFactory) {
         LogUtils::setLoggerFactory(std::move(loggerFactory));
     }
+
+    serviceInfo_.store(std::make_shared<const ServiceInfo>(serviceInfoProvider_->getServiceInfo()));
+    lookupServicePtr_ = createLookup(*serviceInfo_.load());
 }
 
 ClientImpl::~ClientImpl() { shutdown(); }
 
 void ClientImpl::initialize() {
-    serviceInfoProvider_->initialize(
-        [this](ServiceInfo serviceInfo) { updateServiceInfo(std::move(serviceInfo)); });
+    auto weakSelf = weak_from_this();
+    serviceInfoProvider_->initialize([weakSelf](ServiceInfo serviceInfo) {
+        if (auto self = weakSelf.lock()) {
+            self->updateServiceInfo(std::move(serviceInfo));
+        }
+    });
 }
 
 LookupServicePtr ClientImpl::createLookup(ServiceInfo serviceInfo) {

lib/DefaultServiceUrlProvider.h

@@ -31,9 +31,9 @@ class DefaultServiceUrlProvider : public ServiceInfoProvider {
     DefaultServiceUrlProvider(const std::string& serviceUrl, const ClientConfigurationImpl& config)
         : serviceInfo_(config.toServiceInfo(serviceUrl)) {}
 
-    void initialize(std::function<void(ServiceInfo)> onServiceInfoUpdate) override {
-        onServiceInfoUpdate(serviceInfo_);
-    }
+    ServiceInfo getServiceInfo() const override { return serviceInfo_; }
+
+    void initialize(std::function<void(ServiceInfo)> onServiceInfoUpdate) override {}
 
    private:
     ServiceInfo serviceInfo_;

tests/ServiceInfoProviderTest.cc

@@ -74,6 +74,8 @@ class TestServiceInfoProvider : public ServiceInfoProvider {
    public:
     TestServiceInfoProvider(ServiceInfoQueue &queue) : queue_(queue) {}
 
+    ServiceInfo getServiceInfo() const override { return queue_.pop(); }
+
     void initialize(std::function<void(ServiceInfo)> onServiceInfoUpdate) override {
         onServiceInfoUpdate(queue_.pop());
         thread_ = std::thread([this, onServiceInfoUpdate] {
@@ -99,7 +101,6 @@ class TestServiceInfoProvider : public ServiceInfoProvider {
 
     std::thread thread_;
     mutable std::mutex mutex_;
-    std::queue<ServiceInfo> newServiceInfo_;
 };
 
 TEST(ServiceInfoProviderTest, testSwitchCluster) {