def test_get_kernel_info_raises_err_with_abnormal_status(mocker): mock_req_obj = mock.Mock() mock_req_obj.send.return_value = mock.MagicMock(status=400) mocker.patch('ai.backend.client.kernel.Request', return_value=mock_req_obj) kernel_id = token_hex(12) with pytest.raises(BackendAPIError): k = Kernel(kernel_id) k.get_info()
def test_get_kernel_info_url(mocker): mock_req_obj = mock.Mock() mock_req_obj.send.return_value = mock.MagicMock(status=200) mock_req = mocker.patch('ai.backend.client.kernel.Request', return_value=mock_req_obj) kernel_id = token_hex(12) k = Kernel(kernel_id) k.get_info() mock_req.assert_called_once_with('GET', '/kernel/{}'.format(kernel_id)) mock_req_obj.send.assert_called_once_with() mock_req_obj.send.return_value.json.assert_called_once_with()