From 9a775c8a7b650414160131aa2498c523665c17f2 Mon Sep 17 00:00:00 2001
From: Alex Andreev <alex.andreev.email@gmail.com>
Date: Fri, 19 Feb 2021 07:08:46 +0300
Subject: [PATCH] Cleaning up tests a bit

Signed-off-by: Alex Andreev <alex.andreev.email@gmail.com>
---
 src/main/helm/__mocks__/helm-chart-manager.ts | 33 ++++++++-----------
 1 file changed, 13 insertions(+), 20 deletions(-)

diff --git a/src/main/helm/__mocks__/helm-chart-manager.ts b/src/main/helm/__mocks__/helm-chart-manager.ts
index 1f150939d7..ab4cd45bde 100644
--- a/src/main/helm/__mocks__/helm-chart-manager.ts
+++ b/src/main/helm/__mocks__/helm-chart-manager.ts
@@ -10,11 +10,9 @@ export class HelmChartManager {
   }
 
   public async charts(): Promise<any> {
-    return new Promise((resolve, reject) => {
-      let groups: any = {};
-
-      if (this.repo.name == "stable") {
-        groups = {
+    switch (this.repo.name) {
+      case "stable":
+        return Promise.resolve({
           "apm-server": [
             {
               apiVersion: "3.0.0",
@@ -47,11 +45,9 @@ export class HelmChartManager {
               digest: "test"
             }
           ]
-        };
-      }
-
-      if (this.repo.name == "experiment") {
-        groups = {
+        });
+      case "experiment":
+        return Promise.resolve({
           "fairwind": [
             {
               apiVersion: "3.0.0",
@@ -69,11 +65,9 @@ export class HelmChartManager {
               deprecated: true
             }
           ]
-        };
-      }
-
-      if (this.repo.name == "bitnami") {
-        groups = {
+        });
+      case "bitnami":
+        return Promise.resolve({
           "hotdog": [
             {
               apiVersion: "3.0.0",
@@ -106,10 +100,9 @@ export class HelmChartManager {
               digest: "test"
             }
           ]
-        };
-      }
-
-      resolve(groups);
-    });
+        });
+      default:
+        return Promise.resolve({});
+    }
   }
 }