Python set_untargetted_query_callback示例

示例#1

文件： test_operations.py 项目： jameslegg/shardmonster

    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()

示例#2

    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})

示例#3

文件： test_operations.py 项目： alsuren/shardmonster

    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})