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