from polyfactory.factories.pydantic_factory import ModelFactory

from app.adapters.google.types import (
    OfflineJobOperation,
    UploadUsersResponse as GoogleUploadUsersResponse,
    UserDataOfflineJob,
)


class GoogleUploadUsersResponseFactory(ModelFactory[GoogleUploadUsersResponse]):
    __model__ = GoogleUploadUsersResponse


class UserDataOfflineJobFactory(ModelFactory[UserDataOfflineJob]):
    __model__ = UserDataOfflineJob


class OfflineJobOperationFactory(ModelFactory[OfflineJobOperation]):
    __model__ = OfflineJobOperation