513 lines
21 KiB
Python
513 lines
21 KiB
Python
# -*- coding: utf-8; -*-
|
|
|
|
import functools
|
|
from unittest import TestCase
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from pyramid import testing
|
|
from pyramid.response import Response
|
|
from pyramid.httpexceptions import HTTPFound
|
|
|
|
from wuttjamaican.conf import WuttaConfig
|
|
from wuttaweb.views import master
|
|
from wuttaweb.subscribers import new_request_set_user
|
|
|
|
from tests.views.utils import WebTestCase
|
|
|
|
|
|
class TestMasterView(WebTestCase):
|
|
|
|
def test_defaults(self):
|
|
with patch.multiple(master.MasterView, create=True,
|
|
model_name='Widget',
|
|
viewable=False,
|
|
editable=False):
|
|
master.MasterView.defaults(self.pyramid_config)
|
|
|
|
##############################
|
|
# class methods
|
|
##############################
|
|
|
|
def test_get_model_class(self):
|
|
|
|
# no model class by default
|
|
self.assertIsNone(master.MasterView.get_model_class())
|
|
|
|
# subclass may specify
|
|
MyModel = MagicMock()
|
|
master.MasterView.model_class = MyModel
|
|
self.assertIs(master.MasterView.get_model_class(), MyModel)
|
|
del master.MasterView.model_class
|
|
|
|
def test_get_model_name(self):
|
|
|
|
# error by default (since no model class)
|
|
self.assertRaises(AttributeError, master.MasterView.get_model_name)
|
|
|
|
# subclass may specify model name
|
|
master.MasterView.model_name = 'Widget'
|
|
self.assertEqual(master.MasterView.get_model_name(), 'Widget')
|
|
del master.MasterView.model_name
|
|
|
|
# or it may specify model class
|
|
MyModel = MagicMock(__name__='Blaster')
|
|
master.MasterView.model_class = MyModel
|
|
self.assertEqual(master.MasterView.get_model_name(), 'Blaster')
|
|
del master.MasterView.model_class
|
|
|
|
def test_get_model_name_normalized(self):
|
|
|
|
# error by default (since no model class)
|
|
self.assertRaises(AttributeError, master.MasterView.get_model_name_normalized)
|
|
|
|
# subclass may specify *normalized* model name
|
|
master.MasterView.model_name_normalized = 'widget'
|
|
self.assertEqual(master.MasterView.get_model_name_normalized(), 'widget')
|
|
del master.MasterView.model_name_normalized
|
|
|
|
# or it may specify *standard* model name
|
|
master.MasterView.model_name = 'Blaster'
|
|
self.assertEqual(master.MasterView.get_model_name_normalized(), 'blaster')
|
|
del master.MasterView.model_name
|
|
|
|
# or it may specify model class
|
|
MyModel = MagicMock(__name__='Dinosaur')
|
|
master.MasterView.model_class = MyModel
|
|
self.assertEqual(master.MasterView.get_model_name_normalized(), 'dinosaur')
|
|
del master.MasterView.model_class
|
|
|
|
def test_get_model_title(self):
|
|
|
|
# error by default (since no model class)
|
|
self.assertRaises(AttributeError, master.MasterView.get_model_title)
|
|
|
|
# subclass may specify model title
|
|
master.MasterView.model_title = 'Wutta Widget'
|
|
self.assertEqual(master.MasterView.get_model_title(), "Wutta Widget")
|
|
del master.MasterView.model_title
|
|
|
|
# or it may specify model name
|
|
master.MasterView.model_name = 'Blaster'
|
|
self.assertEqual(master.MasterView.get_model_title(), "Blaster")
|
|
del master.MasterView.model_name
|
|
|
|
# or it may specify model class
|
|
MyModel = MagicMock(__name__='Dinosaur')
|
|
master.MasterView.model_class = MyModel
|
|
self.assertEqual(master.MasterView.get_model_title(), "Dinosaur")
|
|
del master.MasterView.model_class
|
|
|
|
def test_get_model_title_plural(self):
|
|
|
|
# error by default (since no model class)
|
|
self.assertRaises(AttributeError, master.MasterView.get_model_title_plural)
|
|
|
|
# subclass may specify *plural* model title
|
|
master.MasterView.model_title_plural = 'People'
|
|
self.assertEqual(master.MasterView.get_model_title_plural(), "People")
|
|
del master.MasterView.model_title_plural
|
|
|
|
# or it may specify *singular* model title
|
|
master.MasterView.model_title = 'Wutta Widget'
|
|
self.assertEqual(master.MasterView.get_model_title_plural(), "Wutta Widgets")
|
|
del master.MasterView.model_title
|
|
|
|
# or it may specify model name
|
|
master.MasterView.model_name = 'Blaster'
|
|
self.assertEqual(master.MasterView.get_model_title_plural(), "Blasters")
|
|
del master.MasterView.model_name
|
|
|
|
# or it may specify model class
|
|
MyModel = MagicMock(__name__='Dinosaur')
|
|
master.MasterView.model_class = MyModel
|
|
self.assertEqual(master.MasterView.get_model_title_plural(), "Dinosaurs")
|
|
del master.MasterView.model_class
|
|
|
|
def test_get_model_key(self):
|
|
|
|
# error by default (since no model class)
|
|
self.assertRaises(AttributeError, master.MasterView.get_model_key)
|
|
|
|
# subclass may specify model key
|
|
master.MasterView.model_key = 'uuid'
|
|
self.assertEqual(master.MasterView.get_model_key(), ('uuid',))
|
|
del master.MasterView.model_key
|
|
|
|
def test_get_route_prefix(self):
|
|
|
|
# error by default (since no model class)
|
|
self.assertRaises(AttributeError, master.MasterView.get_route_prefix)
|
|
|
|
# subclass may specify route prefix
|
|
master.MasterView.route_prefix = 'widgets'
|
|
self.assertEqual(master.MasterView.get_route_prefix(), 'widgets')
|
|
del master.MasterView.route_prefix
|
|
|
|
# subclass may specify *normalized* model name
|
|
master.MasterView.model_name_normalized = 'blaster'
|
|
self.assertEqual(master.MasterView.get_route_prefix(), 'blasters')
|
|
del master.MasterView.model_name_normalized
|
|
|
|
# or it may specify *standard* model name
|
|
master.MasterView.model_name = 'Dinosaur'
|
|
self.assertEqual(master.MasterView.get_route_prefix(), 'dinosaurs')
|
|
del master.MasterView.model_name
|
|
|
|
# or it may specify model class
|
|
MyModel = MagicMock(__name__='Truck')
|
|
master.MasterView.model_class = MyModel
|
|
self.assertEqual(master.MasterView.get_route_prefix(), 'trucks')
|
|
del master.MasterView.model_class
|
|
|
|
def test_get_url_prefix(self):
|
|
|
|
# error by default (since no model class)
|
|
self.assertRaises(AttributeError, master.MasterView.get_url_prefix)
|
|
|
|
# subclass may specify url prefix
|
|
master.MasterView.url_prefix = '/widgets'
|
|
self.assertEqual(master.MasterView.get_url_prefix(), '/widgets')
|
|
del master.MasterView.url_prefix
|
|
|
|
# or it may specify route prefix
|
|
master.MasterView.route_prefix = 'trucks'
|
|
self.assertEqual(master.MasterView.get_url_prefix(), '/trucks')
|
|
del master.MasterView.route_prefix
|
|
|
|
# or it may specify *normalized* model name
|
|
master.MasterView.model_name_normalized = 'blaster'
|
|
self.assertEqual(master.MasterView.get_url_prefix(), '/blasters')
|
|
del master.MasterView.model_name_normalized
|
|
|
|
# or it may specify *standard* model name
|
|
master.MasterView.model_name = 'Dinosaur'
|
|
self.assertEqual(master.MasterView.get_url_prefix(), '/dinosaurs')
|
|
del master.MasterView.model_name
|
|
|
|
# or it may specify model class
|
|
MyModel = MagicMock(__name__='Machine')
|
|
master.MasterView.model_class = MyModel
|
|
self.assertEqual(master.MasterView.get_url_prefix(), '/machines')
|
|
del master.MasterView.model_class
|
|
|
|
def test_get_instance_url_prefix(self):
|
|
|
|
# error by default (since no model class)
|
|
self.assertRaises(AttributeError, master.MasterView.get_instance_url_prefix)
|
|
|
|
# typical example with url_prefix and simple key
|
|
master.MasterView.url_prefix = '/widgets'
|
|
master.MasterView.model_key = 'uuid'
|
|
self.assertEqual(master.MasterView.get_instance_url_prefix(), '/widgets/{uuid}')
|
|
del master.MasterView.url_prefix
|
|
del master.MasterView.model_key
|
|
|
|
# typical example with composite key
|
|
master.MasterView.url_prefix = '/widgets'
|
|
master.MasterView.model_key = ('foo', 'bar')
|
|
self.assertEqual(master.MasterView.get_instance_url_prefix(), '/widgets/{foo}|{bar}')
|
|
del master.MasterView.url_prefix
|
|
del master.MasterView.model_key
|
|
|
|
def test_get_template_prefix(self):
|
|
|
|
# error by default (since no model class)
|
|
self.assertRaises(AttributeError, master.MasterView.get_template_prefix)
|
|
|
|
# subclass may specify template prefix
|
|
master.MasterView.template_prefix = '/widgets'
|
|
self.assertEqual(master.MasterView.get_template_prefix(), '/widgets')
|
|
del master.MasterView.template_prefix
|
|
|
|
# or it may specify url prefix
|
|
master.MasterView.url_prefix = '/trees'
|
|
self.assertEqual(master.MasterView.get_template_prefix(), '/trees')
|
|
del master.MasterView.url_prefix
|
|
|
|
# or it may specify route prefix
|
|
master.MasterView.route_prefix = 'trucks'
|
|
self.assertEqual(master.MasterView.get_template_prefix(), '/trucks')
|
|
del master.MasterView.route_prefix
|
|
|
|
# or it may specify *normalized* model name
|
|
master.MasterView.model_name_normalized = 'blaster'
|
|
self.assertEqual(master.MasterView.get_template_prefix(), '/blasters')
|
|
del master.MasterView.model_name_normalized
|
|
|
|
# or it may specify *standard* model name
|
|
master.MasterView.model_name = 'Dinosaur'
|
|
self.assertEqual(master.MasterView.get_template_prefix(), '/dinosaurs')
|
|
del master.MasterView.model_name
|
|
|
|
# or it may specify model class
|
|
MyModel = MagicMock(__name__='Machine')
|
|
master.MasterView.model_class = MyModel
|
|
self.assertEqual(master.MasterView.get_template_prefix(), '/machines')
|
|
del master.MasterView.model_class
|
|
|
|
def test_get_grid_key(self):
|
|
|
|
# error by default (since no model class)
|
|
self.assertRaises(AttributeError, master.MasterView.get_grid_key)
|
|
|
|
# subclass may specify grid key
|
|
master.MasterView.grid_key = 'widgets'
|
|
self.assertEqual(master.MasterView.get_grid_key(), 'widgets')
|
|
del master.MasterView.grid_key
|
|
|
|
# or it may specify route prefix
|
|
master.MasterView.route_prefix = 'trucks'
|
|
self.assertEqual(master.MasterView.get_grid_key(), 'trucks')
|
|
del master.MasterView.route_prefix
|
|
|
|
# or it may specify *normalized* model name
|
|
master.MasterView.model_name_normalized = 'blaster'
|
|
self.assertEqual(master.MasterView.get_grid_key(), 'blasters')
|
|
del master.MasterView.model_name_normalized
|
|
|
|
# or it may specify *standard* model name
|
|
master.MasterView.model_name = 'Dinosaur'
|
|
self.assertEqual(master.MasterView.get_grid_key(), 'dinosaurs')
|
|
del master.MasterView.model_name
|
|
|
|
# or it may specify model class
|
|
MyModel = MagicMock(__name__='Machine')
|
|
master.MasterView.model_class = MyModel
|
|
self.assertEqual(master.MasterView.get_grid_key(), 'machines')
|
|
del master.MasterView.model_class
|
|
|
|
def test_get_config_title(self):
|
|
|
|
# error by default (since no model class)
|
|
self.assertRaises(AttributeError, master.MasterView.get_config_title)
|
|
|
|
# subclass may specify config title
|
|
master.MasterView.config_title = 'Widgets'
|
|
self.assertEqual(master.MasterView.get_config_title(), "Widgets")
|
|
del master.MasterView.config_title
|
|
|
|
# subclass may specify *plural* model title
|
|
master.MasterView.model_title_plural = 'People'
|
|
self.assertEqual(master.MasterView.get_config_title(), "People")
|
|
del master.MasterView.model_title_plural
|
|
|
|
# or it may specify *singular* model title
|
|
master.MasterView.model_title = 'Wutta Widget'
|
|
self.assertEqual(master.MasterView.get_config_title(), "Wutta Widgets")
|
|
del master.MasterView.model_title
|
|
|
|
# or it may specify model name
|
|
master.MasterView.model_name = 'Blaster'
|
|
self.assertEqual(master.MasterView.get_config_title(), "Blasters")
|
|
del master.MasterView.model_name
|
|
|
|
# or it may specify model class
|
|
MyModel = MagicMock(__name__='Dinosaur')
|
|
master.MasterView.model_class = MyModel
|
|
self.assertEqual(master.MasterView.get_config_title(), "Dinosaurs")
|
|
del master.MasterView.model_class
|
|
|
|
##############################
|
|
# support methods
|
|
##############################
|
|
|
|
def test_render_to_response(self):
|
|
|
|
def widgets(request): return {}
|
|
self.pyramid_config.add_route('widgets', '/widgets/')
|
|
self.pyramid_config.add_view(widgets, route_name='widgets')
|
|
|
|
# basic sanity check using /master/index.mako
|
|
# (nb. it skips /widgets/index.mako since that doesn't exist)
|
|
master.MasterView.model_name = 'Widget'
|
|
view = master.MasterView(self.request)
|
|
response = view.render_to_response('index', {})
|
|
self.assertIsInstance(response, Response)
|
|
del master.MasterView.model_name
|
|
|
|
# basic sanity check using /appinfo/index.mako
|
|
master.MasterView.model_name = 'AppInfo'
|
|
master.MasterView.route_prefix = 'appinfo'
|
|
master.MasterView.url_prefix = '/appinfo'
|
|
view = master.MasterView(self.request)
|
|
response = view.render_to_response('index', {})
|
|
self.assertIsInstance(response, Response)
|
|
del master.MasterView.model_name
|
|
del master.MasterView.route_prefix
|
|
del master.MasterView.url_prefix
|
|
|
|
# bad template name causes error
|
|
master.MasterView.model_name = 'Widget'
|
|
self.assertRaises(IOError, view.render_to_response, 'nonexistent', {})
|
|
del master.MasterView.model_name
|
|
|
|
def test_get_index_title(self):
|
|
master.MasterView.model_title_plural = "Wutta Widgets"
|
|
view = master.MasterView(self.request)
|
|
self.assertEqual(view.get_index_title(), "Wutta Widgets")
|
|
del master.MasterView.model_title_plural
|
|
|
|
def test_get_instance(self):
|
|
view = master.MasterView(self.request)
|
|
self.assertRaises(NotImplementedError, view.get_instance)
|
|
|
|
##############################
|
|
# view methods
|
|
##############################
|
|
|
|
def test_index(self):
|
|
|
|
# sanity/coverage check using /settings/
|
|
master.MasterView.model_name = 'Setting'
|
|
master.MasterView.model_key = 'name'
|
|
master.MasterView.grid_columns = ['name', 'value']
|
|
view = master.MasterView(self.request)
|
|
response = view.index()
|
|
# then again with data, to include view action url
|
|
data = [{'name': 'foo', 'value': 'bar'}]
|
|
with patch.object(view, 'index_get_grid_data', return_value=data):
|
|
response = view.index()
|
|
del master.MasterView.model_name
|
|
del master.MasterView.model_key
|
|
del master.MasterView.grid_columns
|
|
|
|
def test_view(self):
|
|
|
|
# sanity/coverage check using /settings/XXX
|
|
setting = {'name': 'foo.bar', 'value': 'baz'}
|
|
self.request.matchdict = {'name': 'foo.bar'}
|
|
with patch.multiple(master.MasterView, create=True,
|
|
model_name='Setting',
|
|
model_key='name',
|
|
grid_columns=['name', 'value'],
|
|
form_fields=['name', 'value']):
|
|
view = master.MasterView(self.request)
|
|
with patch.object(view, 'get_instance', return_value=setting):
|
|
response = view.view()
|
|
|
|
def test_edit(self):
|
|
model = self.app.model
|
|
self.app.save_setting(self.session, 'foo.bar', 'frazzle')
|
|
self.session.commit()
|
|
|
|
def get_instance():
|
|
setting = self.session.query(model.Setting).get('foo.bar')
|
|
return {
|
|
'name': setting.name,
|
|
'value': setting.value,
|
|
}
|
|
|
|
# sanity/coverage check using /settings/XXX/edit
|
|
self.request.matchdict = {'name': 'foo.bar'}
|
|
with patch.multiple(master.MasterView, create=True,
|
|
model_name='Setting',
|
|
model_key='name',
|
|
form_fields=['name', 'value']):
|
|
view = master.MasterView(self.request)
|
|
with patch.object(view, 'get_instance', new=get_instance):
|
|
|
|
# get the form page
|
|
response = view.edit()
|
|
self.assertIsInstance(response, Response)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertIn('frazzle', response.text)
|
|
# nb. no error
|
|
self.assertNotIn('Required', response.text)
|
|
|
|
def persist(setting):
|
|
self.app.save_setting(self.session, setting['name'], setting['value'])
|
|
self.session.commit()
|
|
|
|
# post request to save settings
|
|
self.request.method = 'POST'
|
|
self.request.POST = {
|
|
'name': 'foo.bar',
|
|
'value': 'froogle',
|
|
}
|
|
with patch.object(view, 'persist', new=persist):
|
|
response = view.edit()
|
|
# nb. should get redirect back to view page
|
|
self.assertIsInstance(response, HTTPFound)
|
|
# setting should be updated in DB
|
|
self.assertEqual(self.app.get_setting(self.session, 'foo.bar'), 'froogle')
|
|
|
|
# try another post with invalid data (name is required)
|
|
self.request.method = 'POST'
|
|
self.request.POST = {
|
|
'value': 'gargoyle',
|
|
}
|
|
with patch.object(view, 'persist', new=persist):
|
|
response = view.edit()
|
|
# nb. should get a form with errors
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertIn('Required', response.text)
|
|
# setting did not change in DB
|
|
self.assertEqual(self.app.get_setting(self.session, 'foo.bar'), 'froogle')
|
|
|
|
def test_configure(self):
|
|
model = self.app.model
|
|
|
|
# setup
|
|
master.MasterView.model_name = 'AppInfo'
|
|
master.MasterView.route_prefix = 'appinfo'
|
|
master.MasterView.template_prefix = '/appinfo'
|
|
|
|
# mock settings
|
|
settings = [
|
|
{'name': 'wutta.app_title'},
|
|
{'name': 'wutta.foo', 'value': 'bar'},
|
|
{'name': 'wutta.flag', 'type': bool},
|
|
{'name': 'wutta.number', 'type': int, 'default': 42},
|
|
{'name': 'wutta.value1', 'save_if_empty': True},
|
|
{'name': 'wutta.value2', 'save_if_empty': False},
|
|
]
|
|
|
|
view = master.MasterView(self.request)
|
|
with patch.object(self.request, 'current_route_url',
|
|
return_value='/appinfo/configure'):
|
|
with patch.object(master.MasterView, 'configure_get_simple_settings',
|
|
return_value=settings):
|
|
with patch.object(master, 'Session', return_value=self.session):
|
|
|
|
# get the form page
|
|
response = view.configure()
|
|
self.assertIsInstance(response, Response)
|
|
|
|
# post request to save settings
|
|
self.request.method = 'POST'
|
|
self.request.POST = {
|
|
'wutta.app_title': 'Wutta',
|
|
'wutta.foo': 'bar',
|
|
'wutta.flag': 'true',
|
|
}
|
|
response = view.configure()
|
|
# nb. should get redirect back to configure page
|
|
self.assertIsInstance(response, HTTPFound)
|
|
|
|
# should now have 5 settings
|
|
count = self.session.query(model.Setting).count()
|
|
self.assertEqual(count, 5)
|
|
get_setting = functools.partial(self.app.get_setting, self.session)
|
|
self.assertEqual(get_setting('wutta.app_title'), 'Wutta')
|
|
self.assertEqual(get_setting('wutta.foo'), 'bar')
|
|
self.assertEqual(get_setting('wutta.flag'), 'true')
|
|
self.assertEqual(get_setting('wutta.number'), '42')
|
|
self.assertEqual(get_setting('wutta.value1'), '')
|
|
self.assertEqual(get_setting('wutta.value2'), None)
|
|
|
|
# post request to remove settings
|
|
self.request.method = 'POST'
|
|
self.request.POST = {'remove_settings': '1'}
|
|
response = view.configure()
|
|
# nb. should get redirect back to configure page
|
|
self.assertIsInstance(response, HTTPFound)
|
|
|
|
# should now have 0 settings
|
|
count = self.session.query(model.Setting).count()
|
|
self.assertEqual(count, 0)
|
|
|
|
# teardown
|
|
del master.MasterView.model_name
|
|
del master.MasterView.route_prefix
|
|
del master.MasterView.template_prefix
|