Skip to content

Commit b794019

Browse files
author
BiffoBear
committed
Updated tests to match changes merged in from main.
1 parent eb3b04b commit b794019

File tree

2 files changed

+13
-13
lines changed

2 files changed

+13
-13
lines changed

adafruit_wiznet5k/adafruit_wiznet5k_dns.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -220,7 +220,7 @@ def __init__(
220220
self,
221221
iface: WIZNET5K,
222222
dns_address: Union[str, Tuple[int, int, int, int]],
223-
debug: bool = True,
223+
debug: bool = False,
224224
) -> None:
225225
"""
226226
:param adafruit_wiznet5k.WIZNET5K: Ethernet network connection.

tests/test_dns_server_nonbreaking_changes.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from adafruit_wiznet5k.adafruit_wiznet5k_socket import socket
1212

1313

14-
DEFAULT_DEBUG_ON = True
14+
DEFAULT_DEBUG_ON = False
1515

1616

1717
@pytest.fixture
@@ -130,13 +130,13 @@ def test_good_domain_names_give_correct_ipv4(
130130
)
131131
# Set up mock server calls.
132132
dns_server = wiz_dns.DNS(wiznet, "8.8.8.8", debug=DEFAULT_DEBUG_ON)
133-
dns_server._sock.available.return_value = len(dns_bytes_recv)
133+
dns_server._sock._available.return_value = len(dns_bytes_recv)
134134
dns_server._sock.recv.return_value = dns_bytes_recv
135135

136136
# Check that the correct IPv4 address was received.
137137
assert dns_server.gethostbyname(bytes(domain, "utf-8")) == ipv4
138138
# Check that correct socket calls were made.
139-
dns_server._sock.bind.assert_called_once_with((None, 0x35))
139+
dns_server._sock.bind.assert_called_once_with(("", 0x35))
140140
dns_server._sock.connect.assert_called_once()
141141
dns_server._sock.send.assert_called_once_with(dns_bytes_sent)
142142

@@ -169,15 +169,15 @@ def test_bad_response_returns_correct_value(
169169
)
170170
# Set up mock server calls.
171171
dns_server = wiz_dns.DNS(wiznet, "8.8.8.8", debug=DEFAULT_DEBUG_ON)
172-
dns_server._sock.available.return_value = len(dns_bytes_recv)
172+
dns_server._sock._available.return_value = len(dns_bytes_recv)
173173
dns_server._sock.recv.return_value = dns_bytes_recv
174174

175175
# Check that the correct response was received.
176176
assert dns_server.gethostbyname(bytes("apple.com", "utf-8")) == response
177177

178178
# Check that the correct number of calls to _sock.available were made.
179-
dns_server._sock.available.assert_called()
180-
assert len(dns_server._sock.available.call_args_list) == 5
179+
dns_server._sock._available.assert_called()
180+
assert len(dns_server._sock._available.call_args_list) == 5
181181

182182
@freezegun.freeze_time("2022-3-4", auto_tick_seconds=0.1)
183183
def test_retries_with_no_data_on_socket(self, wiznet, wrench):
@@ -186,13 +186,13 @@ def test_retries_with_no_data_on_socket(self, wiznet, wrench):
186186
# pylint: disable=unused-argument
187187

188188
dns_server = wiz_dns.DNS(wiznet, "8.8.8.8", debug=DEFAULT_DEBUG_ON)
189-
dns_server._sock.available.return_value = 0
189+
dns_server._sock._available.return_value = 0
190190
dns_server._sock.recv.return_value = b""
191191
dns_server.gethostbyname(bytes("domain.name", "utf-8"))
192192

193193
# Check how many times the socket was polled for data before giving up.
194-
dns_server._sock.available.assert_called()
195-
assert len(dns_server._sock.available.call_args_list) == 12
194+
dns_server._sock._available.assert_called()
195+
assert len(dns_server._sock._available.call_args_list) == 12
196196
# Check that no attempt made to read data from the socket.
197197
dns_server._sock.recv.assert_not_called()
198198

@@ -202,13 +202,13 @@ def test_retries_with_bad_data_on_socket(self, wiznet, wrench):
202202
# pylint: disable=unused-argument
203203

204204
dns_server = wiz_dns.DNS(wiznet, "8.8.8.8", debug=DEFAULT_DEBUG_ON)
205-
dns_server._sock.available.return_value = 7
205+
dns_server._sock._available.return_value = 7
206206
dns_server._sock.recv.return_value = b"\x99\x12\x81\x80\x00\x01\x00"
207207
dns_server.gethostbyname(bytes("domain.name", "utf-8"))
208208

209209
# Check how many times the socket was polled for data before giving up.
210-
dns_server._sock.available.assert_called()
211-
assert len(dns_server._sock.available.call_args_list) == 5
210+
dns_server._sock._available.assert_called()
211+
assert len(dns_server._sock._available.call_args_list) == 5
212212
# Check how many attempts were made to read data from the socket.
213213
dns_server._sock.recv.assert_called()
214214
assert len(dns_server._sock.recv.call_args_list) == 5

0 commit comments

Comments
 (0)