factories

ContractFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class ContractFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.Contract
        django_get_or_create = ("contract_identifier",)

    created = factory.LazyFunction(timezone.now)
    modified = factory.LazyFunction(timezone.now)
    title = factory.Dict({"en": factory.Sequence(lambda n: f"contract-{n}")})
    quota = factory.Faker("random_number")
    validity_start_date = factory.LazyFunction(timezone.now)
    validity_end_date = factory.Faker(
        "date_between", start_date=factory.SelfAttribute("..validity_start_date")
    )

    class Params:
        identifier_uuid = factory.Faker("uuid4")

    @factory.lazy_attribute
    def contract_identifier(self):
        return f"urn:nbn:fi:contract:{self.identifier_uuid}"

PreservationFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class PreservationFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.Preservation

    contract = factory.SubFactory(ContractFactory)
    state = factory.Iterator(
        iter(models.Preservation.PreservationState.choices), getter=lambda value: value[0]
    )
    description = factory.Sequence(lambda n: {"en": f"description-for-preservation-entry-{n}"})
    reason_description = factory.Sequence(
        lambda n: f"reason-description-for-preservation-entry-{n}"
    )

CatalogHomePageFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class CatalogHomePageFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.CatalogHomePage
        django_get_or_create = ("url",)

    title = factory.Dict({"en": factory.Sequence(lambda n: f"catalog-homepage-{n}")})

    @factory.sequence
    def url(self):
        return f"https://catalog-homepage-{self}.fi"

DatasetPublisherFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class DatasetPublisherFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.DatasetPublisher
        django_get_or_create = ("name",)
        skip_postgeneration_save = True

    name = factory.Dict({"en": factory.Sequence(lambda n: f"dataset-publisher-{n}")})

    @factory.post_generation
    def homepages(self, create, extracted, **kwargs):
        if not create:
            return
        if extracted:
            for homepage in extracted:
                self.homepage.add(homepage)

AccessTypeFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class AccessTypeFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.AccessType
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"dataset-access-type-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://dataset-access-type-{self}.fi"

RestrictionGroundsFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class RestrictionGroundsFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.RestrictionGrounds
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"restriction-grounds-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://dataset-restriction-grounds-{self}.fi"

FunderTypeFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class FunderTypeFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.FunderType
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"dataset-funder-type-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://dataset-funder-type-{self}.fi"

FieldOfScienceFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class FieldOfScienceFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.FieldOfScience
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"dataset-field-of-science-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://dataset-field-of-science-{self}.fi"

InfrastructureFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class InfrastructureFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.ResearchInfra
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"dataset-infrastructure-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://dataset-infrastructure-{self}.fi"

LifecycleEventFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class LifecycleEventFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.LifecycleEvent
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"lifecycle-event-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://lifecycle-event-{self}.fi"

PreservationEventFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class PreservationEventFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.PreservationEvent
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"preservation-event-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://preservation-event-{self}.fi"

EventOutcomeFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class EventOutcomeFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.EventOutcome
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"event-outcome-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://event-outcome-{self}.fi"

FileTypeFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class FileTypeFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.FileType
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"dataset-file-type-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://dataset-file-type-{self}.fi"

LanguageFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class LanguageFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.Language
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"dataset-language-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://dataset-language-{self}.fi"

ResourceTypeFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class ResourceTypeFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.ResourceType
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"resource-type-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://resource-type-{self}.fi"

RelationTypeFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class RelationTypeFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.RelationType
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"relation-type-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://relation-type-{self}.fi"

LicenseFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class LicenseFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = refdata.License
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"license-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://license-{self}.fi"

DatasetLicenseFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class DatasetLicenseFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.DatasetLicense

    reference = factory.SubFactory(
        LicenseFactory,
        url="http://uri.suomi.fi/codelist/fairdata/license/code/other",
        in_scheme="http://uri.suomi.fi/codelist/fairdata/license",
        pref_label={
            "en": "Other",
            "fi": "Muu",
        },
    )

ThemeFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class ThemeFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.Theme
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"dataset-theme-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://dataset-theme-{self}.fi"

UseCategoryFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class UseCategoryFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.UseCategory
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"dataset-use-category-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://dataset-use-category-{self}.fi"

AccessRightsFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class AccessRightsFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.AccessRights
        django_get_or_create = ("description",)
        skip_postgeneration_save = True

    access_type = factory.SubFactory(AccessTypeFactory)
    description = factory.Dict({"en": factory.Faker("paragraph")})

    @factory.post_generation
    def license(self, create, extracted, **kwargs):
        if not create:
            return
        if extracted:
            for license in extracted:
                self.license.add(license)
        else:
            self.license.set([DatasetLicenseFactory()])

    @factory.post_generation
    def restriction_grounds(self, create, extracted, **kwargs):
        if not create:
            return
        if extracted:
            for restriction_grounds in extracted:
                self.restriction_grounds.add(restriction_grounds)
        else:
            self.restriction_grounds.set([RestrictionGroundsFactory()])

DataCatalogFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class DataCatalogFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.DataCatalog
        django_get_or_create = ("id",)
        skip_postgeneration_save = True

    @factory.sequence
    def id(self):
        return f"urn:data-catalog-{self}"

    title = factory.Dict({"en": factory.Sequence(lambda n: f"data-catalog-{n}")})
    publisher = factory.SubFactory(DatasetPublisherFactory)
    system_creator = factory.SubFactory(MetaxUserFactory)
    allow_remote_resources = True
    storage_services = ["ida", "pas", "test"]
    publishing_channels = ["etsin", "ttv"]

    @factory.post_generation
    def languages(self, create, extracted, **kwargs):
        if not create:
            return
        if extracted:
            for language in extracted:
                self.language.add(language)

MetadataProviderFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class MetadataProviderFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.MetadataProvider

    user = factory.SubFactory(MetaxUserFactory)
    system_creator = factory.SubFactory(MetaxUserFactory)

CatalogRecordFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class CatalogRecordFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.CatalogRecord

    data_catalog = factory.SubFactory(DataCatalogFactory)
    metadata_owner = factory.SubFactory(MetadataProviderFactory)

DatasetFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class DatasetFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.Dataset
        django_get_or_create = ("title",)

    data_catalog = factory.SubFactory(DataCatalogFactory)
    title = factory.Dict({"en": factory.Sequence(lambda n: f"research-dataset-{n}")})
    description = factory.Dict({"en": factory.Faker("paragraph")})
    preservation = factory.SubFactory(PreservationFactory)
    access_rights = factory.SubFactory(AccessRightsFactory)
    system_creator = factory.SubFactory(MetaxUserFactory)
    metadata_owner = factory.SubFactory(MetadataProviderFactory)

DatasetActorFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class DatasetActorFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.DatasetActor

    person = factory.SubFactory(PersonFactory)
    organization = factory.SubFactory(OrganizationFactory)

PublishedDatasetFactory

Bases: DatasetFactory

Source code in src/apps/core/factories.py

class PublishedDatasetFactory(DatasetFactory):
    class Meta:
        model = models.Dataset
        django_get_or_create = ("title",)
        skip_postgeneration_save = True

    @factory.lazy_attribute
    def persistent_identifier(self):
        return f"urn:nbn:fi:test:{factory.Faker('uuid')}"

    @factory.post_generation
    def actors(self, create, extracted, **kwargs):
        if not create:
            return
        self.actors.set([DatasetActorFactory(roles=["creator", "publisher"])])

    @factory.post_generation
    def state(self, create, extracted, **kwargs):
        if not create:
            return
        if not extracted:
            self.state = "published"
            self.save()

FileSetFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class FileSetFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.FileSet
        django_get_or_create = ("dataset", "storage")
        skip_postgeneration_save = True

    dataset = factory.SubFactory(DatasetFactory)
    storage = factory.SubFactory(FileStorageFactory)

    @factory.post_generation
    def files(self, create, extracted, **kwargs):
        # allow passing files as argument to factory
        if not create:
            return
        if extracted:
            self.files.set(extracted)

LocationFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class LocationFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = refdata.Location
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"location-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://location-{self}.fi"

SpatialFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class SpatialFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.Spatial

    reference = factory.SubFactory(
        LocationFactory, url="http://www.yso.fi/onto/onto/yso/c_9908ce39"
    )

IdentifierTypeFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class IdentifierTypeFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = refdata.IdentifierType
        django_get_or_create = ("url",)

    pref_label = factory.Dict({"en": factory.Sequence(lambda n: f"dataset-identifier-type-{n}")})
    in_scheme = factory.Faker("url")

    @factory.sequence
    def url(self):
        return f"https://dataset-identifier-type-{self}.fi"

ProvenanceFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class ProvenanceFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.Provenance

    dataset = factory.SubFactory(DatasetFactory)
    lifecycle_event = factory.SubFactory(LifecycleEventFactory)
    event_outcome = factory.SubFactory(EventOutcomeFactory)
    spatial = factory.SubFactory(SpatialFactory)

    description = factory.Dict({"en": factory.Sequence(lambda n: f"provenance-desc-{n}")})
    title = factory.Dict({"en": factory.Sequence(lambda n: f"provenance-{n}")})

DatasetMetricsFactory

Bases: DjangoModelFactory

Source code in src/apps/core/factories.py

class DatasetMetricsFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = models.DatasetMetrics

    downloads_complete_failed = factory.fuzzy.FuzzyInteger(1, 1000)
    downloads_complete_successful = factory.fuzzy.FuzzyInteger(1, 1000)
    downloads_file_failed = factory.fuzzy.FuzzyInteger(1, 1000)
    downloads_file_successful = factory.fuzzy.FuzzyInteger(1, 1000)
    downloads_package_failed = factory.fuzzy.FuzzyInteger(1, 1000)
    downloads_package_successful = factory.fuzzy.FuzzyInteger(1, 1000)
    downloads_partial_failed = factory.fuzzy.FuzzyInteger(1, 1000)
    downloads_partial_successful = factory.fuzzy.FuzzyInteger(1, 1000)
    downloads_total_failed = factory.fuzzy.FuzzyInteger(1, 1000)
    downloads_total_requests = factory.fuzzy.FuzzyInteger(1, 1000)
    downloads_total_successful = factory.fuzzy.FuzzyInteger(1, 1000)
    views_data_views = factory.fuzzy.FuzzyInteger(1, 1000)
    views_details_views = factory.fuzzy.FuzzyInteger(1, 1000)
    views_events_views = factory.fuzzy.FuzzyInteger(1, 1000)
    views_maps_views = factory.fuzzy.FuzzyInteger(1, 1000)
    views_total_views = factory.fuzzy.FuzzyInteger(1, 1000)