def test_incorrect_uri_wrong_collection_in_remove(self):
URI = 'mysql://*****:*****@localhost/kvlite_test'
collection_name = 'unknown_collection'
manager = MysqlCollectionManager(URI)
self.assertRaises(RuntimeError, manager.remove, (collection_name))
def test_manager_get_collection(self):
URI = 'mysql://*****:*****@localhost/kvlite_test'
collection = 'kvlite_test'
manager = MysqlCollectionManager(URI)
self.assertEqual(manager.collection_class.__name__, 'MysqlCollection')
def test_parse_url_no_collection(self):
params = MysqlCollectionManager.parse_uri('mysql://*****:*****@localhost:3307/kvlite_test')
self.assertEqual(params['backend'], 'mysql')
self.assertEqual(params['username'], 'kvlite_test')
self.assertEqual(params['password'], 'eixaaghiequ6ZeiBahn0')
self.assertEqual(params['host'], 'localhost')
self.assertEqual(params['port'], 3307)
self.assertEqual(params['db'], 'kvlite_test')
self.assertEqual(params['collection'], None)
def test_manager(self):
URI = 'mysql://*****:*****@localhost/kvlite_test'
collection = 'kvlite_test'
manager = MysqlCollectionManager(URI)
if collection in manager.collections():
manager.remove(collection)
self.assertNotIn(collection, manager.collections())
manager.create(collection)
self.assertIn(collection, manager.collections())
manager.remove(collection)
self.assertNotIn(collection, manager.collections())
manager.close()
