Function TestPostgreSQLDatabase

IO/PostgreSQL/Testing/Cxx/TestPostgreSQLDatabase.cxx:23–492

#include <iostream>

int TestPostgreSQLDatabase(int /*argc*/, char* /*argv*/[])
{
  // This test requires the user in VTK_PSQL_TEST_URL to have
  // permission to create and drop the database named in that URL
  // as well as tables in that database. That user must also be
  // able to connect to the "template1" database (which initdb
  // creates and should be present on all systems -- we do NOT
  // support non-standard configurations where this is not true).
  vtkPostgreSQLDatabase* db =
    vtkPostgreSQLDatabase::SafeDownCast(vtkSQLDatabase::CreateFromURL(VTK_PSQL_TEST_URL));
  std::string realDatabase = db->GetDatabaseName();
  db->SetDatabaseName("template1"); // This is guaranteed to exist
  bool status = db->Open();
  if (!status)
  {
    std::cerr << "Couldn't open database.\nError message: \"" << db->GetLastErrorText() << "\"\n";
    db->Delete();
    return 1;
  }

  vtkStringArray* dbNames = db->GetDatabases();
  std::cout << "Database list:\n";
  for (int dbi = 0; dbi < dbNames->GetNumberOfValues(); ++dbi)
  {
    std::cout << "+ " << dbNames->GetValue(dbi) << std::endl;
  }
  dbNames->Delete();
  if (!db->CreateDatabase(realDatabase.c_str(), true))
  {
    std::cerr << "Error: " << db->GetLastErrorText() << std::endl;
  }

  vtkSQLQuery* query = db->GetQueryInstance();

  // Force a database connection close
  // This also forces us to connect to the database named in the test URL.
  std::string fauxDatabase = realDatabase + "blarney";
  db->SetDatabaseName(fauxDatabase.c_str());
  db->SetDatabaseName(realDatabase.c_str());

  if (!db->Open())
  {
    std::cerr << "Error: " << db->GetLastErrorText() << std::endl;
  }

  // Test that bad queries fail without segfaulting...
  std::string dropQuery("DROP TABLE people");
  std::cout << dropQuery << std::endl;
  query->SetQuery(dropQuery.c_str());
  if (!query->Execute())
  {
    std::cout << "Drop query did not succeed (this result *** was *** expected). The last message: "
              << std::endl;
    std::cout << "   " << query->GetLastErrorText() << std::endl;
  }
  else
  {
    std::cerr << "The query \"DROP TABLE people\" succeeded when it should not have.\n";