Skip to content

Commit 72d2907

Browse files
author
Sergio García Prado
committed
ISSUE #346
* Increase coverage.
1 parent 0aca88f commit 72d2907

File tree

2 files changed

+33
-5
lines changed

2 files changed

+33
-5
lines changed

packages/core/minos-microservice-common/minos/common/pools.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,11 +79,11 @@ def get_pool(self, type_: str, key: Optional[str] = None, **kwargs) -> Pool:
7979

8080
def _create_pool(self, type_: str, **kwargs) -> Pool:
8181
# noinspection PyTypeChecker
82-
pool_cls = self._get_pool_class(type_)
82+
pool_cls = self._get_pool_cls(type_)
8383
pool = pool_cls.from_config(self._config, **kwargs)
8484
return pool
8585

86-
def _get_pool_class(self, type_: str) -> type[Pool]:
86+
def _get_pool_cls(self, type_: str) -> type[Pool]:
8787
pool_cls = self._default_classes.get(type_)
8888

8989
if pool_cls is None:

packages/core/minos-microservice-common/tests/test_common/test_pools.py

Lines changed: 31 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,13 +24,41 @@
2424
)
2525
from tests.utils import (
2626
CommonTestCase,
27+
FakeLockPool,
2728
)
2829

2930

3031
class TestPoolFactory(CommonTestCase):
31-
def test_constructor(self):
32-
factory = PoolFactory.from_config(self.config)
33-
self.assertIsInstance(factory, PoolFactory)
32+
33+
def setUp(self):
34+
super().setUp()
35+
self.factory = PoolFactory(self.config, {"lock": FakeLockPool})
36+
37+
def test_from_config(self):
38+
self.assertIsInstance(PoolFactory.from_config(self.config), PoolFactory)
39+
40+
async def asyncTearDown(self):
41+
await self.factory.destroy()
42+
await super().asyncTearDown()
43+
44+
def test_get_pool(self):
45+
lock = self.factory.get_pool("lock")
46+
self.assertIsInstance(lock, FakeLockPool)
47+
self.assertEqual(lock, self.factory.get_pool("lock"))
48+
49+
def test_get_pool_with_key(self):
50+
lock_a = self.factory.get_pool("lock", "a")
51+
lock_b = self.factory.get_pool("lock", "b")
52+
self.assertIsInstance(lock_a, FakeLockPool)
53+
self.assertIsInstance(lock_b, FakeLockPool)
54+
55+
self.assertNotEqual(lock_a, lock_b)
56+
self.assertEqual(lock_a, self.factory.get_pool("lock", "a"))
57+
self.assertEqual(lock_b, self.factory.get_pool("lock", "b"))
58+
59+
def test_get_pool_raises(self):
60+
with self.assertRaises(ValueError):
61+
self.factory.get_pool("something")
3462

3563

3664
class TestPool(unittest.IsolatedAsyncioTestCase):

0 commit comments

Comments
 (0)