Source rowcount generic test

models/_sector/nft/trades/ethereum/platforms/_schema.yml

@@ -13,10 +13,11 @@ models:
    tests:
      - dbt_utils.unique_combination_of_columns:
          combination_of_columns: ['block_number','tx_hash','sub_tx_trade_id']
-     - dbt_utils.equal_rowcount:
-         compare_model: source('archipelago_ethereum','ArchipelagoMarket_evt_Trade')
      - dbt_utils.equal_rowcount:
          compare_model: ref('archipelago_ethereum_events')
+     - equal_rowcount_with_sources:
+         evt_sources:
+           - source('archipelago_ethereum','ArchipelagoMarket_evt_Trade')
      - check_seed:
          seed_file: ref('archipelago_ethereum_base_trades_seed')
          match_columns:
@@ -75,6 +76,12 @@ models:
    tests:
      - dbt_utils.unique_combination_of_columns:
          combination_of_columns: [ 'block_number','tx_hash','sub_tx_trade_id' ]
+     - equal_rowcount_with_sources:
+         evt_sources:
+           - source('foundation_ethereum','market_evt_ReserveAuctionFinalized')
+           - source('foundation_ethereum','market_evt_BuyPriceAccepted')
+           - source('foundation_ethereum','market_evt_OfferAccepted')
+           - source('foundation_ethereum','market_evt_PrivateSaleFinalized')
      - check_seed:
          seed_file: ref('foundation_ethereum_base_trades_seed')
          match_columns:

tests/generic/equal_rowcount_with_sources.sql

@@ -0,0 +1,27 @@
+{% test equal_rowcount_with_sources(model, evt_sources=[]) %}
+
+    WITH
+    model_count as (
+    select count(*) as count_a from {{ model }}
+    )
+    ,sources_count as (
+    select sum(count_b) as count_b
+    from (
+        {% for source in evt_sources %}
+        select count(*) as count_b
+        from {{ source }}
+        where evt_block_time <= (select max(block_time) from {{ model }})
+        {% if not loop.last %} UNION ALL {% endif %}
+        {% endfor %}
+        ) b
+    )
+
+    ,unit_test as (
+    select count_a, count_b, (count_a - count_b) as diff_count
+    from model_count
+    full outer join sources_count
+    on 1=1
+    )
+
+    select * from unit_test where diff_count > 0
+{% endtest %}