httpie-cli/tests/test_exit_status.py

from unittest import TestCase

import requests

from httpie import ExitStatus
from tests import (
    TestEnvironment,
    http, httpbin, HTTP_OK, skip, skipIf
)


class ExitStatusTest(TestCase):

    def test_ok_response_exits_0(self):
        r = http(
            'GET',
            httpbin('/status/200')
        )
        assert HTTP_OK in r
        assert r.exit_status == ExitStatus.OK

    def test_error_response_exits_0_without_check_status(self):
        r = http(
            'GET',
            httpbin('/status/500')
        )
        assert 'HTTP/1.1 500' in r
        assert r.exit_status == ExitStatus.OK
        assert not r.stderr

    @skip('timeout broken in requests'
          ' (https://github.com/jkbr/httpie/issues/185)')
    def test_timeout_exit_status(self):
        r = http(
            '--timeout=0.5',
            'GET',
            httpbin('/delay/1')
        )
        assert r.exit_status == ExitStatus.ERROR_TIMEOUT

    def test_3xx_check_status_exits_3_and_stderr_when_stdout_redirected(self):
        r = http(
            '--check-status',
            '--headers',  # non-terminal, force headers
            'GET',
            httpbin('/status/301'),
            env=TestEnvironment(stdout_isatty=False,)
        )
        assert 'HTTP/1.1 301' in r
        assert r.exit_status == ExitStatus.ERROR_HTTP_3XX
        assert '301 moved permanently' in r.stderr.lower()

    @skipIf(requests.__version__ == '0.13.6',
            'Redirects with prefetch=False are broken in Requests 0.13.6')
    def test_3xx_check_status_redirects_allowed_exits_0(self):
        r = http(
            '--check-status',
            '--follow',
            'GET',
            httpbin('/status/301')
        )
        # The redirect will be followed so 200 is expected.
        assert 'HTTP/1.1 200 OK' in r
        assert r.exit_status == ExitStatus.OK

    def test_4xx_check_status_exits_4(self):
        r = http(
            '--check-status',
            'GET',
            httpbin('/status/401')
        )
        assert 'HTTP/1.1 401' in r
        assert r.exit_status == ExitStatus.ERROR_HTTP_4XX
        # Also stderr should be empty since stdout isn't redirected.
        assert not r.stderr

    def test_5xx_check_status_exits_5(self):
        r = http(
            '--check-status',
            'GET',
            httpbin('/status/500')
        )
        assert 'HTTP/1.1 500' in r
        assert r.exit_status == ExitStatus.ERROR_HTTP_5XX