# Copyright (c) 2025 Ansible Project
# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
# SPDX-License-Identifier: GPL-3.0-or-later

from __future__ import annotations

import json

import pytest
from ansible.plugins.loader import lookup_loader

from .onepassword_common import SSH_KEY_MOCK_ENTRIES


@pytest.mark.parametrize(
    ("vault", "queries", "kwargs", "output", "expected"),
    (
        (item["vault_name"], item["queries"], item.get("kwargs", {}), item["output"], item["expected"])
        for item in SSH_KEY_MOCK_ENTRIES
    ),
)
def test_ssh_key(mocker, vault, queries, kwargs, output, expected):
    mocker.patch(
        "ansible_collections.community.general.plugins.lookup.onepassword.OnePass.assert_logged_in", return_value=True
    )
    mocker.patch(
        "ansible_collections.community.general.plugins.lookup.onepassword.OnePassCLIBase._run",
        return_value=(0, json.dumps(output), ""),
    )

    op_lookup = lookup_loader.get("community.general.onepassword_ssh_key")
    result = op_lookup.run(queries, vault=vault, **kwargs)

    assert result == expected