1 import testconfig

     2 import unittest

     3 import keywords

     4 import my_db

     5 

     6 class TestKeywords(unittest.TestCase):

     7 	def setUp(self):

     8 		" Setup a test-db environment."

     9 		self.db = my_db.connect(testconfig.CONFIG, "test")

    10 	    

    11 	def tearDown(self):

    12 		" Tear down the test-db environment"

    13 		self.db.delete_user(self.db.userid)

    14 

    15 	def _get_all_kw(self):

    16 		res = self.db.get_all_keywords()[1:]

    17 		res = map(lambda x: x[1], res)

    18 		res.sort()

    19 		return res

    20 	def testAddUniqueKeywords(self):

    21 		kws1 = ['a1','a2', 'a3']

    22 		kws2 = ['a1', 'b1','b2', 'b3']

    23 		all_kw = list(set(kws1+kws2))

    24 		all_kw.sort()

    25 		kws1.sort()

    26 		kws2.sort()

    27 		keywords.add_unique_keywords(self.db, kws1)

    28 		self.assertEqual(kws1, self._get_all_kw())

    29 		keywords.add_unique_keywords(self.db, kws2)

    30 		self.assertEqual(all_kw, self._get_all_kw())

    31 

    32 if __name__ == '__main__':

    33 	unittest.main()