httpie-cli/tests/test_httpie.py

"""High-level tests."""
from unittest import TestCase

from tests import TestEnvironment, http, httpbin, HTTP_OK
from tests.fixtures import FILE_PATH, FILE_CONTENT


class HTTPieTest(TestCase):

    def test_GET(self):
        r = http('GET', httpbin('/get'))
        assert HTTP_OK in r

    def test_DELETE(self):
        r = http('DELETE', httpbin('/delete'))
        assert HTTP_OK in r

    def test_PUT(self):
        r = http('PUT', httpbin('/put'), 'foo=bar')
        assert HTTP_OK in r
        assert r'\"foo\": \"bar\"' in r

    def test_POST_JSON_data(self):
        r = http('POST', httpbin('/post'), 'foo=bar')
        assert HTTP_OK in r
        assert r'\"foo\": \"bar\"' in r

    def test_POST_form(self):
        r = http('--form', 'POST', httpbin('/post'), 'foo=bar')
        assert HTTP_OK in r
        assert '"foo": "bar"' in r

    def test_POST_form_multiple_values(self):
        r = http('--form', 'POST', httpbin('/post'), 'foo=bar', 'foo=baz')
        assert HTTP_OK in r
        assert r.json['form'] == {'foo': ['bar', 'baz']}

    def test_POST_stdin(self):

        with open(FILE_PATH) as f:
            env = TestEnvironment(stdin=f, stdin_isatty=False)
            r = http('--form', 'POST', httpbin('/post'), env=env)
        assert HTTP_OK in r
        assert FILE_CONTENT in r

    def test_headers(self):
        r = http('GET', httpbin('/headers'), 'Foo:bar')
        assert HTTP_OK in r
        assert '"User-Agent": "HTTPie' in r
        assert '"Foo": "bar"' in r