From 5ab0137f274b9c92f0511f7b13f2eb28b2f603b0 Mon Sep 17 00:00:00 2001
From: VakarisZ <vakarisz@yahoo.com>
Date: Tue, 1 Jun 2021 10:31:59 +0300
Subject: [PATCH] Improved a mock endpoint for testing import configuration
 modal

---
 .../monkey_island/cc/resources/temp_configuration.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/monkey/monkey_island/cc/resources/temp_configuration.py b/monkey/monkey_island/cc/resources/temp_configuration.py
index 1c558e7f8..fe9ae09b6 100644
--- a/monkey/monkey_island/cc/resources/temp_configuration.py
+++ b/monkey/monkey_island/cc/resources/temp_configuration.py
@@ -1,6 +1,8 @@
+import json
 from dataclasses import dataclass
 
 import flask_restful
+from flask import request
 
 from common.utils.exceptions import (
     InvalidConfigurationError,
@@ -26,9 +28,9 @@ class TempConfiguration(flask_restful.Resource):
 
     @jwt_required
     def post(self):
-        # request_contents = json.loads(request.data)
+        request_contents = json.loads(request.data)
         try:
-            self.decrypt()
+            self.decrypt(request_contents["password"])
             self.import_config()
             return ResponseContents().form_response()
         except InvalidCredentialsError:
@@ -50,7 +52,11 @@ class TempConfiguration(flask_restful.Resource):
                 status_code=403,
             ).form_response()
 
-    def decrypt(self):
+    def decrypt(self, password=""):
+        if not password:
+            raise NoCredentialsError
+        if not password == "abc":
+            raise InvalidCredentialsError
         return False
 
     def import_config(self):