Skip to content

Commit 487ca41

Browse files
committed
Add e2e test to validate packages
1 parent 5ac7c11 commit 487ca41

2 files changed

Lines changed: 6 additions & 2 deletions

File tree

end2end/django_mysql_test.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,5 +88,6 @@ def test_initial_heartbeat():
8888
"method": "POST",
8989
"path": "/app/create"
9090
}],
91-
{"aborted":0,"attacksDetected":{"blocked":2,"total":2},"total":3}
91+
{"aborted":0,"attacksDetected":{"blocked":2,"total":2},"total":3},
92+
{"django", "aikido_zen"}
9293
)

end2end/server/check_events_from_mock.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,10 @@ def validate_started_event(event, stack, dry_mode=False, serverless=False, os_na
2121
# if stack is not None:
2222
# assert set(event["agent"]["stack"]) == set(stack)
2323

24-
def validate_heartbeat(event, routes, req_stats):
24+
def validate_heartbeat(event, routes, req_stats, packages=None):
25+
if packages:
26+
package_names = set(map(lambda x: x["name"], event["packages"]))
27+
assert package_names == packages, f"Expected {packages} but got {package_names}"
2528
assert event["type"] == "heartbeat", f"Expected event type 'heartbeat', but got '{event['type']}'"
2629
assert event["routes"] == routes, f"Expected routes '{routes}', but got '{event['routes']}'"
2730
assert event["stats"]["requests"] == req_stats, f"Expected request stats '{req_stats}', but got '{event['stats']['requests']}'"

0 commit comments

Comments
 (0)