def test_handle_request(self, stop_mock):
otto = TornadoOctopus(cache=False, auto_start=True)
response = self.get_response()
callback = Mock()
handle_request = otto.handle_request('some url', callback)
handle_request(response)
expect(otto.running_urls).to_equal(-1)
expect(callback.called).to_be_true()
expect(stop_mock.called).to_be_true()
def test_handle_request(self, stop_mock):
otto = TornadoOctopus(cache=False, auto_start=True)
response = self.get_response()
callback = Mock()
handle_request = otto.handle_request('some url', callback)
handle_request(response)
expect(otto.running_urls).to_equal(-1)
expect(callback.called).to_be_true()
expect(stop_mock.called).to_be_true()
def test_handle_request_when_queue_has_no_items_but_running_urls(self):
otto = TornadoOctopus(cache=True, auto_start=True)
otto.response_cache = Mock()
otto.running_urls = 10
response = self.get_response()
callback = Mock()
handle_request = otto.handle_request('some url', callback)
handle_request(response)
expect(otto.running_urls).to_equal(9)
expect(callback.called).to_be_true()
expect(otto.response_cache.put.called).to_be_true()
def test_handle_request_when_queue_has_no_items_but_running_urls(self):
otto = TornadoOctopus(cache=True, auto_start=True)
otto.response_cache = Mock()
otto.running_urls = 10
response = self.get_response()
callback = Mock()
handle_request = otto.handle_request('some url', callback)
handle_request(response)
expect(otto.running_urls).to_equal(9)
expect(callback.called).to_be_true()
expect(otto.response_cache.put.called).to_be_true()
def test_handle_request_when_queue_has_items(self, fetch_mock):
otto = TornadoOctopus(cache=False, auto_start=True)
handler_mock = Mock()
otto.url_queue.append(
('other url', handler_mock, 'POST', {'foo': 'bar'})
)
response = self.get_response()
callback = Mock()
handle_request = otto.handle_request('some url', callback)
handle_request(response)
expect(otto.running_urls).to_equal(-1)
expect(otto.url_queue).to_be_empty()
expect(callback.called).to_be_true()
fetch_mock.assert_called_once_with('other url', handler_mock, 'POST', foo='bar')
def test_handle_request_when_queue_has_items(self, fetch_mock):
otto = TornadoOctopus(cache=False, auto_start=True)
handler_mock = Mock()
otto.url_queue.append(('other url', handler_mock, 'POST', {
'foo': 'bar'
}))
response = self.get_response()
callback = Mock()
handle_request = otto.handle_request('some url', callback)
handle_request(response)
expect(otto.running_urls).to_equal(-1)
expect(otto.url_queue).to_be_empty()
expect(callback.called).to_be_true()
fetch_mock.assert_called_once_with('other url',
handler_mock,
'POST',
foo='bar')
