def test_websocket_view_adds_registry_reference_to_environ(pyramid_request):
pyramid_request.get_response = lambda _: None
views.websocket_view(pyramid_request)
env = pyramid_request.environ
assert env["h.ws.registry"] == pyramid_request.registry
def test_websocket_view_adds_work_queue_to_environ(pyramid_request):
pyramid_request.get_response = lambda _: None
views.websocket_view(pyramid_request)
env = pyramid_request.environ
assert env["h.ws.streamer_work_queue"] == streamer.WORK_QUEUE
def test_websocket_view_adds_work_queue_to_environ(pyramid_request):
pyramid_request.get_response = lambda _: None
views.websocket_view(pyramid_request)
env = pyramid_request.environ
assert env['h.ws.streamer_work_queue'] == streamer.WORK_QUEUE
def test_websocket_view_adds_registry_reference_to_environ(pyramid_request):
pyramid_request.get_response = lambda _: None
views.websocket_view(pyramid_request)
env = pyramid_request.environ
assert env['h.ws.registry'] == pyramid_request.registry
def test_websocket_view_adds_auth_state_to_environ(pyramid_config, pyramid_request):
pyramid_config.testing_securitypolicy('ragnar', groupids=['foo', 'bar'])
pyramid_request.get_response = lambda _: None
views.websocket_view(pyramid_request)
env = pyramid_request.environ
assert env['h.ws.authenticated_userid'] == 'ragnar'
assert env['h.ws.effective_principals'] == pyramid_request.effective_principals
def test_websocket_view_adds_auth_state_to_environ(pyramid_config, pyramid_request):
pyramid_config.testing_securitypolicy("ragnar", groupids=["foo", "bar"])
pyramid_request.get_response = lambda _: None
views.websocket_view(pyramid_request)
env = pyramid_request.environ
assert env["h.ws.authenticated_userid"] == "ragnar"
assert env["h.ws.effective_principals"] == pyramid_request.effective_principals
def test_websocket_view_same_origin(pyramid_request):
# example.com is the dummy request default host URL
pyramid_request.registry.settings.update({'origins': []})
pyramid_request.headers = {'Origin': 'http://example.com'}
pyramid_request.get_response = lambda _: mock.sentinel.good_response
res = views.websocket_view(pyramid_request)
assert res == mock.sentinel.good_response
def test_websocket_view_same_origin(config):
# example.com is the dummy request default host URL
config.registry.settings.update({'origins': []})
req = DummyRequest(headers={'Origin': 'http://example.com'})
req.get_response = lambda _: mock.sentinel.good_response
res = views.websocket_view(req)
assert res == mock.sentinel.good_response
def test_websocket_view_bad_origin(pyramid_request):
pyramid_request.registry.settings.update({'origins': ['http://good']})
pyramid_request.headers = {'Origin': 'http://bad'}
res = views.websocket_view(pyramid_request)
assert res.code == 403
def test_websocket_view_good_origin(pyramid_request):
pyramid_request.registry.settings.update({'origins': ['http://good']})
pyramid_request.headers = {'Origin': 'http://good'}
pyramid_request.get_response = lambda _: mock.sentinel.good_response
res = views.websocket_view(pyramid_request)
assert res == mock.sentinel.good_response
def test_websocket_view_bad_origin(config):
config.registry.settings.update({'origins': ['http://good']})
req = DummyRequest(headers={'Origin': 'http://bad'})
res = views.websocket_view(req)
assert res.code == 403
def test_websocket_view_good_origin(config):
config.registry.settings.update({'origins': ['http://good']})
req = DummyRequest(headers={'Origin': 'http://good'})
req.get_response = lambda _: mock.sentinel.good_response
res = views.websocket_view(req)
assert res == mock.sentinel.good_response
def test_websocket_view_bad_origin(pyramid_request):
pyramid_request.registry.settings.update({'origins': ['http://good']})
pyramid_request.headers = {'Origin': 'http://bad'}
res = views.websocket_view(pyramid_request)
assert res.code == 403
def test_websocket_view_same_origin(pyramid_request):
pyramid_request.get_response = lambda _: mock.sentinel.good_response
res = views.websocket_view(pyramid_request)
assert res == mock.sentinel.good_response
def test_websocket_view_good_origin(pyramid_request):
pyramid_request.registry.settings.update({'origins': ['http://good']})
pyramid_request.headers = {'Origin': 'http://good'}
pyramid_request.get_response = lambda _: mock.sentinel.good_response
res = views.websocket_view(pyramid_request)
assert res == mock.sentinel.good_response
