def batch(self, **kwargs):
if self._server_settings is None:
resp, _ = self.session.request("GET", self.get_endpoint('root'))
self._server_settings = resp['settings']
batch_max_requests = self._server_settings['batch_max_requests']
batch_session = BatchSession(self,
batch_max_requests=batch_max_requests)
batch_client = self.clone(session=batch_session, **kwargs)
yield batch_client
batch_session.send()
batch_session.reset()
def test_reset_empties_the_requests_cache(self):
batch = BatchSession(self.client)
batch.request("GET", "/foobar/baz", permissions=mock.sentinel.permissions)
assert len(batch.requests) == 1
batch.reset()
assert len(batch.requests) == 0
