From 63fb396bbb270b1c19d0b234a5c57c2a06d9212a Mon Sep 17 00:00:00 2001
From: Mike Salvatore <mike.s.salvatore@gmail.com>
Date: Mon, 7 Jun 2021 19:55:33 -0400
Subject: [PATCH] island: Add unit tests for island_config_options_validator

---
 .../test_island_config_options_validator.py   | 62 +++++++++++++++++++
 1 file changed, 62 insertions(+)
 create mode 100644 monkey/tests/unit_tests/monkey_island/cc/setup/test_island_config_options_validator.py

diff --git a/monkey/tests/unit_tests/monkey_island/cc/setup/test_island_config_options_validator.py b/monkey/tests/unit_tests/monkey_island/cc/setup/test_island_config_options_validator.py
new file mode 100644
index 000000000..9c2353c5b
--- /dev/null
+++ b/monkey/tests/unit_tests/monkey_island/cc/setup/test_island_config_options_validator.py
@@ -0,0 +1,62 @@
+import os
+
+import pytest
+
+from common.utils.exceptions import InsecurePermissionsError
+from monkey_island.cc.setup.island_config_options import IslandConfigOptions
+from monkey_island.cc.setup.island_config_options_validator import raise_on_invalid_options
+
+
+@pytest.fixture
+def island_config_options(tmpdir, create_empty_file):
+    crt_file = os.path.join(tmpdir, "test.crt")
+    create_empty_file(crt_file)
+    os.chmod(crt_file, 0o400)
+
+    key_file = os.path.join(tmpdir, "test.key")
+    create_empty_file(key_file)
+    os.chmod(key_file, 0o400)
+    return IslandConfigOptions(
+        {
+            "ssl_certificate": {
+                "ssl_certificate_file": crt_file,
+                "ssl_certificate_key_file": key_file,
+            }
+        }
+    )
+
+
+def test_valid_crt_and_key_paths(island_config_options):
+    try:
+        raise_on_invalid_options(island_config_options)
+    except Exception as ex:
+        print(ex)
+        assert False
+
+
+def test_crt_path_does_not_exist(island_config_options):
+    os.remove(island_config_options.crt_path)
+
+    with pytest.raises(FileNotFoundError):
+        raise_on_invalid_options(island_config_options)
+
+
+def test_crt_path_insecure_permissions(island_config_options):
+    os.chmod(island_config_options.crt_path, 0o777)
+
+    with pytest.raises(InsecurePermissionsError):
+        raise_on_invalid_options(island_config_options)
+
+
+def test_key_path_does_not_exist(island_config_options):
+    os.remove(island_config_options.key_path)
+
+    with pytest.raises(FileNotFoundError):
+        raise_on_invalid_options(island_config_options)
+
+
+def test_key_path_insecure_permissions(island_config_options):
+    os.chmod(island_config_options.key_path, 0o777)
+
+    with pytest.raises(InsecurePermissionsError):
+        raise_on_invalid_options(island_config_options)