def test_multishard_create_index_no_untargetted_query_callback_called( self): _callback = Mock() api.set_untargetted_query_callback(_callback) operations.multishard_create_index('dummy', [('x', ASCENDING)]) _callback.assert_not_called()
def test_untargetted_query_callback(self): _callback = Mock() doc1 = {'x': 1, 'y': 1} doc2 = {'x': 2, 'y': 1} self.db1.dummy.insert(doc1) self.db2.dummy.insert(doc2) api.set_untargetted_query_callback(_callback) list(operations.multishard_find('dummy', {'y': 1})) _callback.assert_called_with('dummy', {'y': 1})
def test_untargetted_query_callback(self): _callback = Mock() doc1 = {'x': 1, 'y': 1} doc2 = {'x': 2, 'y': 1} self.db1.dummy.insert(doc1) self.db2.dummy.insert(doc2) api.set_untargetted_query_callback(_callback) list(operations.multishard_find('dummy', {'y': 1})) _callback.assert_called_with('dummy', {'y': 1})