class TestHaproxyDriverRemoteService(unittest.TestCase): def setUp(self): self.remote_service = RemoteService(device_fake) self.remote_service.ssh = Mock() def test_start_service(self): self.assertTrue(self.remote_service.start()) def test_stop_service(self): self.assertTrue(self.remote_service.stop()) def test_restart_service(self): self.assertTrue(self.remote_service.restart())
def setUp(self): self.remote_service = RemoteService(device_fake) self.remote_service.ssh = Mock()