euscan-ng/euscanwww/djeuscan/tests/euscan_factory.py

import random
from string import letters
from datetime import datetime
from collections import defaultdict

from django.contrib.auth.models import User
from django.db import IntegrityError

import factory

from djeuscan.models import Herd, Maintainer, Package, Version, EuscanResult, \
    Category, Overlay


class UserFactory(factory.Factory):
    FACTORY_FOR = User

    username = factory.Sequence(lambda n: 'user' + n)


def random_string(length=None):
    if length is None:
        length = random.randint(1, 30)
    return "".join([random.choice(letters) for _ in range(length)])


class HerdFactory(factory.Factory):
    FACTORY_FOR = Herd

    herd = factory.LazyAttribute(lambda a: random_string())
    email = factory.LazyAttribute(lambda a: "%s@example.com" % a.herd)


class MaintainerFactory(factory.Factory):
    FACTORY_FOR = Maintainer

    name = factory.LazyAttribute(lambda a: random_string())
    email = factory.LazyAttribute(lambda a: "%s@example.com" % a.name)


class CategoryFactory(factory.Factory):
    FACTORY_FOR = Category

    name = factory.LazyAttribute(lambda a: random_string())


class OverlayFactory(factory.Factory):
    FACTORY_FOR = Overlay

    name = factory.LazyAttribute(lambda a: random_string())


class PackageFactory(factory.Factory):
    FACTORY_FOR = Package

    category = factory.LazyAttribute(
        lambda a: "%s-%s" % (random_string(), random_string())
    )
    name = factory.LazyAttribute(lambda a: random_string())
    description = "This is a test package"
    homepage = "http://testpackage.com"

    @classmethod
    def _prepare(cls, create, **kwargs):
        package = super(PackageFactory, cls)._prepare(create, **kwargs)
        category = kwargs.pop('category', None)
        if create:
            CategoryFactory.create(name=category)
        else:
            CategoryFactory.build(name=category)
        return package


class VersionFactory(factory.Factory):
    FACTORY_FOR = Version

    package = factory.LazyAttribute(lambda a: PackageFactory())
    slot = "1"
    revision = "1"
    version = "0.1"
    packaged = True
    overlay = "gentoo"
    urls = "http://packageurl.com"

    @classmethod
    def _prepare(cls, create, **kwargs):
        version = super(VersionFactory, cls)._prepare(create, **kwargs)
        overlay = kwargs.pop('overlay', None)
        try:
            if create:
                OverlayFactory.create(name=overlay)
            else:
                OverlayFactory.build(name=overlay)
        except IntegrityError:
            pass
        return version


class EuscanResultFactory(factory.Factory):
    FACTORY_FOR = EuscanResult

    package = factory.LazyAttribute(lambda a: PackageFactory())
    datetime = datetime.now()
    result = "this is the result"


def setup_maintainers():
    maintainers = [MaintainerFactory.create() for _ in range(10)]
    packages = []
    for i in range(0, 10, 2):
        p = PackageFactory.create()
        p.maintainers.add(maintainers[i])
        p.maintainers.add(maintainers[i + 1])
        packages.append(p)
    return maintainers, packages


def setup_categories():
    packages = [PackageFactory.create() for _ in range(10)]
    categories = [p.category for p in packages]
    return categories, packages


def setup_herds():
    herds = [HerdFactory.create() for _ in range(10)]
    packages = []

    for i in range(0, 10, 2):
        p = PackageFactory.create()
        p.herds.add(herds[i])
        p.herds.add(herds[i + 1])
        packages.append(p)
    return herds, packages


def setup_overlays():
    overlays = [random_string() for _ in range(3)]
    packages = defaultdict(list)

    for _ in range(3):
        package = PackageFactory.create()
        for overlay in overlays:
            VersionFactory.create(package=package,
                                  overlay=overlay)
            packages[overlay].append(package)

    return overlays, packages