tests/tests/test_actions.py

   1 from django.test import TestCase
   2 from django.contrib.auth.models import User
   3 from ..models import SomeModel
   4
   5
   6 class ActionsTests(TestCase):
   7     def test_csv(self):
   8         u = User(username='user', is_superuser=True, is_staff=True)
   9         u.set_password('test')
  10         u.save()
  11
  12         SomeModel.objects.create()
  13
  14         self.client.login(username='user', password='test')
  15
  16         response = self.client.post(
  17             '/admin/tests/somemodel/',
  18             {
  19                 'action': 'export_as_csv',
  20                 '_selected_action': '1',
  21             }
  22         )
  23         self.assertEqual(
  24             response.content,
  25             b"id,attachment\r\n1,\r\n")
  26