Fix at_least_one test when group_by_columns is configured

integration_tests/data/schema_tests/data_test_at_least_one.csv

@@ -1,2 +1,3 @@
-field
-a
+field,value
+a,1
+b,

integration_tests/models/generic_tests/schema.yml

@@ -14,8 +14,12 @@ seeds:
       - name: field
         data_tests:
           - dbt_utils.at_least_one
+      - name: value
+        data_tests:
           - dbt_utils.at_least_one:
               group_by_columns: ['field']
+              error_if: "<1"
+              warn_if: "<0"
 
   - name: data_test_expression_is_true
     data_tests:

macros/generic_tests/at_least_one.sql

@@ -4,38 +4,20 @@
 
 {% macro default__test_at_least_one(model, column_name, group_by_columns) %}
 
-{% set pruned_cols = [column_name] %}
-
 {% if group_by_columns|length() > 0 %}
-
   {% set select_gb_cols = group_by_columns|join(' ,') + ', ' %}
   {% set groupby_gb_cols = 'group by ' + group_by_columns|join(',') %}
-  {% set pruned_cols = group_by_columns %}
-
-  {% if column_name not in pruned_cols %}
-    {% do pruned_cols.append(column_name) %}
-  {% endif %}
-
 {% endif %}
 
-{% set select_pruned_cols = pruned_cols|join(' ,') %}
-
 select *
 from (
-    with pruned_rows as (
-      select
-        {{ select_pruned_cols }}
-      from {{ model }}
-      where {{ column_name }} is not null
-      limit 1
-    )
     select
         {# In TSQL, subquery aggregate columns need aliases #}
         {# thus: a filler col name, 'filler_column' #}
       {{select_gb_cols}}
       count({{ column_name }}) as filler_column
 
-    from pruned_rows
+    from {{ model }}
 
     {{groupby_gb_cols}}