def test_add_headers_optional(self):
"""It's okay if no headers are passed."""
request_ = add_effectful_headers(self.auth_effect, request)
eff = request_('m', 'u')
self.assertEqual(
resolve_stubs(eff).intent,
Request(method='m', url='u', headers={'x-auth-token': 'abc123'}))
def test_added_headers_win(self):
"""When merging headers together, headers from the effect win."""
request_ = add_effectful_headers(self.auth_effect, request)
eff = request_('m', 'u', headers={'x-auth-token': 'fooey'})
self.assertEqual(
resolve_stubs(eff).intent,
Request(method="m", url="u", headers={"x-auth-token": "abc123"}))
def test_add_headers_optional(self):
"""It's okay if no headers are passed."""
request_ = add_effectful_headers(self.auth_effect, request)
eff = request_('m', 'u')
self.assertEqual(
resolve_stubs(eff).intent,
Request(method='m',
url='u',
headers={'x-auth-token': 'abc123'}))
def test_added_headers_win(self):
"""When merging headers together, headers from the effect win."""
request_ = add_effectful_headers(self.auth_effect, request)
eff = request_('m', 'u', headers={'x-auth-token': 'fooey'})
self.assertEqual(
resolve_stubs(eff).intent,
Request(method="m",
url="u",
headers={"x-auth-token": "abc123"}))
def test_add_headers(self):
"""Headers from the provided effect are inserted."""
request_ = add_effectful_headers(self.auth_effect, request)
eff = request_('m', 'u', headers={'default': 'headers'})
self.assertEqual(
resolve_stubs(eff).intent,
Request(method="m",
url="u",
headers={"x-auth-token": "abc123",
"default": "headers"}))
def test_add_headers(self):
"""Headers from the provided effect are inserted."""
request_ = add_effectful_headers(self.auth_effect, request)
eff = request_('m', 'u', headers={'default': 'headers'})
self.assertEqual(
resolve_stubs(eff).intent,
Request(method="m",
url="u",
headers={
"x-auth-token": "abc123",
"default": "headers"
}))
