11
11
from adafruit_wiznet5k .adafruit_wiznet5k_socket import socket
12
12
13
13
14
- DEFAULT_DEBUG_ON = True
14
+ DEFAULT_DEBUG_ON = False
15
15
16
16
17
17
@pytest .fixture
@@ -130,13 +130,13 @@ def test_good_domain_names_give_correct_ipv4(
130
130
)
131
131
# Set up mock server calls.
132
132
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 )
134
134
dns_server ._sock .recv .return_value = dns_bytes_recv
135
135
136
136
# Check that the correct IPv4 address was received.
137
137
assert dns_server .gethostbyname (bytes (domain , "utf-8" )) == ipv4
138
138
# 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 ))
140
140
dns_server ._sock .connect .assert_called_once ()
141
141
dns_server ._sock .send .assert_called_once_with (dns_bytes_sent )
142
142
@@ -169,15 +169,15 @@ def test_bad_response_returns_correct_value(
169
169
)
170
170
# Set up mock server calls.
171
171
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 )
173
173
dns_server ._sock .recv .return_value = dns_bytes_recv
174
174
175
175
# Check that the correct response was received.
176
176
assert dns_server .gethostbyname (bytes ("apple.com" , "utf-8" )) == response
177
177
178
178
# 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
181
181
182
182
@freezegun .freeze_time ("2022-3-4" , auto_tick_seconds = 0.1 )
183
183
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):
186
186
# pylint: disable=unused-argument
187
187
188
188
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
190
190
dns_server ._sock .recv .return_value = b""
191
191
dns_server .gethostbyname (bytes ("domain.name" , "utf-8" ))
192
192
193
193
# 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
196
196
# Check that no attempt made to read data from the socket.
197
197
dns_server ._sock .recv .assert_not_called ()
198
198
@@ -202,13 +202,13 @@ def test_retries_with_bad_data_on_socket(self, wiznet, wrench):
202
202
# pylint: disable=unused-argument
203
203
204
204
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
206
206
dns_server ._sock .recv .return_value = b"\x99 \x12 \x81 \x80 \x00 \x01 \x00 "
207
207
dns_server .gethostbyname (bytes ("domain.name" , "utf-8" ))
208
208
209
209
# 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
212
212
# Check how many attempts were made to read data from the socket.
213
213
dns_server ._sock .recv .assert_called ()
214
214
assert len (dns_server ._sock .recv .call_args_list ) == 5
0 commit comments