import testconfig
import unittest
import my_db
import pgdb

class TestMyDbConnexion(unittest.TestCase):
	def setUp(self):
		" Setup a test-db environment."
		self.db = my_db.connect(testconfig.CONFIG, "test")
		self.prefs = { 'keywords_box' : 10,
			       'keywords_sort' : 'count',
			       'keywords_reverse': True,
			       'fullname': 'Test User',
			       'default_view': 1 }
	    
	def tearDown(self):
		" Tear down the test-db environment"
		self.db.delete_user(self.db.userid)

	def testConnect(self):
		self.assertNotEqual(self.db, None)
		self.assertNotEqual(self.db.userid, -1)

	def testGetPreferences(self):
		prefs = self.db.get_preferences()
		keys = prefs.keys()
		keys.sort()
		expKeys = self.prefs.keys()
		expKeys.sort()
		self.assertEqual(expKeys, keys)

	def testSetPreferences(self):
		self.db.set_preferences(self.prefs)
		prefs = self.db.get_preferences()
		self.assertEqual(prefs, self.prefs)

	def testAddKeyword(self):
		kw = self.db.add_keyword('keyword')
		self.assertEqual(self.db.get_keyword(kw), 'keyword')

	def testRemoveKeyword(self):
		kw = self.db.add_keyword('keyword')
		self.db.remove_keyword(kw)
		kws = self.db.get_all_keywords()
		# There is always '--' by default
		self.assertEqual(len(kws), 1)

	def testMapKeywords(self):
		kws = ('kw1', 'kw2', 'kw3', 'kw4')
		kwids = map(self.db.add_keyword, kws[:-1])
		expected_map = { kws[-1]: -1 }
		for kw, id in zip(kws[:-1], kwids):
			expected_map[kw] = id
		kw_map = self.db.map_keywords(kws)
		self.assertEqual(expected_map, kw_map)

if __name__ == '__main__':
    unittest.main()