47 lines
1.7 KiB
C++
47 lines
1.7 KiB
C++
// SPDX-FileCopyrightText: 2026 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
// Run with LD_PRELOAD=enospc_shim.so ./jfjoch_hdf5_enospc_test
|
|
|
|
#include <catch2/catch_all.hpp>
|
|
#include "../writer/HDF5Objects.h"
|
|
#include "../writer/FileWriter.h"
|
|
#include "../common/DiffractionExperiment.h"
|
|
|
|
TEST_CASE("HDF5File_enospc") {
|
|
auto file = std::make_unique<HDF5File>("enospc_test.h5");
|
|
|
|
std::vector<uint8_t> small_vector(2056), large_vector(40 * 1024 * 1024);
|
|
REQUIRE_NOTHROW(file->SaveVector("/small", small_vector));
|
|
REQUIRE_NOTHROW(file->SaveVector("/large1", large_vector));
|
|
REQUIRE_THROWS(file->SaveVector("/large2", large_vector));
|
|
REQUIRE_NOTHROW(file.reset());
|
|
// No leftover HDF5 objects
|
|
REQUIRE (H5Fget_obj_count(H5F_OBJ_ALL, H5F_OBJ_ALL) == 0);
|
|
}
|
|
|
|
TEST_CASE("FileWriter_enospc") {
|
|
{
|
|
RegisterHDF5Filter();
|
|
DiffractionExperiment x(DetJF4M());
|
|
|
|
x.FilePrefix("test02_1p10").ImagesPerTrigger(5).ImagesPerFile(2).Compression(CompressionAlgorithm::NO_COMPRESSION);
|
|
StartMessage start_message;
|
|
x.FillMessage(start_message);
|
|
|
|
FileWriter file_set(start_message);
|
|
std::vector<uint16_t> image(x.GetPixelsNum());
|
|
|
|
DataMessage message{};
|
|
message.image = CompressedImage(image, x.GetXPixelsNum(), x.GetYPixelsNum());
|
|
message.number = 0;
|
|
REQUIRE_NOTHROW(file_set.Write(message));
|
|
message.number = 1;
|
|
REQUIRE_THROWS(file_set.Write(message));
|
|
message.number = 2;
|
|
REQUIRE_THROWS(file_set.Write(message));
|
|
}
|
|
// No leftover HDF5 objects
|
|
REQUIRE (H5Fget_obj_count(H5F_OBJ_ALL, H5F_OBJ_ALL) == 0);
|
|
}
|