Misago/misago/users/tests/test_datadownloadsadmin_views.py

import os

from django.contrib.auth import get_user_model
from django.core.files import File
from django.urls import reverse

from misago.admin.testutils import AdminTestCase
from misago.users.datadownloads import request_user_data_download
from misago.users.models import DataDownload


UserModel = get_user_model()

TESTFILES_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "testfiles")
TEST_FILE_PATH = os.path.join(TESTFILES_DIR, "avatar.png")


class DataDownloadAdminViewsTests(AdminTestCase):
    def test_link_registered(self):
        """admin nav contains data downloads link"""
        response = self.client.get(reverse("misago:admin:users:accounts:index"))

        response = self.client.get(response["location"])
        self.assertContains(
            response, reverse("misago:admin:users:data-downloads:index")
        )

    def test_list_view(self):
        """data downloads list view returns 200"""
        response = self.client.get(reverse("misago:admin:users:data-downloads:index"))
        self.assertEqual(response.status_code, 302)

        view_url = response["location"]

        response = self.client.get(view_url)
        self.assertEqual(response.status_code, 200)

        request_user_data_download(self.user)
        response = self.client.get(view_url)
        self.assertEqual(response.status_code, 200)

    def test_expire_action(self):
        """expire action marks data download as expired and deletes its file"""
        data_download = request_user_data_download(self.user)

        with open(TEST_FILE_PATH, "rb") as upload:
            data_download.file = File(upload)
            data_download.save()

        self.assertIsNotNone(data_download.file)
        self.assertTrue(os.path.isfile(data_download.file.path))

        response = self.client.post(
            reverse("misago:admin:users:data-downloads:index"),
            data={"action": "expire", "selected_items": [data_download.pk]},
        )
        self.assertEqual(response.status_code, 302)

        updated_download = DataDownload.objects.get(pk=data_download.pk)
        self.assertEqual(updated_download.status, DataDownload.STATUS_EXPIRED)
        self.assertFalse(updated_download.file)

        self.assertFalse(os.path.isfile(data_download.file.path))

    def test_delete_action(self):
        """dele action deletes data download together with its file"""
        data_download = request_user_data_download(self.user)

        with open(TEST_FILE_PATH, "rb") as upload:
            data_download.file = File(upload)
            data_download.save()

        self.assertIsNotNone(data_download.file)
        self.assertTrue(os.path.isfile(data_download.file.path))

        response = self.client.post(
            reverse("misago:admin:users:data-downloads:index"),
            data={"action": "delete", "selected_items": [data_download.pk]},
        )
        self.assertEqual(response.status_code, 302)

        self.assertEqual(DataDownload.objects.count(), 0)
        self.assertFalse(os.path.isfile(data_download.file.path))

    def test_request_view(self):
        """request data downloads view initializes new downloads"""
        response = self.client.get(reverse("misago:admin:users:data-downloads:request"))
        self.assertEqual(response.status_code, 200)

        other_user = UserModel.objects.create_user("bob", "bob@boberson.com")

        response = self.client.post(
            reverse("misago:admin:users:data-downloads:request"),
            data={
                "user_identifiers": "\n".join([self.user.username, other_user.email])
            },
        )
        self.assertEqual(response.status_code, 302)

        self.assertEqual(DataDownload.objects.count(), 2)

    def test_request_view_empty_data(self):
        """request data downloads view handles empty data"""
        response = self.client.get(reverse("misago:admin:users:data-downloads:request"))
        self.assertEqual(response.status_code, 200)

        response = self.client.post(
            reverse("misago:admin:users:data-downloads:request"),
            data={"user_identifiers": ""},
        )
        self.assertEqual(response.status_code, 200)

        self.assertEqual(DataDownload.objects.count(), 0)

    def test_request_view_user_not_found(self):
        """request data downloads view handles empty data"""
        response = self.client.get(reverse("misago:admin:users:data-downloads:request"))
        self.assertEqual(response.status_code, 200)

        response = self.client.post(
            reverse("misago:admin:users:data-downloads:request"),
            data={"user_identifiers": "not@found.com"},
        )
        self.assertEqual(response.status_code, 200)

        self.assertEqual(DataDownload.objects.count(), 0)