diff options
| author | Jesse Morgan <jesse@jesterpm.net> | 2015-11-24 11:12:49 -0800 |
|---|---|---|
| committer | Jesse Morgan <jesse@jesterpm.net> | 2015-11-24 11:12:49 -0800 |
| commit | 6259efbc44be3e21e42b29246941cd300e79200f (patch) | |
| tree | 67532f37aeadc45e959fffd1d05ab45ecd72e114 /tests/test_canarydb.py | |
| parent | 51129fda2179286143921db55a69b3f4dba03c0e (diff) | |
Adding emailcanary script
Adding the main launch script.
Adding the listAddress to the pings table.
Fixing the Makefile
Diffstat (limited to 'tests/test_canarydb.py')
| -rw-r--r-- | tests/test_canarydb.py | 26 |
1 files changed, 15 insertions, 11 deletions
diff --git a/tests/test_canarydb.py b/tests/test_canarydb.py index 8a29d23..a51afdc 100644 --- a/tests/test_canarydb.py +++ b/tests/test_canarydb.py @@ -14,23 +14,25 @@ class TestCanaryDB(unittest.TestCase): shutil.rmtree(self.tempdir) def testPingCheckPong(self): + listAddress = "list@example.com" address = "test@example.com" time = datetime.datetime(2015, 10, 24, 9, 00) uuid = "1234" expectedDelta = datetime.datetime.now() - time # Record a Ping - self.db.ping(address, time, uuid) + self.db.ping(listAddress, address, time, uuid) # Check for missing pongs missing = self.db.get_missing_pongs() self.assertEqual(1, len(missing)) firstMissing = missing[0] - self.assertEqual(3, len(firstMissing)) - self.assertEqual(uuid, firstMissing[0]) - self.assertEqual(address, firstMissing[1]) - delta = firstMissing[2].total_seconds() - expectedDelta.total_seconds() + self.assertEqual(4, len(firstMissing)) + self.assertEqual(listAddress, firstMissing[0]) + self.assertEqual(uuid, firstMissing[1]) + self.assertEqual(address, firstMissing[2]) + delta = firstMissing[3].total_seconds() - expectedDelta.total_seconds() self.assertTrue(delta <= 10) # Record a pong @@ -42,13 +44,14 @@ class TestCanaryDB(unittest.TestCase): self.assertEqual(0, len(missing)) def testCloseReopen(self): + listAddress = "list@example.com" address = "test@example.com" time = datetime.datetime(2015, 10, 24, 9, 00) uuid = "1234" expectedDelta = datetime.datetime.now() - time # Record a Ping - self.db.ping(address, time, uuid) + self.db.ping(listAddress, address, time, uuid) # Close, Reopen self.db.close() @@ -59,10 +62,11 @@ class TestCanaryDB(unittest.TestCase): self.assertEqual(1, len(missing)) firstMissing = missing[0] - self.assertEqual(3, len(firstMissing)) - self.assertEqual(uuid, firstMissing[0]) - self.assertEqual(address, firstMissing[1]) - delta = firstMissing[2].total_seconds() - expectedDelta.total_seconds() + self.assertEqual(4, len(firstMissing)) + self.assertEqual(listAddress, firstMissing[0]) + self.assertEqual(uuid, firstMissing[1]) + self.assertEqual(address, firstMissing[2]) + delta = firstMissing[3].total_seconds() - expectedDelta.total_seconds() self.assertTrue(delta <= 10) def testAccounts(self): @@ -87,7 +91,7 @@ class TestCanaryDB(unittest.TestCase): self.assertEqual(password, accounts[0][3]) # Remove the account - self.db.remove_account(address) + self.db.remove_account(listAddress, address) accounts = self.db.get_accounts() self.assertEqual(0, len(accounts)) |