testUnmergedData.cpp

Go to the documentation of this file.

/**
 * @file
 * @copyright
 * (c) Copyright ESO 2024
 * All Rights Reserved
 * ESO (eso.org) is an Intergovernmental Organisation, and therefore special legal conditions apply.
 */
 
#include <merge.hpp>
 
#include <daq/fits/cfitsio.hpp>
#include <gmock/gmock.h>
#include <gtest/gtest.h>
 
using namespace ::testing;
namespace daq::dpm::merge {
namespace {
struct MockReporter : StatusReporter {
    MOCK_METHOD(void, PostAlert, (std::string const& id, std::string const& message), (override));
};
 
}  // namespace
 
class TestUnmergedAlert : public ::testing::Test {
public:
    TestUnmergedAlert()
        : m_reporter()
        , m_logger(log4cplus::Logger::getInstance("test.dpm.merge"))
        , m_fits_buffer(4096u)
        , m_file("NAME",
                 "/foo/bar.fits",
                 "/location/file.fits",
                 {},
                 {},
                 true,
                 std::move(m_fits_buffer).GetOwnedFile()) {
    }
    void SetUp() {
        ASSERT_TRUE(m_file.GetFitsFile());
        ASSERT_TRUE(m_file.AlertUnmergeable());
    }
    MockReporter m_reporter;
    log4cplus::Logger m_logger;
    fits::MemoryFitsFile m_fits_buffer;
    FitsFileSource m_file;
};
 
class TestUnmergedNoAlert : public ::testing::Test {
public:
    TestUnmergedNoAlert()
        : m_reporter()
        , m_logger(log4cplus::Logger::getInstance("test.dpm.merge"))
        , m_fits_buffer(4096u)
        , m_file("NAME",
                 "/foo/bar.fits",
                 "/location/file.fits",
                 {},
                 {},
                 false,
                 std::move(m_fits_buffer).GetOwnedFile()) {
    }
    void SetUp() {
        ASSERT_TRUE(m_file.GetFitsFile());
        ASSERT_FALSE(m_file.AlertUnmergeable());
    }
    MockReporter m_reporter;
    log4cplus::Logger m_logger;
    fits::MemoryFitsFile m_fits_buffer;
    FitsFileSource m_file;
};
 
TEST_F(TestUnmergedAlert, EmptyFitsProduceNoAlert) {
    fits::InitPrimaryHduNoImage(m_file.GetFitsFile());
    EXPECT_CALL(m_reporter, PostAlert(_, _)).Times(0);
    CheckUnmergedPrimaryHduData(m_logger, m_reporter, m_file);
}
 
TEST_F(TestUnmergedAlert, NonEmptyFitsProduceAlert) {
    int status = 0;
    long naxes[2] = {2, 2};
    fits_create_img(m_file.GetFitsFile(), BYTE_IMG, 2, naxes, &status);
    ASSERT_EQ(0, status);
 
    // Test
    EXPECT_CALL(m_reporter, PostAlert(_, MatchesRegex("Unmerged data: Source file has data.*")))
        .Times(1);
    CheckUnmergedPrimaryHduData(m_logger, m_reporter, m_file);
}
 
TEST_F(TestUnmergedNoAlert, NonEmptyFitsProduceNoAlertIfDisabled) {
    int status = 0;
    long naxes[2] = {2, 2};
    fits_create_img(m_file.GetFitsFile(), BYTE_IMG, 2, naxes, &status);
    ASSERT_EQ(0, status);
 
    // Test
    EXPECT_CALL(m_reporter, PostAlert(_, _)).Times(0);
    CheckUnmergedPrimaryHduData(m_logger, m_reporter, m_file);
}
 
}  // namespace daq::dpm::merge