class TxGSMServiceTestCase(TestCase): def setUp(self): self.mock_serial = Mock() self.service = TxGSMService('/dev/foo', bar='baz') self.service.serial_port_class = self.mock_serial @inlineCallbacks def test_start_service(self): d = self.service.onProtocol self.service.startService() protocol = yield d self.assertTrue(isinstance(protocol, TxGSMProtocol)) self.assertTrue(self.mock_serial.called) [init_call] = self.mock_serial.call_args_list args, kwargs = init_call proto, device, reactor = args self.assertEqual(device, '/dev/foo') self.assertEqual(kwargs, {'bar': 'baz'}) def test_stop_service(self): self.service.startService() self.service.port.loseConnection = Mock() self.service.stopService() self.assertTrue(self.service.port.loseConnection.called)
def setUp(self): self.mock_serial = Mock() self.service = TxGSMService('/dev/foo', bar='baz') self.service.serial_port_class = self.mock_serial