From 5a76df8489cfbd33a828dc84b0eb1d97146932fb Mon Sep 17 00:00:00 2001
From: Koen Vlaswinkel <koesie10@github.com>
Date: Mon, 24 Oct 2022 16:09:26 +0200
Subject: [PATCH] Load response bodies from files

---
 extensions/ql-vscode/src/mocks/request-handlers.ts | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/extensions/ql-vscode/src/mocks/request-handlers.ts b/extensions/ql-vscode/src/mocks/request-handlers.ts
index 7da2d0f51..602322c9a 100644
--- a/extensions/ql-vscode/src/mocks/request-handlers.ts
+++ b/extensions/ql-vscode/src/mocks/request-handlers.ts
@@ -39,8 +39,17 @@ async function readRequestFiles(scenarioDirPath: string): Promise<GitHubApiReque
 
   const requests: GitHubApiRequest[] = [];
   for (const file of orderedFiles) {
+    if (!file.endsWith('.json')) {
+      continue;
+    }
+
     const filePath = path.join(scenarioDirPath, file);
     const request: GitHubApiRequest = await fs.readJson(filePath, { encoding: 'utf8' });
+
+    if (typeof request.response.body === 'string' && request.response.body.startsWith('file:')) {
+      request.response.body = await fs.readFile(path.join(scenarioDirPath, request.response.body.substring(5)));
+    }
+
     requests.push(request);
   }