1111import debug # pyflakes:ignore
1212
1313from ietf .doc .expire import expire_draft
14- from ietf .doc .factories import DocumentFactory
14+ from ietf .doc .factories import WgDraftFactory
1515from ietf .doc .models import Document
16+ from ietf .group .factories import RoleFactory
1617from ietf .meeting .factories import MeetingFactory
17- from ietf .person .factories import PersonFactory
18+ from ietf .person .factories import PersonFactory , EmailFactory
1819from ietf .person .models import Person
1920from ietf .submit .models import Preapproval
2021from ietf .utils .mail import outbox
2122from ietf .utils .test_utils import TestCase , login_testing_unauthorized
22- from ietf .utils .test_data import make_test_data
2323from ietf .secr .drafts .email import get_email_initial
2424
2525
@@ -48,14 +48,13 @@ def tearDown(self):
4848 settings .IDSUBMIT_MANUAL_STAGING_DIR = self .saved_idsubmit_manual_staging_dir
4949
5050 def test_abstract (self ):
51- draft = make_test_data ()
51+ draft = WgDraftFactory ()
5252 url = urlreverse ('ietf.secr.drafts.views.abstract' , kwargs = {'id' :draft .name })
5353 self .client .login (username = "secretary" , password = "secretary+password" )
5454 response = self .client .get (url )
5555 self .assertEqual (response .status_code , 200 )
5656
5757 def test_approvals (self ):
58- make_test_data ()
5958 Preapproval .objects .create (name = 'draft-dummy' ,
6059 by = Person .objects .get (name = "(System)" ))
6160 url = urlreverse ('ietf.secr.drafts.views.approvals' )
@@ -65,7 +64,7 @@ def test_approvals(self):
6564 self .assertTrue ('draft-dummy' in response .content )
6665
6766 def test_edit (self ):
68- draft = make_test_data ( )
67+ draft = WgDraftFactory ( states = [( 'draft' , 'active' ),( 'draft-stream-ietf' , 'wg-doc' ),( 'draft-iesg' , 'ad-eval' )], shepherd = EmailFactory () )
6968 url = urlreverse ('ietf.secr.drafts.views.edit' , kwargs = {'id' :draft .name })
7069 self .client .login (username = "secretary" , password = "secretary+password" )
7170 response = self .client .get (url )
@@ -83,20 +82,23 @@ def test_email(self):
8382 def test_get_email_initial (self ):
8483 # Makes sure that a manual posting by the Secretariat of an I-D that is
8584 # in the RFC Editor Queue will result in notification of the RFC Editor
86- draft = make_test_data ()
85+ draft = WgDraftFactory (authors = PersonFactory .create_batch (1 ),shepherd = EmailFactory ())
86+ RoleFactory (group = draft .group , name_id = 'chair' )
8787 data = get_email_initial (draft ,action = 'extend' ,input = {'expiration_date' : '2050-01-01' })
8888 self .assertTrue ('Extension of Expiration Date' in data ['subject' ])
8989
9090 def test_makerfc (self ):
91- draft = make_test_data ( )
91+ draft = WgDraftFactory ( intended_std_level_id = 'ps' )
9292 url = urlreverse ('ietf.secr.drafts.views.edit' , kwargs = {'id' :draft .name })
9393 self .client .login (username = "secretary" , password = "secretary+password" )
9494 response = self .client .get (url )
9595 self .assertEqual (response .status_code , 200 )
96+ # It's not clear what this is testing. Was there supposed to be a POST here?
9697 self .assertTrue (draft .intended_std_level )
9798
9899 def test_search (self ):
99- draft = make_test_data ()
100+ WgDraftFactory () # Test exercises branch that requires >1 doc found
101+ draft = WgDraftFactory ()
100102 url = urlreverse ('ietf.secr.drafts.views.search' )
101103 self .client .login (username = "secretary" , password = "secretary+password" )
102104 response = self .client .get (url )
@@ -108,14 +110,14 @@ def test_search(self):
108110 self .assertTrue (draft .name in response .content )
109111
110112 def test_view (self ):
111- draft = make_test_data ()
113+ draft = WgDraftFactory ()
112114 url = urlreverse ('ietf.secr.drafts.views.view' , kwargs = {'id' :draft .name })
113115 self .client .login (username = "secretary" , password = "secretary+password" )
114116 response = self .client .get (url )
115117 self .assertEqual (response .status_code , 200 )
116118
117119 def test_author_delete (self ):
118- draft = make_test_data ( )
120+ draft = WgDraftFactory ( authors = PersonFactory . create_batch ( 2 ) )
119121 author = draft .documentauthor_set .first ()
120122 id = author .id
121123 url = urlreverse ('ietf.secr.drafts.views.author_delete' , kwargs = {'id' :draft .name , 'oid' :id })
@@ -128,7 +130,7 @@ def test_author_delete(self):
128130 self .assertFalse (draft .documentauthor_set .filter (id = id ))
129131
130132 def test_resurrect (self ):
131- draft = make_test_data ()
133+ draft = WgDraftFactory ()
132134 path = os .path .join (self .repository_dir , draft .filename_with_rev ())
133135 with open (path , 'w' ) as file :
134136 file .write ('test' )
@@ -164,7 +166,7 @@ def test_resurrect(self):
164166 self .assertEqual (recv ['Subject' ], subject )
165167
166168 def test_extend (self ):
167- draft = make_test_data ()
169+ draft = WgDraftFactory ()
168170 url = urlreverse ('ietf.secr.drafts.views.extend' , kwargs = {'id' :draft .name })
169171 email_url = urlreverse ('ietf.secr.drafts.views.email' , kwargs = {'id' :draft .name })
170172 confirm_url = urlreverse ('ietf.secr.drafts.views.confirm' , kwargs = {'id' :draft .name })
@@ -203,7 +205,7 @@ def test_extend(self):
203205 self .assertEqual (recv ['Subject' ], subject )
204206
205207 def test_withdraw (self ):
206- draft = make_test_data ()
208+ draft = WgDraftFactory ()
207209 url = urlreverse ('ietf.secr.drafts.views.withdraw' , kwargs = {'id' :draft .name })
208210 email_url = urlreverse ('ietf.secr.drafts.views.email' , kwargs = {'id' :draft .name })
209211 confirm_url = urlreverse ('ietf.secr.drafts.views.confirm' , kwargs = {'id' :draft .name })
@@ -237,7 +239,7 @@ def test_withdraw(self):
237239 self .assertEqual (recv ['Subject' ], subject )
238240
239241 def test_authors (self ):
240- draft = DocumentFactory ()
242+ draft = WgDraftFactory ()
241243 person = PersonFactory ()
242244 url = urlreverse ('ietf.secr.drafts.views.authors' ,kwargs = {'id' :draft .name })
243245 login_testing_unauthorized (self , "secretary" , url )
0 commit comments