How to Mock Pulumi Stack Outputs in Python

from collections import defaultdict

class StackOutputMock:
    def __init__(self, outputs: dict):
        self.outputs = dict(outputs)
    
    def export(self):
        return self.outputs
    
    def get(self, key: str, default=None):
        return self.outputs.get(key, default)
    
    def keys(self):
        return self.outputs.keys()
    
    def values(self):
        return self.outputs.values()
    
    def items(self):
        return self.outputs.items()
    
    def __getitem__(self, key):
        return self.outputs[key]
    
    def __contains__(self, key):
        return key in self.outputs
    

if __name__ == "__main__":
    stack_output = StackOutputMock({
        "bucket_name": "my-app-assets",
        "api_url": "https://api.example.com/v1",
        "region": "us-east-1",
        "lambda_arn": "arn:aws:lambda:us-east-1:123456789012:function:processor"
    })
    
    print(stack_output.get("bucket_name"))
    print(stack_output["api_url"])
    print("region" in stack_output)
    print(stack_output.keys())
    print(list(stack_output.items()))