Skip to content

Commit d9b7434

Browse files
committed
Fix identifying fake IP in log tests
1 parent 6a542a4 commit d9b7434

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

tests/unit/test_pvapacket.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import logging
22
import unittest
33
from struct import pack
4+
from unittest.mock import patch
45

56
from snowsignal.packet import BadPacketException, Packet
67
from snowsignal.pva_packet import (
@@ -183,7 +184,8 @@ def test_log_packet_not_validpva(self):
183184
captured_logs.records[0].getMessage(),
184185
)
185186

186-
def test_log_beacon_packet(self):
187+
@patch("socket.gethostbyaddr", return_value=("example.com", [], []))
188+
def test_log_beacon_packet(self, _):
187189
# Valid beacon packet
188190
beacon_packet = Packet(
189191
b"\xff\xff\xff\xff\xff\xff\x02B\xac\x16\x00\x03\x08\x00E\x00\x00K3\x15@\x00@\x11\xaf^\xac\x16\x00\x02\xac\x16\xff\xff\xc5\xfb\x13\xd4\x007\xd6V\xca\x02\xc0\x00\x00\x00\x00')\x9bb\xff\xf3\xa5\x9a\x8b\xd7\xc1\x00\xb9\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\x00\x00\x00\x00\x13\xd3\x03tcp\xff"
@@ -196,11 +198,12 @@ def test_log_beacon_packet(self):
196198
log_pvaccess_packet(beacon_packet)
197199

198200
self.assertEqual(
199-
"Received BEACON (v2) [Flags: APPLICATION,NOT_SEGMENTED,SERVER,BIGEND] from 172.22.0.2 --> probe1.snowsignal_udptest1: self-identifies as tcp 00000000000000000000ffff00000000:5075;299b62fff3a59a8bd7c100b9 with update counters beacon:0, PVs:1",
201+
"Received BEACON (v2) [Flags: APPLICATION,NOT_SEGMENTED,SERVER,BIGEND] from 172.22.0.2 --> example.com: self-identifies as tcp 00000000000000000000ffff00000000:5075;299b62fff3a59a8bd7c100b9 with update counters beacon:0, PVs:1",
200202
captured_logs.records[0].getMessage(),
201203
)
202204

203-
def test_log_search_packet(self):
205+
@patch("socket.gethostbyaddr", return_value=("example.com", [], []))
206+
def test_log_search_packet(self, _):
204207
# Valid search packet
205208
beacon_packet = Packet(
206209
b"\xff\xff\xff\xff\xff\xff\x02B\xac\x16\x00\x02\x08\x00E\x00\x00T\x17\x99@\x00@\x11\xca\xd1\xac\x16\x00\x02\xac\x16\xff\xff\xa0\x04\x13\xd4\x00@\xa7\x99\xca\x02\x80\x03\x00\x00\x000find\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xa0\x04\x01\x03tcp\x00\x01\x124Vx\nmy:pv:name"
@@ -213,7 +216,7 @@ def test_log_search_packet(self):
213216
log_pvaccess_packet(beacon_packet)
214217

215218
self.assertEqual(
216-
"Received SEARCH_REQUEST (v2) [Flags: APPLICATION,NOT_SEGMENTED,CLIENT,BIGEND] from 172.22.0.2 --> probe1.snowsignal_udptest1: self-identifies as 00000000000000000000000000000000:40964 (seq id 1718185572) with protocols ['tcp'] searching for [305419896 / my:pv:name]",
219+
"Received SEARCH_REQUEST (v2) [Flags: APPLICATION,NOT_SEGMENTED,CLIENT,BIGEND] from 172.22.0.2 --> example.com: self-identifies as 00000000000000000000000000000000:40964 (seq id 1718185572) with protocols ['tcp'] searching for [305419896 / my:pv:name]",
217220
captured_logs.records[0].getMessage(),
218221
)
219222

0 commit comments

Comments
 (0)