import pytest
from django.core.files.uploadedfile import SimpleUploadedFile

from openslides.mediafiles.models import Mediafile

from ..helpers import count_queries


@pytest.mark.django_db(transaction=False)
def test_mediafiles_db_queries():
    """
    Tests that only the following db queries are done:
    * 1 requests to get the list of all files.
    """
    for index in range(10):
        Mediafile.objects.create(
            title="some_file{}".format(index),
            mediafile=SimpleUploadedFile("some_file{}".format(index), b"some content."),
        )

    assert count_queries(Mediafile.get_elements) == 1