Pass all diagnose fields correctly

DataDog · Jul 21, 2023 · fe548ce · fe548ce
1 parent 91f1b94
commit fe548ce
Showing 1 changed file with 16 additions and 10 deletions.
diff --git a/datadog_checks_base/datadog_checks/base/utils/diagnose.py b/datadog_checks_base/datadog_checks/base/utils/diagnose.py
@@ -63,19 +63,25 @@ def success(self, name, diagnosis, category=None, description=None, remediation=
             self._result(
                 self.DIAGNOSIS_SUCCESS,
                 name,
-                diagnosis,
-                category=None,
-                description=None,
-                remediation=None,
-                raw_error=None,
+                diagnosis=diagnosis,
+                category=category,
+                description=description,
+                remediation=remediation,
+                raw_error=raw_error,
             )
         )
 
     def fail(self, name, diagnosis, category=None, description=None, remediation=None, raw_error=None):
         """Register a failing diagnostic result."""
         self._diagnoses.append(
             self._result(
-                self.DIAGNOSIS_FAIL, name, diagnosis, category=None, description=None, remediation=None, raw_error=None
+                self.DIAGNOSIS_FAIL,
+                name,
+                diagnosis,
+                category=category,
+                description=description,
+                remediation=remediation,
+                raw_error=raw_error,
             )
         )
 
@@ -86,10 +92,10 @@ def warning(self, name, diagnosis, category=None, description=None, remediation=
                 self.DIAGNOSIS_WARNING,
                 name,
                 diagnosis,
-                category=None,
-                description=None,
-                remediation=None,
-                raw_error=None,
+                category=category,
+                description=description,
+                remediation=remediation,
+                raw_error=raw_error,
             )
         )