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()