def testCreateDropCollection(self):
"""Test case - create/drop collection -
"""
try:
with Connection(cfgOb=self.__cfgOb,
resourceName=self.__resourceName) as client:
mg = MongoDbUtil(client)
ok = mg.createCollection(self.__dbName, self.__collectionName)
self.assertTrue(ok)
ok = mg.databaseExists(self.__dbName)
self.assertTrue(ok)
ok = mg.collectionExists(self.__dbName, self.__collectionName)
self.assertTrue(ok)
#
logger.debug("Databases = %r", mg.getDatabaseNames())
logger.debug("Collections = %r",
mg.getCollectionNames(self.__dbName))
ok = mg.dropCollection(self.__dbName, self.__collectionName)
self.assertTrue(ok)
logger.debug("Databases = %r", mg.getDatabaseNames())
logger.debug("Collections = %r",
mg.getCollectionNames(self.__dbName))
# Removing the last collection will remove the database (results appear differ between mac and linux - )
ok = mg.databaseExists(self.__dbName)
# self.assertFalse(ok)
#
ok = mg.collectionExists(self.__dbName, self.__collectionName)
self.assertFalse(ok)
logger.debug("Collections = %r",
mg.getCollectionNames(self.__dbName))
except Exception as e:
logger.exception("Failing with %s", str(e))
self.fail()
def __removeCollection(self, dbName, collectionName):
"""Drop collection within database
"""
try:
with Connection(cfgOb=self.__cfgOb,
resourceName=self.__resourceName) as client:
mg = MongoDbUtil(client)
#
logger.debug("Remove collection database %s collection %s",
dbName, collectionName)
logger.debug("Starting databases = %r", mg.getDatabaseNames())
logger.debug("Starting collections = %r",
mg.getCollectionNames(dbName))
ok = mg.dropCollection(dbName, collectionName)
logger.debug("Databases = %r", mg.getDatabaseNames())
logger.debug("Post drop collections = %r",
mg.getCollectionNames(dbName))
ok = mg.collectionExists(dbName, collectionName)
logger.debug("Post drop collections = %r",
mg.getCollectionNames(dbName))
return ok
except Exception as e:
logger.exception("Failing with %s", str(e))
return False