Fix test_infinite_slots

apache · Jun 17, 2021 · 42b7ef8 · 42b7ef8
1 parent 21f59f8
commit 42b7ef8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/airflow/models/pool.py b/airflow/models/pool.py
@@ -117,8 +117,10 @@ def slots_stats(
         ).all()
 
         # calculate queued and running metrics
-        count: int
         for (pool_name, state, count) in state_count_by_pool:
+            # Some databases return decimal.Decimal here.
+            count = int(count)
+
             stats_dict: Optional[PoolStats] = pools.get(pool_name)
             if not stats_dict:
                 continue

diff --git a/tests/models/test_pool.py b/tests/models/test_pool.py
@@ -110,10 +110,10 @@ def test_infinite_slots(self):
                 "running": 0,
             },
             "test_pool": {
-                "open": -1,
+                "open": float('inf'),
                 "queued": 1,
                 "running": 1,
-                "total": -1,
+                "total": float('inf'),
             },
         } == pool.slots_stats()