def test_multi_remove(self): # Tests remove across multiple clusters doc1 = {'x': 1, 'y': 1} doc2 = {'x': 2, 'y': 1} self.db1.dummy.insert(doc1) self.db2.dummy.insert(doc2) operations.multishard_remove('dummy', {'y': 1}) self.assertEquals(0, self.db1.dummy.find({}).count()) self.assertEquals(0, self.db2.dummy.find({}).count())
def test_remove(self): # Perform removes with shards set to specific locations. doc1 = {'x': 1, 'y': 1} doc2 = {'x': 2, 'y': 1} self.db1.dummy.insert(doc1) self.db1.dummy.insert(doc2) self.db2.dummy.insert(doc1) self.db2.dummy.insert(doc2) operations.multishard_remove('dummy', {'x': 1, 'y': 1}) self.assertEquals(0, self.db1.dummy.find({'x': 1}).count()) self.assertEquals(1, self.db2.dummy.find({'x': 1}).count())
def test_multi_remove(self): # Tests remove across multiple clusters api.set_shard_at_rest('dummy', 1, "dest1/test_sharding") api.set_shard_at_rest('dummy', 2, "dest2/test_sharding") doc1 = {'x': 1, 'y': 1} doc2 = {'x': 2, 'y': 1} self.db1.dummy.insert(doc1) self.db2.dummy.insert(doc2) operations.multishard_remove('dummy', {'y': 1}) self.assertEquals(0, self.db1.dummy.find({}).count()) self.assertEquals(0, self.db2.dummy.find({}).count())
def remove(self, *args, **kwargs): return operations.multishard_remove( self.collection_name, *args, **kwargs)