mirror of
https://github.com/osm2pgsql-dev/osm2pgsql.git
synced 2025-08-19 16:28:16 +00:00
111 lines
3.0 KiB
C++
111 lines
3.0 KiB
C++
#include <iostream>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <cassert>
|
|
#include <sstream>
|
|
#include <stdexcept>
|
|
#include <memory>
|
|
|
|
#include "osmtypes.hpp"
|
|
#include "output-null.hpp"
|
|
#include "options.hpp"
|
|
#include "middle-pgsql.hpp"
|
|
|
|
#include <libpq-fe.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include "tests/middle-tests.hpp"
|
|
#include "tests/common-pg.hpp"
|
|
|
|
/* This is basically the same as test-middle-pgsql, but with flat nodes. */
|
|
|
|
void run_tests(options_t options, const std::string cache_type) {
|
|
options.append = false;
|
|
options.create = true;
|
|
options.flat_node_cache_enabled = true;
|
|
// flat nodes truncates the file each time it's started, so we can reuse the same file
|
|
options.flat_node_file = boost::optional<std::string>("tests/test_middle_flat.flat.nodes.bin");
|
|
|
|
{
|
|
middle_pgsql_t mid_pgsql;
|
|
output_null_t out_test(&mid_pgsql, options);
|
|
|
|
mid_pgsql.start(&options);
|
|
|
|
if (test_node_set(&mid_pgsql) != 0) { throw std::runtime_error("test_node_set failed."); }
|
|
|
|
mid_pgsql.commit();
|
|
mid_pgsql.stop();
|
|
}
|
|
{
|
|
middle_pgsql_t mid_pgsql;
|
|
output_null_t out_test(&mid_pgsql, options);
|
|
|
|
mid_pgsql.start(&options);
|
|
|
|
if (test_nodes_comprehensive_set(&mid_pgsql) != 0) { throw std::runtime_error("test_nodes_comprehensive_set failed."); }
|
|
|
|
mid_pgsql.commit();
|
|
mid_pgsql.stop();
|
|
}
|
|
/* This should work, but doesn't. More tests are needed that look at updates
|
|
without the complication of ways.
|
|
*/
|
|
/* {
|
|
middle_pgsql_t mid_pgsql;
|
|
output_null_t out_test(&mid_pgsql, options);
|
|
|
|
mid_pgsql.start(&options);
|
|
mid_pgsql.commit();
|
|
mid_pgsql.stop();
|
|
// Switch to append mode because this tests updates
|
|
options.append = true;
|
|
options.create = false;
|
|
mid_pgsql.start(&options);
|
|
if (test_way_set(&mid_pgsql) != 0) { throw std::runtime_error("test_way_set failed."); }
|
|
|
|
mid_pgsql.commit();
|
|
mid_pgsql.stop();
|
|
}*/
|
|
}
|
|
int main(int argc, char *argv[]) {
|
|
std::unique_ptr<pg::tempdb> db;
|
|
|
|
try {
|
|
db.reset(new pg::tempdb);
|
|
} catch (const std::exception &e) {
|
|
std::cerr << "Unable to setup database: " << e.what() << "\n";
|
|
return 77; // <-- code to skip this test.
|
|
}
|
|
|
|
try {
|
|
options_t options;
|
|
options.conninfo = db->conninfo().c_str();
|
|
options.scale = 10000000;
|
|
options.cache = 1;
|
|
options.num_procs = 1;
|
|
options.prefix = "osm2pgsql_test";
|
|
options.slim = true;
|
|
|
|
options.alloc_chunkwise = ALLOC_SPARSE | ALLOC_DENSE; // what you get with optimized
|
|
run_tests(options, "optimized");
|
|
options.alloc_chunkwise = ALLOC_SPARSE;
|
|
run_tests(options, "sparse");
|
|
|
|
options.alloc_chunkwise = ALLOC_DENSE;
|
|
run_tests(options, "dense");
|
|
|
|
options.alloc_chunkwise = ALLOC_DENSE | ALLOC_DENSE_CHUNK; // what you get with chunk
|
|
run_tests(options, "chunk");
|
|
} catch (const std::exception &e) {
|
|
std::cerr << "ERROR: " << e.what() << std::endl;
|
|
return 1;
|
|
} catch (...) {
|
|
std::cerr << "UNKNOWN ERROR" << std::endl;
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|