httpie-cli/tests/tests.py

190 lines
5.8 KiB
Python
Raw Normal View History

2012-03-15 00:11:49 +01:00
# coding:utf-8
import os
2012-03-04 12:02:30 +01:00
import sys
import unittest
2012-03-04 10:50:23 +01:00
import argparse
2012-03-15 00:11:49 +01:00
from requests.compat import StringIO, is_py26, str
TESTS_ROOT = os.path.dirname(__file__)
2012-03-14 19:21:47 +01:00
sys.path.insert(0, os.path.realpath(os.path.join(TESTS_ROOT, '..')))
from httpie import __main__
2012-03-04 10:50:23 +01:00
from httpie import cli
2012-03-14 19:21:47 +01:00
TEST_FILE = os.path.join(TESTS_ROOT, 'file.txt')
2012-03-15 00:11:49 +01:00
TERMINAL_COLOR_PRESENCE_CHECK = '\x1b['
2012-03-02 01:40:40 +01:00
def http(*args, **kwargs):
http_kwargs = {
'stdin_isatty': True,
'stdout_isatty': False
}
http_kwargs.update(kwargs)
stdout = http_kwargs.setdefault('stdout', StringIO())
__main__.main(args=args, **http_kwargs)
2012-03-02 01:40:40 +01:00
return stdout.getvalue()
2012-03-04 11:22:21 +01:00
class BaseTest(unittest.TestCase):
2012-03-15 00:11:49 +01:00
if is_py26:
2012-03-04 12:02:30 +01:00
def assertIn(self, member, container, msg=None):
2012-03-04 11:22:21 +01:00
self.assert_(member in container, msg)
2012-03-04 12:02:30 +01:00
def assertNotIn(self, member, container, msg=None):
2012-03-04 11:22:21 +01:00
self.assert_(member not in container, msg)
2012-03-04 12:02:30 +01:00
def assertDictEqual(self, d1, d2, msg=None):
2012-03-04 11:58:23 +01:00
self.assertEqual(set(d1.keys()), set(d2.keys()), msg)
self.assertEqual(sorted(d1.values()), sorted(d2.values()), msg)
2012-03-04 11:22:21 +01:00
class TestItemParsing(BaseTest):
2012-03-04 10:50:23 +01:00
def setUp(self):
self.kv = cli.KeyValueType(
cli.SEP_HEADERS,
cli.SEP_DATA,
2012-03-14 19:21:47 +01:00
cli.SEP_DATA_RAW_JSON,
cli.SEP_FILES,
2012-03-04 10:50:23 +01:00
)
def test_invalid_items(self):
items = ['no-separator']
for item in items:
2012-03-04 11:22:21 +01:00
self.assertRaises(argparse.ArgumentTypeError,
lambda: self.kv(item))
2012-03-04 10:50:23 +01:00
def test_escape(self):
headers, data, files = cli.parse_items([
# headers
self.kv('foo\\:bar:baz'),
self.kv('jack\\@jill:hill'),
# data
self.kv('baz\\=bar=foo'),
# files
self.kv('bar\\@baz@%s' % TEST_FILE)
])
self.assertDictEqual(headers, {
'foo:bar': 'baz',
'jack@jill': 'hill',
})
self.assertDictEqual(data, {
'baz=bar': 'foo',
})
self.assertIn('bar@baz', files)
2012-04-19 01:18:00 +02:00
def test_escape_longsep(self):
headers, data, files = cli.parse_items([
self.kv('bob\\:==foo'),
])
self.assertDictEqual(data, {
'bob:=': 'foo',
})
2012-03-04 10:50:23 +01:00
def test_valid_items(self):
2012-03-14 19:21:47 +01:00
headers, data, files = cli.parse_items([
2012-03-04 10:50:23 +01:00
self.kv('string=value'),
self.kv('header:value'),
self.kv('list:=["a", 1, {}, false]'),
self.kv('obj:={"a": "b"}'),
self.kv('eh:'),
self.kv('ed='),
self.kv('bool:=true'),
2012-03-14 19:21:47 +01:00
self.kv('test-file@%s' % TEST_FILE),
2012-03-04 10:50:23 +01:00
])
self.assertDictEqual(headers, {
'header': 'value',
'eh': ''
})
self.assertDictEqual(data, {
"ed": "",
"string": "value",
"bool": True,
"list": ["a", 1, {}, False],
"obj": {"a": "b"}
})
2012-03-14 19:21:47 +01:00
self.assertIn('test-file', files)
2012-03-04 10:50:23 +01:00
2012-03-02 01:40:40 +01:00
2012-03-04 11:22:21 +01:00
class TestHTTPie(BaseTest):
2012-03-02 01:40:40 +01:00
def test_get(self):
http('GET', 'http://httpbin.org/get')
2012-03-15 00:11:49 +01:00
def test_verbose(self):
r = http('--verbose', 'GET', 'http://httpbin.org/get', 'test-header:__test__')
self.assertEqual(r.count('__test__'), 2)
2012-03-02 01:40:40 +01:00
def test_json(self):
response = http('POST', 'http://httpbin.org/post', 'foo=bar')
self.assertIn('"foo": "bar"', response)
response2 = http('-j', 'GET', 'http://httpbin.org/headers')
self.assertIn('"Accept": "application/json"', response2)
response3 = http('-j', 'GET', 'http://httpbin.org/headers', 'Accept:application/xml')
self.assertIn('"Accept": "application/xml"', response3)
2012-03-02 01:40:40 +01:00
def test_form(self):
response = http('--form', 'POST', 'http://httpbin.org/post', 'foo=bar')
2012-03-02 01:40:40 +01:00
self.assertIn('"foo": "bar"', response)
def test_headers(self):
response = http('GET', 'http://httpbin.org/headers', 'Foo:bar')
self.assertIn('"User-Agent": "HTTPie', response)
self.assertIn('"Foo": "bar"', response)
2012-03-04 11:22:21 +01:00
class TestPrettyFlag(BaseTest):
"""Test the --pretty / --ugly flag handling."""
def test_pretty_enabled_by_default(self):
r = http('GET', 'http://httpbin.org/get', stdout_isatty=True)
2012-03-15 00:11:49 +01:00
self.assertIn(TERMINAL_COLOR_PRESENCE_CHECK, r)
def test_pretty_enabled_by_default_unless_stdin_redirected(self):
r = http('GET', 'http://httpbin.org/get', stdout_isatty=False)
2012-03-15 00:11:49 +01:00
self.assertNotIn(TERMINAL_COLOR_PRESENCE_CHECK, r)
def test_force_pretty(self):
r = http('--pretty', 'GET', 'http://httpbin.org/get', stdout_isatty=False)
2012-03-15 00:11:49 +01:00
self.assertIn(TERMINAL_COLOR_PRESENCE_CHECK, r)
def test_force_ugly(self):
r = http('--ugly', 'GET', 'http://httpbin.org/get', stdout_isatty=True)
2012-03-15 00:11:49 +01:00
self.assertNotIn(TERMINAL_COLOR_PRESENCE_CHECK, r)
class TestFileUpload(BaseTest):
def test_non_existent_file_raises_parse_error(self):
self.assertRaises(cli.ParseError, http,
'--form', '--traceback',
'POST', 'http://httpbin.org/post',
'foo@/__does_not_exist__')
def test_upload_ok(self):
r = http('--form', 'POST', 'http://httpbin.org/post',
2012-03-14 19:21:47 +01:00
'test-file@%s' % TEST_FILE)
self.assertIn('"test-file": "__test_file_content__', r)
class TestAuth(BaseTest):
def test_basic_auth(self):
r = http('--auth', 'user:password',
'GET', 'httpbin.org/basic-auth/user/password')
self.assertIn('"authenticated": true', r)
self.assertIn('"user": "user"', r)
def test_digest_auth(self):
r = http('--auth-type=digest', '--auth', 'user:password',
'GET', 'httpbin.org/digest-auth/auth/user/password')
self.assertIn('"authenticated": true', r)
self.assertIn('"user": "user"', r)
2012-03-02 01:40:40 +01:00
if __name__ == '__main__':
unittest.main()