From d3ff56138fe57ba40260ba5e79fc76fcdc991a32 Mon Sep 17 00:00:00 2001
From: Mike Salvatore <mike.s.salvatore@gmail.com>
Date: Mon, 3 Oct 2022 13:08:52 -0400
Subject: [PATCH] Agent: Remove disused ExploitInterceptingTelemetryMessenger

---
 ...xploit_intercepting_telemetry_messenger.py | 41 ------------
 ...xploit_intercepting_telemetry_messenger.py | 62 -------------------
 2 files changed, 103 deletions(-)
 delete mode 100644 monkey/infection_monkey/telemetry/messengers/exploit_intercepting_telemetry_messenger.py
 delete mode 100644 monkey/tests/unit_tests/infection_monkey/telemetry/messengers/test_exploit_intercepting_telemetry_messenger.py

diff --git a/monkey/infection_monkey/telemetry/messengers/exploit_intercepting_telemetry_messenger.py b/monkey/infection_monkey/telemetry/messengers/exploit_intercepting_telemetry_messenger.py
deleted file mode 100644
index cb59aa0ae..000000000
--- a/monkey/infection_monkey/telemetry/messengers/exploit_intercepting_telemetry_messenger.py
+++ /dev/null
@@ -1,41 +0,0 @@
-from functools import singledispatch
-from ipaddress import IPv4Address
-
-from infection_monkey.network.relay import TCPRelay
-from infection_monkey.telemetry.exploit_telem import ExploitTelem
-from infection_monkey.telemetry.i_telem import ITelem
-from infection_monkey.telemetry.messengers.i_telemetry_messenger import ITelemetryMessenger
-
-
-class ExploitInterceptingTelemetryMessenger(ITelemetryMessenger):
-    def __init__(self, telemetry_messenger: ITelemetryMessenger, relay: TCPRelay):
-        self._telemetry_messenger = telemetry_messenger
-        self._relay = relay
-
-    def send_telemetry(self, telemetry: ITelem):
-        _send_telemetry(telemetry, self._telemetry_messenger, self._relay)
-
-
-# Note: We can use @singledispatchmethod instead of @singledispatch if we migrate to Python 3.8 or
-# later.
-@singledispatch
-def _send_telemetry(
-    telemetry: ITelem,
-    telemetry_messenger: ITelemetryMessenger,
-    relay: TCPRelay,
-):
-    telemetry_messenger.send_telemetry(telemetry)
-
-
-@_send_telemetry.register
-def _(
-    telemetry: ExploitTelem,
-    telemetry_messenger: ITelemetryMessenger,
-    relay: TCPRelay,
-):
-    if telemetry.propagation_result is True:
-        if relay:
-            address = IPv4Address(str(telemetry.host["ip_addr"]))
-            relay.add_potential_user(address)
-
-    telemetry_messenger.send_telemetry(telemetry)
diff --git a/monkey/tests/unit_tests/infection_monkey/telemetry/messengers/test_exploit_intercepting_telemetry_messenger.py b/monkey/tests/unit_tests/infection_monkey/telemetry/messengers/test_exploit_intercepting_telemetry_messenger.py
deleted file mode 100644
index 61ca1b971..000000000
--- a/monkey/tests/unit_tests/infection_monkey/telemetry/messengers/test_exploit_intercepting_telemetry_messenger.py
+++ /dev/null
@@ -1,62 +0,0 @@
-from unittest.mock import MagicMock
-
-from infection_monkey.i_puppet.i_puppet import ExploiterResultData
-from infection_monkey.model.host import VictimHost
-from infection_monkey.telemetry.exploit_telem import ExploitTelem
-from infection_monkey.telemetry.messengers.exploit_intercepting_telemetry_messenger import (
-    ExploitInterceptingTelemetryMessenger,
-)
-
-
-class MockExploitTelem(ExploitTelem):
-    def __init__(self, propagation_success):
-        erd = ExploiterResultData()
-        erd.propagation_success = propagation_success
-        super().__init__("TestExploiter", VictimHost("127.0.0.1"), erd)
-
-    def get_data(self):
-        return {}
-
-
-def test_generic_telemetry(TestTelem):
-    mock_telemetry_messenger = MagicMock()
-    mock_relay = MagicMock()
-
-    telemetry_messenger = ExploitInterceptingTelemetryMessenger(
-        mock_telemetry_messenger, mock_relay
-    )
-
-    telemetry_messenger.send_telemetry(TestTelem())
-
-    assert mock_telemetry_messenger.send_telemetry.called
-    assert not mock_relay.add_potential_user.called
-
-
-def test_propagation_successful_exploit_telemetry():
-    mock_telemetry_messenger = MagicMock()
-    mock_relay = MagicMock()
-    mock_exploit_telem = MockExploitTelem(True)
-
-    telemetry_messenger = ExploitInterceptingTelemetryMessenger(
-        mock_telemetry_messenger, mock_relay
-    )
-
-    telemetry_messenger.send_telemetry(mock_exploit_telem)
-
-    assert mock_telemetry_messenger.send_telemetry.called
-    assert mock_relay.add_potential_user.called
-
-
-def test_propagation_failed_exploit_telemetry():
-    mock_telemetry_messenger = MagicMock()
-    mock_relay = MagicMock()
-    mock_exploit_telem = MockExploitTelem(False)
-
-    telemetry_messenger = ExploitInterceptingTelemetryMessenger(
-        mock_telemetry_messenger, mock_relay
-    )
-
-    telemetry_messenger.send_telemetry(mock_exploit_telem)
-
-    assert mock_telemetry_messenger.send_telemetry.called
-    assert not mock_relay.add_potential_user.called