Admin: add a disable action

1 files changed, 29 insertions, 0 deletions

diff --git a/chimere/tests.py b/chimere/tests.py
index 5e0ba7c..eeb970c 100644
--- a/chimere/tests.py
+++ b/chimere/tests.py
@@ -212,6 +212,35 @@ class FeedsTest(TestCase):
         doc = lxml.etree.fromstring(response.content)
         self.assertEqual(int(doc.xpath('count(//item)')), 2)
 
+class AdminTest(TestCase):
+    def setUp(self):
+        self.areas = areas_setup()
+        self.markers = marker_setup()
+        self.adminuser = User.objects.create_superuser('admin',
+                                                  'admin@test.com',
+                                                  'pass')
+        self.client.login(username='admin', password='pass')
+
+    def test_actions(self):
+        q_markers = Marker.objects.filter(pk__in=[m.pk for m in self.markers])
+        # disable
+        response = self.client.post('/admin/chimere/marker/',
+                       data={'action':['disable'],
+                             '_selected_action':[unicode(m.pk)
+                                                 for m in self.markers],
+                            })
+        self.assertEqual(q_markers.count(),
+                         q_markers.filter(status='D').count())
+        # validate
+        response = self.client.post('/admin/chimere/marker/',
+                       data={'action':['validate'],
+                             '_selected_action':[unicode(m.pk)
+                                                 for m in self.markers]
+                            })
+        self.assertEqual(q_markers.count(),
+                         q_markers.filter(status='A').count())
+
+
 class MarkerFormTest(TestCase):
     def setUp(self):
         self.subcategories = subcategory_setup()