forked from osm2pgsql-dev/osm2pgsql
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest-output-multi-point-multi-table.cpp
More file actions
115 lines (89 loc) · 3.29 KB
/
Copy pathtest-output-multi-point-multi-table.cpp
File metadata and controls
115 lines (89 loc) · 3.29 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
#include <iostream>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <cassert>
#include <sstream>
#include <stdexcept>
#include <memory>
#include "osmtypes.hpp"
#include "osmdata.hpp"
#include "output-multi.hpp"
#include "options.hpp"
#include "middle-pgsql.hpp"
#include "taginfo_impl.hpp"
#include <sys/types.h>
#include <unistd.h>
#include <boost/lexical_cast.hpp>
#include "tests/middle-tests.hpp"
#include "tests/common-pg.hpp"
#include "tests/common.hpp"
int main(int argc, char *argv[]) {
(void)argc;
(void)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.database_options = db->database_options;
options.num_procs = 1;
options.prefix = "osm2pgsql_test";
options.slim = true;
export_list columns;
{
taginfo info;
info.name = "amenity";
info.type = "text";
columns.add(osmium::item_type::node, info);
}
std::shared_ptr<middle_t> mid_pgsql(new middle_pgsql_t(&options));
mid_pgsql->start();
auto midq = mid_pgsql->get_query_instance(mid_pgsql);
std::vector<std::shared_ptr<output_t> > outputs;
// let's make lots of tables!
for (int i = 0; i < 10; ++i) {
std::string name = (boost::format("foobar_%d") % i).str();
std::shared_ptr<geometry_processor> processor =
geometry_processor::create("point", &options);
auto out_test = std::make_shared<output_multi_t>(
name, processor, columns, midq, options);
outputs.push_back(out_test);
}
osmdata_t osmdata(mid_pgsql, outputs);
testing::parse("tests/liechtenstein-2013-08-03.osm.pbf", "pbf",
options, &osmdata);
for (int i = 0; i < 10; ++i) {
std::string name = (boost::format("foobar_%d") % i).str();
db->check_count(1,
(boost::format("select count(*) from pg_catalog.pg_class "
"where relname = 'foobar_%d'")
% i).str());
db->check_count(244,
(boost::format("select count(*) from foobar_%d")
% i).str());
db->check_count(36,
(boost::format("select count(*) from foobar_%d "
"where amenity='parking'")
% i).str());
db->check_count(34,
(boost::format("select count(*) from foobar_%d "
"where amenity='bench'")
% i).str());
db->check_count(1,
(boost::format("select count(*) from foobar_%d "
"where amenity='vending_machine'")
% i).str());
}
return 0;
} catch (const std::exception &e) {
std::cerr << "ERROR: " << e.what() << std::endl;
} catch (...) {
std::cerr << "UNKNOWN ERROR" << std::endl;
}
return 1;
}