def test_function(self): """register_view works as a function.""" site = AdminSitePlus() def foo(request): return 'foo' site.register_view('foo', view=foo) urls = site.get_urls() assert any(u.resolve('foo') for u in urls)
def test_visibility(self): """Make sure visibility works.""" site = AdminSitePlus() req_factory = RequestFactory() def always_visible(request): return 'i am here' site.register_view('always-visible', view=always_visible, visible=True) def always_hidden(request): return 'i am here, but not shown' site.register_view('always-hidden', view=always_visible, visible=False) cond = lambda req: req.user.pk == 1 b = lambda s: s.encode('ascii') if hasattr(s, 'encode') else s @site.register_view(r'conditional-view', visible=cond) class ConditionallyVisible(View): def get(self, request): return 'hi there' urls = site.get_urls() assert any(u.resolve('always-visible') for u in urls) assert any(u.resolve('always-hidden') for u in urls) assert any(u.resolve('conditional-view') for u in urls) class MockUser(object): is_active = True is_staff = True def __init__(self, pk): self.pk = pk self.id = pk req_show = req_factory.get('/admin/') req_show.user = MockUser(1) result = site.index(req_show).render().content assert b('always-visible') in result assert b('always-hidden') not in result assert b('conditional-view') in result req_hide = req_factory.get('/admin/') req_hide.user = MockUser(2) result = site.index(req_hide).render().content assert b('always-visible') in result assert b('always-hidden') not in result assert b('conditional-view') not in result
def test_path(self): """Setting the path works correctly.""" site = AdminSitePlus() def foo(request): return 'foo' site.register_view('foo', view=foo) site.register_view('bar/baz', view=foo) site.register_view('baz-qux', view=foo) urls = site.get_urls() foo_urls = [u for u in urls if u.resolve('foo')] self.assertEqual(1, len(foo_urls)) bar_urls = [u for u in urls if u.resolve('bar/baz')] self.assertEqual(1, len(bar_urls)) qux_urls = [u for u in urls if u.resolve('baz-qux')] self.assertEqual(1, len(qux_urls))
def test_path(self): """Setting the path works correctly.""" site = AdminSitePlus() def foo(request): return 'foo' site.register_view('foo', view=foo) site.register_view('bar/baz', view=foo) site.register_view('baz-qux', view=foo) urls = site.get_urls() matches = lambda u: lambda p: p.resolve(u) foo_urls = filter(matches('foo'), urls) self.assertEqual(1, len(foo_urls)) bar_urls = filter(matches('bar/baz'), urls) self.assertEqual(1, len(bar_urls)) qux_urls = filter(matches('baz-qux'), urls) self.assertEqual(1, len(qux_urls))
actions = [ export_to_app_excel, export_all_info_excel, email_guests_initial, email_guests_reminder, statistics_admin_action, export_rides_action, set_to_default_action ] def save_model(self, request, obj, form, change): """Add and remove guest person using the checkbox""" obj.save() has_guest = obj.has_guest_person() person_name = request.POST["person_set-0-name"] is_english = str_is_english(person_name) if not has_guest and obj.with_guest: obj.create_guest(is_english) if has_guest and not obj.with_guest: people = Person.objects.filter(invitation=obj.id) for person in people: if person.is_guest(): person.delete() obj.save() def statistics(request): stats = Statistics() return render(request, 'admin/statistics.html', {'stats': stats}) site.register_view('statistics.html', view=statistics) site.register(Invitation, InvitationAdmin)