monkeyking/notebook/services/api/tests/test_api.py

"""Test the basic /api endpoints"""

import requests

from notebook._tz import isoformat
from notebook.utils import url_path_join
from notebook.tests.launchnotebook import NotebookTestBase


class KernelAPITest(NotebookTestBase):
    """Test the kernels web service API"""
    
    def _req(self, verb, path, **kwargs):
        r = self.request(verb, url_path_join('api', path))
        r.raise_for_status()
        return r
    
    def get(self, path, **kwargs):
        return self._req('GET', path)
    
    def test_get_spec(self):
        r = self.get('spec.yaml')
        assert r.text
    
    def test_get_status(self):
        r = self.get('status')
        data = r.json()
        assert data['connections'] == 0
        assert data['kernels'] == 0
        assert data['last_activity'].endswith('Z')
        assert data['started'].endswith('Z')
        assert data['started'] == isoformat(self.notebook.web_app.settings['started'])