collerek/ormar

import random
import string

import pytest

from benchmarks.conftest import Author, Book, Publisher

pytestmark = pytest.mark.asyncio


@pytest.mark.parametrize("num_models", [10, 20, 40])
async def test_saving_models_individually(aio_benchmark, num_models: int):
    @aio_benchmark
    async def make_and_insert(num_models: int):
        authors = [
            Author(
                name="".join(random.sample(string.ascii_letters, 5)),
                score=int(random.random() * 100),
            )
            for i in range(0, num_models)
        ]
        assert len(authors) == num_models

        ids = []
        for author in authors:
            a = await author.save()
            ids.append(a)
        return ids

    ids = make_and_insert(num_models)
    for id in ids:
        assert id is not None


@pytest.mark.parametrize("num_models", [10, 20, 40])
async def test_saving_models_individually_with_related_models(
    aio_benchmark, num_models: int, author: Author, publisher: Publisher
):
    @aio_benchmark
    async def making_and_inserting_related_models_one_by_one(
        author: Author, publisher: Publisher, num_models: int
    ):
        books = [
            Book(
                author=author,
                publisher=publisher,
                title="".join(random.sample(string.ascii_letters, 5)),
                year=random.randint(0, 2000),
            )
            for i in range(0, num_models)
        ]

        ids = []
        for book in books:
            await book.save()
            ids.append(book.id)

        return ids

    ids = making_and_inserting_related_models_one_by_one(
        author=author, publisher=publisher, num_models=num_models
    )

    for id in ids:
        assert id is not None