format

DataDog · Jan 9, 2025 · ffab521 · ffab521
1 parent 3b8962f
commit ffab521
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/tests/appsec/iast/test_security_controls.py b/tests/appsec/iast/test_security_controls.py
@@ -43,7 +43,9 @@ def setup_no_vulnerability_suppression_with_an_input_validator_configured_for_a_
     def test_no_vulnerability_suppression_with_an_input_validator_configured_for_a_different_vulnerability(self):
         self.assert_iast_is_enabled(self.check_r)
         assert_iast_vulnerability(
-            request=self.r, vulnerability_count=1, vulnerability_type="SQL_INJECTION",
+            request=self.r,
+            vulnerability_count=1,
+            vulnerability_type="SQL_INJECTION",
         )
 
     def setup_vulnerability_suppression_with_an_input_validator_configured_for_all_vulnerabilities(self):
@@ -77,7 +79,9 @@ def test_no_vulnerability_suppression_with_an_input_validator_configured_for_an_
     ):
         self.assert_iast_is_enabled(self.check_r)
         assert_iast_vulnerability(
-            request=self.r, vulnerability_count=1, vulnerability_type="SQL_INJECTION",
+            request=self.r,
+            vulnerability_count=1,
+            vulnerability_type="SQL_INJECTION",
         )
 
     def setup_vulnerability_suppression_with_a_sanitizer_configured_for_a_specific_vulnerability(self):
@@ -95,7 +99,9 @@ def setup_no_vulnerability_suppression_with_a_sanitizer_configured_for_a_differe
     def test_no_vulnerability_suppression_with_a_sanitizer_configured_for_a_different_vulnerability(self):
         self.assert_iast_is_enabled(self.check_r)
         assert_iast_vulnerability(
-            request=self.r, vulnerability_count=1, vulnerability_type="SQL_INJECTION",
+            request=self.r,
+            vulnerability_count=1,
+            vulnerability_type="SQL_INJECTION",
         )
 
     def setup_vulnerability_suppression_with_a_sanitizer_configured_for_all_vulnerabilities(self):
@@ -129,5 +135,7 @@ def test_no_vulnerability_suppression_with_a_sanitizer_configured_for_an_overloa
     ):
         self.assert_iast_is_enabled(self.check_r)
         assert_iast_vulnerability(
-            request=self.r, vulnerability_count=1, vulnerability_type="COMMAND_INJECTION",
+            request=self.r,
+            vulnerability_count=1,
+            vulnerability_type="COMMAND_INJECTION",
         )
diff --git a/utils/_features.py b/utils/_features.py
@@ -2295,12 +2295,12 @@ def agent_host_ipv6(test_object):
 
     @staticmethod
     def iast_security_controls(test_object):
-        """
-        IAST: Security Controls
+        """IAST: Security Controls
 
         https://feature-parity.us1.prod.dog/#/?feature=343
         """
         pytest.mark.features(feature_id=343)(test_object)
         return test_object
 
+
 features = _Features()