ppf: Add tests for date_time methods.

diff --git a/ppf/db/tests/test_access_sqlite.py b/ppf/db/tests/test_access_sqlite.py
index 76d7a67..3a4bb3e 100644 (file)
--- a/ppf/db/tests/test_access_sqlite.py
+++ b/ppf/db/tests/test_access_sqlite.py
@@ -8,6 +8,7 @@ import unittest
 import os
 import sqlite3
 import access
+import datetime
 
 sql_create_script = """
 DROP TABLE IF EXISTS status_messages;
@@ -260,6 +261,46 @@ class AccessTest(unittest.TestCase):
         # Increment number of entries in client_sessions table.
         self.expected_sessions_count = self.expected_sessions_count + 2
 
+    def insert_test_client_sessions_date_time(self):
+        """
+        Insert test entries through insert_client_sessions_date_time().
+        """
+        # First insert appropiate swarm entries.
+        self.insert_test_swarms()
+
+        # Both client sessions are part of the same swarm and
+        # use the same client.
+        self.dba.insert_client_sessions_date_time('3', '5', 'Linux',
+                '2.6.26', '512', '1500', '141.85.224.209', '50500',
+                '512', '64', '2010-08-20', '15:34:23')
+        self.dba.insert_client_sessions_date_time('3', '5', 'Linux',
+                '2.6.26', '512', '1500', '141.85.224.202', '50500',
+                '512', '64', '2010-10-02', '09:23:23')
+
+        # Increment number of entries in client_sessions table.
+        self.expected_sessions_count = self.expected_sessions_count + 2
+
+    def insert_test_client_sessions_datetime(self):
+        """
+        Insert test entries through insert_client_sessions_datetime().
+        """
+        # First insert appropiate swarm entries.
+        self.insert_test_swarms()
+
+        # Both client sessions are part of the same swarm and
+        # use the same client.
+        dt = datetime.datetime(2010, 8, 20, 15, 34, 23)
+        self.dba.insert_client_sessions_datetime('3', '5', 'Linux',
+                '2.6.26', '512', '1500', '141.85.224.209', '50500',
+                '512', '64', dt)
+        dt = datetime.datetime(2010, 10, 2, 9, 23, 23)
+        self.dba.insert_client_sessions_datetime('3', '5', 'Linux',
+                '2.6.26', '512', '1500', '141.85.224.202', '50500',
+                '512', '64', dt)
+
+        # Increment number of entries in client_sessions table.
+        self.expected_sessions_count = self.expected_sessions_count + 2
+
     def insert_test_status_messages(self):
         """ Insert test entries through insert_status_messages(). """
         # First insert appropiate swarm entries.
@@ -288,6 +329,40 @@ class AccessTest(unittest.TestCase):
         # Increment number of entries in status_messages table.
         self.expected_statmsg_count = self.expected_statmsg_count + 2
 
+    def insert_test_status_messages_date_time(self):
+        """
+        Insert test entries through insert_status_messages_date_time().
+        """
+        # First insert appropiate swarm entries.
+        self.insert_test_client_sessions()
+
+        # Both entries are part of client session 3.
+        self.dba.insert_status_messages_date_time('3', '2010-09-13',
+            '03:23:43', '222', '0', '213', '56', '200', '300', '121.324');
+        self.dba.insert_status_messages_date_time('3', '2010-10-23',
+            '16:32:12', '223', '0', '213', '56', '300', '400', '121.324');
+
+        # Increment number of entries in status_messages table.
+        self.expected_statmsg_count = self.expected_statmsg_count + 2
+
+    def insert_test_status_messages_datetime(self):
+        """
+        Insert test entries through insert_status_messages_datetime().
+        """
+        # First insert appropiate swarm entries.
+        self.insert_test_client_sessions()
+
+        # Both entries are part of client session 3.
+        dt = datetime.datetime(2010, 9, 13, 3, 23, 43)
+        self.dba.insert_status_messages_datetime('3', dt, '222', '0',
+                '213', '56', '200', '300', '121.324');
+        dt = datetime.datetime(2010, 10, 23, 16, 32, 12)
+        self.dba.insert_status_messages_datetime('3', dt, '223', '0',
+                '213', '56', '300', '400', '121.324');
+
+        # Increment number of entries in status_messages table.
+        self.expected_statmsg_count = self.expected_statmsg_count + 2
+
     def insert_test_verbose_messages(self):
         """ Insert test entries through insert_verbose_messages(). """
         # First insert appropiate client sessions entries.
@@ -316,6 +391,40 @@ class AccessTest(unittest.TestCase):
         # Increment number of entries in verbose_messsages table.
         self.expected_verbmsg_count = self.expected_verbmsg_count + 2
 
+    def insert_test_verbose_messages_date_time(self):
+        """
+        Insert test entries through insert_verbose_messages_date_time().
+        """
+        # First insert appropiate client sessions entries.
+        self.insert_test_client_sessions()
+
+        # Both entries are part of client session 3.
+        self.dba.insert_verbose_messages_date_time('3', '2011-07-20',
+            '12:01:02', '0', '127.0.0.1', '1345', '0', '3', '4', '13', '777');
+        self.dba.insert_verbose_messages_date_time('3', '2011-05-28',
+            '14:20:32', '0', '127.0.1.1', '1345', '0', '3', '4', '13', '778');
+
+        # Increment number of entries in verbose_messsages table.
+        self.expected_verbmsg_count = self.expected_verbmsg_count + 2
+
+    def insert_test_verbose_messages_datetime(self):
+        """
+        Insert test entries through insert_verbose_messages_datetime().
+        """
+        # First insert appropiate client sessions entries.
+        self.insert_test_client_sessions()
+
+        # Both entries are part of client session 3.
+        dt = datetime.datetime(2011, 7, 20, 12, 1, 2)
+        self.dba.insert_verbose_messages_datetime('3', dt, '0',
+                '127.0.0.1', '1345', '0', '3', '4', '13', '777');
+        dt = datetime.datetime(2011, 5, 28, 14, 20, 32)
+        self.dba.insert_verbose_messages_datetime('3', dt, '0',
+                '127.0.1.1', '1345', '0', '3', '4', '13', '778');
+
+        # Increment number of entries in verbose_messsages table.
+        self.expected_verbmsg_count = self.expected_verbmsg_count + 2
+
     def test_connect(self):
         """ If connection is successful, no exception is raised. """
         try:
@@ -503,6 +612,30 @@ class AccessTest(unittest.TestCase):
 
         self.assertEqual(num_rows, self.expected_sessions_count)
 
+    def test_insert_client_sessions_date_time(self):
+        """
+        Use insert_test_client_sessions_date_time() to add new entries.
+        Test number of rows selected against expected_sessions_count.
+        """
+        self.insert_test_client_sessions_date_time()
+
+        self.dba.select_client_sessions_by_id()
+        num_rows = self.get_cursor_num_rows()
+
+        self.assertEqual(num_rows, self.expected_sessions_count)
+
+    def test_insert_client_sessions_datetime(self):
+        """
+        Use insert_test_client_sessions_datetime() to add new entries.
+        Test number of rows selected against expected_sessions_count.
+        """
+        self.insert_test_client_sessions_datetime()
+
+        self.dba.select_client_sessions_by_id()
+        num_rows = self.get_cursor_num_rows()
+
+        self.assertEqual(num_rows, self.expected_sessions_count)
+
     def test_delete_client_sessions_by_id(self):
         """
         Use insert_test_client_sessions to add new entries.
@@ -560,11 +693,31 @@ class AccessTest(unittest.TestCase):
         Use insert_test_status_messages to add new entries.
         Test number of rows selected against expected_statmsg_count.
         """
+        self.insert_test_status_messages()
+
+        self.dba.select_status_messages()
+        num_rows = self.get_cursor_num_rows()
+
+        self.assertEqual(num_rows, self.expected_statmsg_count)
+
+    def test_insert_status_messages_date_time(self):
         """
-        Use insert_test_swarms to add new entries in swarms table.
-        Test number of rows selected against expected_swarms_count.
+        Use insert_test_status_messages_date_time() to add new entries.
+        Test number of rows selected against expected_statmsg_count.
         """
-        self.insert_test_status_messages()
+        self.insert_test_status_messages_date_time()
+
+        self.dba.select_status_messages()
+        num_rows = self.get_cursor_num_rows()
+
+        self.assertEqual(num_rows, self.expected_statmsg_count)
+
+    def test_insert_status_messages_datetime(self):
+        """
+        Use insert_test_status_messages_datetime() to add new entries.
+        Test number of rows selected against expected_statmsg_count.
+        """
+        self.insert_test_status_messages_datetime()
 
         self.dba.select_status_messages()
         num_rows = self.get_cursor_num_rows()
@@ -619,6 +772,30 @@ class AccessTest(unittest.TestCase):
 
         self.assertEqual(num_rows, self.expected_verbmsg_count)
 
+    def test_insert_verbose_messages_date_time(self):
+        """
+        Use insert_test_verbose_messages_date_time() to add new entries.
+        Test number of rows selected against expected_verbmsg_count.
+        """
+        self.insert_test_verbose_messages_date_time()
+
+        self.dba.select_verbose_messages()
+        num_rows = self.get_cursor_num_rows()
+
+        self.assertEqual(num_rows, self.expected_verbmsg_count)
+
+    def test_insert_verbose_messages_datetime(self):
+        """
+        Use insert_test_verbose_messages_datetime() to add new entries.
+        Test number of rows selected against expected_verbmsg_count.
+        """
+        self.insert_test_verbose_messages_datetime()
+
+        self.dba.select_verbose_messages()
+        num_rows = self.get_cursor_num_rows()
+
+        self.assertEqual(num_rows, self.expected_verbmsg_count)
+
     def test_delete_verbose_messages(self):
         """
         Use insert_test_verbose_nessages to add new entries.