From 642e543fd6e3ea210f76ce83351172888a498685 Mon Sep 17 00:00:00 2001
From: Cadence Ember <cloudrac3r@vivaldi.net>
Date: Tue, 7 Jul 2020 22:08:47 +1200
Subject: [PATCH] Automatically download test files when required

---
 src/lib/utils/saved_requests/manager.js | 28 +++++++++++++++++++++++--
 test/screenshots/commands.js            |  7 +++++--
 2 files changed, 31 insertions(+), 4 deletions(-)

diff --git a/src/lib/utils/saved_requests/manager.js b/src/lib/utils/saved_requests/manager.js
index a45d6aa..29e6b77 100644
--- a/src/lib/utils/saved_requests/manager.js
+++ b/src/lib/utils/saved_requests/manager.js
@@ -5,6 +5,7 @@ const {promisify} = require("util")
 const crypto = require("crypto")
 const stream = require("stream")
 
+const constants = require("../../constants")
 const db = require("../../db")
 const Saved = require("../request_backends/saved")
 const NodeFetch = require("../request_backends/node-fetch")
@@ -56,9 +57,32 @@ class SavedRequestManager {
 	request() {
 		const row = db.prepare("SELECT * FROM SavedRequests WHERE url = ?").get(this.url)
 		if (row) {
-			console.log("Found, using saved request for "+row.path)
 			const base = pj(folder, row.path)
-			return new Saved(base)
+			try {
+				fs.accessSync(base)
+				console.log("Found, using saved request for "+row.path)
+				return new Saved(base)
+			} catch (e) {
+				console.log("Known, but not downloaded  for "+row.path)
+
+				function download(name) {
+					return new NodeFetch(constants.external.saved_requests_location+name).stream().then(readable => {
+						return promisify(stream.pipeline)(
+							readable,
+							fs.createWriteStream(pj(folder, name))
+						)
+					})
+				}
+
+				return new DelayedBackend(
+					Promise.all([
+						download(row.path),
+						download(row.path + ".meta.json")
+					]).then(() => {
+						return this.clone().request()
+					})
+				)
+			}
 		} else {
 			const name = generateName()
 			console.log("Not found, saving now as "+name)
diff --git a/test/screenshots/commands.js b/test/screenshots/commands.js
index f1776c5..73b4c20 100644
--- a/test/screenshots/commands.js
+++ b/test/screenshots/commands.js
@@ -18,12 +18,13 @@ const cookies = {
 	}
 }
 
-function generateSetup(pageName, url, sizeName, cookiesName) {
+function generateSetup(pageName, url, sizeName, cookiesName, scrolls) {
 	return {
 		url: url,
 		filename: `${pageName}-${sizeName}-${cookiesName}`,
 		size: sizes[sizeName],
-		cookies: cookies[cookiesName]
+		cookies: cookies[cookiesName],
+		scrolls: scrolls
 	}
 }
 
@@ -31,6 +32,8 @@ module.exports = [
 	generateSetup("home", "/", "laptop", "default"),
 	generateSetup("settings", "/settings", "laptop", "default"),
 	generateSetup("home", "/", "phone", "default"),
+	generateSetup("settings", "/settings", "phone", "default"),
+
 	generateSetup("home", "/", "laptop", "ptc"),
 	generateSetup("settings", "/settings", "laptop", "ptc"),
 	generateSetup("home", "/", "phone", "ptc"),