can_ssh_patch.py

# CheckConnect
# basic check ssh connection
# --------------------------------------------------------------

# bulldozer.py imports uberlin.operations.ssh.SSH
mock_can_ssh_patcher = patch('uberlin.bulldozer.SSH.can_ssh',
            return_value=True)
self.mock_can_ssh = mock_can_ssh_patcher.start()
self.patchers.append(mock_can_ssh_patcher)

self.mock_check_connect_init = Mock(
            spec=uberlin.bulldozer.CheckConnect.__init__,
            side_effect=uberlin.bulldozer.CheckConnect.__init__)

# run_step calls SSH.can_ssh
# but isn't getting the mocked method
cc = uberlin.bulldozer.CheckConnect(self.environment,
            self.environment.host)
mock_check_connect_patcher = patch(
            'uberlin.bulldozer.CheckConnect.run_step',
            side_effect=cc.run_step,
            return_value=AttributeDict({'success': True}))
self.mock_check_connect = mock_check_connect_patcher.start()
self.patchers.append(mock_check_connect_patcher)