From 85fa69450859ad0c8714bc10ff13e1c8e7082c91 Mon Sep 17 00:00:00 2001 From: Wout Feys Date: Mon, 16 Sep 2024 19:02:10 +0200 Subject: [PATCH] Add tests for mathematics, small payloads and userinput thats not in inj --- .../detect_code_injection_test.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/aikido_zen/vulnerabilities/code_injection/detect_code_injection_test.py b/aikido_zen/vulnerabilities/code_injection/detect_code_injection_test.py index 33820daf..99d5d8d7 100644 --- a/aikido_zen/vulnerabilities/code_injection/detect_code_injection_test.py +++ b/aikido_zen/vulnerabilities/code_injection/detect_code_injection_test.py @@ -39,3 +39,20 @@ def test_booleans_and_numbers(): is_not_injection("0123456") is_not_injection("01234567.18234") is_not_injection("False True 012345.6789") + + +def test_maths_not_injection(): + is_not_injection("1 + 2 + 3 + 4 + 5") + is_not_injection("1 + 2 + 3 + 4", "1 + 2 + 3") + + +def test_small_not_injection(): + is_not_injection("a.b") + is_not_injection("abc") + is_not_injection("abcdefghijklmnopabc", "abc") + is_not_injection("def test_injection();", "();") + + +def test_not_in_userinput_not_injection(): + is_not_injection("def test_function(a=True, b=False): pass", "test_function_not") + is_not_injection("def test_injection();", "injection_none")