# -*- 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() self.request = self.make_request() self.pyramid_config = testing.setUp(request=self.request, settings={ 'wutta_config': self.config, }) def tearDown(self): testing.tearDown() def make_request(self): request = testing.DummyRequest() # request.registry.settings = {'wutta_config': self.config} return request def test_wutta_config(self): event = MagicMock(request=self.request) # request gets a new attr self.assertFalse(hasattr(self.request, 'wutta_config')) subscribers.new_request(event) self.assertTrue(hasattr(self.request, 'wutta_config')) self.assertIs(self.request.wutta_config, self.config) def test_use_oruga_default(self): event = MagicMock(request=self.request) # request gets a new attr, false by default self.assertFalse(hasattr(self.request, 'use_oruga')) subscribers.new_request(event) self.assertFalse(self.request.use_oruga) def test_use_oruga_custom(self): self.config.setdefault('wuttaweb.oruga_detector.spec', 'tests.test_subscribers:custom_oruga_detector') event = MagicMock(request=self.request) # request gets a new attr, which should be true self.assertFalse(hasattr(self.request, 'use_oruga')) subscribers.new_request(event) self.assertTrue(self.request.use_oruga) def test_get_referrer(self): event = MagicMock(request=self.request) def home(request): pass self.pyramid_config.add_route('home', '/') self.pyramid_config.add_view(home, route_name='home') self.assertFalse(hasattr(self.request, 'get_referrer')) subscribers.new_request(event) self.assertTrue(hasattr(self.request, 'get_referrer')) # default if no referrer, is home route url = self.request.get_referrer() self.assertEqual(url, self.request.route_url('home')) # can specify another default url = self.request.get_referrer(default='https://wuttaproject.org') self.assertEqual(url, 'https://wuttaproject.org') # or referrer can come from user session self.request.session['referrer'] = 'https://rattailproject.org' self.assertIn('referrer', self.request.session) url = self.request.get_referrer() self.assertEqual(url, 'https://rattailproject.org') # nb. referrer should also have been removed from user session self.assertNotIn('referrer', self.request.session) # or referrer can come from request params self.request.params['referrer'] = 'https://kernel.org' url = self.request.get_referrer() self.assertEqual(url, 'https://kernel.org') 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(wutta_config=self.config) 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) def test_is_admin(self): event = MagicMock(request=self.request) # anonymous user self.assertFalse(hasattr(self.request, 'user')) self.assertFalse(hasattr(self.request, 'is_admin')) subscribers.new_request_set_user(event, db_session=self.session) self.assertIsNone(self.request.user) self.assertFalse(self.request.is_admin) # reset del self.request.is_admin # authenticated user, but still not an admin self.request.user = self.user subscribers.new_request_set_user(event, db_session=self.session) self.assertIs(self.request.user, self.user) self.assertFalse(self.request.is_admin) # reset del self.request.is_admin # but if we make them an admin, it changes auth = self.app.get_auth_handler() admin = auth.get_role_administrator(self.session) self.user.roles.append(admin) self.session.commit() subscribers.new_request_set_user(event, db_session=self.session) self.assertIs(self.request.user, self.user) self.assertTrue(self.request.is_admin) def test_is_root(self): event = MagicMock(request=self.request) # anonymous user self.assertFalse(hasattr(self.request, 'user')) self.assertFalse(hasattr(self.request, 'is_root')) subscribers.new_request_set_user(event, db_session=self.session) self.assertIsNone(self.request.user) self.assertFalse(self.request.is_root) # reset del self.request.is_admin del self.request.is_root # authenticated user, but still not an admin self.request.user = self.user subscribers.new_request_set_user(event, db_session=self.session) self.assertIs(self.request.user, self.user) self.assertFalse(self.request.is_root) # reset del self.request.is_admin del self.request.is_root # even if we make them an admin, still not yet root auth = self.app.get_auth_handler() admin = auth.get_role_administrator(self.session) self.user.roles.append(admin) self.session.commit() subscribers.new_request_set_user(event, db_session=self.session) self.assertIs(self.request.user, self.user) self.assertTrue(self.request.is_admin) self.assertFalse(self.request.is_root) # reset del self.request.is_admin del self.request.is_root # root status flag lives in user session self.request.session['is_root'] = True subscribers.new_request_set_user(event, db_session=self.session) self.assertTrue(self.request.is_admin) self.assertTrue(self.request.is_root) 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')