@@ -33,6 +33,7 @@ def __init__(self):
3333 self .threads = []
3434 self .context = {}
3535
36+
3637#---------------------------------------------------------------------------#
3738# Fixture
3839#---------------------------------------------------------------------------#
@@ -280,24 +281,30 @@ def testUdpServerProcess(self):
280281 #-----------------------------------------------------------------------#
281282 def testSerialServerConnect (self ):
282283 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")
291296
292297 with patch .object (serial , 'Serial' ) as mock_serial :
298+ mock_serial .write = lambda x : len (x )
299+ mock_serial .read = lambda size : '\x00 ' * size
293300 mock_serial .side_effect = serial .SerialException ()
294- server = ModbusSerialServer (None )
301+ server = ModbusSerialServer (None , port = "dummy" )
295302 self .assertEqual (server .socket , None )
296303
297304 def testSerialServerServeForever (self ):
298305 ''' test that the synchronous serial server closes correctly '''
299306 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 :
301308 server = ModbusSerialServer (None )
302309 instance = mock_handler .return_value
303310 instance .handle .side_effect = server .server_close
0 commit comments