chore(test): make global setup safe for running in shards

sanity-io · Oct 24, 2023 · 3281d2b · 3281d2b
1 parent 596fd8f
commit 3281d2b
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/packages/@sanity/cli/test/shared/globalSetup.ts b/packages/@sanity/cli/test/shared/globalSetup.ts
@@ -140,12 +140,18 @@ async function prepareDatasets() {
     token: cliUserToken,
   })
 
+  const existingDatasets = await client.datasets.list()
+  const datasetSet = new Set(existingDatasets.map((ds) => ds.name))
+
   for (const version of studioVersions) {
     const args = getTestRunArgs(version)
     const datasets = [args.documentsDataset, args.graphqlDataset, args.aclDataset]
 
     await Promise.all(
       datasets.map((ds) => {
+        if (datasetSet.has(ds)) {
+          return Promise.resolve()
+        }
         // eslint-disable-next-line no-console
         console.log(`Creating dataset ${ds}...`)
         return client.datasets.create(ds, {aclMode: 'public'}).catch((err) => {

diff --git a/packages/@sanity/cli/test/shared/globalTeardown.ts b/packages/@sanity/cli/test/shared/globalTeardown.ts
@@ -84,7 +84,15 @@ async function deleteDatasets(args: ReturnType<typeof getTestRunArgs>) {
     args.aclDataset,
   ]
 
+  const existingDatasets = await testClient.datasets.list()
+  const datasetSet = new Set(existingDatasets.map((ds) => ds.name))
+
   await Promise.all(
-    datasets.map((ds) => testClient.datasets.delete(ds).catch(getErrorWarner('dataset', ds))),
+    datasets.map((ds) => {
+      if (!datasetSet.has(ds)) {
+        return Promise.resolve()
+      }
+      return testClient.datasets.delete(ds).catch(getErrorWarner('dataset', ds))
+    }),
   )
 }