def test_send_captcha_wrong_format(self):
ag = AntiCaptcha(api_key, loop=self.loop)
ag.close()
with self.assertRaises(ServiceError) as cm:
self.loop.run_until_complete(ag._send_captcha('str'))
self.assertIn('Unsupported captcha type', str(cm.exception))
def test_send_captcha_ok(self):
ag = AntiCaptcha(api_key, loop=self.loop)
ag.close()
ag._session = fake_client_session(200, 'OK|123')
cid = self.loop.run_until_complete(ag._send_captcha(b'id'))
self.assertEqual(cid, '123')
def test_send_captcha_inv_reply(self):
ag = AntiCaptcha(api_key, loop=self.loop)
ag.close()
ag._session = fake_client_session(200, 'abc')
with self.assertRaises(ServiceError) as cm:
self.loop.run_until_complete(ag._send_captcha(b'id'))
self.assertIn('Invalid server reply', str(cm.exception))
def test_send_captcha_http_err(self):
ag = AntiCaptcha(api_key, loop=self.loop)
ag.close()
ag._session = fake_client_session(400, 'OK|123')
with self.assertRaises(ServiceError) as cm:
self.loop.run_until_complete(ag._send_captcha(b'base64'))
self.assertIn('HTTP error', str(cm.exception))
def test_send_captcha_base64(self, add_fields):
ag = AntiCaptcha(api_key, loop=self.loop)
ag.close()
ag._session = fake_client_session(200, 'OK|123')
self.loop.run_until_complete(ag._send_captcha(b'base64'))
add_fields.assert_called_with(('method', 'base64'),
('body', b64encode(b'base64').decode()))
def test_send_captcha_handle_err(self):
ag = AntiCaptcha(api_key, loop=self.loop)
ag.close()
ag._session = fake_client_session(200,
'ERROR_NO_REQUEST_ACTION_RECEIVED')
with self.assertRaises(ServiceError) as cm:
self.loop.run_until_complete(ag._send_captcha(b'id'))
self.assertIn('No request action received', str(cm.exception))
def test_send_captcha_ext_opts(self, add_fields):
ag = AntiCaptcha(api_key, loop=self.loop)
ag.close()
ag._session = fake_client_session(200, 'OK|123')
self.loop.run_until_complete(ag._send_captcha(b'base64', b=2))
self.assertEqual(add_fields.call_count, 3)
add_fields.assert_any_call(('method', 'base64'),
('body', b64encode(b'base64').decode()))
add_fields.assert_any_call([('b', 2)])
def test_send_captcha_not_ready(self, sleep_mock):
sleep_mock.side_effect = fake_coroutine(1)
ag = AntiCaptcha(api_key, loop=self.loop)
ag.close()
ag._session = fake_client_session(
200, ['ERROR_NO_SLOT_AVAILABLE', 'OK|123'], iter_v=True)
cid = self.loop.run_until_complete(ag._send_captcha(b'id'))
self.assertEqual(cid, '123')
self.assertEqual(ag._session.post.call_count, 2)
sleep_mock.assert_called_with(ag._send_interval, loop=ag._loop)
def test_send_captcha_client_error(self):
ag = AntiCaptcha(api_key, loop=self.loop)
ag.close()
ag._session, resp = fake_client_session(200,
aiohttp.ClientError(),
ret_resp=True)
with self.assertRaises(ServiceError) as cm:
self.loop.run_until_complete(ag._send_captcha(b'id'))
self.assertIn('Network error', str(cm.exception))
self.assertTrue(resp.release.called)
self.assertTrue(resp.close.called)
def test_send_captcha_io_base(self, add_field):
ag = AntiCaptcha(api_key, loop=self.loop)
ag.close()
ag._session = fake_client_session(200, 'OK|123')
f = io.StringIO()
self.loop.run_until_complete(ag._send_captcha(f))
self.assertEqual(add_field.call_count, 3)
add_field.assert_any_call('method', 'post')
add_field.assert_any_call('key', api_key)
add_field.assert_any_call('file',
f,
filename='cap',
content_type='multipart/form-data')
