@cumulus/integration-tests 20.1.1 → 20.1.2

package/cmr.js

@@ -223,12 +223,10 @@ function generateCmrXml(granule, collection, additionalUrls) {
  * @returns {Promise<Array<string>>} - Promise of a list of granule files including the created
  * CMR xml files
  */
-async function generateAndStoreCmrXml(granule, collection, bucket, additionalUrls) {
+async function generateAndStoreCmrXml(granule, collection, bucket, additionalUrls, stagingDir = 'file-staging') {
   const xml = generateCmrXml(granule, collection, additionalUrls);
   const granuleFiles = granule.files.map((f) => `s3://${f.bucket}/${f.key}`);
 
-  const stagingDir = 'file-staging';
-
   const fileKey = `${stagingDir}/${granule.granuleId}.cmr.xml`;
 
   const params = {
@@ -357,7 +355,8 @@ async function generateAndStoreCmrUmmJson(
   collection,
   bucket,
   additionalUrls,
-  cmrMetadataFormat
+  cmrMetadataFormat,
+  stagingDir = 'file-staging'
 ) {
   const versionString = metadataFormatToVersion(cmrMetadataFormat);
   const jsonObject = sampleUmmGranule;
@@ -385,8 +384,6 @@ async function generateAndStoreCmrUmmJson(
     };
   }
 
-  const stagingDir = 'file-staging';
-
   const fileKey = `${stagingDir}/${granule.granuleId}.cmr.json`;
 
   const params = {
@@ -419,26 +416,40 @@ async function generateAndStoreCmrUmmJson(
  * @param {Array<string>} additionalUrls - URLs to convert to online resources or related urls
  * @returns {Array<string>} list of S3 locations for CMR xml files
  */
-async function generateCmrFilesForGranules(
+async function generateCmrFilesForGranules({
   granules,
   collection,
   bucket,
   cmrMetadataFormat,
-  additionalUrls
-) {
+  additionalUrls,
+  stagingDir,
+}) {
   let files;
 
   log.info(`Generating fake CMR file with type ${cmrMetadataFormat}`);
 
   if (isUMMGMetadataFormat(cmrMetadataFormat)) {
     files = await Promise.all(
-      granules.map(
-        (g) => generateAndStoreCmrUmmJson(g, collection, bucket, additionalUrls, cmrMetadataFormat)
-      )
+      granules.map((g) =>
+        generateAndStoreCmrUmmJson(
+          g,
+          collection,
+          bucket,
+          additionalUrls,
+          cmrMetadataFormat,
+          stagingDir
+        ))
     );
   } else {
     files = await Promise.all(
-      granules.map((g) => generateAndStoreCmrXml(g, collection, bucket, additionalUrls))
+      granules.map((g) =>
+        generateAndStoreCmrXml(
+          g,
+          collection,
+          bucket,
+          additionalUrls,
+          stagingDir
+        ))
     );
   }