Add e2e tests

Author: Michal Ploski

tests/e2e/event_handler/files/schema.graphql

@@ -0,0 +1,19 @@
+schema {
+    query: Query
+}
+
+type Query {
+    getPost(id:ID!): Post
+    allPosts: [Post]
+}
+
+type Post {
+    id: ID!
+    author: String!
+    title: String
+    content: String
+    url: String
+    ups: Int
+    downs: Int
+    relatedPosts: [Post]
+}

tests/e2e/event_handler/handlers/appsync_handler.py renamed to tests/e2e/event_handler/handlers/appsync_resolver_handler.py

@@ -1,13 +1,16 @@
+from pathlib import Path
 from typing import Dict, Optional
 
-from aws_cdk import CfnOutput
+from aws_cdk import CfnOutput, Duration, Expiration
 from aws_cdk import aws_apigateway as apigwv1
 from aws_cdk import aws_apigatewayv2_alpha as apigwv2
 from aws_cdk import aws_apigatewayv2_authorizers_alpha as apigwv2authorizers
 from aws_cdk import aws_apigatewayv2_integrations_alpha as apigwv2integrations
 from aws_cdk import aws_ec2 as ec2
 from aws_cdk import aws_elasticloadbalancingv2 as elbv2
 from aws_cdk import aws_elasticloadbalancingv2_targets as targets
+from aws_cdk import aws_appsync_alpha as appsync
+from aws_cdk import aws_iam
 from aws_cdk.aws_lambda import Function, FunctionUrlAuthType
 
 from tests.e2e.utils.infrastructure import BaseInfrastructure
@@ -17,10 +20,11 @@ class EventHandlerStack(BaseInfrastructure):
     def create_resources(self):
         functions = self.create_lambda_functions()
 
-        self._create_alb(function=functions["AlbHandler"])
-        self._create_api_gateway_rest(function=functions["ApiGatewayRestHandler"])
-        self._create_api_gateway_http(function=functions["ApiGatewayHttpHandler"])
-        self._create_lambda_function_url(function=functions["LambdaFunctionUrlHandler"])
+        # self._create_alb(function=functions["AlbHandler"])
+        # self._create_api_gateway_rest(function=functions["ApiGatewayRestHandler"])
+        # self._create_api_gateway_http(function=functions["ApiGatewayHttpHandler"])
+        # self._create_lambda_function_url(function=functions["LambdaFunctionUrlHandler"])
+        self._create_appsync_endpoint(function=functions["AppsyncResolverHandler"])
 
     def _create_alb(self, function: Function):
         vpc = ec2.Vpc.from_lookup(
@@ -84,3 +88,40 @@ def _create_lambda_function_url(self, function: Function):
         # Maintenance: move auth to IAM when we create sigv4 builders
         function_url = function.add_function_url(auth_type=FunctionUrlAuthType.AWS_IAM)
         CfnOutput(self.stack, "LambdaFunctionUrl", value=function_url.url)
+
+    def _create_appsync_endpoint(self, function: Function):
+        api = appsync.GraphqlApi(
+            self.stack,
+            "Api",
+            name="e2e-tests",
+            schema=appsync.SchemaFile.from_asset(str(Path(self.feature_path, "files/schema.graphql"))),
+            authorization_config=appsync.AuthorizationConfig(
+                default_authorization=appsync.AuthorizationMode(
+                    authorization_type=appsync.AuthorizationType.API_KEY,
+                    api_key_config=appsync.ApiKeyConfig(
+                        description="public key for getting data",
+                        expires=Expiration.after(Duration.hours(25)),
+                        name="API Token",
+                    ),
+                )
+            ),
+            xray_enabled=False,
+        )
+        lambda_datasource = api.add_lambda_data_source("DataSource", lambda_function=function)
+
+        lambda_datasource.create_resolver(
+            "QueryGetAllPostsResolver",
+            type_name="Query",
+            field_name="allPosts",
+        )
+        lambda_datasource.create_resolver(
+            "QueryGetPostResolver",
+            type_name="Query",
+            field_name="getPost",
+        )
+        lambda_datasource.create_resolver(
+            "QueryGetPostRelatedResolver", type_name="Post", field_name="relatedPosts", max_batch_size=10
+        )
+
+        CfnOutput(self.stack, "GraphQLHTTPUrl", value=api.graphql_url)
+        CfnOutput(self.stack, "GraphQLAPIKey", value=api.api_key)

tests/e2e/event_handler/infrastructure.py

@@ -0,0 +1,108 @@
+import json
+import pytest
+from requests import HTTPError, Request
+
+from tests.e2e.utils import data_fetcher
+from tests.e2e.utils.auth import build_iam_auth
+
+
+@pytest.fixture
+def appsync_endpoint(infrastructure: dict) -> str:
+    return infrastructure["GraphQLHTTPUrl"]
+
+
+@pytest.fixture
+def appsync_access_key(infrastructure: dict) -> str:
+    return infrastructure["GraphQLAPIKey"]
+
+
+@pytest.mark.xdist_group(name="event_handler")
+def test_appsync_get_all_posts(appsync_endpoint, appsync_access_key):
+    # GIVEN
+    body = {
+        "query": "query MyQuery { allPosts { id }}",
+        "variables": None,
+        "operationName": "MyQuery",
+    }
+
+    # WHEN
+    response = data_fetcher.get_http_response(
+        Request(
+            method="POST",
+            url=appsync_endpoint,
+            json=body,
+            headers={"x-api-key": appsync_access_key, "Content-Type": "application/json"},
+        )
+    )
+
+    # THEN expect a HTTP 200 response and content return list of Posts
+    assert response.status_code == 200
+    assert response.content is not None
+
+    data = json.loads(response.content.decode("ascii"))["data"]
+
+    assert data["allPosts"] is not None
+    assert len(data["allPosts"]) > 0
+
+
+@pytest.mark.xdist_group(name="event_handler")
+def test_appsync_get_post(appsync_endpoint, appsync_access_key):
+    # GIVEN
+    post_id = "1"
+    body = {
+        "query": f'query MyQuery {{ getPost(id: "{post_id}") {{ id }} }}',
+        "variables": None,
+        "operationName": "MyQuery",
+    }
+
+    # WHEN
+    response = data_fetcher.get_http_response(
+        Request(
+            method="POST",
+            url=appsync_endpoint,
+            json=body,
+            headers={"x-api-key": appsync_access_key, "Content-Type": "application/json"},
+        )
+    )
+
+    # THEN expect a HTTP 200 response and content return Post id
+    assert response.status_code == 200
+    assert response.content is not None
+
+    data = json.loads(response.content.decode("ascii"))["data"]
+
+    assert data["getPost"]["id"] == post_id
+
+
+@pytest.mark.xdist_group(name="event_handler")
+def test_appsync_get_related_posts_batch(appsync_endpoint, appsync_access_key):
+    # GIVEN
+    post_id = "2"
+    related_posts_ids = ["3", "5"]
+
+    body = {
+        "query": f'query MyQuery {{ getPost(id: "{post_id}") {{ id relatedPosts {{ id }} }} }}',
+        "variables": None,
+        "operationName": "MyQuery",
+    }
+
+    # WHEN
+    response = data_fetcher.get_http_response(
+        Request(
+            method="POST",
+            url=appsync_endpoint,
+            json=body,
+            headers={"x-api-key": appsync_access_key, "Content-Type": "application/json"},
+        )
+    )
+
+    # THEN expect a HTTP 200 response and content return Post id with dependent Posts id's
+    assert response.status_code == 200
+    assert response.content is not None
+
+    data = json.loads(response.content.decode("ascii"))["data"]
+
+    assert data["getPost"]["id"] == post_id
+    assert len(data["getPost"]["relatedPosts"]) == len(related_posts_ids)
+    for post in data["getPost"]["relatedPosts"]:
+        assert post["id"] in related_posts_ids