mirror of
https://github.com/suitenumerique/drive.git
synced 2026-04-25 17:15:19 +02:00
a8e71cdb38
The search e2e fixture items lacked a filename, which made the seeded file items diverge from real uploads where the filename is present. We need the filename to be predictive so we can make assertion in the e2e tests about the viewer that should open or not.
109 lines
4.0 KiB
Python
109 lines
4.0 KiB
Python
"""E2E fixture search."""
|
|
|
|
from django.core.management.base import BaseCommand
|
|
|
|
from core import factories, models
|
|
|
|
from e2e.utils import get_or_create_e2e_user
|
|
|
|
|
|
class Command(BaseCommand):
|
|
"""E2E fixture search."""
|
|
|
|
help = "Generates E2E search fixtures."
|
|
|
|
def handle(self, *args, **options):
|
|
"""E2E fixture search."""
|
|
user = get_or_create_e2e_user("drive@example.com")
|
|
content = [
|
|
{
|
|
"title": "Project 2025",
|
|
"type": models.ItemTypeChoices.FOLDER,
|
|
"creator": user,
|
|
"children": [
|
|
{
|
|
"title": "Budget report",
|
|
"type": models.ItemTypeChoices.FILE,
|
|
"creator": user,
|
|
"filename": "budget-report.pdf",
|
|
},
|
|
{
|
|
"title": "Sales report",
|
|
"type": models.ItemTypeChoices.FILE,
|
|
"creator": user,
|
|
"filename": "sales-report.pdf",
|
|
},
|
|
{
|
|
"title": "I am deleted",
|
|
"type": models.ItemTypeChoices.FOLDER,
|
|
"creator": user,
|
|
"deleted": True,
|
|
},
|
|
{
|
|
"title": "Resume",
|
|
"type": models.ItemTypeChoices.FILE,
|
|
"creator": user,
|
|
"deleted": True,
|
|
"filename": "resume.pdf",
|
|
},
|
|
],
|
|
},
|
|
{
|
|
"title": "Dev Team",
|
|
"type": models.ItemTypeChoices.FOLDER,
|
|
"creator": user,
|
|
"children": [
|
|
{
|
|
"title": "Backlog",
|
|
"type": models.ItemTypeChoices.FILE,
|
|
"creator": user,
|
|
"filename": "backlog.pdf",
|
|
},
|
|
{
|
|
"title": "Meetings",
|
|
"type": models.ItemTypeChoices.FOLDER,
|
|
"creator": user,
|
|
"children": [
|
|
{
|
|
"title": "Meeting notes 5th September",
|
|
"type": models.ItemTypeChoices.FILE,
|
|
"creator": user,
|
|
"filename": "meeting-notes-5th-september.pdf",
|
|
},
|
|
{
|
|
"title": "Meeting notes 15th September",
|
|
"type": models.ItemTypeChoices.FILE,
|
|
"creator": user,
|
|
"filename": "meeting-notes-15th-september.pdf",
|
|
},
|
|
],
|
|
},
|
|
],
|
|
},
|
|
]
|
|
|
|
self._create_item(None, content)
|
|
|
|
def _create_item(self, parent, content, depth=0):
|
|
if content is None:
|
|
return
|
|
for data in content:
|
|
item = factories.ItemFactory(
|
|
title=data["title"],
|
|
type=data["type"],
|
|
creator=data["creator"],
|
|
parent=parent,
|
|
filename=data.get("filename"),
|
|
update_upload_state=models.ItemUploadStateChoices.READY
|
|
if data["type"] == models.ItemTypeChoices.FILE
|
|
else None,
|
|
users=[(data["creator"], models.RoleChoices.OWNER)] if depth == 0 else None,
|
|
)
|
|
if data.get("deleted"):
|
|
item.soft_delete()
|
|
self.stdout.write(
|
|
f"Item created: {item.title} with parent: {parent.title if parent else None} "
|
|
f"and depth: {depth} and deleted: {data.get('deleted')}"
|
|
)
|
|
self._create_item(item, data.get("children"), depth + 1)
|