1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
// Copyright (C) 2013-2025 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#include <config.h>

#include <asiolink/io_service.h>
#include <asiolink/interval_timer.h>

#include <gtest/gtest.h>
#include <functional>
#include <thread>
#include <vector>

using namespace isc::asiolink;

namespace {

void
postedEvent(std::vector<int>* destination, int value) {
    destination->push_back(value);
}

// Check the posted events are called, in the same order they are posted.
TEST(IOService, post) {
    std::vector<int> called;
    IOService service;
    // Post two events
    service.post(std::bind(&postedEvent, &called, 1));
    service.post(std::bind(&postedEvent, &called, 2));
    service.post(std::bind(&postedEvent, &called, 3));
    // They have not yet been called
    EXPECT_TRUE(called.empty());
    // Process two events
    service.runOne();
    service.runOne();
    // Both events were called in the right order
    ASSERT_EQ(2, called.size());
    EXPECT_EQ(1, called[0]);
    EXPECT_EQ(2, called[1]);

    // Test that poll() executes the last handler.
    service.poll();
    ASSERT_EQ(3, called.size());
    EXPECT_EQ(3, called[2]);
}

// Verify that runOneFor() operates correctly.
TEST(IOService, runOneFor) {
    IOServicePtr io_service(new IOService());

    // Setup up a timer to expire in 200 ms.
    IntervalTimer timer(io_service);
    size_t wait_ms = 200;
    bool timer_fired = false;
    timer.setup([&timer_fired] { timer_fired = true; },
                wait_ms, IntervalTimer::ONE_SHOT);

    size_t time_outs = 0;
    while (timer_fired == false && time_outs < 5) {<--- Assuming that condition 'timer_fired==false' is not redundant
        // Call runOneFor() with 1/4 of the timer duration.
        bool timed_out = false;
        auto cnt = io_service->runOneFor(50 * 1000, timed_out);
        if (cnt || timer_fired) {<--- Condition 'timer_fired' is always false
            ASSERT_FALSE(timed_out);
        } else {
            ASSERT_TRUE(timed_out);
            ++time_outs;
        }
    }

    // Should have had at least two time outs.
    EXPECT_GE(time_outs, 2);

    // Timer should have fired.
    EXPECT_EQ(timer_fired, 1);
}

// Verify that runOneFor() handles service stop correctly.
TEST(IOService, runOneForStopped) {
    IOServicePtr io_service(new IOService());

    // Setup up a timer to expire in 200 ms.
    IntervalTimer timer(io_service);
    size_t wait_ms = 200;
    bool timer_fired = false;
    timer.setup([&timer_fired] { timer_fired = true; },
                wait_ms, IntervalTimer::ONE_SHOT);

    // Call runOneFor() with runtime of 500ms in a thread.
    size_t cnt = 0;
    bool timed_out = false;
    std::thread th([io_service, &timed_out, &cnt]() {
        cnt = io_service->runOneFor(500 * 1000, timed_out);
    });

    // Sleep for 5 ms.
    usleep(5 * 1000);

    // Stop the service.
    io_service->stop();

    // Wait for the thread to complete.
    th.join();

    // Handle count should be 0, we should not have timed out, and
    // the timer should not have fired.
    EXPECT_EQ(0, cnt);
    EXPECT_FALSE(timed_out);
    EXPECT_FALSE(timer_fired);
}

}