Fix API test

BrunoQuaresma · BrunoQuaresma · commit ae42dcc085c7 · 2022-05-13T14:16:29.000Z
diff --git a/site/src/api/api.test.ts b/site/src/api/api.test.ts
@@ -2,29 +2,20 @@ import axios from "axios"
 import { getApiKey, login, logout } from "./api"
 import * as TypesGen from "./typesGenerated"
 
-// Mock the axios module so that no real network requests are made, but rather
-// we swap in a resolved or rejected value
-//
-// See: https://jestjs.io/docs/mock-functions#mocking-modules
-jest.mock("axios")
-
 describe("api.ts", () => {
   describe("login", () => {
     it("should return LoginResponse", async () => {
       // given
       const loginResponse: TypesGen.LoginWithPasswordResponse = {
         session_token: "abc_123_test",
       }
-      const axiosMockPost = jest.fn().mockImplementationOnce(() => {
-        return Promise.resolve({ data: loginResponse })
-      })
-      axios.post = axiosMockPost
+      jest.spyOn(axios, "post").mockResolvedValueOnce({ data: loginResponse })
 
       // when
       const result = await login("test", "123")
 
       // then
-      expect(axiosMockPost).toHaveBeenCalled()
+      expect(axios.post).toHaveBeenCalled()
       expect(result).toStrictEqual(loginResponse)
     })
 
