Add vue test for DOI on draft dandiset

Author: asmacdo

dandiapi/api/management/commands/inject_doi.py

@@ -0,0 +1,56 @@
+from __future__ import annotations
+
+from django.core.management.base import BaseCommand
+
+from dandiapi.api.models import Dandiset, Version
+
+
+class Command(BaseCommand):
+    help = 'Inject a DOI into a dandiset version for testing'
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            'dandiset_identifier', type=str, help='Dandiset identifier (e.g., 000001)'
+        )
+        parser.add_argument(
+            '--dandiset-version', type=str, default='draft', help='Version (default: draft)'
+        )
+        parser.add_argument(
+            '--doi', type=str, help='DOI to inject (if not provided, will generate one)'
+        )
+
+    def handle(self, *args, **options):
+        dandiset_identifier = options['dandiset_identifier']
+        version = options['dandiset_version']
+        doi = options['doi']
+
+        try:
+            try:
+                dandiset = Dandiset.objects.get(id=int(dandiset_identifier))
+            except (ValueError, Dandiset.DoesNotExist):
+                numeric_id = (
+                    int(dandiset_identifier.lstrip('0')) if dandiset_identifier.lstrip('0') else 0
+                )
+                dandiset = Dandiset.objects.get(id=numeric_id)
+
+            version_obj = Version.objects.get(dandiset=dandiset, version=version)
+
+            if not doi:
+                # TODO: this prefix needs to be updated for non-dandi deployments
+                doi = f'10.80507/dandi.{dandiset_identifier}'
+
+            version_obj.metadata['doi'] = doi
+            version_obj.save()
+
+            self.stdout.write(
+                self.style.SUCCESS(
+                    f'Successfully injected DOI "{doi}" into {dandiset_identifier}/{version}'
+                )
+            )
+
+        except Dandiset.DoesNotExist:
+            self.stdout.write(self.style.ERROR(f'Dandiset {dandiset_identifier} not found'))
+        except Version.DoesNotExist:
+            self.stdout.write(
+                self.style.ERROR(f'Version {version} not found for dandiset {dandiset_identifier}')
+            )

e2e/tests/dandisetLandingPage.spec.ts

@@ -1,6 +1,7 @@
 import { expect, test } from "@playwright/test";
 import { clientUrl, registerNewUser, LOGOUT_BUTTON_TEXT, registerDandiset } from "../utils.ts";
 import { faker } from "@faker-js/faker";
+import { execSync } from "child_process";
 
 test.describe("dandiset landing page", async () => {
   test("add an owner to the dandiset", async ({ page, browser }) => {
@@ -40,9 +41,42 @@ test.describe("dandiset landing page", async () => {
     await expect(newPage.getByText(otherUserName)).toHaveCount(1);
     await context.close();
   });
+
   test("navigate to an invalid dandiset URL", async ({ page }) => {
     await page.goto(`${clientUrl}/dandiset/1`);
     await page.waitForLoadState("networkidle");
     await expect(page.getByText("Error: Dandiset does not exist")).toHaveCount(1);
   });
+
+  test("draft dandiset shows dandiset DOI", async ({ page }) => {
+    // Register a new user and create a dandiset
+    await registerNewUser(page);
+    const dandisetName = faker.lorem.words();
+    const dandisetDescription = faker.lorem.sentences();
+    const dandisetId = await registerDandiset(page, dandisetName, dandisetDescription);
+
+    // Inject a DOI directly using Django management command
+    const testDoi = `10.80507/dandi.${dandisetId}`;
+
+    // Execute the Django management command to inject DOI
+    try {
+      execSync(`cd .. && python manage.py inject_doi ${dandisetId} --dandiset-version=draft --doi="${testDoi}"`, {
+        stdio: 'inherit',
+        timeout: 10000
+      });
+    } catch (error) {
+      console.error('Failed to inject DOI:', error);
+    }
+
+    // Refresh the page to see the updated DOI
+    await page.reload();
+    await page.waitForLoadState("networkidle");
+
+    // The draft version should show the injected Dandiset DOI
+    await expect(page.getByText(testDoi)).toHaveCount(1);
+
+    // Should not show a version DOI (since it's a draft)
+    const versionDoiPattern = new RegExp(`10\\.(48324|80507)/dandi\\.${dandisetId}/`);
+    await expect(page.getByText(versionDoiPattern)).toHaveCount(0);
+  });
 });