testDBConnection.cc

Go to the documentation of this file.

#include <utl/Reader.h>
#include <cppunit/extensions/HelperMacros.h>
#include <mysql/mysql.h>
#include <string>
#include <iostream>

using namespace std;
using namespace utl;


class DBConnectionTest : public CppUnit::TestFixture {

  CPPUNIT_TEST_SUITE(DBConnectionTest);
  CPPUNIT_TEST(testMasterConnection);
  CPPUNIT_TEST_SUITE_END();

public:
  virtual ~DBConnectionTest() { }

  void setUp() override { }

  void tearDown() override { }

  void
  testMasterConnection()
  {
    Reader reader(MASTERDBLIST, Reader::eSCHEMA);

    const string database = "Atm_Aerosol_1_A";

    bool connectionPresent = false;

    // top branch = "masterDBList"
    for (Branch dbB = reader.GetTopBranch().GetFirstChild();
         dbB; dbB = dbB.GetNextSibling()) {

      const auto servers = dbB.GetChild("server").Get<vector<string>>();

      for (const auto& ser : servers) {

        cerr << "Testing connection to " << ser << endl;

        const auto userName = dbB.GetChild("userName").Get<string>();
        const auto password = dbB.GetChild("password").Get<string>();
        const auto port = dbB.GetChild("port").Get<unsigned int>();

        MYSQL* const mySQL = mysql_init(nullptr);
        CPPUNIT_ASSERT(mySQL);

        if (mysql_real_connect(mySQL, ser.c_str(), userName.c_str(), password.c_str(),
                               database.c_str(), port, nullptr, 0)) {
          connectionPresent = true;
          mysql_close(mySQL);
          break;
        } else {
          cerr << "Connection to the database '" << database << "' at "
               << userName << ':' << password << '@' << ser << ':' << port << " "
                  "failed. Error message: \"" << mysql_error(mySQL) << '"' << endl;
        }

        mysql_close(mySQL);

      }

      if (connectionPresent)
        break;

    }

    CPPUNIT_ASSERT(connectionPresent);
  }

};


CPPUNIT_TEST_SUITE_REGISTRATION(DBConnectionTest);