fix test

cowprotocol · Jun 26, 2024 · f1587cc · f1587cc
1 parent e2c4db9
commit f1587cc
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/src/sql/orderbook/barn_order_rewards.sql b/src/sql/orderbook/barn_order_rewards.sql
@@ -365,7 +365,7 @@ select
     cast(oq.buy_amount as numeric(78, 0)) :: text as quote_buy_amount,
     oq.gas_amount * oq.gas_price as quote_gas_cost,
     oq.sell_token_price as quote_sell_token_price,
-    opfp.partner_fee,
+    cast(coalesce(opfp.partner_fee, 0) as numeric(78, 0)) :: text as partner_fee,
     opfp.partner_fee_recipient,
     opfp.protocol_fee_kind
 from

diff --git a/src/sql/orderbook/prod_order_rewards.sql b/src/sql/orderbook/prod_order_rewards.sql
@@ -365,7 +365,7 @@ select
     cast(oq.buy_amount as numeric(78, 0)) :: text as quote_buy_amount,
     oq.gas_amount * oq.gas_price as quote_gas_cost,
     oq.sell_token_price as quote_sell_token_price,
-    opfp.partner_fee,
+    cast(coalesce(opfp.partner_fee, 0) as numeric(78, 0)) :: text as partner_fee,
     opfp.partner_fee_recipient,
     opfp.protocol_fee_kind
 from

diff --git a/tests/integration/test_fetch_orderbook.py b/tests/integration/test_fetch_orderbook.py
@@ -146,6 +146,15 @@ def test_get_order_rewards(self):
                     1.0,
                     1.0,
                 ],
+                "partner_fee": [
+                    "0",
+                    "0",
+                    "0",
+                    "0",
+                    "0",
+                    "0",
+                    "0",
+                ],
                 "partner_fee_recipient": [
                     None,
                     None,
@@ -395,6 +404,11 @@ def test_get_order_rewards_with_integrator_fee(self):
                     0.000234609143374563,
                     0.000160802298220274,
                 ],
+                "partner_fee": [
+                    "346011",
+                    "0",
+                    "0",
+                ],
                 "partner_fee_recipient": [
                     "0x9FA3c00a92Ec5f96B1Ad2527ab41B3932EFEDa58",
                     None,