def test_get_existing(self):
conn = MagicMock()
conn.check.return_value = True
pool = ConnectionPool(None)
pool._get_pool().put_nowait(conn)
self.assertEqual(conn, pool.get())
conn.check.assert_called_with()
def test_get_existing(self):
conn = MagicMock()
conn.check.return_value = True
pool = ConnectionPool(None)
pool._get_pool().put_nowait(conn)
self.assertEqual(conn, pool.get())
conn.check.assert_called_with()
def test_get_existing_broken(self):
conn_factory = MagicMock()
conn_factory.return_value = new_conn = MagicMock()
old_conn = MagicMock()
old_conn.check.return_value = False
pool = ConnectionPool(conn_factory)
pool._get_pool().put_nowait(old_conn)
self.assertEqual(new_conn, pool.get())
conn_factory.assert_called_with()
old_conn.check.assert_called_with()
def test_get_existing_broken(self):
conn_factory = MagicMock()
conn_factory.return_value = new_conn = MagicMock()
old_conn = MagicMock()
old_conn.check.return_value = False
pool = ConnectionPool(conn_factory)
pool._get_pool().put_nowait(old_conn)
self.assertEqual(new_conn, pool.get())
conn_factory.assert_called_with()
old_conn.check.assert_called_with()
def test_get_new(self):
conn_factory = MagicMock()
conn_factory.return_value = conn = MagicMock()
pool = ConnectionPool(conn_factory, (1, 2), {'three': 4})
self.assertEqual(conn, pool.get())
conn_factory.assert_called_with(1, 2, three=4)
def test_get_new(self):
conn_factory = MagicMock()
conn_factory.return_value = conn = MagicMock()
pool = ConnectionPool(conn_factory, (1, 2), {'three': 4})
self.assertEqual(conn, pool.get())
conn_factory.assert_called_with(1, 2, three=4)
