forked from extern/httpie-cli
75 lines
2.6 KiB
Python
75 lines
2.6 KiB
Python
|
from httpie.cliparse import HTTPieArgumentParser, KeyValue
|
||
|
|
from mock import Mock
|
||
|
|
|
||
|
|
__author__ = 'vladimir'
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
|
||
|
|
class HTTPieArgumentParserTestCase(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
self.HTTPieArgumentParserStub = type(HTTPieArgumentParser.__name__, (HTTPieArgumentParser,), {})
|
||
|
|
self.HTTPieArgumentParserStub.__init__ = lambda self: None
|
||
|
|
self.httpie_argument_parser = self.HTTPieArgumentParserStub()
|
||
|
|
|
||
|
|
def test_suggest_when_method_set_and_valid(self):
|
||
|
|
args = Mock()
|
||
|
|
args.method = 'GET'
|
||
|
|
args.url = 'http://example.com/'
|
||
|
|
args.items = []
|
||
|
|
|
||
|
|
self.httpie_argument_parser.suggest_method(args)
|
||
|
|
|
||
|
|
self.assertEquals(args.method, 'GET')
|
||
|
|
self.assertEquals(args.url, 'http://example.com/')
|
||
|
|
self.assertEquals(args.items, [])
|
||
|
|
|
||
|
|
def test_suggest_when_method_not_set(self):
|
||
|
|
args = Mock()
|
||
|
|
args.method = None
|
||
|
|
args.url = 'http://example.com/'
|
||
|
|
args.items = []
|
||
|
|
|
||
|
|
self.httpie_argument_parser.suggest_method(args)
|
||
|
|
|
||
|
|
self.assertEquals(args.method, 'GET')
|
||
|
|
self.assertEquals(args.url, 'http://example.com/')
|
||
|
|
self.assertEquals(args.items, [])
|
||
|
|
|
||
|
|
def test_suggest_when_method_set_but_invalid_and_data_field(self):
|
||
|
|
args = Mock()
|
||
|
|
args.method = 'http://example.com/'
|
||
|
|
args.url = 'data=field'
|
||
|
|
args.items = []
|
||
|
|
|
||
|
|
self.httpie_argument_parser.suggest_method(args)
|
||
|
|
|
||
|
|
self.assertEquals(args.method, 'POST')
|
||
|
|
self.assertEquals(args.url, 'http://example.com/')
|
||
|
|
self.assertEquals(args.items, [KeyValue(key='data', value='field', sep='=', orig='data=field')])
|
||
|
|
|
||
|
|
def test_suggest_when_method_set_but_invalid_and_header_field(self):
|
||
|
|
args = Mock()
|
||
|
|
args.method = 'http://example.com/'
|
||
|
|
args.url = 'test:header'
|
||
|
|
args.items = []
|
||
|
|
|
||
|
|
self.httpie_argument_parser.suggest_method(args)
|
||
|
|
|
||
|
|
self.assertEquals(args.method, 'GET')
|
||
|
|
self.assertEquals(args.url, 'http://example.com/')
|
||
|
|
self.assertEquals(args.items, [KeyValue(key='test', value='header', sep=':', orig='test:header')])
|
||
|
|
|
||
|
|
def test_suggest_when_method_set_but_invalid_and_item_exists(self):
|
||
|
|
args = Mock()
|
||
|
|
args.method = 'http://example.com/'
|
||
|
|
args.url = 'new_item=a'
|
||
|
|
args.items = [KeyValue(key='old_item', value='b', sep='=', orig='old_item=b')]
|
||
|
|
|
||
|
|
self.httpie_argument_parser.suggest_method(args)
|
||
|
|
|
||
|
|
self.assertEquals(args.items, [
|
||
|
|
KeyValue(key='new_item', value='a', sep='=', orig='new_item=a'),
|
||
|
|
KeyValue(key='old_item', value='b', sep='=', orig='old_item=b'),
|
||
|
|
])
|