Skip to content

Commit 6f011a3

Browse files
committed
Use factories for ietf.doc.tests_draft. Commit ready for merge.
- Legacy-Id: 15225
1 parent dcf9e6b commit 6f011a3

3 files changed

Lines changed: 83 additions & 108 deletions

File tree

ietf/doc/factories.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,8 @@
11
import debug # pyflakes:ignore
22
import factory
3+
import datetime
4+
5+
from django.conf import settings
36

47
from ietf.doc.models import Document, DocEvent, NewRevisionDocEvent, DocAlias, State, DocumentAuthor
58
from ietf.group.models import Group
@@ -21,6 +24,8 @@ class Meta:
2124
rev = '00'
2225
std_level_id = None
2326
intended_std_level_id = None
27+
time = datetime.datetime.now()
28+
expires = factory.LazyAttribute(lambda o: o.time+datetime.timedelta(days=settings.INTERNET_DRAFT_DAYS_TO_EXPIRE))
2429

2530
@factory.lazy_attribute_sequence
2631
def name(self, n):
@@ -98,6 +103,7 @@ class WgDraftFactory(BaseDocumentFactory):
98103

99104
type_id = 'draft'
100105
group = factory.SubFactory('ietf.group.factories.GroupFactory',type_id='wg')
106+
stream_id = 'ietf'
101107

102108
@factory.post_generation
103109
def states(obj, create, extracted, **kwargs):

ietf/doc/tests.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -882,7 +882,7 @@ def test_document_bibtex(self):
882882
#other_aliases = ['rfc6020',],
883883
states = [('draft','rfc'),('draft-iesg','pub')],
884884
std_level_id = 'ps',
885-
time = '2010-10-10',
885+
time = datetime.datetime(2010,10,10),
886886
)
887887
num = rfc.rfc_number()
888888
DocEventFactory.create(doc=rfc, type='published_rfc', time = '2010-10-10')
@@ -902,7 +902,7 @@ def test_document_bibtex(self):
902902
stream_id = 'rse',
903903
states = [('draft','rfc'),('draft-iesg','pub')],
904904
std_level_id = 'ind',
905-
time = '1990-04-01',
905+
time = datetime.datetime(1990,04,01),
906906
)
907907
num = april1.rfc_number()
908908
DocEventFactory.create(doc=april1, type='published_rfc', time = '1990-04-01')

0 commit comments

Comments
 (0)