forked from extern/httpie-cli
160 lines
4.4 KiB
Python
160 lines
4.4 KiB
Python
import os
|
|
import shutil
|
|
from unittest import TestCase
|
|
|
|
from tests import TestEnvironment, mk_config_dir, http, httpbin, HTTP_OK
|
|
|
|
|
|
class SessionsTest(TestCase):
|
|
|
|
@property
|
|
def env(self):
|
|
return TestEnvironment(config_dir=self.config_dir)
|
|
|
|
def setUp(self):
|
|
# Start a full-blown session with a custom request header,
|
|
# authorization, and response cookies.
|
|
self.config_dir = mk_config_dir()
|
|
r = http(
|
|
'--follow',
|
|
'--session=test',
|
|
'--auth=username:password',
|
|
'GET',
|
|
httpbin('/cookies/set?hello=world'),
|
|
'Hello:World',
|
|
env=self.env
|
|
)
|
|
assert HTTP_OK in r
|
|
|
|
def tearDown(self):
|
|
shutil.rmtree(self.config_dir)
|
|
|
|
def test_session_create(self):
|
|
# Verify that the session has been created.
|
|
r = http(
|
|
'--session=test',
|
|
'GET',
|
|
httpbin('/get'),
|
|
env=self.env
|
|
)
|
|
assert HTTP_OK in r
|
|
assert r.json['headers']['Hello'] == 'World'
|
|
assert r.json['headers']['Cookie'] == 'hello=world'
|
|
assert 'Basic ' in r.json['headers']['Authorization']
|
|
|
|
def test_session_ignored_header_prefixes(self):
|
|
r = http(
|
|
'--session=test',
|
|
'GET',
|
|
httpbin('/get'),
|
|
'Content-Type: text/plain',
|
|
'If-Unmodified-Since: Sat, 29 Oct 1994 19:43:31 GMT',
|
|
env=self.env
|
|
)
|
|
assert HTTP_OK in r
|
|
|
|
r2 = http(
|
|
'--session=test',
|
|
'GET',
|
|
httpbin('/get')
|
|
)
|
|
assert HTTP_OK in r2
|
|
assert 'Content-Type' not in r2.json['headers']
|
|
assert 'If-Unmodified-Since' not in r2.json['headers']
|
|
|
|
def test_session_update(self):
|
|
# Get a response to a request from the original session.
|
|
r1 = http(
|
|
'--session=test',
|
|
'GET',
|
|
httpbin('/get'),
|
|
env=self.env
|
|
)
|
|
assert HTTP_OK in r1
|
|
|
|
# Make a request modifying the session data.
|
|
r2 = http(
|
|
'--follow',
|
|
'--session=test',
|
|
'--auth=username:password2',
|
|
'GET',
|
|
httpbin('/cookies/set?hello=world2'),
|
|
'Hello:World2',
|
|
env=self.env
|
|
)
|
|
assert HTTP_OK in r2
|
|
|
|
# Get a response to a request from the updated session.
|
|
r3 = http(
|
|
'--session=test',
|
|
'GET',
|
|
httpbin('/get'),
|
|
env=self.env
|
|
)
|
|
assert HTTP_OK in r3
|
|
assert r3.json['headers']['Hello'] == 'World2'
|
|
assert r3.json['headers']['Cookie'] == 'hello=world2'
|
|
assert (r1.json['headers']['Authorization'] !=
|
|
r3.json['headers']['Authorization'])
|
|
|
|
def test_session_read_only(self):
|
|
# Get a response from the original session.
|
|
r1 = http(
|
|
'--session=test',
|
|
'GET',
|
|
httpbin('/get'),
|
|
env=self.env
|
|
)
|
|
assert HTTP_OK in r1
|
|
|
|
# Make a request modifying the session data but
|
|
# with --session-read-only.
|
|
r2 = http(
|
|
'--follow',
|
|
'--session-read-only=test',
|
|
'--auth=username:password2',
|
|
'GET',
|
|
httpbin('/cookies/set?hello=world2'),
|
|
'Hello:World2',
|
|
env=self.env
|
|
)
|
|
assert HTTP_OK in r2
|
|
|
|
# Get a response from the updated session.
|
|
r3 = http(
|
|
'--session=test',
|
|
'GET',
|
|
httpbin('/get'),
|
|
env=self.env
|
|
)
|
|
assert HTTP_OK in r3
|
|
|
|
# Origin can differ on Travis.
|
|
del r1.json['origin'], r3.json['origin']
|
|
# Different for each request.
|
|
del r1.json['headers']['X-Request-Id'], \
|
|
r3.json['headers']['X-Request-Id']
|
|
# Should be the same as before r2.
|
|
assert r1.json == r3.json
|
|
|
|
def test_session_by_path(self):
|
|
session_path = os.path.join(self.config_dir, 'session-by-path.json')
|
|
r1 = http(
|
|
'--session=' + session_path,
|
|
'GET',
|
|
httpbin('/get'),
|
|
'Foo:Bar',
|
|
env=self.env
|
|
)
|
|
assert HTTP_OK in r1
|
|
|
|
r2 = http(
|
|
'--session=' + session_path,
|
|
'GET',
|
|
httpbin('/get'),
|
|
env=self.env
|
|
)
|
|
assert HTTP_OK in r2
|
|
|
|
assert r2.json['headers']['Foo'] in 'Bar'
|