report the same test status as dbt

Author: haritamar

integration_tests/tests/test_failed_row_count.py

@@ -54,3 +54,19 @@ def test_custom_failed_row_count(test_id: str, dbt_project: DbtProject):
     )
     assert test_result["status"] == "fail"
     assert test_result["failed_row_count"] == overwrite_failed_row_count
+
+
+def test_warn_if_0(test_id: str, dbt_project: DbtProject):
+    # Edge case that we want to verify
+
+    null_count = 50
+    data = [{COLUMN_NAME: "pasten"} for _ in range(null_count)]
+    test_result = dbt_project.test(
+        test_id,
+        "not_null",
+        dict(column_name=COLUMN_NAME, warn_if="=0"),
+        data=data,
+        test_vars={"enable_elementary_test_materialization": True},
+    )
+    assert test_result["status"] == "warn"
+    assert test_result["failed_row_count"] == 0

macros/edr/tests/on_run_end/handle_tests_results.sql

@@ -41,7 +41,7 @@
 
     {% for elementary_test_results_row in elementary_test_results_rows %}
       {% set failures = elementary_test_results_row.get("failures", result.failures) %}
-      {% set status = "pass" if failures == 0 else result.status %}
+      {% set status = result.status %}
       {% do elementary_test_results_row.update({'status': status, 'failures': failures, 'invocation_id': invocation_id, 
                                                 'failed_row_count': elementary_test_failed_row_count}) %}
       {% do elementary_test_results_row.setdefault('test_results_description', result.message) %}