d516e1c736
The latest releases have some critical fixes, see the changelog: https://github.com/osmcode/libosmium/blob/v2.5.4/CHANGELOG.md Merge commit 'afdf8e7b21fbaf597e91d9d8a7542635e60ee9a1' into use_libosmium_2_5_4
118 lines
3.6 KiB
C++
118 lines
3.6 KiB
C++
#include "catch.hpp"
|
|
#include "utils.hpp"
|
|
|
|
#include <algorithm>
|
|
|
|
#include <osmium/io/any_compression.hpp>
|
|
#include <osmium/io/xml_input.hpp>
|
|
#include <osmium/io/xml_output.hpp>
|
|
#include <osmium/io/output_iterator.hpp>
|
|
#include <osmium/memory/buffer.hpp>
|
|
|
|
TEST_CASE("Writer") {
|
|
|
|
osmium::io::Header header;
|
|
header.set("generator", "test_writer.cpp");
|
|
|
|
osmium::io::Reader reader(with_data_dir("t/io/data.osm"));
|
|
osmium::memory::Buffer buffer = reader.read();
|
|
REQUIRE(buffer);
|
|
REQUIRE(buffer.committed() > 0);
|
|
auto num = std::distance(buffer.cbegin<osmium::OSMObject>(), buffer.cend<osmium::OSMObject>());
|
|
REQUIRE(num > 0);
|
|
REQUIRE(buffer.cbegin<osmium::OSMObject>()->id() == 1);
|
|
|
|
std::string filename;
|
|
|
|
SECTION("Empty writes") {
|
|
|
|
SECTION("Empty buffer") {
|
|
filename = "test-writer-out-empty-buffer.osm";
|
|
osmium::io::Writer writer(filename, header, osmium::io::overwrite::allow);
|
|
osmium::memory::Buffer empty_buffer(1024);
|
|
writer(std::move(empty_buffer));
|
|
writer.close();
|
|
}
|
|
|
|
SECTION("Invalid buffer") {
|
|
filename = "test-writer-out-invalid-buffer.osm";
|
|
osmium::io::Writer writer(filename, header, osmium::io::overwrite::allow);
|
|
osmium::memory::Buffer invalid_buffer;
|
|
writer(std::move(invalid_buffer));
|
|
writer.close();
|
|
}
|
|
|
|
osmium::io::Reader reader_check(filename);
|
|
osmium::memory::Buffer buffer_check = reader_check.read();
|
|
REQUIRE(!buffer_check);
|
|
}
|
|
|
|
SECTION("Successfull writes") {
|
|
|
|
SECTION("Writer buffer") {
|
|
filename = "test-writer-out-buffer.osm";
|
|
osmium::io::Writer writer(filename, header, osmium::io::overwrite::allow);
|
|
writer(std::move(buffer));
|
|
writer.close();
|
|
|
|
REQUIRE_THROWS_AS({
|
|
writer(osmium::memory::Buffer{});
|
|
}, osmium::io_error);
|
|
}
|
|
|
|
SECTION("Writer item") {
|
|
filename = "test-writer-out-item.osm";
|
|
osmium::io::Writer writer(filename, header, osmium::io::overwrite::allow);
|
|
for (const auto& item : buffer) {
|
|
writer(item);
|
|
}
|
|
writer.close();
|
|
}
|
|
|
|
SECTION("Writer output iterator") {
|
|
filename = "test-writer-out-iterator.osm";
|
|
osmium::io::Writer writer(filename, header, osmium::io::overwrite::allow);
|
|
auto it = osmium::io::make_output_iterator(writer);
|
|
std::copy(buffer.cbegin(), buffer.cend(), it);
|
|
writer.close();
|
|
}
|
|
|
|
osmium::io::Reader reader_check(filename);
|
|
osmium::memory::Buffer buffer_check = reader_check.read();
|
|
REQUIRE(buffer_check);
|
|
REQUIRE(buffer_check.committed() > 0);
|
|
REQUIRE(std::distance(buffer_check.cbegin<osmium::OSMObject>(), buffer_check.cend<osmium::OSMObject>()) == num);
|
|
REQUIRE(buffer_check.cbegin<osmium::OSMObject>()->id() == 1);
|
|
|
|
}
|
|
|
|
SECTION("Interrupted write") {
|
|
|
|
int error = 0;
|
|
try {
|
|
|
|
SECTION("fail after open") {
|
|
filename = "test-writer-out-fail1.osm";
|
|
osmium::io::Writer writer(filename, header, osmium::io::overwrite::allow);
|
|
throw 1;
|
|
}
|
|
|
|
SECTION("fail after write") {
|
|
filename = "test-writer-out-fail2.osm";
|
|
osmium::io::Writer writer(filename, header, osmium::io::overwrite::allow);
|
|
writer(std::move(buffer));
|
|
throw 2;
|
|
}
|
|
|
|
} catch (int e) {
|
|
error = e;
|
|
}
|
|
|
|
REQUIRE(error > 0);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|