mirror of
https://github.com/ansible-collections/community.general.git
synced 2025-04-06 10:40:32 -07:00
410cf72aec
Some checks failed
EOL CI / EOL Sanity (Ⓐ2.15) (push) Has been cancelled
EOL CI / EOL Units (Ⓐ2.15+py2.7) (push) Has been cancelled
EOL CI / EOL Units (Ⓐ2.15+py3.10) (push) Has been cancelled
EOL CI / EOL Units (Ⓐ2.15+py3.5) (push) Has been cancelled
EOL CI / EOL I (Ⓐ2.15+alpine3+py:azp/posix/1/) (push) Has been cancelled
EOL CI / EOL I (Ⓐ2.15+alpine3+py:azp/posix/2/) (push) Has been cancelled
EOL CI / EOL I (Ⓐ2.15+alpine3+py:azp/posix/3/) (push) Has been cancelled
EOL CI / EOL I (Ⓐ2.15+fedora37+py:azp/posix/1/) (push) Has been cancelled
EOL CI / EOL I (Ⓐ2.15+fedora37+py:azp/posix/2/) (push) Has been cancelled
EOL CI / EOL I (Ⓐ2.15+fedora37+py:azp/posix/3/) (push) Has been cancelled
import-galaxy / Test to import built collection artifact with Galaxy importer (push) Has been cancelled
Verify REUSE / check (push) Has been cancelled
* Replace compat with equivalent from community.internal_test_tools. * Replace mock with equivalent from community.internal_test_tools.
70 lines
2.5 KiB
Python
70 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright (c) 2023, Poh Wei Sheng <weisheng-p@hotmail.sg>
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
import json
|
|
|
|
from ansible_collections.community.internal_test_tools.tests.unit.compat import unittest
|
|
from ansible_collections.community.internal_test_tools.tests.unit.compat.mock import (
|
|
patch,
|
|
MagicMock,
|
|
mock_open
|
|
)
|
|
from ansible.plugins.loader import lookup_loader
|
|
|
|
|
|
class MockJWT(MagicMock):
|
|
def encode(self, payload, key, alg):
|
|
return 'Foobar'
|
|
|
|
|
|
class MockResponse(MagicMock):
|
|
response_token = 'Bar'
|
|
|
|
def read(self):
|
|
return json.dumps({
|
|
"token": self.response_token,
|
|
}).encode('utf-8')
|
|
|
|
|
|
class TestLookupModule(unittest.TestCase):
|
|
|
|
def test_get_token_with_file(self):
|
|
with patch.multiple("ansible_collections.community.general.plugins.lookup.github_app_access_token",
|
|
open=mock_open(read_data="foo_bar"),
|
|
open_url=MagicMock(return_value=MockResponse()),
|
|
jwk_from_pem=MagicMock(return_value='private_key'),
|
|
jwt_instance=MockJWT(),
|
|
HAS_JWT=True):
|
|
lookup = lookup_loader.get('community.general.github_app_access_token')
|
|
self.assertListEqual(
|
|
[MockResponse.response_token],
|
|
lookup.run(
|
|
[],
|
|
key_path="key",
|
|
app_id="app_id",
|
|
installation_id="installation_id",
|
|
token_expiry=600
|
|
)
|
|
)
|
|
|
|
def test_get_token_with_fact(self):
|
|
with patch.multiple("ansible_collections.community.general.plugins.lookup.github_app_access_token",
|
|
open_url=MagicMock(return_value=MockResponse()),
|
|
jwk_from_pem=MagicMock(return_value='private_key'),
|
|
jwt_instance=MockJWT(),
|
|
HAS_JWT=True):
|
|
lookup = lookup_loader.get('community.general.github_app_access_token')
|
|
self.assertListEqual(
|
|
[MockResponse.response_token],
|
|
lookup.run(
|
|
[],
|
|
app_id="app_id",
|
|
installation_id="installation_id",
|
|
private_key="foo_bar",
|
|
token_expiry=600
|
|
)
|
|
)
|