Function derive_pubkey_from_same_key

tests/unit/test_wireguard_key_generation.py:279–311

()

    try:

        def derive_pubkey_from_same_key():
            class MockModule:
                def __init__(self, params):
                    self.params = params
                    self.result = {}

                def fail_json(self, **kwargs):
                    raise Exception(f"Module failed: {kwargs}")

                def exit_json(self, **kwargs):
                    self.result = kwargs

            import x25519_pubkey
            from x25519_pubkey import run_module

            original_AnsibleModule = x25519_pubkey.AnsibleModule

            try:
                mock_module = MockModule(
                    {
                        "private_key_b64": b64_key,  # SAME key each time
                        "private_key_path": None,
                        "public_key_path": None,
                    }
                )

                x25519_pubkey.AnsibleModule = lambda **kwargs: mock_module
                run_module()

                return mock_module.result["public_key"]

            finally:
                x25519_pubkey.AnsibleModule = original_AnsibleModule

        # Derive public key multiple times from same private key
        pubkey1 = derive_pubkey_from_same_key()