From 42b7ef86a9a0dd5fad6e8d398914a24eb0a12001 Mon Sep 17 00:00:00 2001
From: Benoit Person <benoit.person@gmail.com>
Date: Sat, 12 Jun 2021 05:45:08 +0000
Subject: [PATCH] Fix `test_infinite_slots`

---
 airflow/models/pool.py    | 4 +++-
 tests/models/test_pool.py | 4 ++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/airflow/models/pool.py b/airflow/models/pool.py
index 5c42e0ad9d19f..3d152eee80fe3 100644
--- 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
index f4c7626615bda..7981e230f2af7 100644
--- 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()