VladPolskiy/osm2pgsql
1
0
Fork 0
mirror of https://github.com/osm2pgsql-dev/osm2pgsql.git synced 2025-07-28 04:21:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

move new tests into place

Browse Source
This commit is contained in:
Sarah Hoffmann
2019-11-05 23:32:37 +01:00
parent 8678be3374
commit 8859b6eb8e
65 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

139
tests/test-db-copy-thread.cpp Normal file
View File

@ -0,0 +1,139 @@
#include "catch.hpp"
#include "db-copy.hpp"
#include "common-pg.hpp"
static pg::tempdb_t db;
static int table_count(pg::conn_t const &conn, std::string const &where = "")
{
return conn.require_scalar<int>("SELECT count(*) FROM test_copy_thread " +
where);
}
TEST_CASE("db_copy_thread_t")
{
auto conn = db.connect();
conn.exec("DROP TABLE IF EXISTS test_copy_thread");
conn.exec("CREATE TABLE test_copy_thread (id int8)");
auto table = std::make_shared<db_target_descr_t>();
table->name = "test_copy_thread";
table->id = "id";
db_copy_thread_t t(db.conninfo());
auto cmd = std::unique_ptr<db_cmd_copy_t>(new db_cmd_copy_t(table));
SECTION("simple copy command")
{
SECTION("add one copy line and sync")
{
cmd->buffer += "42\n";
t.add_buffer(std::unique_ptr<db_cmd_t>(cmd.release()));
t.sync_and_wait();
REQUIRE(conn.require_scalar<int>(
"SELECT id FROM test_copy_thread") == 42);
}
SECTION("add multiple rows and sync")
{
cmd->buffer += "101\n 23\n 900\n";
t.add_buffer(std::unique_ptr<db_cmd_t>(cmd.release()));
t.sync_and_wait();
REQUIRE(table_count(conn) == 3);
}
SECTION("add one line and finish")
{
cmd->buffer += "2\n";
t.add_buffer(std::unique_ptr<db_cmd_t>(cmd.release()));
t.finish();
REQUIRE(conn.require_scalar<int>(
"SELECT id FROM test_copy_thread") == 2);
}
}
SECTION("delete command")
{
cmd->buffer += "42\n43\n133\n223\n224\n";
t.add_buffer(std::unique_ptr<db_cmd_t>(cmd.release()));
t.sync_and_wait();
cmd = std::unique_ptr<db_cmd_copy_t>(new db_cmd_copy_t(table));
SECTION("simple delete of existing rows")
{
cmd->deletables.push_back(223);
cmd->deletables.push_back(42);
t.add_buffer(std::unique_ptr<db_cmd_t>(cmd.release()));
t.sync_and_wait();
REQUIRE(table_count(conn, "WHERE id = 42") == 0);
REQUIRE(table_count(conn, "WHERE id = 223") == 0);
}
SECTION("delete one and add another")
{
cmd->deletables.push_back(133);
cmd->buffer += "134\n";
t.add_buffer(std::unique_ptr<db_cmd_t>(cmd.release()));
t.sync_and_wait();
REQUIRE(table_count(conn, "WHERE id = 133") == 0);
REQUIRE(table_count(conn, "WHERE id = 134") == 1);
}
SECTION("delete one and add the same")
{
cmd->deletables.push_back(133);
cmd->buffer += "133\n";
t.add_buffer(std::unique_ptr<db_cmd_t>(cmd.release()));
t.sync_and_wait();
REQUIRE(table_count(conn, "WHERE id = 133") == 1);
}
}
SECTION("multi buffer add without delete")
{
cmd->buffer += "542\n5543\n10133\n";
t.add_buffer(std::unique_ptr<db_cmd_t>(cmd.release()));
cmd = std::unique_ptr<db_cmd_copy_t>(new db_cmd_copy_t(table));
cmd->buffer += "12\n784\n523\n";
t.add_buffer(std::unique_ptr<db_cmd_t>(cmd.release()));
t.finish();
REQUIRE(table_count(conn) == 6);
REQUIRE(table_count(conn, "WHERE id = 10133") == 1);
REQUIRE(table_count(conn, "WHERE id = 523") == 1);
}
SECTION("multi buffer add with delete")
{
cmd->buffer += "542\n5543\n10133\n";
t.add_buffer(std::unique_ptr<db_cmd_t>(cmd.release()));
cmd = std::unique_ptr<db_cmd_copy_t>(new db_cmd_copy_t(table));
cmd->deletables.push_back(542);
cmd->buffer += "12\n";
t.add_buffer(std::unique_ptr<db_cmd_t>(cmd.release()));
t.finish();
REQUIRE(table_count(conn) == 3);
REQUIRE(table_count(conn, "WHERE id = 542") == 0);
REQUIRE(table_count(conn, "WHERE id = 12") == 1);
}
}