From 2f0903e0ebba95bcf0ece45045dc18e8a988b930 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Karol=20S=C3=B3jko?= <karolsojko@proton.me>
Date: Thu, 13 Jul 2023 10:58:35 +0200
Subject: [PATCH] feat(home-server): add activating premium features during an
 e2e test suite run

---
 packages/home-server/.env.sample              |  1 +
 packages/home-server/src/Server/HomeServer.ts | 12 ++++++++++++
 2 files changed, 13 insertions(+)

diff --git a/packages/home-server/.env.sample b/packages/home-server/.env.sample
index bfb99a2eb..33bc977b0 100644
--- a/packages/home-server/.env.sample
+++ b/packages/home-server/.env.sample
@@ -1,5 +1,6 @@
 LOG_LEVEL=debug
 NODE_ENV=development
+E2E_TESTING=false
 
 JWT_SECRET=
 AUTH_JWT_SECRET=
diff --git a/packages/home-server/src/Server/HomeServer.ts b/packages/home-server/src/Server/HomeServer.ts
index e9327f47d..ae0722cc3 100644
--- a/packages/home-server/src/Server/HomeServer.ts
+++ b/packages/home-server/src/Server/HomeServer.ts
@@ -138,6 +138,18 @@ export class HomeServer implements HomeServerInterface {
             Disallow: '/',
           }),
         )
+
+        if (env.get('E2E_TESTING', true) === 'true') {
+          app.post('/e2e/activate-premium', (request: Request, response: Response) => {
+            void this.activatePremiumFeatures(request.body.username).then((result) => {
+              if (result.isFailed()) {
+                response.status(400).send({ error: { message: result.getError() } })
+              } else {
+                response.status(200).send({ message: result.getValue() })
+              }
+            })
+          })
+        }
       })
 
       const logger: winston.Logger = winston.loggers.get('home-server')