From b7375412ab5c570de6fdf3ad5986268778ed8c23 Mon Sep 17 00:00:00 2001
From: Robert Sparks <rjsparks@nostrum.com>
Date: Tue, 24 Apr 2018 19:22:09 +0000
Subject: [PATCH] Improved testing for docs_for_ad.  - Legacy-Id: 15075

---
 ietf/doc/tests.py | 33 +++++++++++++++++++++++++++------
 1 file changed, 27 insertions(+), 6 deletions(-)

diff --git a/ietf/doc/tests.py b/ietf/doc/tests.py
index 5d56ac43b..723efd03e 100644
--- a/ietf/doc/tests.py
+++ b/ietf/doc/tests.py
@@ -22,7 +22,7 @@ import debug                            # pyflakes:ignore
 
 from ietf.doc.models import ( Document, DocAlias, DocRelationshipName, RelatedDocument, State,
     DocEvent, BallotPositionDocEvent, LastCallDocEvent, WriteupDocEvent, NewRevisionDocEvent )
-from ietf.doc.factories import DocumentFactory, DocEventFactory
+from ietf.doc.factories import DocumentFactory, DocEventFactory, CharterFactory
 from ietf.doc.utils import create_ballot_if_not_open
 from ietf.group.models import Group
 from ietf.group.factories import GroupFactory
@@ -201,13 +201,34 @@ class SearchTests(TestCase):
         self.assertEqual(r.status_code, 200)
         self.assertTrue("Document Search" in unicontent(r))
 
-    def test_drafts_pages(self):
-        draft = make_test_data()
-
-        r = self.client.get(urlreverse('ietf.doc.views_search.docs_for_ad', kwargs=dict(name=draft.ad.full_name_as_key())))
+    def test_docs_for_ad(self):
+        ad = PersonFactory()
+        draft = DocumentFactory(type_id='draft',ad=ad)
+        draft.set_state(State.objects.get(type='draft', slug='active'))
+        draft.set_state(State.objects.get(type='draft-iesg', slug='lc'))
+        rfc = DocumentFactory(type_id='draft',ad=ad)
+        rfc.set_state(State.objects.get(type='draft', slug='rfc'))
+        rfc.docalias_set.create(name='rfc6666')
+        conflrev = DocumentFactory(type_id='conflrev',ad=ad)
+        conflrev.set_state(State.objects.get(type='conflrev', slug='iesgeval'))
+        statchg = DocumentFactory(type_id='statchg',ad=ad)
+        statchg.set_state(State.objects.get(type='statchg', slug='iesgeval'))
+        charter = CharterFactory(ad=ad)
+        charter.set_state(State.objects.get(type='charter', slug='iesgrev'))
+        
+        r = self.client.get(urlreverse('ietf.doc.views_search.docs_for_ad', kwargs=dict(name=ad.full_name_as_key())))
         self.assertEqual(r.status_code, 200)
-        self.assertTrue(draft.title in unicontent(r))
+        response_content = unicontent(r)
+        #debug.show('response_content')
+        self.assertTrue(draft.name in response_content)
+        self.assertTrue(rfc.canonical_name() in response_content)
+        self.assertTrue(conflrev.name in response_content)
+        self.assertTrue(statchg.name in response_content)
+        self.assertTrue(charter.name in response_content)
+        
 
+    def test_drafts_in_last_call(self):
+        draft = make_test_data()
         draft.set_state(State.objects.get(type="draft-iesg", slug="lc"))
         r = self.client.get(urlreverse('ietf.doc.views_search.drafts_in_last_call'))
         self.assertEqual(r.status_code, 200)