def test_normal_send(self):
queue_to_solr(self.queue, 1, self.solr_connection)
expected = [
mock.call([{
"foo": "bar"
}], commit=False),
mock.call([], commit=False),
]
calls = self.solr_connection.add.call_args_list
self.assertEqual(calls, expected)
def test_queue_drained_send(self):
queue_to_solr(self.queue, 2, self.solr_connection)
self.solr_connection.add.assert_called_once_with([{"foo": "bar"}])
def test_normal_send(self):
queue_to_solr(self.queue, 1, self.solr_connection)
expected = [mock.call([{"foo": "bar"}]), mock.call([]), ]
calls = self.solr_connection.add.call_args_list
self.assertEqual(calls, expected)
def test_queue_drained_send(self):
queue_to_solr(self.queue, 2, self.solr_connection)
self.solr_connection.add_many.assert_called_once_with([{"foo": "bar"}])
def test_bad_request_ignored(self):
self.solr_connection.add_many.side_effect = [SolrException(400), None]
queue_to_solr(self.queue, 1, self.solr_connection)