add the widget check asserts

marazmiki · Jan 20, 2015 · b68d857 · b68d857
1 parent d620fc3
commit b68d857
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/disguise/tests/__init__.py b/disguise/tests/__init__.py
@@ -175,12 +175,14 @@ def test_mask_myself(self):
         self.assertEquals(200, resp.status_code)
         self.assertEquals(self.user, resp.context['request'].user)
         self.assertEquals(self.root, resp.context['request'].original_user)
+        self.assertContains(resp, '<form ')
 
         # Disguise to superuser again
         resp = self.mask_request(username='root', follow=True)
         self.assertEquals(200, resp.status_code)
         self.assertEquals(self.root, resp.context['request'].original_user)
         self.assertEquals(self.root, resp.context['request'].user)
+        self.assertContains(resp, '<form ')
 
     def test_unmask(self):
         self.mask_request(username='user', follow=True)