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" alive = True @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