From e0d65bc269036db0968e3388c670f271db31e469 Mon Sep 17 00:00:00 2001
From: Bethany <bethanyj28@users.noreply.github.com>
Date: Mon, 7 Aug 2023 14:24:58 -0700
Subject: [PATCH] wip

---
 __tests__/artifact-http-client.test.ts | 142 ++++++++++++++++++++++++-
 1 file changed, 139 insertions(+), 3 deletions(-)

diff --git a/__tests__/artifact-http-client.test.ts b/__tests__/artifact-http-client.test.ts
index 14c0a9c..44469ee 100644
--- a/__tests__/artifact-http-client.test.ts
+++ b/__tests__/artifact-http-client.test.ts
@@ -1,8 +1,26 @@
+import * as http from "http"
+import * as net from "net"
+import { HttpClient } from "@actions/http-client"
 import * as config from "../src/internal/shared/config"
 import { createArtifactTwirpClient } from "../src/internal/shared/artifact-twirp-client"
 import * as core from "@actions/core"
 
-jest.mock("@actions/http-client")
+
+const mockPost = jest.fn((statusCode: number, body: string) => {
+  const msg = new http.IncomingMessage(new net.Socket())
+  msg.statusCode = statusCode
+  return {
+    message: msg,
+    readBody: () => {return Promise.resolve(body)}
+  } 
+})
+jest.mock("@actions/http-client", () => {
+  return jest.fn().mockImplementation(() => {
+    return {
+      post: mockPost
+    }
+  })
+})
 
 describe("artifact-http-client", () => {
   beforeAll(() => {
@@ -15,14 +33,33 @@ describe("artifact-http-client", () => {
     jest.spyOn(config, "getRuntimeToken").mockReturnValue("token")
   })
 
+  beforeEach(() => {
+    mockPost.mockClear();
+    (HttpClient as unknown as jest.Mock).mockClear()
+  })
+
   it("should successfully create a client", () => {
     const client = createArtifactTwirpClient("upload")
     expect(client).toBeDefined()
   })
 
   it("should make a request", async () => {
-    const client = createArtifactTwirpClient("upload")
-    const artifact = client.CreateArtifact(
+    /*
+    const mockHttpClient = (HttpClient as unknown as jest.Mock).mockImplementation(() => {
+      return {
+        post: () => { 
+          const msg = new http.IncomingMessage(new net.Socket())
+          msg.statusCode = 200
+          return {
+            message: msg,
+            readBody: () => {return Promise.resolve(`{"ok": true, "signedUploadUrl": "http://localhost:8080/lol/upload"}`)}
+          } 
+        }
+      }
+    })
+    */
+    const client = createArtifactTwirpClient("upload", new HttpClient())
+    const artifact = await client.CreateArtifact(
       {
         workflowRunBackendId: "1234", 
         workflowJobRunBackendId: "5678", 
@@ -30,6 +67,105 @@ describe("artifact-http-client", () => {
         version: 4
       }
     )
+
+    expect(mockHttpClient).toHaveBeenCalledTimes(1)
     expect(artifact).toBeDefined()
+    expect(artifact.ok).toBe(true)
+    expect(artifact.signedUploadUrl).toBe("http://localhost:8080/upload")
+  })
+
+  it("should retry if the request fails", async () => {
+    /*
+    const mockPost = jest.fn(() => {
+      const msg = new http.IncomingMessage(new net.Socket())
+      if (mockPost.mock.calls.length > 1) {
+        msg.statusCode = 200
+        return {
+          message: msg,
+          readBody: () => {return Promise.resolve(`{"ok": true, "signedUploadUrl": "http://localhost:8080/upload"}`)}
+        } 
+      }
+      msg.statusCode = 500
+      return {
+        message: msg,
+        readBody: () => {return Promise.resolve(`{"ok": false}`)}
+      } 
+    })
+    */
+    const msgFailed = new http.IncomingMessage(new net.Socket())
+    msgFailed.statusCode = 500
+    const msgSucceeded = new http.IncomingMessage(new net.Socket())
+    msgSucceeded.statusCode = 200
+
+    const mockPost = jest.fn()
+    mockPost.mockReturnValueOnce({
+      message: msgFailed,
+      readBody: () => {return Promise.resolve(`{"ok": false}`)}
+    }).mockReturnValue({
+      message: msgSucceeded,
+      readBody: () => {return Promise.resolve(`{"ok": true, "signedUploadUrl": "http://localhost:8080/upload"}`)}
+    })
+
+    /*
+    mockPost.mockImplementationOnce(() => {
+      const msg = new http.IncomingMessage(new net.Socket())
+      msg.statusCode = 500
+      return {
+        message: msg,
+        readBody: () => {return Promise.resolve(`{"ok": false}`)}
+      } 
+    })
+
+    mockPost.mockImplementation(() => {
+      const msg = new http.IncomingMessage(new net.Socket())
+      msg.statusCode = 200
+      return {
+        message: msg,
+        readBody: () => {return Promise.resolve(`{"ok": true, "signedUploadUrl": "http://localhost:8080/lol/upload"}`)}
+      }
+    })
+    */
+
+    jest.mock("@actions/http-client", () => {
+      return jest.fn().mockImplementation(() => {
+        return {
+          post: mockPost
+        }
+      })
+    })
+    /*
+    jest.mock("@actions/http-client", () => {
+      return {
+        HttpClient: jest.fn().mockImplementation(() => {
+          return {
+            post: mockPost
+          }
+        })
+      }
+    })
+    jest.mock("@actions/http-client", () => {
+      return jest.fn().mockImplementation(() => {
+        return {
+          post: mockPost
+        }
+      })
+    })
+    */
+
+    const client = createArtifactTwirpClient("upload", new HttpClient())
+    const artifact = await client.CreateArtifact(
+      {
+        workflowRunBackendId: "1234", 
+        workflowJobRunBackendId: "5678", 
+        name: "artifact", 
+        version: 4
+      }
+    )
+
+    expect(artifact).toBeDefined()
+    expect(artifact.ok).toBe(true)
+    expect(artifact.signedUploadUrl).toBe("http://localhost:8080/upload")
+    expect(mockPost).toHaveBeenCalledTimes(2)
   })
 })
+