def test_get(self, mock_scp_client_type): auto_connect_ssh = AutoConnectSSH('user1', 'host1') auto_connect_ssh._client = mock.Mock() auto_connect_ssh.get('a', 'z') with mock_scp_client_type() as mock_scp_client: mock_scp_client.get.assert_called_once()
def test_get(self, mock_scp_client_type): auto_connect_ssh = AutoConnectSSH('user1', 'host1') auto_connect_ssh._client = mock.Mock() auto_connect_ssh.get('a', 'z') with mock_scp_client_type() as mock_scp_client: self.assertEqual(mock_scp_client.get.call_count, 1)