def test_signal_sent(self):
def handler_factory(name):
def myhandler(sender, action, request, queryset, **kwargs):
handler_factory.invoked[name] = True
self.assertEqual(action, self.action_name)
self.assertSequenceEqual(queryset.order_by('id').values_list('id', flat=True),
sorted(self._selected_values))
return myhandler
handler_factory.invoked = {}
try:
m1 = handler_factory('adminaction_requested')
adminaction_requested.connect(m1, sender=self.sender_model)
m2 = handler_factory('adminaction_start')
adminaction_start.connect(m2, sender=self.sender_model)
m3 = handler_factory('adminaction_end')
adminaction_end.connect(m3, sender=self.sender_model)
self._run_action()
self.assertIn('adminaction_requested', handler_factory.invoked)
self.assertIn('adminaction_start', handler_factory.invoked)
self.assertIn('adminaction_end', handler_factory.invoked)
finally:
adminaction_requested.disconnect(m1, sender=self.sender_model)
adminaction_start.disconnect(m2, sender=self.sender_model)
adminaction_end.disconnect(m3, sender=self.sender_model)
def test_signal_sent(self):
def handler_factory(name):
def myhandler(sender, action, request, queryset, **kwargs):
handler_factory.invoked[name] = True
self.assertEqual(action, self.action_name)
self.assertSequenceEqual(
queryset.order_by('id').values_list('id', flat=True),
sorted(self._selected_values))
return myhandler
handler_factory.invoked = {}
try:
m1 = handler_factory('adminaction_requested')
adminaction_requested.connect(m1, sender=self.sender_model)
m2 = handler_factory('adminaction_start')
adminaction_start.connect(m2, sender=self.sender_model)
m3 = handler_factory('adminaction_end')
adminaction_end.connect(m3, sender=self.sender_model)
self._run_action()
self.assertIn('adminaction_requested', handler_factory.invoked)
self.assertIn('adminaction_start', handler_factory.invoked)
self.assertIn('adminaction_end', handler_factory.invoked)
finally:
adminaction_requested.disconnect(m1, sender=self.sender_model)
adminaction_start.disconnect(m2, sender=self.sender_model)
adminaction_end.disconnect(m3, sender=self.sender_model)
def test_transaction_mass_update(app, users, administrator):
assert User.objects.filter(is_staff=True).count() == 1 # sanity check
def _handler(*args, **kwargs):
raise ActionInterrupted()
res = app.get('/admin/', user=administrator.username)
res = res.click('Users')
form = res.forms['changelist-form']
form['action'] = 'mass_update'
form.get('_selected_action', index=0).checked = True
form.get('_selected_action', index=1).checked = True
form.get('_selected_action', index=2).checked = True
res = form.submit()
res.form['chk_id_is_staff'].checked = True
res.form['is_staff'].checked = True
# res.form.submit('apply').follow()
#
# assert User.objects.filter(is_staff=True).count() == 1
with pytest.raises(Exception):
adminaction_end.connect(_handler)
res.form.submit('apply').follow()
adminaction_end.disconnect(_handler)
assert User.objects.filter(is_staff=True).count() == 1
def test_signal_end(self):
# test if adminaction_start Signal can stop the action
def myhandler(sender, action, request, queryset, **kwargs):
myhandler.invoked = True
self.assertEqual(action, self.action_name)
self.assertSequenceEqual(queryset.order_by('id').values_list('id', flat=True), self.selected_rows)
try:
adminaction_end.connect(myhandler, sender=self.sender_model)
self._run_action(code3=200)
self.assertTrue(myhandler.invoked)
finally:
adminaction_end.disconnect(myhandler, sender=self.sender_model)
def test_happy_path(self):
url = reverse('admin:auth_user_changelist')
def myhandler(sender, action, request, queryset, **kwargs):
self.assertEqual(action, self.action_name)
self.assertSequenceEqual(queryset.order_by('id').values_list('id', flat=True), self.selected_rows)
try:
adminaction_start.connect(myhandler, sender=User)
adminaction_end.connect(myhandler, sender=User)
adminaction_requested.connect(myhandler, sender=User)
response = self._run_action(url=url, header=1)
self.assertIn('Content-Disposition', response)
self.assertIn('attachment;filename=', response['Content-Disposition'])
finally:
adminaction_start.disconnect(myhandler, sender=User)
adminaction_end.disconnect(myhandler, sender=User)
adminaction_requested.disconnect(myhandler, sender=User)
def test_happy_path(self):
url = reverse('admin:auth_user_changelist')
def myhandler(sender, action, request, queryset, **kwargs):
self.assertEqual(action, self.action_name)
self.assertSequenceEqual(
queryset.order_by('id').values_list('id', flat=True),
self.selected_rows)
try:
adminaction_start.connect(myhandler, sender=User)
adminaction_end.connect(myhandler, sender=User)
adminaction_requested.connect(myhandler, sender=User)
response = self._run_action(url=url, header=1)
self.assertIn('Content-Disposition', response)
self.assertIn('attachment;filename=',
response['Content-Disposition'])
finally:
adminaction_start.disconnect(myhandler, sender=User)
adminaction_end.disconnect(myhandler, sender=User)
adminaction_requested.disconnect(myhandler, sender=User)
