Skip to content

Send AgentlessAPICall Lambda ARN to backend #255

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
diogocp merged 2 commits into from
Nov 11, 2025
Merged

Send AgentlessAPICall Lambda ARN to backend #255

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
2d89c7d
Send AgentlessAPICall Lambda ARN to backend
diogocp Nov 10, 2025
64a7d6e
Bump aws_quickstart version to 4.2.1
diogocp Nov 11, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 4 additions & 3 deletions aws_quickstart/datadog_agentless_api_call.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
LOGGER = logging.getLogger()


def call_datadog_agentless_api(event, method):
def call_datadog_agentless_api(context, event, method):
template_version = event["ResourceProperties"]["TemplateVersion"]
api_key = event["ResourceProperties"]["APIKey"]
app_key = event["ResourceProperties"]["APPKey"]
Expand Down Expand Up @@ -69,6 +69,7 @@ def call_datadog_agentless_api(event, method):
"orchestrator_policy_arn": orchestrator_policy_arn,
"worker_policy_arn": worker_policy_arn,
"worker_dspm_policy_arn": worker_dspm_policy_arn,
"invoked_function_arn": context.invoked_function_arn,
},
},
"data": {
Expand Down Expand Up @@ -116,7 +117,7 @@ def handler(event, context):
try:
if event["RequestType"] == "Create":
LOGGER.info("Received Create request.")
response = call_datadog_agentless_api(event, "POST")
response = call_datadog_agentless_api(context, event, "POST")
send_response(
event,
context,
Expand All @@ -135,7 +136,7 @@ def handler(event, context):
)
elif event["RequestType"] == "Delete":
LOGGER.info("Received Delete request.")
response = call_datadog_agentless_api(event, "DELETE")
response = call_datadog_agentless_api(context, event, "DELETE")
send_response(
event,
context,
Expand Down
28 changes: 15 additions & 13 deletions aws_quickstart/datadog_agentless_api_call_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@

import json
import unittest
from types import SimpleNamespace
from unittest.mock import patch, Mock
from urllib.error import HTTPError

Expand All @@ -17,6 +18,7 @@ class TestCallDatadogAgentlessAPI(unittest.TestCase):

def setUp(self):
"""Set up test fixtures"""
self.context = SimpleNamespace(invoked_function_arn="arn:aws:lambda:us-east-1:012345678901:function:DatadogAgentlessAPICallFunction")
self.base_event = {
"ResourceProperties": {
"TemplateVersion": "1.0.0",
Expand Down Expand Up @@ -55,7 +57,7 @@ def test_post_success_200(self, mock_is_enabled, mock_urlopen):
mock_response = self.create_mock_response(200)
mock_urlopen.return_value = mock_response

result = call_datadog_agentless_api(self.base_event, "POST")
result = call_datadog_agentless_api(self.context, self.base_event, "POST")

self.assertEqual(result.status, 200)

Expand All @@ -71,7 +73,7 @@ def test_post_success_201(self, mock_is_enabled, mock_urlopen):
mock_response = self.create_mock_response(201)
mock_urlopen.return_value = mock_response

result = call_datadog_agentless_api(self.base_event, "POST")
result = call_datadog_agentless_api(self.context, self.base_event, "POST")

self.assertEqual(result.status, 201)

Expand All @@ -83,7 +85,7 @@ def test_post_success_204(self, mock_is_enabled, mock_urlopen):
mock_response = self.create_mock_response(204)
mock_urlopen.return_value = mock_response

result = call_datadog_agentless_api(self.base_event, "POST")
result = call_datadog_agentless_api(self.context, self.base_event, "POST")

self.assertEqual(result.status, 204)

Expand All @@ -96,7 +98,7 @@ def test_post_error_400(self, mock_is_enabled, mock_urlopen):
mock_urlopen.side_effect = mock_error

with self.assertRaises(HTTPError):
call_datadog_agentless_api(self.base_event, "POST")
call_datadog_agentless_api(self.context, self.base_event, "POST")

@patch("datadog_agentless_api_call.urllib.request.urlopen")
@patch("datadog_agentless_api_call.is_agentless_scanning_enabled")
Expand All @@ -107,7 +109,7 @@ def test_post_error_404(self, mock_is_enabled, mock_urlopen):
mock_urlopen.side_effect = mock_error

with self.assertRaises(HTTPError):
call_datadog_agentless_api(self.base_event, "POST")
call_datadog_agentless_api(self.context, self.base_event, "POST")

@patch("datadog_agentless_api_call.urllib.request.urlopen")
@patch("datadog_agentless_api_call.is_agentless_scanning_enabled")
Expand All @@ -118,7 +120,7 @@ def test_post_error_500(self, mock_is_enabled, mock_urlopen):
mock_urlopen.side_effect = mock_error

with self.assertRaises(HTTPError):
call_datadog_agentless_api(self.base_event, "POST")
call_datadog_agentless_api(self.context, self.base_event, "POST")

@patch("datadog_agentless_api_call.urllib.request.urlopen")
@patch("datadog_agentless_api_call.is_agentless_scanning_enabled")
Expand All @@ -128,7 +130,7 @@ def test_post_patch_when_enabled(self, mock_is_enabled, mock_urlopen):
mock_response = self.create_mock_response(200)
mock_urlopen.return_value = mock_response

result = call_datadog_agentless_api(self.base_event, "POST")
result = call_datadog_agentless_api(self.context, self.base_event, "POST")

self.assertEqual(result.status, 200)

Expand All @@ -142,7 +144,7 @@ def test_delete_success_200(self, mock_urlopen):
mock_response = self.create_mock_response(200)
mock_urlopen.return_value = mock_response

result = call_datadog_agentless_api(self.base_event, "DELETE")
result = call_datadog_agentless_api(self.context, self.base_event, "DELETE")

self.assertEqual(result.status, 200)

Expand All @@ -152,7 +154,7 @@ def test_delete_success_204(self, mock_urlopen):
mock_response = self.create_mock_response(204)
mock_urlopen.return_value = mock_response

result = call_datadog_agentless_api(self.base_event, "DELETE")
result = call_datadog_agentless_api(self.context, self.base_event, "DELETE")

self.assertEqual(result.status, 204)

Expand All @@ -162,7 +164,7 @@ def test_delete_error_404_returns_error(self, mock_urlopen):
mock_error = self.create_mock_http_error(404)
mock_urlopen.side_effect = mock_error

result = call_datadog_agentless_api(self.base_event, "DELETE")
result = call_datadog_agentless_api(self.context, self.base_event, "DELETE")

self.assertEqual(result.status, 404)

Expand All @@ -173,11 +175,11 @@ def test_delete_error_500_raises_exception(self, mock_urlopen):
mock_urlopen.side_effect = mock_error

with self.assertRaises(HTTPError):
call_datadog_agentless_api(self.base_event, "DELETE")
call_datadog_agentless_api(self.context, self.base_event, "DELETE")

def test_unsupported_method_returns_none(self):
"""Test that unsupported HTTP methods return None"""
result = call_datadog_agentless_api(self.base_event, "PUT")
result = call_datadog_agentless_api(self.context, self.base_event, "PUT")
self.assertIsNone(result)

@patch("datadog_agentless_api_call.urllib.request.urlopen")
Expand All @@ -188,7 +190,7 @@ def test_post_request_payload_structure(self, mock_is_enabled, mock_urlopen):
mock_response = self.create_mock_response(200)
mock_urlopen.return_value = mock_response

call_datadog_agentless_api(self.base_event, "POST")
call_datadog_agentless_api(self.context, self.base_event, "POST")

# Get the request that was made
call_args = mock_urlopen.call_args[0][0]
Expand Down
2 changes: 1 addition & 1 deletion aws_quickstart/version.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1 +1 @@
v4.2.0
v4.2.1
Loading