Merge pull request #4 from maxhoheiser/feature/localStackIntegrationTests

Author: maxhoheiser

bin/deploy.sh

@@ -0,0 +1,50 @@
+os=$(uname -s)
+if [ "$os" == "Linux" ]; then
+    (
+        cd lambdas/updateDockerHubImageStats
+        rm -rf package lambda.zip
+        mkdir package
+        pip install -r requirements.txt -t package
+        cp -r ../../src libs/src
+        zip lambda.zip lambda_function.py
+        cd package
+        zip -r ../lambda.zip *;
+    )
+else
+    (
+    cd lambdas/updateDockerHubImageStats
+    rm -rf libs lambda.zip
+    docker run --platform linux/x86_64 -v "$PWD":/var/task "public.ecr.aws/sam/build-python3.10" /bin/sh -c "pip install -r requirements.txt -t libs; exit"
+    cp -r ../../src libs/src
+    cd libs && zip -r ../lambda.zip . && cd ..
+    zip lambda.zip lambda_function.py
+    rm -rf libs
+    )
+fi
+
+awslocal dynamodb create-table \
+    --table-name localstack \
+    --key-schema AttributeName=user_name,KeyType=HASH AttributeName=image_name,KeyType=RANGE \
+    --attribute-definitions AttributeName=user_name,AttributeType=S AttributeName=image_name,AttributeType=S \
+    --billing-mode PAY_PER_REQUEST \
+    --region ap-south-1
+
+awslocal lambda create-function \
+    --function-name update_docker_hub_image_stats \
+    --runtime python3.10 \
+    --timeout 10 \
+    --zip-file fileb://lambdas/updateDockerHubImageStats/lambda.zip \
+    --handler lambda_function.lambda_handler \
+    --dead-letter-config TargetArn=arn:aws:sns:eu-west-1:000000000000:failed-resize-topic \
+    --role arn:aws:iam::000000000000:role/lambda-role \
+    --environment Variables="{STAGE=local}"
+
+rm -rf lambdas/updateDockerHubImageStats/lambda.zip
+
+awslocal lambda wait function-active-v2 --function-name update_docker_hub_image_stats
+awslocal lambda create-function-url-config \
+    --function-name update_docker_hub_image_stats \
+    --auth-type NONE
+
+lambda_url=$(awslocal lambda list-function-url-configs --function-name update_docker_hub_image_stats | jq -r '.FunctionUrlConfigs[0].FunctionUrl')
+curl -v $lambda_url

lambdas/updateDockerHubImageStats/requirements.txt

@@ -0,0 +1,11 @@
+boto3==1.29.3
+botocore==1.32.3
+certifi==2023.11.17
+charset-normalizer==3.3.2
+idna==3.4
+jmespath==1.0.1
+python-dateutil==2.8.2
+requests==2.31.0
+s3transfer==0.7.0
+six==1.16.0
+urllib3==2.0.7

tests/integration/__init__.py

@@ -0,0 +1,49 @@
+import json
+from datetime import datetime
+
+import boto3
+import pytest
+
+IMAGE_PULL = 100
+IMAGE_STAR = 50
+
+input_string = "2022-01-01T00:00:00Z"
+datetime_object = datetime.strptime(input_string, "%Y-%m-%dT%H:%M:%SZ")
+DATE_TIME = datetime_object.isoformat()
+
+
+awslambda = boto3.client(
+    "lambda", endpoint_url="http://localhost.localstack.cloud:4566"
+)
+
+
+@pytest.fixture(autouse=True)
+def _wait_for_lambdas():
+    awslambda.get_waiter("function_active").wait(
+        FunctionName="update_docker_hub_image_stats"
+    )
+
+
+def test_lambda_handler():
+    response = awslambda.invoke(
+        FunctionName="update_docker_hub_image_stats",
+        InvocationType="RequestResponse",
+        Payload=json.dumps({"None": "None"}),
+    )
+
+    payload = json.loads(response["Payload"].read())
+    # need to use payload status code, since response status code is always 200
+    statusCode = payload.get("statusCode")
+    # body = payload.get("body")
+
+    assert statusCode == 200
+    # extend this test with mock DockerHubService and get_iso_date_time_now return values
+    # assert (
+    #    body
+    #    == f"successful update for dynamoDB table: localstack, for pull_count: {IMAGE_PULL}
+    # and star_count: {IMAGE_STAR}, at date_time: {DATE_TIME}"
+    # )
+
+
+if __name__ == "__main__":
+    pytest.main()