def test_get_with_superuser(self): user = UserF.create(**{'is_superuser': True}) inactive_users = UserF.create_batch(3, **{'is_active': False}) self.request.user = user response = self.view(self.request).render() rendered = render_to_string( 'superusertools/manage_inactiveusers.html', { 'inactive_users': inactive_users, 'user': user, 'PLATFORM_NAME': get_current_site(self.request).name, 'GEOKEY_VERSION': version.get_version() } ) self.assertEqual(response.status_code, 200) self.assertEqual(response.content.decode('utf-8'), rendered)
def test_get_context_data(self): UserF.create_batch(2, **{'is_superuser': True}) UserF.create_batch(2, **{'is_superuser': False}) view = ManageSuperUsers() context = view.get_context_data() self.assertEqual(len(context.get('superusers')), 2)