1 import os, shutil, unittest
3 from db_test_base import config
5 class SessionTest(unittest.TestCase):
6 def setUp(self):
7 # remove previous test, ignore errors
8 if os.path.exists(config.DATABASE):
9 shutil.rmtree(config.DATABASE)
10 os.makedirs(config.DATABASE + '/files')
11 self.db = self.module.Database(config, 'admin')
12 self.sessions = self.sessions_module.Sessions(self.db)
13 self.otks = self.sessions_module.OneTimeKeys(self.db)
15 def tearDown(self):
16 del self.otks
17 del self.sessions
18 if hasattr(self, 'db'):
19 self.db.close()
20 if os.path.exists(config.DATABASE):
21 shutil.rmtree(config.DATABASE)
23 def testSetSession(self):
24 self.sessions.set('random_key', text='hello, world!')
25 self.assertEqual(self.sessions.get('random_key', 'text'),
26 'hello, world!')
28 def testUpdateSession(self):
29 self.sessions.set('random_key', text='hello, world!')
30 self.assertEqual(self.sessions.get('random_key', 'text'),
31 'hello, world!')
32 self.sessions.set('random_key', text='nope')
33 self.assertEqual(self.sessions.get('random_key', 'text'), 'nope')
35 class DBMTest(SessionTest):
36 import roundup.backends.sessions_dbm as sessions_module
38 class RDBMSTest(SessionTest):
39 import roundup.backends.sessions_rdbms as sessions_module