def setUp(self):
self.config = config_manager.get_configuration(
section_name='persistence',
config_file='tests/files/config.ini',
variables_file='tests/files/envvars.ini')
self.config['database'] = 'metahosting_tests'
self.store = MongoStore(config=self.config)
class MongoStoreTest(unittest.TestCase):
def setUp(self):
self.config = config_manager.get_configuration(
section_name='persistence',
config_file='tests/files/config.ini',
variables_file='tests/files/envvars.ini')
self.config['database'] = 'metahosting_tests'
self.store = MongoStore(config=self.config)
def tearDown(self):
MongoClient(host=self.config['host'],
port=int(self.config['port']))\
.drop_database('metahosting_tests')
def test_update(self):
self.assertIsNone(self.store.get('name'))
self.store.update('name', 'value')
self.assertIsNotNone(self.store.get('name'))
self.store.update('name', 'new_value')
self.assertEqual(self.store.get('name'), 'new_value')
def test_get(self):
self.assertIsNone(self.store.get('foo'))
self.store.update('foo', 'bar')
self.assertIsNotNone(self.store.get('foo'))
self.assertEqual('bar', self.store.get('foo'))
def test_get_all(self):
ret = self.store.get_all()
self.assertDictEqual({}, ret)
a = dict()
a['foo'] = 'bar'
a['foo2'] = 'bar2'
a[1] = 'some'
a['foo3'] = 21211
for key, value in a.iteritems():
self.store.update(key, value)
self.assertDictEqual(a, self.store.get_all())
def test_update_with_dict(self):
simple_dict = dict()
simple_dict['a'] = {'foo': 'bar'}
self.store.update('foo', simple_dict)
simple_dict['a'] = {'ooo': 'barr'}
ret = self.store.get('foo')
self.assertTrue('a' in ret)
self.assertFalse('b' in ret)
ret['c'] = {'foo': 'ba'}
ret2 = self.store.get('foo')
self.assertFalse('c' in ret2)
