Synnefo

import factory

import models

def prefix_seq(x):

    return lambda n: x + '-{0}'.format(n)

def user_seq():

    return lambda n: 'user-{0}.example.com'.format(n)

def round_seq(x):

    size = len(x)

    return lambda n: x[int(n) % size]

def round_seq_first(x):

    size = len(x)

    return lambda n: x[int(n) % size][0]

class NetworkFactory(factory.DjangoModelFactory):

    FACTORY_FOR = models.Network

    name = factory.Sequence(prefix_seq('network'))

    userid = factory.Sequence(user_seq())

    flavor = factory.Sequence(round_seq(models.Network.FLAVORS.keys()))

    mode = factory.LazyAttribute(lambda a:

                                 models.Network.FLAVORS[a.flavor]['mode'])

    link = factory.Sequence(prefix_seq('link'))

    mac_prefix = 'aa:00:0'

    tags = factory.LazyAttribute(lambda a:

                                 models.Network.FLAVORS[a.flavor]['tags'])

    public = False

    deleted = False

    state = factory.Sequence(round_seq_first(models.Network.OPER_STATES))