143 lines
4.2 KiB
Python
143 lines
4.2 KiB
Python
# -*- coding: utf-8; -*-
|
|
|
|
import json
|
|
from unittest import TestCase
|
|
from unittest.mock import MagicMock
|
|
|
|
from wuttjamaican.conf import WuttaConfig
|
|
|
|
from pyramid import testing
|
|
from pyramid.security import remember
|
|
|
|
from wuttaweb import subscribers
|
|
from wuttaweb import helpers
|
|
from wuttaweb.auth import WuttaSecurityPolicy
|
|
|
|
|
|
class TestNewRequest(TestCase):
|
|
|
|
def setUp(self):
|
|
self.config = WuttaConfig()
|
|
|
|
def make_request(self):
|
|
request = testing.DummyRequest()
|
|
request.registry.settings = {'wutta_config': self.config}
|
|
return request
|
|
|
|
def test_wutta_config(self):
|
|
request = self.make_request()
|
|
event = MagicMock(request=request)
|
|
|
|
# request gets a new attr
|
|
self.assertFalse(hasattr(request, 'wutta_config'))
|
|
subscribers.new_request(event)
|
|
self.assertTrue(hasattr(request, 'wutta_config'))
|
|
self.assertIs(request.wutta_config, self.config)
|
|
|
|
def test_use_oruga_default(self):
|
|
request = self.make_request()
|
|
event = MagicMock(request=request)
|
|
|
|
# request gets a new attr, false by default
|
|
self.assertFalse(hasattr(request, 'use_oruga'))
|
|
subscribers.new_request(event)
|
|
self.assertFalse(request.use_oruga)
|
|
|
|
def test_use_oruga_custom(self):
|
|
self.config.setdefault('wuttaweb.oruga_detector.spec', 'tests.test_subscribers:custom_oruga_detector')
|
|
request = self.make_request()
|
|
event = MagicMock(request=request)
|
|
|
|
# request gets a new attr, which should be true
|
|
self.assertFalse(hasattr(request, 'use_oruga'))
|
|
subscribers.new_request(event)
|
|
self.assertTrue(request.use_oruga)
|
|
|
|
|
|
def custom_oruga_detector(request):
|
|
return True
|
|
|
|
|
|
class TestNewRequestSetUser(TestCase):
|
|
|
|
def setUp(self):
|
|
self.config = WuttaConfig(defaults={
|
|
'wutta.db.default.url': 'sqlite://',
|
|
})
|
|
|
|
self.request = testing.DummyRequest()
|
|
self.pyramid_config = testing.setUp(request=self.request, settings={
|
|
'wutta_config': self.config,
|
|
})
|
|
|
|
self.app = self.config.get_app()
|
|
model = self.app.model
|
|
model.Base.metadata.create_all(bind=self.config.appdb_engine)
|
|
self.session = self.app.make_session()
|
|
self.user = model.User(username='barney')
|
|
self.session.add(self.user)
|
|
self.session.commit()
|
|
|
|
self.pyramid_config.set_security_policy(WuttaSecurityPolicy(db_session=self.session))
|
|
|
|
def tearDown(self):
|
|
testing.tearDown()
|
|
|
|
def test_anonymous(self):
|
|
self.assertFalse(hasattr(self.request, 'user'))
|
|
event = MagicMock(request=self.request)
|
|
subscribers.new_request_set_user(event)
|
|
self.assertIsNone(self.request.user)
|
|
|
|
def test_authenticated(self):
|
|
uuid = self.user.uuid
|
|
self.assertIsNotNone(uuid)
|
|
remember(self.request, uuid)
|
|
event = MagicMock(request=self.request)
|
|
subscribers.new_request_set_user(event, db_session=self.session)
|
|
self.assertIs(self.request.user, self.user)
|
|
|
|
|
|
class TestBeforeRender(TestCase):
|
|
|
|
def setUp(self):
|
|
self.config = WuttaConfig()
|
|
|
|
def make_request(self):
|
|
request = testing.DummyRequest()
|
|
request.registry.settings = {'wutta_config': self.config}
|
|
request.wutta_config = self.config
|
|
return request
|
|
|
|
def test_basic(self):
|
|
request = self.make_request()
|
|
event = {'request': request}
|
|
|
|
# event dict will get populated with more context
|
|
subscribers.before_render(event)
|
|
|
|
self.assertIn('config', event)
|
|
self.assertIs(event['config'], self.config)
|
|
|
|
self.assertIn('app', event)
|
|
self.assertIs(event['app'], self.config.get_app())
|
|
|
|
self.assertIn('h', event)
|
|
self.assertIs(event['h'], helpers)
|
|
|
|
self.assertIn('url', event)
|
|
# TODO: not sure how to test this?
|
|
# self.assertIs(event['url'], request.route_url)
|
|
|
|
self.assertIn('json', event)
|
|
self.assertIs(event['json'], json)
|
|
|
|
|
|
class TestIncludeMe(TestCase):
|
|
|
|
def test_basic(self):
|
|
with testing.testConfig() as pyramid_config:
|
|
|
|
# just ensure no error happens when included..
|
|
pyramid_config.include('wuttaweb.subscribers')
|