IPython/html/services/kernels/tests/test_kernels_api.py

"""Test the kernels service API."""


import os
import sys
import json

import requests

from IPython.html.utils import url_path_join
from IPython.html.tests.launchnotebook import NotebookTestBase


class KernelAPITest(NotebookTestBase):
    """Test the kernels web service API"""

    def base_url(self):
        return url_path_join(super(KernelAPITest,self).base_url(), 'api/kernels')

    def mkkernel(self):
        r = requests.post(self.base_url())
        return r.json()

    def test__no_kernels(self):
        """Make sure there are no kernels running at the start"""
        url = self.base_url()
        r = requests.get(url)
        self.assertEqual(r.json(), [])

    def test_main_kernel_handler(self):
        # POST request
        r = requests.post(self.base_url())
        data = r.json()
        assert isinstance(data, dict)

        # GET request
        r = requests.get(self.base_url())
        assert isinstance(r.json(), list)
        self.assertEqual(r.json()[0], data['id'])

    def test_kernel_handler(self):
        # GET kernel with id
        data = self.mkkernel()
        url = self.base_url() +'/' + data['id']
        r = requests.get(url)
        assert isinstance(r.json(), dict)
        self.assertIn('id', r.json())
        self.assertEqual(r.json()['id'], data['id'])
        
        # DELETE kernel with id
        r = requests.delete(url)
        self.assertEqual(r.status_code, 204)
        r = requests.get(self.base_url())
        self.assertEqual(r.json(), [])