From 8c5fa9e22f8d4f6fc7880aaa1faa2c9f1e906723 Mon Sep 17 00:00:00 2001
From: Brandon Clayton <bclayton@usgs.gov>
Date: Tue, 23 Apr 2024 12:47:40 -0600
Subject: [PATCH] add primers

---
 .../earthquake/nshmp/www/PrimingResource.java | 29 ++++++++++++++++++-
 1 file changed, 28 insertions(+), 1 deletion(-)

diff --git a/src/main/java/gov/usgs/earthquake/nshmp/www/PrimingResource.java b/src/main/java/gov/usgs/earthquake/nshmp/www/PrimingResource.java
index dbe5f9d3..087fcded 100644
--- a/src/main/java/gov/usgs/earthquake/nshmp/www/PrimingResource.java
+++ b/src/main/java/gov/usgs/earthquake/nshmp/www/PrimingResource.java
@@ -5,10 +5,12 @@ import java.util.logging.Logger;
 import org.crac.Context;
 import org.crac.Resource;
 
+import com.amazonaws.serverless.proxy.internal.testutils.MockLambdaContext;
 import com.amazonaws.serverless.proxy.model.AwsProxyRequest;
 
 import io.micronaut.context.annotation.Value;
 import io.micronaut.crac.OrderedResource;
+import io.micronaut.function.aws.proxy.MicronautLambdaHandler;
 import jakarta.inject.Singleton;
 
 /**
@@ -23,8 +25,33 @@ public class PrimingResource implements OrderedResource {
   @Override
   public void beforeCheckpoint(Context<? extends Resource> context) throws Exception {
     Logger log = Logger.getAnonymousLogger();
-
     log.info("Before checkpoint");
+
+    try (MicronautLambdaHandler handler = new MicronautLambdaHandler()) {
+      handler.handleRequest(
+          getAwsProxyRequest(contextPath + "/hazard"),
+          new MockLambdaContext());
+
+      handler.handleRequest(
+          getAwsProxyRequest(contextPath + "/probability"),
+          new MockLambdaContext());
+
+      handler.handleRequest(
+          getAwsProxyRequest(contextPath + "/rate"),
+          new MockLambdaContext());
+
+      handler.handleRequest(
+          getAwsProxyRequest(contextPath + "/disagg"),
+          new MockLambdaContext());
+
+      handler.handleRequest(
+          getAwsProxyRequest(contextPath + "/map"),
+          new MockLambdaContext());
+
+      handler.handleRequest(
+          getAwsProxyRequest(contextPath),
+          new MockLambdaContext());
+    }
   }
 
   @Override
-- 
GitLab