From: Stephen Morris <stephen@isc.org>
Date: Wed, 24 Oct 2012 21:14:42 +0000 (+0100)
Subject: [2342] Extend MySQL tests
X-Git-Tag: trac2487_base~1^2~47
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=89c8076272fb81a9e49b3bdbcd704103707db83f;p=thirdparty%2Fkea.git

[2342] Extend MySQL tests

Clear up after every test and check that commit works.
---

diff --git a/src/lib/dhcp/mysql_lease_mgr.cc b/src/lib/dhcp/mysql_lease_mgr.cc
index e14630b5e8..6627dfb235 100644
--- a/src/lib/dhcp/mysql_lease_mgr.cc
+++ b/src/lib/dhcp/mysql_lease_mgr.cc
@@ -90,6 +90,7 @@ public:
         memset(bind_, 0, sizeof(bind_));
 
         // address: varchar(40)
+
         addr6_ = lease_->addr_.toText();
         addr6_length_ = addr6_.size();
 
diff --git a/src/lib/dhcp/tests/mysql_lease_mgr_unittest.cc b/src/lib/dhcp/tests/mysql_lease_mgr_unittest.cc
index be67866e1e..b267bc3dd9 100644
--- a/src/lib/dhcp/tests/mysql_lease_mgr_unittest.cc
+++ b/src/lib/dhcp/tests/mysql_lease_mgr_unittest.cc
@@ -30,7 +30,10 @@ using namespace std;
 
 namespace {
 
-// Connection strings
+// Connection strings.  Assume:
+// Database: keatest
+// Username: keatest
+// Password: keatest
 const char* VALID_TYPE = "type=mysql";
 const char* INVALID_TYPE = "type=unknown";
 const char* VALID_NAME = "name=keatest";
@@ -84,11 +87,31 @@ string connectionString(const char* type, const char* name, const char* host,
 }
 
 // Return valid connection string
-string validConnectionString() {
+string
+validConnectionString() {
     return (connectionString(VALID_TYPE, VALID_NAME, VALID_HOST,
                              VALID_USER, VALID_PASSWORD));
 }
 
+// Clear everything from the database tables
+void
+clearAll() {
+        // Initialise
+        MYSQL handle;
+        (void) mysql_init(&handle);
+
+        // Open database
+        (void) mysql_real_connect(&handle, "localhost", "keatest", "keatest",
+                                  "keatest", 0, NULL, 0);
+
+        // Clear the database
+        (void) mysql_query(&handle, "DELETE FROM lease4");
+        (void) mysql_query(&handle, "DELETE FROM lease6");
+
+        // ... and close
+        (void) mysql_close(&handle);
+}
+
 /// @brief Test Fixture Class
 ///
 /// Opens the database prior to each test and closes it afterwards.
@@ -98,9 +121,19 @@ class MySqlLeaseMgrTest : public ::testing::Test {
 public:
     /// @brief Constructor
     ///
-    /// Open the database.
-
+    /// Deletes everything from the database and opens it.
     MySqlLeaseMgrTest() {
+        clearAll();
+        LeaseMgrFactory::create(validConnectionString());
+        lmptr_ = &(LeaseMgrFactory::instance());
+    }
+
+    /// @brief Reopen the database
+    ///
+    /// Closes the database and re-open it.  Anything committed should be
+    /// visible.
+    void reopen() {
+        LeaseMgrFactory::destroy();
         LeaseMgrFactory::create(validConnectionString());
         lmptr_ = &(LeaseMgrFactory::instance());
     }
@@ -108,11 +141,12 @@ public:
     /// @brief Destructor
     ///
     /// Rolls back all pending transactions.  The deletion of the
-    /// lmptr_ member variable will close the database.
-
+    /// lmptr_ member variable will close the database.  Then
+    /// reopen it and delete everything created by the test.
     virtual ~MySqlLeaseMgrTest() {
         lmptr_->rollback();
         LeaseMgrFactory::destroy();
+        clearAll();
     }
 
     LeaseMgr*   lmptr_;         // Pointer to the lease manager
@@ -284,16 +318,42 @@ TEST_F(MySqlLeaseMgrTest, BasicLease6) {
     l2->cltt_ = 234567;         // Current time of day
     l2->subnet_id_ = l1->subnet_id_;    // Same as l1
 
+    const IOAddress L3_ADDRESS(std::string("2001:db8::3"));
+    Lease6Ptr l3(new Lease6());
+    initializeUnusedLease6(l3);
+
+    l3->type_ = Lease6::LEASE_IA_NA;
+    l3->addr_ = L3_ADDRESS;
+    l3->prefixlen_ = 28;
+    l3->iaid_ = 0xfffffffe;
+    vector<uint8_t> duid;
+    for (uint8_t i = 0; i < 128; ++i) {
+        duid.push_back(i + 5);
+    }
+    l3->duid_ = boost::shared_ptr<DUID>(new DUID(duid));
+    l3->preferred_lft_ = 0xfffffffc;    // Preferred lifetime
+    l3->valid_lft_ = 0xfffffffd;        // Actual lifetime
+    l3->cltt_ = 234567;                 // Current time of day
+    l3->subnet_id_ = l1->subnet_id_;    // Same as l1
+
     // Sanity check that the leases are different
     ASSERT_TRUE(*l1 != *l2);
+    ASSERT_TRUE(*l1 != *l3);
+    ASSERT_TRUE(*l2 != *l3);
 
-    // Start the tests.  Add two leases to the database, read them back and
+    // Start the tests.  Add three leases to the database, read them back and
     // check they are what we think they are.
     Lease6Ptr l_returned;
 
     EXPECT_TRUE(lmptr_->addLease(l1));
     EXPECT_TRUE(lmptr_->addLease(l2));
+    EXPECT_TRUE(lmptr_->addLease(l3));
+    lmptr_->commit();
+
+    // Reopen the database to ensure that they actually got stored.
+    reopen();
 
+    // check that the values returned are as expected.
     l_returned = lmptr_->getLease6(L1_ADDRESS);
     EXPECT_TRUE(l_returned);
     detailCompareLease6(l1, l_returned);
@@ -302,6 +362,10 @@ TEST_F(MySqlLeaseMgrTest, BasicLease6) {
     EXPECT_TRUE(l_returned);
     detailCompareLease6(l2, l_returned);
 
+    l_returned = lmptr_->getLease6(L3_ADDRESS);
+    EXPECT_TRUE(l_returned);
+    detailCompareLease6(l3, l_returned);
+
     // Check that we can't add a second lease with the same address
     EXPECT_FALSE(lmptr_->addLease(l1));