Merge branch 'main' into quinna/update-pymongo-test

DataDog · Sep 27, 2024 · fd3561f · fd3561f
2 parents efe9aac + 0bd5ea0
commit fd3561f
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/ddtrace/appsec/_iast/_taint_tracking/tests/test_stringutils.cpp b/ddtrace/appsec/_iast/_taint_tracking/tests/test_stringutils.cpp
@@ -40,6 +40,30 @@ TEST_F(PyReMatchCheck, TEstPyReMatchNullObject)
     ASSERT_FALSE(PyReMatch_Check(null_obj));
 }
 
+using PyIOBaseCheck = PyEnvWithContext;
+
+TEST_F(PyIOBaseCheck, TestPyIOBaseValidObject)
+{
+    py::object io_module = py::module_::import("io");
+    py::object stringio_obj = io_module.attr("StringIO")("a");
+
+    ASSERT_TRUE(PyIOBase_Check(stringio_obj.ptr()));
+}
+
+TEST_F(PyIOBaseCheck, TestPyIOBaseInvalidObject)
+{
+    py::object non_io_obj = py::int_(42); // Not a `_io._IOBase` object
+
+    ASSERT_FALSE(PyIOBase_Check(non_io_obj.ptr()));
+}
+
+TEST_F(PyIOBaseCheck, TestPyIOBaseNullObject)
+{
+    PyObject* null_obj = Py_None;
+
+    ASSERT_FALSE(PyIOBase_Check(null_obj));
+}
+
 using IsFastTaintedCheck = PyEnvCheck;
 
 TEST_F(IsFastTaintedCheck, FastTaintedNullptrReturnsTrue)