def test_invalid_url(self): """Test connecting to an invalid URL.""" client = WebsocketClient('wss://127.0.0.1:9876', None) with self.assertRaises(WebsocketError): asyncio.get_event_loop().run_until_complete( client.get_job_status('job_id'))
def test_timeout(self): """Test timeout during retrieving a job status.""" client = WebsocketClient( 'ws://{}:{}'.format(TEST_IP_ADDRESS, VALID_PORT), TOKEN_TIMEOUT) with self.assertRaises(WebsocketTimeoutError): _ = asyncio.get_event_loop().run_until_complete( client.get_job_status('job_id', timeout=2))
def test_job_transition(self): """Test retrieving a job that transitions to final status.""" client = WebsocketClient('ws://127.0.0.1:8765', TOKEN_JOB_TRANSITION) response = asyncio.get_event_loop().run_until_complete( client.get_job_status('job_id')) self.assertIsInstance(response, dict) self.assertIn('status', response) self.assertEqual(response['status'], 'COMPLETED')
def test_invalid_url(self): """Test connecting to an invalid URL.""" client = WebsocketClient( 'wss://{}:{}'.format(TEST_IP_ADDRESS, INVALID_PORT), None) with self.assertRaises(WebsocketError): asyncio.get_event_loop().run_until_complete( client.get_job_status('job_id'))
def test_invalid_response(self): """Test unparseable response from the server.""" client = WebsocketClient( 'ws://{}:{}'.format(TEST_IP_ADDRESS, VALID_PORT), TOKEN_WRONG_FORMAT) with self.assertRaises(WebsocketIBMQProtocolError): _ = asyncio.get_event_loop().run_until_complete( client.get_job_status('job_id'))
def test_job_final_status(self): """Test retrieving a job already in final status.""" client = WebsocketClient( 'ws://{}:{}'.format(TEST_IP_ADDRESS, VALID_PORT), TOKEN_JOB_COMPLETED) response = asyncio.get_event_loop().run_until_complete( client.get_job_status('job_id')) self.assertIsInstance(response, dict) self.assertIn('status', response) self.assertEqual(response['status'], 'COMPLETED')
def test_invalid_response(self): """Test unparseable response from the server.""" client = WebsocketClient('ws://127.0.0.1:8765', TOKEN_WRONG_FORMAT) with self.assertRaises(WebsocketIBMQProtocolError): _ = asyncio.get_event_loop().run_until_complete( client.get_job_status('job_id'))
def test_timeout(self): """Test timeout during retrieving a job status.""" client = WebsocketClient('ws://127.0.0.1:8765', TOKEN_TIMEOUT) with self.assertRaises(WebsocketTimeoutError): _ = asyncio.get_event_loop().run_until_complete( client.get_job_status('job_id', timeout=2))