diff --git a/ietf/doc/tests_draft.py b/ietf/doc/tests_draft.py index 6efa627f3..3a0f0315c 100644 --- a/ietf/doc/tests_draft.py +++ b/ietf/doc/tests_draft.py @@ -1949,3 +1949,17 @@ class MoreReplacesTests(TestCase): old_doc = Document.objects.get(name=old_doc.name) self.assertEqual(old_doc.get_state_slug('draft'),'repl') self.assertEqual(old_doc.get_state_slug('draft-stream-%s'%stream),'repl') + +class ShepherdWriteupTests(TestCase): + + def test_shepherd_writeup_generation(self): + ind_draft = IndividualDraftFactory(stream_id='ietf') + wg_draft = WgDraftFactory() + + url = urlreverse('ietf.doc.views_draft.edit_shepherd_writeup', kwargs=dict(name=ind_draft.name)) + login_testing_unauthorized(self, "secretary", url) + r = self.client.get(url) + self.assertContains(r, "for Individual Documents", status_code=200) + url = urlreverse('ietf.doc.views_draft.edit_shepherd_writeup', kwargs=dict(name=wg_draft.name)) + r = self.client.get(url) + self.assertContains(r, "for Group Documents", status_code=200) diff --git a/ietf/doc/views_draft.py b/ietf/doc/views_draft.py index 1806720c3..e0d73855b 100644 --- a/ietf/doc/views_draft.py +++ b/ietf/doc/views_draft.py @@ -1025,7 +1025,7 @@ def edit_shepherd_writeup(request, name): dict( doc=doc, type="individ" - if not doc.group.type.slug or doc.group.type.slug != "ietf" + if not doc.group.type.slug or doc.group.type.slug != "wg" else "group", stream=doc.stream.slug, group=doc.group.type.slug,