import testconfig
import unittest
import keywords
import my_db

class TestKeywordsWithDB(unittest.TestCase):
	def setUp(self):
		" Setup a test-db environment."
		self.db = my_db.connect(testconfig.CONFIG, "test")
	    
	def tearDown(self):
		" Tear down the test-db environment"
		self.db.delete_user(self.db.userid)

	def _get_all_kw(self):
		res = self.db.get_all_keywords()[1:]
		res = map(lambda x: x[1], res)
		res.sort()
		return res
	def testAddUniqueKeywords(self):
		kws1 = ['a1','a2', 'a3']
		kws2 = ['a1', 'b1','b2', 'b3']
		all_kw = list(set(kws1+kws2))
		all_kw.sort()
		kws1.sort()
		kws2.sort()
		keywords.add_unique_keywords(self.db, kws1)
		self.assertEqual(kws1, self._get_all_kw())
		keywords.add_unique_keywords(self.db, kws2)
		self.assertEqual(all_kw, self._get_all_kw())

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