From a3d2d7f6a1f216eff71f103acb9ce50d96bf9aaf Mon Sep 17 00:00:00 2001
From: Kekoa Kaaikala <kekoa.kaaikala@gmail.com>
Date: Fri, 30 Sep 2022 17:25:12 +0000
Subject: [PATCH] UT: Add tests for Machines resource

---
 .../cc/resources/test_machines.py             | 80 +++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100644 monkey/tests/unit_tests/monkey_island/cc/resources/test_machines.py

diff --git a/monkey/tests/unit_tests/monkey_island/cc/resources/test_machines.py b/monkey/tests/unit_tests/monkey_island/cc/resources/test_machines.py
new file mode 100644
index 000000000..7aab32a1c
--- /dev/null
+++ b/monkey/tests/unit_tests/monkey_island/cc/resources/test_machines.py
@@ -0,0 +1,80 @@
+from http import HTTPStatus
+from ipaddress import IPv4Interface
+from unittest.mock import MagicMock
+
+import pytest
+from tests.common import StubDIContainer
+from tests.unit_tests.monkey_island.conftest import get_url_for_resource
+
+from common import OperatingSystem
+from monkey_island.cc.models import Machine
+from monkey_island.cc.repository import IMachineRepository, RetrievalError
+from monkey_island.cc.resources import Machines
+
+MACHINES_URL = get_url_for_resource(Machines)
+MACHINES = [
+    Machine(
+        id=1,
+        hardware_id=101,
+        island=True,
+        network_interfaces=[IPv4Interface("10.10.10.1")],
+        operating_system=OperatingSystem.WINDOWS,
+    ),
+    Machine(
+        id=2,
+        hardware_id=102,
+        island=False,
+        network_interfaces=[IPv4Interface("10.10.10.2/24")],
+        operating_system=OperatingSystem.LINUX,
+    ),
+]
+
+
+@pytest.fixture
+def mock_machine_repository() -> IMachineRepository:
+    machine_repository = MagicMock(spec=IMachineRepository)
+    return machine_repository
+
+
+@pytest.fixture
+def flask_client(build_flask_client, mock_machine_repository):
+    container = StubDIContainer()
+
+    container.register_instance(IMachineRepository, mock_machine_repository)
+
+    with build_flask_client(container) as flask_client:
+        yield flask_client
+
+
+def test_machines_get__status_ok(flask_client, mock_machine_repository):
+    mock_machine_repository.get_machines = MagicMock(return_value=MACHINES)
+
+    resp = flask_client.get(MACHINES_URL)
+
+    assert resp.status_code == HTTPStatus.OK
+
+
+def test_machines_get__gets_machines(flask_client, mock_machine_repository):
+    mock_machine_repository.get_machines = MagicMock(return_value=MACHINES)
+
+    resp = flask_client.get(MACHINES_URL)
+
+    response_machines = [Machine(**m) for m in resp.json]
+    for machine in response_machines:
+        assert machine in MACHINES
+
+
+def test_machines_get__gets_no_machines(flask_client, mock_machine_repository):
+    mock_machine_repository.get_machines = MagicMock(return_value=[])
+
+    resp = flask_client.get(MACHINES_URL)
+
+    assert resp.json == []
+
+
+def test_machines_get__retrieval_error(flask_client, mock_machine_repository):
+    mock_machine_repository.get_machines = MagicMock(side_effect=RetrievalError)
+
+    resp = flask_client.get(MACHINES_URL)
+
+    assert resp.status_code == HTTPStatus.INTERNAL_SERVER_ERROR