test2.py

import json
from lambda_function import lambda_handler

# Create a dummy context class for testing
class DummyContext:
    def __init__(self):
        self.function_name = "test_lambda"
        self.memory_limit_in_mb = 128
        self.invoked_function_arn = "arn:aws:lambda:region:account-id:function:test_lambda"
        self.aws_request_id = "test_request_id"

if __name__ == '__main__':
    # Define a test event with required parameters.
    test_event = {
        "job_id": "762e7ae1-a7ff-46cf-934e-c5f8f9c1b810",
        "monitored_shortcode": "DExfnDVxaHe",
        "watched_shortcodes": ["DGAtueNiuUe", "DG1KbHkqD5k", "DF9t1vuoxgm"],
        "platform": "instagram",
        "extension": "png"
    }

    # Call lambda_handler with the test event and dummy context
    result = lambda_handler(test_event, DummyContext())

    # Pretty-print the result
    print(json.dumps(result, indent=4))