def test_client_returns_none_on_timeout(self, mocker, cnpj):
mocker.patch('requests.get', side_effect=Timeout)
client = Client(cnpj)
data = client.get()
# Checks
assert data is None
assert requests.get.call_count == 1
def test_client_returns_none_on_timeout(self, mocker, cnpj):
mocker.patch('requests.get', side_effect=Timeout)
client = Client(cnpj)
data = client.get()
# Checks
assert data is None
assert requests.get.call_count == 1
def test_client_returns_json_object(self, mocker, response, cnpj):
mocker.patch('requests.get', side_effect=response)
client = Client(cnpj)
data = client.get()
# Checks
assert isinstance(data, dict)
assert 'status' in data
assert requests.get.call_count == 1
def test_client_returns_json_object(self, mocker, response, cnpj):
mocker.patch('requests.get', side_effect=response)
client = Client(cnpj)
data = client.get()
# Checks
assert isinstance(data, dict)
assert 'status' in data
assert requests.get.call_count == 1
def run(self):
for cnpj in self._list:
self._semaphore.acquire()
thread = threading.Thread(target=self.run_client,
args=(Client(cnpj), ))
thread.start()
failed = []
for cnpj in self._list:
data = self._queue.get()
if data is None:
failed.append(cnpj)
else:
self._res[cnpj] = data
if failed:
self._list = failed
return self.run()
return self._res
