summaryrefslogtreecommitdiffstats
path: root/test/units/ansible_test/ci/test_azp.py
blob: 69c4fa49b4b2a6354ccf8984819ae87703c98355 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type

from .util import common_auth_test


def test_auth():
    # noinspection PyProtectedMember
    from ansible_test._internal.ci.azp import (
        AzurePipelinesAuthHelper,
    )

    class TestAzurePipelinesAuthHelper(AzurePipelinesAuthHelper):
        def __init__(self):
            self.public_key_pem = None
            self.private_key_pem = None

        def publish_public_key(self, public_key_pem):
            # avoid publishing key
            self.public_key_pem = public_key_pem

        def initialize_private_key(self):
            # cache in memory instead of on disk
            if not self.private_key_pem:
                self.private_key_pem = self.generate_private_key()

            return self.private_key_pem

    auth = TestAzurePipelinesAuthHelper()

    common_auth_test(auth)