Adjusts in test files

thewca · Sep 19, 2020 · 41815b9 · 41815b9
1 parent 04d81c4
commit 41815b9
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 12 deletions.
diff --git a/tnoodle-ui/src/test/App.test.js b/tnoodle-ui/src/test/App.test.js
@@ -41,17 +41,15 @@ beforeEach(() => {
     ).mockImplementation(() => Promise.resolve(languages));
 
     jest.spyOn(tnoodleApi, "fetchRunningVersion").mockImplementation(() =>
-        Promise.resolve({
-            json: () => Promise.resolve(version),
-        })
+        Promise.resolve(version)
     );
 
     jest.spyOn(tnoodleApi, "fetchZip").mockImplementation((...payload) => {
         wcif = payload[0];
         mbld = payload[1];
         password = payload[2];
         translations = payload[3];
-        return Promise.resolve(scrambleProgram);
+        return Promise.resolve(new Blob([]));
     });
 
     jest.spyOn(wcaApi, "fetchVersionInfo").mockImplementation(() =>
@@ -293,11 +291,6 @@ it("Online user", async () => {
         "fetchSuggestedFmcTranslations"
     ).mockImplementation(() => Promise.resolve(["de", "es", "pt-BR"]));
 
-    jest.spyOn(tnoodleApi, "fetchZip").mockImplementation(() => {
-        console.log("Generating zip");
-        return Promise.resolve(new Blob());
-    });
-
     // Render component
     await act(async () => {
         render(
@@ -331,6 +324,5 @@ it("Online user", async () => {
     wcaApi.fetchMe.mockRestore();
     wcaApi.getCompetitionJson.mockRestore();
     tnoodleApi.fetchBestMbldAttempt.mockRestore();
-    tnoodleApi.fetchSuggestedFmcTranslations.mockRestore();
-    tnoodleApi.fetchZip.mockRestore();*/
+    tnoodleApi.fetchSuggestedFmcTranslations.mockRestore();*/
 });
diff --git a/tnoodle-ui/src/test/Main.test.js b/tnoodle-ui/src/test/Main.test.js
@@ -13,22 +13,35 @@ import {
 
 import Main from "../main/components/Main";
 
-import { events, languages, formats } from "./mock/tnoodle.api.mock";
+import { events, languages, formats, version } from "./mock/tnoodle.api.mock";
+import { scrambleProgram } from "./mock/wca.api.mock";
 
 const tnoodleApi = require("../main/api/tnoodle.api");
+const wcaApi = require("../main/api/wca.api");
 
 let container = null;
 beforeEach(() => {
     // setup a DOM element as a render target
     container = document.createElement("div");
     document.body.appendChild(container);
+
+    jest.spyOn(tnoodleApi, "fetchRunningVersion").mockImplementation(() =>
+        Promise.resolve(version)
+    );
+
+    jest.spyOn(wcaApi, "fetchVersionInfo").mockImplementation(() =>
+        Promise.resolve(scrambleProgram)
+    );
 });
 
 afterEach(() => {
     // cleanup on exiting
     unmountComponentAtNode(container);
     container.remove();
     container = null;
+
+    tnoodleApi.fetchRunningVersion.mockRestore();
+    wcaApi.fetchVersionInfo.mockRestore();
 });
 
 it("There should be only 1 button of type submit, check FMC changes", async () => {