From da3b9a77541f05f2f979f7f56a2922731a884da8 Mon Sep 17 00:00:00 2001 From: Razvan Deaconescu Date: Fri, 19 Aug 2011 21:48:40 +0300 Subject: [PATCH] ppf: Add tests for date_time methods. --- ppf/db/tests/test_access_sqlite.py | 183 ++++++++++++++++++++++++++++- 1 file changed, 180 insertions(+), 3 deletions(-) diff --git a/ppf/db/tests/test_access_sqlite.py b/ppf/db/tests/test_access_sqlite.py index 76d7a67..3a4bb3e 100644 --- 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. -- 2.20.1