tests.py

import json
import pkg_resources

from tornado import testing, web

import examples.app
import sprockets_status.handlers


class SimpleStatusTests(testing.AsyncHTTPTestCase):

    @classmethod
    def setUpClass(cls):
        super(SimpleStatusTests, cls).setUpClass()
        sprockets_status.handlers.StatusHandler._package_name = None

    def get_app(self):
        return examples.app.make_app()

    def test_that_content_type_is_set(self):
        response = self.fetch('/status')
        self.assertEqual(response.code, 200)
        self.assertTrue(
            response.headers['Content-Type'].startswith('application/json'))

    def test_that_application_name_is_included(self):
        response = self.fetch('/status')
        self.assertEqual(response.code, 200)
        body = json.loads(response.body.decode('utf-8'))
        self.assertEqual(body['name'], examples.app.name)

    def test_that_application_version_is_included(self):
        response = self.fetch('/status')
        self.assertEqual(response.code, 200)
        body = json.loads(response.body.decode('utf-8'))
        self.assertEqual(body['version'], examples.app.version)

    def test_that_application_status_is_included(self):
        response = self.fetch('/status')
        self.assertEqual(response.code, 200)
        body = json.loads(response.body.decode('utf-8'))
        self.assertEqual(body['status'], 'ok')


class PackageLookupTests(testing.AsyncHTTPTestCase):

    @classmethod
    def setUpClass(cls):
        super(PackageLookupTests, cls).setUpClass()
        sprockets_status.handlers.StatusHandler._package_name = None

    def setUp(self):
        self.application = None
        super(PackageLookupTests, self).setUp()
        self.package_info = pkg_resources.get_distribution('tornado')

    def get_app(self):
        self.application = web.Application([
            web.url('/status', sprockets_status.handlers.StatusHandler,
                    {'package': 'tornado'}),
        ])
        return self.application

    def test_that_application_name_is_included(self):
        response = self.fetch('/status')
        self.assertEqual(response.code, 200)
        body = json.loads(response.body.decode('utf-8'))
        self.assertEqual(body['name'], self.package_info.project_name)

    def test_that_application_version_is_included(self):
        response = self.fetch('/status')
        self.assertEqual(response.code, 200)
        body = json.loads(response.body.decode('utf-8'))
        self.assertEqual(body['version'], self.package_info.version)

    def test_that_package_lookup_failure_results_in_server_error(self):
        _, spec_list = self.application.handlers[-1]
        spec_list[0].kwargs['package'] = 'not-a-package'
        response = self.fetch('/status')
        self.assertEqual(response.code, 500)