@@ -364,18 +364,20 @@ def test_edit_relations(self):
364364 statchg_relation_row_blah = "toexp" ,
365365 new_relation_row_foo = "rfc9998" ,
366366 statchg_relation_row_foo = "tobcp" ,
367+ new_relation_row_nob = "rfc14" ,
368+ statchg_relation_row_nob = "tohist" ,
367369 Submit = "Submit" ))
368370 self .assertEqual (r .status_code , 302 )
369371 doc = Document .objects .get (name = 'status-change-imaginary-mid-review' )
370- self .assertEqual (doc .relateddocument_set .count (),2 )
371- verify9999 = doc . relateddocument_set . filter ( target__name = 'rfc9999' )
372- self . assertTrue ( verify9999 )
373- self .assertEqual ( verify9999 . count (), 1 )
374- self .assertEqual (verify9999 [ 0 ]. relationship . slug , 'toexp' )
375- verify9998 = doc . relateddocument_set . filter ( target__name = 'rfc9998' )
376- self . assertTrue ( verify9998 )
377- self . assertEqual ( verify9998 . count (), 1 )
378- self . assertEqual ( verify9998 [ 0 ]. relationship . slug , 'tobcp ' )
372+ self .assertEqual (doc .relateddocument_set .count (),3 )
373+ def verify_relations ( doc , target_name , status ):
374+ target_doc = doc . relateddocument_set . filter ( target__name = target_name )
375+ self .assertTrue ( target_doc )
376+ self .assertEqual (target_doc . count (), 1 )
377+ self . assertEqual ( target_doc [ 0 ]. relationship . slug , status )
378+ verify_relations ( doc , 'rfc9999' , 'toexp' )
379+ verify_relations ( doc , 'rfc9998' , 'tobcp' )
380+ verify_relations ( doc , 'rfc14' , 'tohist ' )
379381 self .assertTrue (doc .latest_event (DocEvent ,type = "added_comment" ).desc .startswith ('Affected RFC list changed.' ))
380382
381383 def setUp (self ):
0 commit comments