move test to event trigger files

Author: e-n-0

bottlecap/src/lifecycle/invocation/span_inferrer.rs

@@ -627,29 +627,6 @@ mod tests {
         })
     }
 
-    #[test]
-    fn test_api_gateway_sets_dd_resource_key_for_rest_event() {
-        let payload = api_gateway_rest_payload();
-        let aws_config = aws_config("us-east-1");
-        let mut inferrer = SpanInferrer::new(Arc::new(Config::default()));
-
-        inferrer.infer_span(&payload, &aws_config);
-
-        let inferred_span = inferrer
-            .inferred_span
-            .as_ref()
-            .expect("Should have inferred API Gateway span");
-
-        assert_eq!(
-            inferred_span
-                .meta
-                .get("dd_resource_key")
-                .cloned()
-                .unwrap_or_default(),
-            "arn:aws:apigateway:us-east-1::/restapis/id"
-        );
-    }
-
     #[test]
     fn test_complete_inferred_spans_propagates_appsec_from_invocation() {
         let payload = api_gateway_rest_payload();

bottlecap/src/lifecycle/invocation/triggers/api_gateway_http_event.rs

@@ -441,6 +441,18 @@ mod tests {
         );
     }
 
+    #[test]
+    fn test_get_dd_resource_key() {
+        let json = read_json_file("api_gateway_http_event.json");
+        let payload = serde_json::from_str(&json).expect("Failed to deserialize into Value");
+        let event =
+            APIGatewayHttpEvent::new(payload).expect("Failed to deserialize APIGatewayHttpEvent");
+        assert_eq!(
+            event.get_dd_resource_key("sa-east-1"),
+            Some("arn:aws:apigateway:sa-east-1::/apis/x02yirxc7a".to_string())
+        );
+    }
+
     #[test]
     fn test_resolve_service_name_with_representation_enabled() {
         let json = read_json_file("api_gateway_http_event.json");

bottlecap/src/lifecycle/invocation/triggers/api_gateway_rest_event.rs

@@ -465,6 +465,18 @@ mod tests {
         );
     }
 
+    #[test]
+    fn test_get_dd_resource_key() {
+        let json = read_json_file("api_gateway_rest_event.json");
+        let payload = serde_json::from_str(&json).expect("Failed to deserialize into Value");
+        let event =
+            APIGatewayRestEvent::new(payload).expect("Failed to deserialize APIGatewayRestEvent");
+        assert_eq!(
+            event.get_dd_resource_key("us-east-1"),
+            Some("arn:aws:apigateway:us-east-1::/restapis/id".to_string())
+        );
+    }
+
     #[test]
     fn test_resolve_service_name_with_representation_enabled() {
         let json = read_json_file("api_gateway_rest_event.json");