OpenSlides/server/tests/integration/mediafiles/test_viewset.py

import json

import pytest
from django.core.files.uploadedfile import SimpleUploadedFile
from django.urls import reverse
from rest_framework import status
from rest_framework.test import APIClient

from openslides.mediafiles.models import Mediafile
from tests.count_queries import count_queries
from tests.test_case import TestCase


@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
    * 1 request to get all lists of speakers.
    * 1 request to get all groups
    * 1 request to prefetch parents
    """
    for index in range(10):
        Mediafile.objects.create(
            title=f"some_file{index}",
            original_filename=f"some_file{index}",
            mediafile=SimpleUploadedFile(f"some_file{index}", b"some content."),
        )

    assert count_queries(Mediafile.get_elements)() == 4


class TestCreation(TestCase):
    def setUp(self):
        self.client = APIClient()
        self.client.login(username="admin", password="admin")
        self.file = SimpleUploadedFile("some_file.ext", b"some content.")

    def test_simple_file(self):
        response = self.client.post(
            reverse("mediafile-list"),
            {"title": "test_title_ahyo1uifoo9Aiph2av5a", "mediafile": self.file},
            format="multipart",
        )
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        mediafile = Mediafile.objects.get()
        self.assertEqual(mediafile.title, "test_title_ahyo1uifoo9Aiph2av5a")
        self.assertFalse(mediafile.is_directory)
        self.assertTrue(mediafile.mediafile.name)
        self.assertEqual(mediafile.path, mediafile.original_filename)

    def test_simple_directory(self):
        response = self.client.post(
            reverse("mediafile-list"),
            {"title": "test_title_ahyo1uifoo9Aiph2av5a", "is_directory": True},
        )
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        mediafile = Mediafile.objects.get()
        self.assertEqual(mediafile.title, "test_title_ahyo1uifoo9Aiph2av5a")
        self.assertTrue(mediafile.is_directory)
        self.assertEqual(mediafile.mediafile.name, "")
        self.assertEqual(mediafile.original_filename, "")
        self.assertEqual(mediafile.path, mediafile.title + "/")

    def test_file_and_directory(self):
        response = self.client.post(
            reverse("mediafile-list"),
            {
                "title": "test_title_ahyo1uifoo9Aiph2av5a",
                "is_directory": True,
                "mediafile": self.file,
            },
            format="multipart",
        )
        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
        self.assertFalse(Mediafile.objects.exists())

    def test_no_extension(self):
        file = SimpleUploadedFile("no_extension", b"some content.")
        response = self.client.post(
            reverse("mediafile-list"),
            {"title": "test_title_vai8oDogohheideedie4", "mediafile": file},
            format="multipart",
        )
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        mediafile = Mediafile.objects.get()
        self.assertEqual(mediafile.title, "test_title_vai8oDogohheideedie4")

    def test_mediafile_twice_different_title(self):
        file1 = SimpleUploadedFile("file.ext", b"some content.")
        file2 = SimpleUploadedFile("file.ext", b"some content.")
        response = self.client.post(
            reverse("mediafile-list"),
            {"title": "test_title_Zeicheipeequie3ohfid", "mediafile": file1},
            format="multipart",
        )
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        mediafile = Mediafile.objects.get()
        self.assertEqual(mediafile.title, "test_title_Zeicheipeequie3ohfid")

        response = self.client.post(
            reverse("mediafile-list"),
            {"title": "test_title_aiChaetohs0quicee9eb", "mediafile": file2},
            format="multipart",
        )
        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
        self.assertEqual(Mediafile.objects.count(), 1)

    def test_directory_twice(self):
        title = "test_title_kFJq83fjmqo2babfqk3f"
        Mediafile.objects.create(is_directory=True, title=title)
        response = self.client.post(
            reverse("mediafile-list"), {"title": title, "is_directory": True}
        )
        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
        self.assertEqual(Mediafile.objects.count(), 1)

    def test_without_mediafile(self):
        response = self.client.post(reverse("mediafile-list"), {})
        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
        self.assertFalse(Mediafile.objects.exists())

    def test_without_title(self):
        response = self.client.post(reverse("mediafile-list"), {"is_directory": True})
        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
        self.assertFalse(Mediafile.objects.exists())

    def test_with_empty_title(self):
        response = self.client.post(
            reverse("mediafile-list"), {"is_directory": True, "title": ""}
        )
        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
        self.assertFalse(Mediafile.objects.exists())

    def test_directory_with_slash(self):
        response = self.client.post(
            reverse("mediafile-list"),
            {"title": "test_title_with_/", "is_directory": True},
        )
        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
        self.assertFalse(Mediafile.objects.exists())

    def test_with_parent(self):
        parent_title = "test_title_3q0cqghZRFewocjwferT"
        title = "test_title_gF3if8jmvrbnwdksg4je"
        Mediafile.objects.create(is_directory=True, title=parent_title)
        response = self.client.post(
            reverse("mediafile-list"),
            {"title": title, "is_directory": True, "parent_id": 1},
        )
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        self.assertEqual(Mediafile.objects.count(), 2)
        mediafile = Mediafile.objects.get(title="test_title_gF3if8jmvrbnwdksg4je")
        self.assertEqual(mediafile.parent.title, "test_title_3q0cqghZRFewocjwferT")
        self.assertEqual(mediafile.path, parent_title + "/" + title + "/")

    def test_with_file_as_parent(self):
        Mediafile.objects.create(
            title="test_title_qejOVM84gw8ghwpKnqeg", mediafile=self.file
        )
        response = self.client.post(
            reverse("mediafile-list"),
            {
                "title": "test_title_poejvvlmmorsgeroemr9",
                "is_directory": True,
                "parent_id": 1,
            },
        )
        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
        self.assertEqual(Mediafile.objects.count(), 1)

    def test_with_access_groups(self):
        response = self.client.post(
            reverse("mediafile-list"),
            {
                "title": "test_title_dggjwevBnUngelkdviom",
                "is_directory": True,
                "access_groups_id": [2, 4],
            },
        )
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        self.assertTrue(Mediafile.objects.exists())
        mediafile = Mediafile.objects.get()
        self.assertEqual(
            sorted([group.id for group in mediafile.access_groups.all()]), [2, 4]
        )
        self.assertEqual(mediafile.mediafile.name, "")
        self.assertEqual(mediafile.original_filename, "")
        self.assertEqual(mediafile.path, mediafile.title + "/")

    def test_with_access_groups_wrong_json(self):
        response = self.client.post(
            reverse("mediafile-list"),
            {
                "title": "test_title_dggjwevBnUngelkdviom",
                "is_directory": True,
                "access_groups_id": json.dumps({"a": 324}),
            },
        )
        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
        self.assertFalse(Mediafile.objects.exists())

    def test_with_access_groups_wrong_json2(self):
        response = self.client.post(
            reverse("mediafile-list"),
            {
                "title": "test_title_dggjwevBnUngelkdviom",
                "is_directory": True,
                "access_groups_id": "_FWEpwwfkwk",
            },
        )
        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
        self.assertFalse(Mediafile.objects.exists())


class TestUpdate(TestCase):
    """
    Tree:
    -dir
      -mediafileA
    -mediafileB
    """

    def setUp(self):
        self.client = APIClient()
        self.client.login(username="admin", password="admin")
        self.dir = Mediafile.objects.create(title="dir", is_directory=True)
        fileA_name = "some_fileA.ext"
        self.fileA = SimpleUploadedFile(fileA_name, b"some content.")
        self.mediafileA = Mediafile.objects.create(
            title="mediafileA",
            original_filename=fileA_name,
            mediafile=self.fileA,
            parent=self.dir,
        )
        fileB_name = "some_fileB.ext"
        self.fileB = SimpleUploadedFile(fileB_name, b"some content.")
        self.mediafileB = Mediafile.objects.create(
            title="mediafileB", original_filename=fileB_name, mediafile=self.fileB
        )

    def test_update(self):
        response = self.client.put(
            reverse("mediafile-detail", args=[self.mediafileA.pk]),
            {"title": "test_title_gpasgrmg*miGUM)EAyGO", "access_groups_id": [2, 4]},
        )
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        mediafile = Mediafile.objects.get(pk=self.mediafileA.pk)
        self.assertEqual(mediafile.title, "test_title_gpasgrmg*miGUM)EAyGO")
        self.assertEqual(mediafile.path, "dir/some_fileA.ext")
        self.assertEqual(
            sorted([group.id for group in mediafile.access_groups.all()]), [2, 4]
        )

    def test_update_directory(self):
        response = self.client.put(
            reverse("mediafile-detail", args=[self.dir.pk]),
            {"title": "test_title_seklMOIGGihdjJBNaflkklnlg"},
        )
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        dir = Mediafile.objects.get(pk=self.dir.pk)
        self.assertEqual(dir.title, "test_title_seklMOIGGihdjJBNaflkklnlg")
        mediafile = Mediafile.objects.get(pk=self.mediafileA.pk)
        self.assertEqual(
            mediafile.path, "test_title_seklMOIGGihdjJBNaflkklnlg/some_fileA.ext"
        )

    def test_update_parent_id(self):
        """Assert, that the parent id does not change"""
        response = self.client.put(
            reverse("mediafile-detail", args=[self.mediafileA.pk]),
            {"title": self.mediafileA.title, "parent_id": None},
        )
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        mediafile = Mediafile.objects.get(pk=self.mediafileA.pk)
        self.assertTrue(mediafile.parent)
        self.assertEqual(mediafile.parent.pk, self.dir.pk)