From e77932f7d61660b385bdf50cf2aedcc9788b742c Mon Sep 17 00:00:00 2001
From: Mike Salvatore <mike.s.salvatore@gmail.com>
Date: Fri, 7 Oct 2022 09:49:57 -0400
Subject: [PATCH] Island: Add NodeUpdateFacade

---
 .../node_update_facade.py                     | 21 ++++++++
 .../test_node_update_facade.py                | 51 +++++++++++++++++++
 2 files changed, 72 insertions(+)
 create mode 100644 monkey/monkey_island/cc/agent_event_handlers/node_update_facade.py
 create mode 100644 monkey/tests/unit_tests/monkey_island/cc/agent_event_handlers/test_node_update_facade.py

diff --git a/monkey/monkey_island/cc/agent_event_handlers/node_update_facade.py b/monkey/monkey_island/cc/agent_event_handlers/node_update_facade.py
new file mode 100644
index 000000000..9659444a2
--- /dev/null
+++ b/monkey/monkey_island/cc/agent_event_handlers/node_update_facade.py
@@ -0,0 +1,21 @@
+from ipaddress import IPv4Address, IPv4Interface
+
+from monkey_island.cc.models import Machine
+from monkey_island.cc.repository import IMachineRepository, UnknownRecordError
+
+
+class NodeUpdateFacade:
+    def __init__(self, machine_repository: IMachineRepository):
+        self._machine_repository = machine_repository
+
+    def get_or_create_target_machine(self, target: IPv4Address):
+        try:
+            target_machines = self._machine_repository.get_machines_by_ip(target)
+            return target_machines[0]
+        except UnknownRecordError:
+            machine = Machine(
+                id=self._machine_repository.get_new_id(),
+                network_interfaces=[IPv4Interface(target)],
+            )
+            self._machine_repository.upsert_machine(machine)
+            return machine
diff --git a/monkey/tests/unit_tests/monkey_island/cc/agent_event_handlers/test_node_update_facade.py b/monkey/tests/unit_tests/monkey_island/cc/agent_event_handlers/test_node_update_facade.py
new file mode 100644
index 000000000..f8e6387ab
--- /dev/null
+++ b/monkey/tests/unit_tests/monkey_island/cc/agent_event_handlers/test_node_update_facade.py
@@ -0,0 +1,51 @@
+from ipaddress import IPv4Address, IPv4Interface
+from unittest.mock import MagicMock
+
+import pytest
+
+from monkey_island.cc.agent_event_handlers.node_update_facade import NodeUpdateFacade
+from monkey_island.cc.models import Machine
+from monkey_island.cc.repository import IMachineRepository, UnknownRecordError
+
+SEED_ID = 99
+IP_ADDRESS = IPv4Address("10.10.10.99")
+
+EXISTING_MACHINE = Machine(
+    id=1,
+    hardware_id=5,
+    network_interfaces=[IPv4Interface(IP_ADDRESS)],
+)
+
+EXPECTED_CREATED_MACHINE = Machine(
+    id=SEED_ID,
+    network_interfaces=[IPv4Interface(IP_ADDRESS)],
+)
+
+
+@pytest.fixture
+def machine_repository() -> IMachineRepository:
+    machine_repository = MagicMock(spec=IMachineRepository)
+    machine_repository.get_new_id = MagicMock(return_value=SEED_ID)
+    return machine_repository
+
+
+@pytest.fixture
+def node_update_facade(machine_repository) -> NodeUpdateFacade:
+    return NodeUpdateFacade(machine_repository)
+
+
+def test_return_existing_machine(node_update_facade, machine_repository):
+    machine_repository.get_machines_by_ip = MagicMock(return_value=[EXISTING_MACHINE])
+
+    target_machine = node_update_facade.get_or_create_target_machine(IP_ADDRESS)
+
+    assert target_machine == EXISTING_MACHINE
+
+
+def test_create_new_machine(node_update_facade, machine_repository):
+    machine_repository.get_machines_by_ip = MagicMock(side_effect=UnknownRecordError)
+
+    target_machine = node_update_facade.get_or_create_target_machine(IP_ADDRESS)
+
+    assert target_machine == EXPECTED_CREATED_MACHINE
+    assert machine_repository.upsert_machine.called_once_with(target_machine)