@@ -18,7 +18,7 @@ def _makeOneWithMap(self, adj=None):
18
18
map = {}
19
19
inst = self ._makeOne (sock , "127.0.0.1" , adj , map = map )
20
20
inst .outbuf_lock = DummyLock ()
21
- return inst , sock , map
21
+ return inst , sock . local () , map
22
22
23
23
def test_ctor (self ):
24
24
inst , _ , map = self ._makeOneWithMap ()
@@ -218,7 +218,7 @@ def test_write_soon_nonempty_byte(self):
218
218
def send (_ ):
219
219
return 0
220
220
221
- sock .send = send
221
+ sock .remote . send = send
222
222
223
223
wrote = inst .write_soon (b"a" )
224
224
self .assertEqual (wrote , 1 )
@@ -236,7 +236,7 @@ def test_write_soon_filewrapper(self):
236
236
def send (_ ):
237
237
return 0
238
238
239
- sock .send = send
239
+ sock .remote . send = send
240
240
241
241
outbufs = inst .outbufs
242
242
wrote = inst .write_soon (wrapper )
@@ -270,7 +270,7 @@ def test_write_soon_rotates_outbuf_on_overflow(self):
270
270
def send (_ ):
271
271
return 0
272
272
273
- sock .send = send
273
+ sock .remote . send = send
274
274
275
275
inst .adj .outbuf_high_watermark = 3
276
276
inst .current_outbuf_count = 4
@@ -286,7 +286,7 @@ def test_write_soon_waits_on_backpressure(self):
286
286
def send (_ ):
287
287
return 0
288
288
289
- sock .send = send
289
+ sock .remote . send = send
290
290
291
291
inst .adj .outbuf_high_watermark = 3
292
292
inst .total_outbufs_len = 4
@@ -315,7 +315,7 @@ def send(_):
315
315
inst .connected = False
316
316
raise Exception ()
317
317
318
- sock .send = send
318
+ sock .remote . send = send
319
319
320
320
inst .adj .outbuf_high_watermark = 3
321
321
inst .total_outbufs_len = 4
@@ -345,7 +345,7 @@ def send(_):
345
345
inst .connected = False
346
346
raise Exception ()
347
347
348
- sock .send = send
348
+ sock .remote . send = send
349
349
350
350
wrote = inst .write_soon (b"xyz" )
351
351
self .assertEqual (wrote , 3 )
@@ -376,7 +376,7 @@ def test_handle_write_no_notify_after_flush(self):
376
376
inst .total_outbufs_len = len (inst .outbufs [0 ])
377
377
inst .adj .send_bytes = 1
378
378
inst .adj .outbuf_high_watermark = 2
379
- sock .send = lambda x , do_close = True : False
379
+ sock .remote . send = lambda x , do_close = True : False
380
380
inst .will_close = False
381
381
inst .last_activity = 0
382
382
result = inst .handle_write ()
@@ -400,7 +400,7 @@ def test__flush_some_full_outbuf_socket_returns_nonzero(self):
400
400
401
401
def test__flush_some_full_outbuf_socket_returns_zero (self ):
402
402
inst , sock , map = self ._makeOneWithMap ()
403
- sock .send = lambda x : False
403
+ sock .remote . send = lambda x : False
404
404
inst .outbufs [0 ].append (b"abc" )
405
405
inst .total_outbufs_len = sum (len (x ) for x in inst .outbufs )
406
406
result = inst ._flush_some ()
@@ -907,7 +907,8 @@ class DummySock:
907
907
closed = False
908
908
909
909
def __init__ (self ):
910
- self .sent = b""
910
+ self .local_sent = b""
911
+ self .remote_sent = b""
911
912
912
913
def setblocking (self , * arg ):
913
914
self .blocking = True
@@ -925,14 +926,44 @@ def close(self):
925
926
self .closed = True
926
927
927
928
def send (self , data ):
928
- self .sent += data
929
+ self .remote_sent += data
929
930
return len (data )
930
931
931
932
def recv (self , buffer_size ):
932
- result = self .sent [:buffer_size ]
933
- self .sent = self .sent [buffer_size :]
933
+ result = self .local_sent [:buffer_size ]
934
+ self .local_sent = self .local_sent [buffer_size :]
934
935
return result
935
936
937
+ def local (self ):
938
+ outer = self
939
+
940
+ class LocalDummySock :
941
+ def send (self , data ):
942
+ outer .local_sent += data
943
+ return len (data )
944
+
945
+ def recv (self , buffer_size ):
946
+ result = outer .remote_sent [:buffer_size ]
947
+ outer .remote_sent = outer .remote_sent [buffer_size :]
948
+ return result
949
+
950
+ def close (self ):
951
+ outer .closed = True
952
+
953
+ @property
954
+ def sent (self ):
955
+ return outer .remote_sent
956
+
957
+ @property
958
+ def closed (self ):
959
+ return outer .closed
960
+
961
+ @property
962
+ def remote (self ):
963
+ return outer
964
+
965
+ return LocalDummySock ()
966
+
936
967
937
968
class DummyLock :
938
969
notified = False
0 commit comments