daqController.hpp

Go to the documentation of this file.

/**
 * @file
 * @ingroup daq_test
 * @copyright 2022 ESO - European Southern Observatory
 *
 * @brief Mock of DaqController
 */
#ifndef OCM_DAQ_TEST_MOCK_DAQ_CONTROLLER_HPP_
#define OCM_DAQ_TEST_MOCK_DAQ_CONTROLLER_HPP_
// GCOVR_EXCL_START
#include <chrono>
 
#include <daq/daqController.hpp>
#include <gmock/gmock.h>
 
namespace daq {
 
 
/**
 * Mock version of daq::DaqController
 */
struct DaqControllerMock : public daq::DaqController {
    MOCK_METHOD0(StartAsync, boost::future<daq::State>());
    MOCK_METHOD1(StopAsync, boost::future<daq::Status>(ErrorPolicy));
    MOCK_METHOD1(AbortAsync, boost::future<daq::Status>(ErrorPolicy));
    MOCK_METHOD0(ScheduleMergeAsync, boost::future<daq::State>());
    MOCK_METHOD1(UpdateKeywords, void(fits::KeywordVector const&));
    MOCK_METHOD2(AwaitAsync,
                 boost::future<daq::State>(std::vector<std::string>, std::chrono::milliseconds));
    MOCK_CONST_METHOD0(GetErrorFlag, bool());
    MOCK_METHOD0(GetStatus, std::shared_ptr<daq::ObservableStatus>());
    MOCK_CONST_METHOD0(GetStatus, std::shared_ptr<daq::ObservableStatus const>());
    MOCK_METHOD0(GetEventLog, std::shared_ptr<daq::ObservableEventLog>());
    MOCK_CONST_METHOD0(GetId, std::string const&());
    MOCK_CONST_METHOD0(GetContext, DaqContext const&());
    // Implement by delegating to mock method
    daq::State GetState() const override {
        return GetStatus()->GetState();
    }
 
    boost::signals2::connection
    ConnectContext(ContextSignal::slot_type const& slot) override {
        return signal.connect(slot);
    }
    ContextSignal signal;
};
 
/**
 * Factory that creates mock versions
 */
struct DaqControllerFactoryFake : public DaqControllerFactory {
    auto MakeOcmPhase(DaqContext daq_ctx,
                      std::shared_ptr<ObservableStatus> status,
                      std::shared_ptr<ObservableEventLog> event_log,
                      fits::KeywordFormatter const& kw_formatter)
        -> std::shared_ptr<DaqController> override {
        return ocm_mocks.at(daq_ctx.id);
    }
    auto MakeDpmPhase(DaqContext daq_ctx,
                      std::shared_ptr<ObservableStatus> status,
                      std::shared_ptr<ObservableEventLog> event_log)
        -> std::shared_ptr<DaqController> override {
        return dpm_mocks.at(daq_ctx.id);
    }
 
    std::map<std::string, std::shared_ptr<DaqControllerMock>> ocm_mocks;
    std::map<std::string, std::shared_ptr<DaqControllerMock>> dpm_mocks;
};
 
} // namespace daq
 
// GCOVR_EXCL_STOP
#endif  // #define OCM_DAQ_TEST_MOCK_DAQ_CONTROLLER_HPP_