import json
from uuid import uuid4

import pytest

from hello_world import app

def lambda_context():
    class LambdaContext:
        def __init__(self):
            self.function_name = "test-func"
            self.memory_limit_in_mb = 128
            self.invoked_function_arn = "arn:aws:lambda:eu-west-1:809313241234:function:test-func"
            self.aws_request_id = "52fdfc07-2182-154f-163f-5f0f9a621d72"

        def get_remaining_time_in_millis(self) -> int:
            return 1000

    return LambdaContext()


@pytest.fixture()
def apigw_event():
    """ Generates API GW Event"""

    return {
   "body":"",
   "headers":{
      "Accept":"text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9",
      "Accept-Encoding":"gzip, deflate, br",
      "Accept-Language":"pt-BR,pt;q=0.9,en-US;q=0.8,en;q=0.7",
      "Cache-Control":"max-age=0",
      "Connection":"keep-alive",
      "Host":"127.0.0.1:3000",
      "Sec-Ch-Ua":"\"Google Chrome\";v=\"105\", \"Not)A;Brand\";v=\"8\", \"Chromium\";v=\"105\"",
      "Sec-Ch-Ua-Mobile":"?0",
      "Sec-Ch-Ua-Platform":"\"Linux\"",
      "Sec-Fetch-Dest":"document",
      "Sec-Fetch-Mode":"navigate",
      "Sec-Fetch-Site":"none",
      "Sec-Fetch-User":"?1",
      "Upgrade-Insecure-Requests":"1",
      "User-Agent":"Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/105.0.0.0 Safari/537.36",
      "X-Forwarded-Port":"3000",
      "X-Forwarded-Proto":"http"
   },
   "httpMethod":"GET",
   "isBase64Encoded":False,
   "multiValueHeaders":{
      "Accept":[
         "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9"
      ],
      "Accept-Encoding":[
         "gzip, deflate, br"
      ],
      "Accept-Language":[
         "pt-BR,pt;q=0.9,en-US;q=0.8,en;q=0.7"
      ],
      "Cache-Control":[
         "max-age=0"
      ],
      "Connection":[
         "keep-alive"
      ],
      "Host":[
         "127.0.0.1:3000"
      ],
      "Sec-Ch-Ua":[
         "\"Google Chrome\";v=\"105\", \"Not)A;Brand\";v=\"8\", \"Chromium\";v=\"105\""
      ],
      "Sec-Ch-Ua-Mobile":[
         "?0"
      ],
      "Sec-Ch-Ua-Platform":[
         "\"Linux\""
      ],
      "Sec-Fetch-Dest":[
         "document"
      ],
      "Sec-Fetch-Mode":[
         "navigate"
      ],
      "Sec-Fetch-Site":[
         "none"
      ],
      "Sec-Fetch-User":[
         "?1"
      ],
      "Upgrade-Insecure-Requests":[
         "1"
      ],
      "User-Agent":[
         "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/105.0.0.0 Safari/537.36"
      ],
      "X-Forwarded-Port":[
         "3000"
      ],
      "X-Forwarded-Proto":[
         "http"
      ]
   },
   "multiValueQueryStringParameters":"",
   "path":"/hello",
   "pathParameters":"",
   "queryStringParameters":"",
   "requestContext":{
      "accountId":"123456789012",
      "apiId":"1234567890",
      "domainName":"127.0.0.1:3000",
      "extendedRequestId":"",
      "httpMethod":"GET",
      "identity":{
         "accountId":"",
         "apiKey":"",
         "caller":"",
         "cognitoAuthenticationProvider":"",
         "cognitoAuthenticationType":"",
         "cognitoIdentityPoolId":"",
         "sourceIp":"127.0.0.1",
         "user":"",
         "userAgent":"Custom User Agent String",
         "userArn":""
      },
      "path":"/hello",
      "protocol":"HTTP/1.1",
      "requestId":"a3590457-cac2-4f10-8fc9-e47114bf7c62",
      "requestTime":"02/Feb/2023:11:45:26 +0000",
      "requestTimeEpoch":1675338326,
      "resourceId":"123456",
      "resourcePath":"/hello",
      "stage":"Prod"
   },
   "resource":"/hello",
   "stageVariables":"",
   "version":"1.0"
}


def test_lambda_handler(apigw_event):

    ret = app.lambda_handler(apigw_event, lambda_context())
    data = json.loads(ret["body"])

    assert ret["statusCode"] == 200
    assert "message" in ret["body"]
    assert data["message"] == "hello world"