pulsar-client-cpp/tests/LatchTest.cc

/**
 * Copyright 2016 Yahoo Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <gtest/gtest.h>
#include <lib/Latch.h>
#include <boost/thread.hpp>
#include <boost/date_time/posix_time/posix_time.hpp>
#include "LogUtils.h"

DECLARE_LOG_OBJECT()

using namespace pulsar;
using namespace boost::posix_time;

class Service {
 private:
    std::string serviceName_;
    time_duration sleepDuration_;
    Latch latch_;
    boost::thread thread_;

 public:
    Service(const std::string& serviceName, time_duration sleepDuration, const Latch& latch)
            : serviceName_(serviceName),
              sleepDuration_(sleepDuration),
              latch_(latch) {
        thread_ = boost::thread(&Service::run, this);
    }

    void run() {
        boost::this_thread::sleep(sleepDuration_);
        LOG_INFO("Service " << serviceName_ << " is up");
        latch_.countdown();
    }

    ~Service() {
        thread_.join();
    }
};

TEST(LatchTest, testCountDown) {
    Latch latch(3);
    Service service1("service1", millisec(50), latch);
    Service service2("service2", millisec(30), latch);
    Service service3("service3", millisec(20), latch);
    latch.wait();
}

TEST(LatchTest, testLatchCount) {
    Latch latch(3);
    Service service1("service1", millisec(50), latch);
    Service service2("service2", millisec(30), latch);
    Service service3("service3", millisec(20), latch);
    ASSERT_EQ(3, latch.getCount());
    latch.wait();
    ASSERT_EQ(0, latch.getCount());
}

TEST(LatchTest, testTimedWait) {
    // Wait for 7 seconds which is more than the maximum sleep time (5 seconds)
    Latch latch1(3);
    Service service1("service1", millisec(50), latch1);
    Service service2("service2", millisec(30), latch1);
    Service service3("service3", millisec(50), latch1);
    ASSERT_TRUE(latch1.wait(millisec(70)));

    // Wait for 3 seconds which is less than the maximum sleep time (5 seconds)
    Latch latch2(3);
    Service service4("service4", millisec(50), latch2);
    Service service5("service5", millisec(30), latch2);
    Service service6("service6", millisec(50), latch2);
    ASSERT_FALSE(latch2.wait(millisec(30)));

    // After the assert is passed and Service is destroyed because of join, the
    // main thread would not exit until service4 thread is returned.
}