def test_delete_pattern_calls_scan_iter(self, init_mock, make_pattern_mock):
client = ShardClient()
client._backend = Mock()
client._backend.key_prefix = ''
connection = Mock()
connection.scan_iter.return_value = []
client._serverdict = {'test': connection}
client.delete_pattern(pattern='foo*')
connection.scan_iter.assert_called_once_with(match=make_pattern_mock.return_value)
def test_delete_pattern_calls_delete_for_given_keys(self, init_mock, make_pattern_mock):
client = ShardClient()
client._backend = Mock()
client._backend.key_prefix = ''
connection = Mock()
connection.scan_iter.return_value = [Mock(), Mock()]
connection.delete.return_value = 0
client._serverdict = {'test': connection}
client.delete_pattern(pattern='foo*')
connection.delete.assert_called_once_with(*connection.scan_iter.return_value)
def test_delete_pattern_calls_scan_iter(self, init_mock, make_key_mock):
client = ShardClient()
client._backend = Mock()
connection = Mock()
connection.scan_iter.return_value = []
client._serverdict = {'test': connection}
client.delete_pattern(pattern='foo*')
connection.scan_iter.assert_called_once_with(match=make_key_mock.return_value)
def test_delete_pattern_calls_delete_for_given_keys(self, init_mock, make_key_mock):
client = ShardClient()
client._backend = Mock()
connection = Mock()
connection.scan_iter.return_value = [Mock(), Mock()]
connection.delete.return_value = 0
client._serverdict = {'test': connection}
client.delete_pattern(pattern='foo*')
connection.delete.assert_called_once_with(*connection.scan_iter.return_value)
def test_delete_pattern_calls_scan_iter_with_count_if_itersize_given(
self, init_mock, make_pattern_mock):
client = ShardClient()
client._backend = Mock()
client._backend.key_prefix = ''
connection = Mock()
connection.scan_iter.return_value = []
client._serverdict = {'test': connection}
client.delete_pattern(pattern='foo*', itersize=10)
connection.scan_iter.assert_called_once_with(count=10, match=make_pattern_mock.return_value)