osm2pgsql/tests/test-output-pgsql-schema.cpp

71 lines
2.5 KiB
C++

/**
* SPDX-License-Identifier: GPL-2.0-or-later
*
* This file is part of osm2pgsql (https://osm2pgsql.org/).
*
* Copyright (C) 2006-2024 by the osm2pgsql developer community.
* For a full list of authors see the git log.
*/
#include <catch.hpp>
#include "common-import.hpp"
#include "common-options.hpp"
namespace {
testing::db::import_t db;
} // anonymous namespace
TEST_CASE("schema separation")
{
{
auto conn = db.db().connect();
conn.exec("CREATE SCHEMA IF NOT EXISTS myschema;"
"CREATE TABLE myschema.osm2pgsql_test_point (id bigint);"
"CREATE TABLE myschema.osm2pgsql_test_line (id bigint);"
"CREATE TABLE myschema.osm2pgsql_test_polygon (id bigint);"
"CREATE TABLE myschema.osm2pgsql_test_roads (id bigint)");
}
REQUIRE_NOTHROW(
db.run_file(testing::opt_t().slim(), "test_output_pgsql_z_order.osm"));
auto conn = db.db().connect();
conn.require_has_table("public.osm2pgsql_test_point");
conn.require_has_table("public.osm2pgsql_test_line");
conn.require_has_table("public.osm2pgsql_test_polygon");
conn.require_has_table("public.osm2pgsql_test_roads");
conn.require_has_table("myschema.osm2pgsql_test_point");
conn.require_has_table("myschema.osm2pgsql_test_line");
conn.require_has_table("myschema.osm2pgsql_test_polygon");
conn.require_has_table("myschema.osm2pgsql_test_roads");
REQUIRE(2 == conn.get_count("osm2pgsql_test_point"));
REQUIRE(11 == conn.get_count("osm2pgsql_test_line"));
REQUIRE(1 == conn.get_count("osm2pgsql_test_polygon"));
REQUIRE(8 == conn.get_count("osm2pgsql_test_roads"));
REQUIRE(0 == conn.get_count("myschema.osm2pgsql_test_point"));
REQUIRE(0 == conn.get_count("myschema.osm2pgsql_test_line"));
REQUIRE(0 == conn.get_count("myschema.osm2pgsql_test_polygon"));
REQUIRE(0 == conn.get_count("myschema.osm2pgsql_test_roads"));
}
TEST_CASE("liechtenstein slim with schema")
{
options_t options = testing::opt_t().slim();
options.output_dbschema = "myschema";
auto conn = db.db().connect();
conn.exec("CREATE SCHEMA IF NOT EXISTS myschema;");
REQUIRE_NOTHROW(db.run_file(options, "liechtenstein-2013-08-03.osm.pbf"));
REQUIRE(1342 == conn.get_count("myschema.osm2pgsql_test_point"));
REQUIRE(3231 == conn.get_count("myschema.osm2pgsql_test_line"));
REQUIRE(375 == conn.get_count("myschema.osm2pgsql_test_roads"));
REQUIRE(4130 == conn.get_count("myschema.osm2pgsql_test_polygon"));
}