diff --git a/test/test_flood_moderation.py b/test/test_flood_moderation.py index 44645b20b..21b41c5bf 100644 --- a/test/test_flood_moderation.py +++ b/test/test_flood_moderation.py @@ -13,6 +13,7 @@ from test_serial_wifi_sessions import SANITIZER_FLAGS ROOT = Path(__file__).resolve().parents[1] HARNESS = r''' +#include #include #include #include diff --git a/test/test_retry_cr_presets.py b/test/test_retry_cr_presets.py index fe10fd635..5b34d6e7f 100644 --- a/test/test_retry_cr_presets.py +++ b/test/test_retry_cr_presets.py @@ -34,6 +34,7 @@ class RetryCrPresetsTest(unittest.TestCase): #include #include #include +#include "src/helpers/CLIReplyText.h" #define constrain(x, low, high) ((x) < (low) ? (low) : (x) > (high) ? (high) : (x)) struct NodePrefs { uint8_t retry_preset, direct_retry_attempts, direct_retry_enabled; @@ -150,6 +151,7 @@ int main() { source.write_text(generated, encoding="ascii") binary = work / "retry-cr" result = subprocess.run([compiler, "-std=c++17", "-Wall", "-Wextra", + "-I", str(ROOT), str(source), "-o", str(binary)], capture_output=True, text=True, timeout=60) self.assertEqual(result.returncode, 0, result.stdout + result.stderr)