testRsyncAsyncProcess.cpp

Go to the documentation of this file.

/**
 * @file
 * @ingroup daq_test
 * @copyright (c) Copyright ESO 2022
 * All Rights Reserved
 * ESO (eso.org) is an Intergovernmental Organisation, and therefore special legal conditions apply.
 *
 * @brief daq::RsyncAsyncProcess integration tests
 *
 * This starts subprocesses that depend on @c rsync.
 */
#include <stdlib.h>
 
#include <filesystem>
#include <sstream>
#include <thread>
 
#include <gtest/gtest.h>
 
#include <rad/ioExecutor.hpp>
 
#include <daq/process/rsyncAsyncProcess.hpp>
 
namespace fs = std::filesystem;
 
class RsyncAsyncProcessTest : public ::testing::Test {
public:
    RsyncAsyncProcessTest() : m_ctx(), m_exec(m_ctx) {
    }
 
    /**
     * Creates a temporary file source (m_source) and destination (m_dest) that can be used by test
     * cases.
     */
    void SetUp() override {
        m_source = "rsync-XXXXXX";
        m_dest = "rsync-XXXXXX";
        // Create a source file with known size
        int fd = mkstemp(m_source.data());
        close(fd);
        truncate(m_source.c_str(), 4 * 1024);
        // Dest
        fd = mkstemp(m_dest.data());
        close(fd);
    }
    /**
     * Delete temp files.
     */
    void TearDown() override {
        unlink(m_source.c_str());
        unlink(m_dest.c_str());
    }
 
protected:
    boost::asio::io_context m_ctx;
    rad::IoExecutor m_exec;
    std::string m_source;
    std::string m_dest;
};
 
TEST_F(RsyncAsyncProcessTest, Copy) {
    daq::RsyncOptions opts;
    opts.bw_limit = 200;  // 200 kB/s
    opts.whole_file = true;
    auto proc = daq::RsyncAsyncProcess(m_ctx, m_source, m_dest, opts);
 
    std::stringstream stderr;
 
    unsigned num_progresses = 0;
    proc.ConnectProgress([&](pid_t, daq::RsyncProgress const& p) { num_progresses++; });
    proc.ConnectStderr([&](pid_t, std::string const& line) { stderr << line; });
 
    bool done = false;
    proc.Initiate().then(m_exec, [&](boost::future<int> exit_code) {
        done = true;
        EXPECT_FALSE(proc.IsRunning());
        auto code = exit_code.get();
        EXPECT_EQ(code, 0) << "Unexpected exit code for SIGTERM, rsync: " << stderr.str();
    });
 
    EXPECT_TRUE(proc.IsRunning());
 
    m_ctx.run();  // runs until there's no more work
 
    EXPECT_TRUE(done)
        << "Expected continuation to have been executed as io_context::run has exited";
 
    EXPECT_EQ(4 * 1024, fs::file_size(m_dest));
    EXPECT_GT(num_progresses, 0) << "There should have been at least one progress report..";
}