From 51129fda2179286143921db55a69b3f4dba03c0e Mon Sep 17 00:00:00 2001
From: Jesse Morgan <jesse@jesterpm.net>
Date: Sun, 25 Oct 2015 19:42:14 -0700
Subject: Adding Canary Check and Accounts

Adding the canary check method and tests for the Canary class.
Adding accounts to CanaryDB.
---
 tests/test_canarydb.py | 53 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)

(limited to 'tests/test_canarydb.py')

diff --git a/tests/test_canarydb.py b/tests/test_canarydb.py
index 5a0fc23..8a29d23 100644
--- a/tests/test_canarydb.py
+++ b/tests/test_canarydb.py
@@ -64,3 +64,56 @@ class TestCanaryDB(unittest.TestCase):
 		self.assertEqual(address, firstMissing[1])
 		delta = firstMissing[2].total_seconds() - expectedDelta.total_seconds()
 		self.assertTrue(delta <= 10)
+
+	def testAccounts(self):
+		listAddress = "list@example.org"
+		address = "user@example.net"
+		imapserver = "imap.example.net"
+		password = "secretpassword"
+
+		# Verify that no accounts exist
+		accounts = self.db.get_accounts()
+		self.assertEqual(0, len(accounts))
+
+		# Add one account
+		self.db.add_account(listAddress, address, imapserver, password)
+
+		# Verify that the account exists
+		accounts = self.db.get_accounts()
+		self.assertEqual(1, len(accounts))
+		self.assertEqual(listAddress, accounts[0][0])
+		self.assertEqual(address, accounts[0][1])
+		self.assertEqual(imapserver, accounts[0][2])
+		self.assertEqual(password, accounts[0][3])
+
+		# Remove the account
+		self.db.remove_account(address)
+		accounts = self.db.get_accounts()
+		self.assertEqual(0, len(accounts))
+
+	def testGetRecipientsForList(self):
+		listAddress1 = "list1@example.org"
+		listAddress2 = "list2@example.org"
+		imapserver = "imap.example.net"
+		password = "secretpassword"
+		address1 = "user1@example.net"
+		address2 = "user2@example.net"
+
+		# No accounts
+		self.assertEqual([], self.db.get_recipients_for_list(listAddress1));
+		self.assertEqual([], self.db.get_recipients_for_list(listAddress2));
+
+		# One account
+		self.db.add_account(listAddress1, address1, imapserver, password)
+		self.assertEqual([address1], self.db.get_recipients_for_list(listAddress1));
+		self.assertEqual([], self.db.get_recipients_for_list(listAddress2));
+
+		# Two accounts
+		self.db.add_account(listAddress1, address2, imapserver, password)
+		self.assertEqual([address1, address2], self.db.get_recipients_for_list(listAddress1));
+		self.assertEqual([], self.db.get_recipients_for_list(listAddress2));
+
+		# Two lists
+		self.db.add_account(listAddress2, address1, imapserver, password)
+		self.assertEqual([address1, address2], self.db.get_recipients_for_list(listAddress1));
+		self.assertEqual([address1], self.db.get_recipients_for_list(listAddress2));
-- 
cgit v1.2.3