@@ -33,6 +33,7 @@ def __init__(self):
33
33
self .threads = []
34
34
self .context = {}
35
35
36
+
36
37
#---------------------------------------------------------------------------#
37
38
# Fixture
38
39
#---------------------------------------------------------------------------#
@@ -280,24 +281,30 @@ def testUdpServerProcess(self):
280
281
#-----------------------------------------------------------------------#
281
282
def testSerialServerConnect (self ):
282
283
with patch .object (serial , 'Serial' ) as mock_serial :
283
- mock_serial .return_value = "socket"
284
- identity = ModbusDeviceIdentification (info = {0x00 : 'VendorName' })
285
- server = ModbusSerialServer (context = None , identity = identity )
286
- self .assertEqual (server .socket , "socket" )
287
- self .assertEqual (server .control .Identity .VendorName , 'VendorName' )
288
-
289
- server ._connect ()
290
- self .assertEqual (server .socket , "socket" )
284
+ # mock_serial.return_value = "socket"
285
+ mock_serial .write = lambda x : len (x )
286
+ mock_serial .read = lambda size : '\x00 ' * size
287
+ identity = ModbusDeviceIdentification (info = {0x00 : 'VendorName' })
288
+ server = ModbusSerialServer (context = None , identity = identity , port = "dummy" )
289
+ # # mock_serial.return_value = "socket"
290
+ # self.assertEqual(server.socket.port, "dummy")
291
+ self .assertEquals (server .handler .__class__ .__name__ , "CustomSingleRequestHandler" )
292
+ self .assertEqual (server .control .Identity .VendorName , 'VendorName' )
293
+
294
+ server ._connect ()
295
+ # self.assertEqual(server.socket, "socket")
291
296
292
297
with patch .object (serial , 'Serial' ) as mock_serial :
298
+ mock_serial .write = lambda x : len (x )
299
+ mock_serial .read = lambda size : '\x00 ' * size
293
300
mock_serial .side_effect = serial .SerialException ()
294
- server = ModbusSerialServer (None )
301
+ server = ModbusSerialServer (None , port = "dummy" )
295
302
self .assertEqual (server .socket , None )
296
303
297
304
def testSerialServerServeForever (self ):
298
305
''' test that the synchronous serial server closes correctly '''
299
306
with patch .object (serial , 'Serial' ) as mock_serial :
300
- with patch ('pymodbus.server.sync.ModbusSingleRequestHandler ' ) as mock_handler :
307
+ with patch ('pymodbus.server.sync.CustomSingleRequestHandler ' ) as mock_handler :
301
308
server = ModbusSerialServer (None )
302
309
instance = mock_handler .return_value
303
310
instance .handle .side_effect = server .server_close
0 commit comments