class MyMocks(pulumi.runtime.Mocks):
    def new_resource(self, args: pulumi.runtime.MockResourceArgs):
        outputs = args.inputs
        # if args.typ == "aws:route53/getZone:getZone":
        #     outputs = {
        #         **args.inputs,
        #         "zone_id": "TEST",
        #     }
        return [args.name + "_id", outputs]

    def call(self, args: pulumi.runtime.MockCallArgs):
        if args.token == "aws:iam/getPolicyDocument:getPolicyDocument":
            return {"json": json.dumps({})}
        if args.token == "aws:route53/getZone:getZone":
            return {"zone_id": "TEST"}
        return {}