Synnefo

import factory

import models

from synnefo.db.models_factory import VirtualMachineFactory

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))

class NetworkInterfaceFactory(factory.DjangoModelFactory):

    FACTORY_FOR = models.NetworkInterface

    machine = factory.SubFactory(VirtualMachineFactory)

    network = factory.SubFactory(NetworkFactory)