/home/martin/workspace/OpenStreetNav/src/test/psql.cpp

Go to the documentation of this file.

#include <boost/test/unit_test.hpp>

#include "../psql/psql.h"
#include "../sqllib/sqllib.h"
#include <tuple>

class PSqlFixture
{
public:
    psql::Database db;
    PSqlFixture():
        db("", true)
    {
        psql::Statement<psql::BindTypes<>, psql::RetTypes<> > st("CREATE SCHEMA testing", db);
        st.execute();
        st = psql::Statement<psql::BindTypes<>, psql::RetTypes<> >("SET search_path TO testing", db);
        st.execute();
    }
    ~PSqlFixture()
    {
        try
        {
            psql::Statement<psql::BindTypes<>, psql::RetTypes<> > st("DROP SCHEMA testing CASCADE", db);
            st.execute();
        }
        catch (psql::PgSqlException& ex)
        {
            std::cout << "WARNING: problem deleting testing schema " << ex.what() << std::endl;
        }
    }
};

BOOST_AUTO_TEST_SUITE(psql_test)

BOOST_AUTO_TEST_CASE(db)
{
    psql::Database db("");
    db.get_db();
}

BOOST_AUTO_TEST_CASE(sync)
{
    psql::Database db("", true);
    db.get_db();
}

BOOST_AUTO_TEST_CASE(dbfail)
{
    BOOST_CHECK_THROW(psql::Database db("asdf"), psql::PgSqlException);
}

BOOST_AUTO_TEST_CASE(dbfail2)
{
    BOOST_CHECK_THROW(psql::Database("user=asdfgasdafgasdghafasasa", true), psql::PgSqlException);
}

BOOST_AUTO_TEST_CASE(dbfailasync2)
{
    psql::Database db("user=asdfgasdafgasdghafasasa", false);
    BOOST_CHECK_THROW(db.get_db(), psql::PgSqlException);
}

BOOST_AUTO_TEST_SUITE_END()

BOOST_FIXTURE_TEST_SUITE(psql2, PSqlFixture)

BOOST_AUTO_TEST_CASE(statement)
{
    auto st1(sqllib::get_create_test_table(db));
    st1.execute();
    auto st2(sqllib::get_insert_test_table(db));
    st2.execute(32, "asdf", 52341093);
    auto st3(sqllib::get_test_select(db));
    st3.execute(32);
    auto res(st3.get_row(0));
    std::tuple<int, std::string, int64_t> tup(32,  "asdf", 52341093);
    BOOST_CHECK(res == tup);
}

BOOST_AUTO_TEST_CASE(named_statement)
{
    auto st1(sqllib::get_create_test_table(db, true, "asdf"));
    st1.execute();
    auto st2(sqllib::get_insert_test_table(db, true, "bvba"));
    st2.execute(32, "asdf", 52341093);
    auto st3(sqllib::get_test_select(db, true, "agsdf"));
    st3.execute(32);
    auto res(st3.get_row(0));
    std::tuple<int, std::string, int64_t> tup(32, "asdf", 52341093);
    BOOST_CHECK(res == tup);
}

BOOST_AUTO_TEST_CASE(utils)
{
    auto st1(sqllib::get_create_test_table(db));
    st1.execute();
    auto st2(sqllib::get_insert_test_table(db));
    st2.execute(32, "asdf", 52341093);
    auto st3(sqllib::get_test_select(db));
    auto ret = psql::exec_statement(st3, 32);
    std::tuple<int, std::string, int64_t> tup(32, "asdf", 52341093);
    std::vector<std::tuple<int, std::string, int64_t> > vect {tup};
    std::vector<int64_t> vect2 {52341093};
    auto ret2 = psql::exec_statement_col<2>(st3, 32);
    BOOST_CHECK(ret == vect);
    BOOST_CHECK(ret2 == vect2);
}

BOOST_AUTO_TEST_CASE(copy)
{
    auto st1(sqllib::get_create_test_table(db));
    st1.execute();
    auto st2(sqllib::get_copy_test_table(db));
    BOOST_CHECK(!st2.copying());
    st2.execute();
    BOOST_CHECK(st2.copying());
    st2.copy_data(32, "asdf", 52341093);
    st2.end_copy();
    BOOST_CHECK(!st2.copying());
    auto st3(sqllib::get_test_select(db));
    st3.execute(32);
    BOOST_CHECK(st3.row_count() == 1);
    auto res(st3.get_row(0));
    std::tuple<int, std::string, int64_t> tup(32,  "asdf", 52341093);
    BOOST_CHECK(res == tup);
}

BOOST_AUTO_TEST_CASE(cursor)
{
    auto st1(sqllib::get_create_test_table(db));
    st1.execute();
    auto st2(sqllib::get_insert_test_table(db));
    st2.execute(32, "asdf", 52341093);
    st2.execute(34, "asdfa", 52341093);
    st2.execute(35, "asdfb", 52341093);
    st2.execute(36, "asdfc", 52341093);
    st2.execute(37, "asdfd", 52341093);
    auto st3(sqllib::get_test_select_gt(db));
    db.begin_transaction();
    try
    {
        auto curs = psql::make_cursor(db, "test_crs", st3);
        curs.open(34);
        curs.fetch(3);
        std::vector<std::tuple<int, std::string, int64_t> > exp
        {
            std::make_tuple(35, "asdfb", 52341093),
            std::make_tuple(36, "asdfc", 52341093),
            std::make_tuple(37, "asdfd", 52341093)
        };
        BOOST_CHECK(curs.get_buffer() == exp);
        curs.close();
        db.rollback_transaction();
    }
    catch (...)
    {
        try
        {
            db.rollback_transaction();
        }
        catch (...)
        {
            std::cout << "WARNING: Failed to rollback" << std::endl;
        }
        throw;
    }
}

BOOST_AUTO_TEST_SUITE_END()