Synnefo

import factory

import models

from random import choice

from string import letters, digits

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]

def random_string(x):

    '''Returns a random string of length x'''

    return ''.join([choice(digits + letters) for i in range(x)])

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)

    name = random_string(5)

class SubnetFactory(factory.DjangoModelFactory):

    FACTORY_FOR = models.Subnet

    name = factory.LazyAttribute(lambda self: random_string(30))

    ipversion = 4

    cidr = factory.Sequence(lambda n: '192.168.{0}.0/24'.format(n))

    dhcp = True

    gateway = factory.Sequence(lambda n: '192.168.{0}.1'.format(n))

class SecurityGroupFactory(factory.DjangoModelFactory):

    FACTORY_FOR = models.SecurityGroup

    name = factory.LazyAttribute(lambda self: random_string(30))