test(MongoBinaryDownloadUrl): getDownloadUrl: add more tests for DOWNLOAD_MIRROR

Author: hasezoey

packages/mongodb-memory-server-core/src/util/__tests__/MongoBinaryDownloadUrl.test.ts

@@ -477,19 +477,49 @@ describe('MongoBinaryDownloadUrl', () => {
       delete process.env[envName(ResolveConfigVariables.DOWNLOAD_URL)];
     });
 
-    it('should allow mirror overwrite with "DOWNLOAD_MIRROR"', async () => {
-      const archiveName = 'mongodb-linux-x86_64-4.0.0.tgz';
-      const mirror = 'https://custom.org';
-      process.env[envName(ResolveConfigVariables.DOWNLOAD_MIRROR)] = mirror;
+    describe('overwrite with "DOWNLOAD_MIRROR" option', () => {
+      const archiveName = 'mongodb-linux-x86_64-ubuntu1804-4.0.0.tgz';
+      let mbdu: MongoBinaryDownloadUrl;
 
-      const du = new MongoBinaryDownloadUrl({
-        platform: 'linux',
-        arch: 'x64',
-        version: '3.6.3',
+      beforeAll(() => {
+        mbdu = new MongoBinaryDownloadUrl({
+          platform: 'linux',
+          arch: 'x64',
+          version: '4.0.0',
+        });
+
+        jest.spyOn(mbdu, 'getArchiveName').mockImplementation(() => Promise.resolve(archiveName));
+      });
+
+      afterAll(() => {
+        jest.restoreAllMocks();
+      });
+
+      afterEach(() => {
+        delete process.env[envName(ResolveConfigVariables.DOWNLOAD_MIRROR)];
+      });
+
+      it('should allow mirror overwrite (only server name) with "DOWNLOAD_MIRROR"', async () => {
+        const mirror = 'https://custom.org';
+        process.env[envName(ResolveConfigVariables.DOWNLOAD_MIRROR)] = mirror;
+
+        expect(await mbdu.getDownloadUrl()).toBe(`${mirror}/linux/${archiveName}`);
+      });
+
+      it('should allow mirror overwrite (with extra path) with "DOWNLOAD_MIRROR" without extra "/"', async () => {
+        const mirror = 'https://custom.org/extra/path';
+        process.env[envName(ResolveConfigVariables.DOWNLOAD_MIRROR)] = mirror;
+
+        expect(await mbdu.getDownloadUrl()).toBe(`${mirror}/linux/${archiveName}`);
+      });
+
+      it('should allow mirror overwrite (with extra path) with "DOWNLOAD_MIRROR" with extra "/"', async () => {
+        const mirror = 'https://custom.org/extra/path/';
+        process.env[envName(ResolveConfigVariables.DOWNLOAD_MIRROR)] = mirror;
+
+        // no extra "/" because "mirror" already contains it
+        expect(await mbdu.getDownloadUrl()).toBe(`${mirror}linux/${archiveName}`);
       });
-      jest.spyOn(du, 'getArchiveName').mockImplementationOnce(() => Promise.resolve(archiveName));
-      expect(await du.getDownloadUrl()).toBe(`${mirror}/linux/${archiveName}`);
-      delete process.env[envName(ResolveConfigVariables.ARCHIVE_NAME)];
     });
 
     it('should throw an error if platform is unknown (getArchiveName)', async () => {