import datetime

from src import models


def shipment_factory(**kwargs):
    data = dict(
        id=1,
        name="test shipment",
        date=datetime.date(2025, 10, 10),
        form_id=1,
    )
    data.update(kwargs)
    return models.Shipment(**data)


def shipment_public_factory(**kwargs):
    data = dict(
        id=1,
        name="test shipment",
        date=datetime.date(2025, 10, 10),
        form_id=1,
        products=[],
        form=models.Form(id=1, name="test")
    )
    data.update(kwargs)
    return models.ShipmentPublic(**data)


def shipment_create_factory(**kwargs):
    data = dict(
        name="test shipment",
        form_id=1,
        date='2025-10-10',
        product_ids=[],
    )
    data.update(kwargs)
    return models.ShipmentCreate(**data)


def shipment_update_factory(**kwargs):
    data = dict(
        name="test shipment",
        form_id=1,
        date='2025-10-10',
        product_ids=[],
    )
    data.update(kwargs)
    return models.ShipmentUpdate(**data)


def shipment_body_factory(**kwargs):
    data = dict(
        name="test shipment",
        form_id=1,
        date="2025-10-10",
    )
    data.update(kwargs)
    return data