test_data.py

from django.conf import settings
from django.contrib.auth.models import User

from ietf.iesg.models import TelechatDate, WGAction
from ietf.ipr.models import IprDetail, IprDocAlias
from ietf.meeting.models import Meeting
from ietf.doc.models import *
from ietf.doc.utils import *
from ietf.name.models import *
from ietf.group.models import *
from ietf.person.models import *

def make_test_data():
    # telechat dates
    t = datetime.date.today()
    old = TelechatDate.objects.create(date=t - datetime.timedelta(days=14)).date
    date1 = TelechatDate.objects.create(date=t).date
    date2 = TelechatDate.objects.create(date=t + datetime.timedelta(days=14)).date
    date3 = TelechatDate.objects.create(date=t + datetime.timedelta(days=14 * 2)).date
    date4 = TelechatDate.objects.create(date=t + datetime.timedelta(days=14 * 3)).date

    # groups
    secretariat = Group.objects.create(
        name="Secretariat",
        acronym="secretariat",
        state_id="active",
        type_id="ietf",
        parent=None)
    ietf = Group.objects.create(
        name="IETF",
        acronym="ietf",
        state_id="active",
        type_id="ietf",
        parent=None)
    for x in ["irtf", "iab", "ise", "iesg"]:
        Group.objects.create(
            name=x.upper(),
            acronym=x,
            state_id="active",
            type_id="ietf",
            parent=None)
    area = Group.objects.create(
        name="Far Future",
        acronym="farfut",
        state_id="active",
        type_id="area",
        parent=ietf)
    individ = Group.objects.create(
        name="Individual submissions",
        acronym="none",
        state_id="active",
        type_id="individ",
        parent=None)
    # mars WG
    mars_wg = group = Group.objects.create(
        name="Martian Special Interest Group",
        acronym="mars",
        state_id="active",
        type_id="wg",
        parent=area,
        )
    charter = Document.objects.create(
        name="charter-ietf-" + group.acronym,
        type_id="charter",
        title=group.name,
        group=group,
        rev="00",
        )
    charter.set_state(State.objects.get(used=True, slug="approved", type="charter"))
    group.charter = charter
    group.save()
    DocAlias.objects.create(
        name=charter.name,
        document=charter
        )
    # ames WG
    group = Group.objects.create(
        name="Asteroid Mining Equipment Standardization Group",
        acronym="ames",
        state_id="proposed",
        type_id="wg",
        parent=area,
        )
    charter = Document.objects.create(
        name="charter-ietf-" + group.acronym,
        type_id="charter",
        title=group.name,
        group=group,
        rev="00",
        )
    charter.set_state(State.objects.get(used=True, slug="infrev", type="charter"))
    DocAlias.objects.create(
        name=charter.name,
        document=charter
        )
    group.charter = charter
    group.save()
    WGAction.objects.create(
        pk=group.pk,
        note="",
        status_date=datetime.date.today(),
        agenda=1,
        token_name="Aread",
        category=13,
        telechat_date=date2
        )

    # persons

    # system
    system_person = Person.objects.create(
        id=0, # special value
        name="(System)",
        ascii="(System)",
        address="",
        )

    # IANA and RFC Editor groups
    iana = Group.objects.create(
        name="IANA",
        acronym="iana",
        state_id="active",
        type_id="ietf",
        parent=None,
        )
    rfc_editor = Group.objects.create(
        name="RFC Editor",
        acronym="rfceditor",
        state_id="active",
        type_id="ietf",
        parent=None,
        )
    
    if system_person.id != 0: # work around bug in Django
        Person.objects.filter(id=system_person.id).update(id=0)
        system_person = Person.objects.get(id=0)

    Alias.objects.get_or_create(person=system_person, name=system_person.name)
    Email.objects.get_or_create(address="", person=system_person)

    # plain IETF'er
    u = User.objects.create(username="plain")
    plainman = Person.objects.create(
        name="Plain Man",
        ascii="Plain Man",
        user=u)
    email = Email.objects.create(
        address="plain@example.com",
        person=plainman)
    
    # ad
    u = User.objects.create(username="ad")
    ad = p = Person.objects.create(
        name="Aread Irector",
        ascii="Aread Irector",
        user=u)
    email = Email.objects.create(
        address="aread@ietf.org",
        person=p)
    Role.objects.create(
        name_id="ad",
        group=area,
        person=p,
        email=email)

    # create a bunch of ads for swarm tests
    for i in range(1, 10):
        u = User.objects.create(username="ad%s" % i)
        p = Person.objects.create(
            name="Ad No%s" % i,
            ascii="Ad No%s" % i,
            user=u)
        email = Email.objects.create(
            address="ad%s@ietf.org" % i,
            person=p)
        if i < 6:
            # active
            Role.objects.create(
                name_id="ad",
                group=area,
                person=p,
                email=email)
        else:
            areahist = GroupHistory.objects.create(
                group=area,
                name=area.name,
                acronym=area.acronym,
                type_id=area.type_id,
                state_id=area.state_id,
                parent=area.parent
                )
            RoleHistory.objects.create(
                name_id="ad",
                group=areahist,
                person=p,
                email=email)
    
    # stream chairs
    for stream in ['ietf','irtf','iab','iesg']:
        u = User.objects.create( username= ("%schair"%stream) )
        p = Person.objects.create(
            name="%s chair"%stream,
            ascii="%s chair"%stream,
            user = u,
            )
        chairmail = Email.objects.create(
            address="%schair@ietf.org"%stream,
            person = p,
            )
        Role.objects.create(
            name_id = "chair",
            group = Group.objects.get(acronym=stream),
            person = p,
            email = chairmail,
            )

    # group chair
    u = User.objects.create(username="marschairman")
    p = Person.objects.create(
        name="WG Chair Man",
        ascii="WG Chair Man",
        user=u
        )
    wgchair = Email.objects.create(
        address="wgchairman@ietf.org",
        person=p)
    Role.objects.create(
        name_id="chair",
        group=mars_wg,
        person=p,
        email=wgchair,
        )

    # group delegate
    u = User.objects.create(username="wgdelegate")
    p = Person.objects.create(
        name="WG Delegate",
        ascii="WG Delegate",
        user=u
        )
    email = Email.objects.create(
        address="wgdelegate@ietf.org",
        person=p)
    Role.objects.create(
        name_id="delegate",
        group=mars_wg,
        person=p,
        email=email,
        )

    # secretary
    u = User.objects.create(username="secretary")
    p = Person.objects.create(
        name="Sec Retary",
        ascii="Sec Retary",
        user=u)
    email = Email.objects.create(
        address="sec.retary@ietf.org",
        person=p)
    Role.objects.create(
        name_id="secr",
        group=secretariat,
        person=p,
        email=email,
        )

    # IANA user
    u = User.objects.create(username="iana")
    p = Person.objects.create(
        name="Ina Iana",
        ascii="Ina Iana",
        user=u)
    Alias.objects.create(
        name=p.name,
        person=p)
    email = Email.objects.create(
        address="iana@ia.na",
        person=p)
    Role.objects.create(
        name_id="auth",
        group=iana,
        email=email,
        person=p,
        )

    # RFC Editor user
    u = User.objects.create(username="rfc")
    p = Person.objects.create(
        name="Rfc Editor",
        ascii="Rfc Editor",
        user=u)
    email = Email.objects.create(
        address="rfc@edit.or",
        person=p)
    Role.objects.create(
        name_id="auth",
        group=rfc_editor,
        email=email,
        person=p,
        )

    
    # draft
    draft = Document.objects.create(
        name="draft-ietf-mars-test",
        time=datetime.datetime.now(),
        type_id="draft",
        title="Optimizing Martian Network Topologies",
        stream_id="ietf",
        group=mars_wg,
        abstract="Techniques for achieving near-optimal Martian networks.",
        rev="01",
        pages=2,
        intended_std_level_id="ps",
        shepherd=plainman,
        ad=ad,
        expires=datetime.datetime.now() + datetime.timedelta(days=settings.INTERNET_DRAFT_DAYS_TO_EXPIRE),
        notify="aliens@example.mars",
        note="",
        )

    draft.set_state(State.objects.get(used=True, type="draft", slug="active"))
    draft.set_state(State.objects.get(used=True, type="draft-iesg", slug="pub-req"))
    draft.set_state(State.objects.get(used=True, type="draft-stream-%s" % draft.stream_id, slug="wg-doc"))

    doc_alias = DocAlias.objects.create(
        document=draft,
        name=draft.name,
        )

    DocumentAuthor.objects.create(
        document=draft,
        author=Email.objects.get(address="aread@ietf.org"),
        order=1
        )

    # fill in some useful default events
    DocEvent.objects.create(
        type="started_iesg_process",
        by=ad,
        doc=draft,
        desc="Started IESG process",
        )

    BallotDocEvent.objects.create(
        type="created_ballot",
        ballot_type=BallotType.objects.get(doc_type="draft", slug="approve"),
        by=ad,
        doc=draft,
        desc="Created ballot",
        )

    # IPR
    ipr = IprDetail.objects.create(
        title="Statement regarding rights",
        legal_name="Native Martians United",
        is_pending=0,
        applies_to_all=1,
        licensing_option=1,
        lic_opt_a_sub=2,
        lic_opt_b_sub=2,
        lic_opt_c_sub=2,
        comments="",
        lic_checkbox=True,
        other_notes="",
        status=1,
        submitted_date=datetime.date.today(),
        )

    IprDocAlias.objects.create(
        ipr=ipr,
        doc_alias=doc_alias,
        rev="00",
        )
    
    # meeting
    Meeting.objects.create(
        number="42",
        type_id="ietf",
        date=datetime.date.today() + datetime.timedelta(days=180),
        city="New York",
        country="US",
        time_zone="US/Eastern",
        break_area="Lounge",
        reg_area="Lobby",
        )

    # an independent submission before review
    doc = Document.objects.create(name='draft-imaginary-independent-submission',type_id='draft')
    DocAlias.objects.create(      name='draft-imaginary-independent-submission',document=doc)

    # an irtf submission mid review
    doc = Document.objects.create(     name='draft-imaginary-irtf-submission',type_id='draft')
    docalias = DocAlias.objects.create(name='draft-imaginary-irtf-submission',document=doc)
    doc.stream = StreamName.objects.get(slug='irtf')
    doc.save()
    crdoc = Document.objects.create(name='conflict-review-imaginary-irtf-submission',type_id='conflrev',rev='00',notify="fsm@ietf.org")
    DocAlias.objects.create(        name='conflict-review-imaginary-irtf-submission',document=crdoc)
    crdoc.set_state(State.objects.get(used=True, name='Needs Shepherd',type__slug='conflrev'))
    crdoc.save()
    crdoc.relateddocument_set.create(target=docalias,relationship_id='conflrev')
    
    return draft