コード例 #1
0
ファイル: test_units.py プロジェクト: urbanairship/kiddiepool
class TestKiddiePool(unittest.TestCase):
    def setUp(self):
        super(TestKiddiePool, self).setUp()
        self.pool = KiddiePool(
            ['foo:123', 'bar:321'],
            connection_factory=FakeConnection,
            connection_options={'tcp_keepalives': False},
            max_size=2,
            pool_timeout=.1,
            connect_attempts=2,
        )

    def test_max_size_and_pool_timeout(self):
        # Try to get more connections than max_size
        self.pool.get()
        self.pool.get()

        # Start timer
        start = time.time()

        # Make sure getting next connection throws KiddiePoolEmpty
        self.assertRaises(
            KiddiePoolEmpty,
            self.pool.get
        )

        # Make sure it took at least the timeout we set seconds
        self.assertTrue(time.time() - start > self.pool.pool_timeout)

    def test_connect_attempts(self):
        # Make a kiddiepool and mock the connection
        conn = FakeConnection()

        with patch.object(conn, 'connect') as mock_connect:
            mock_connect.return_value = False

            with self.assertRaises(KiddiePoolMaxAttempts):
                self.pool._connect(conn)

            self.assertEqual(sorted(mock_connect.call_args_list), sorted([
                call('foo', 123),
                call('foo', 123),
                call('bar', 321),
                call('bar', 321),
            ]))

    def test_connection_options(self):
        # Pass connection options into kiddiepool (in setUp)
        # Make sure they are applied to internal connections
        self.assertFalse(self.pool.connection_pool.get().tcp_keepalives)

    def test_reset_hosts(self):
        # Get connections
        orig_conn1 = self.pool.get()
        orig_conn2 = self.pool.get()

        self.pool.put(orig_conn1)
        self.pool.put(orig_conn2)

        # Change hosts
        self.pool.set_hosts(['baz:666'])

        # Get connections
        conn1 = self.pool.get()
        conn2 = self.pool.get()

        # make sure they changed to the new hosts
        self.assertEqual(conn1.host, 'baz')
        self.assertEqual(conn2.host, 'baz')
        self.assertEqual(conn1.port, 666)
        self.assertEqual(conn2.port, 666)
コード例 #2
0
class TestKiddiePool(mimic.MimicTestBase):
    def setUp(self):
        super(TestKiddiePool, self).setUp()
        self.pool = KiddiePool(
            ['foo:123', 'bar:321'],
            connection_factory=FakeConnection,
            connection_options={'tcp_keepalives': False},
            max_size=2,
            pool_timeout=.1,
            connect_attempts=2,
        )

    def test_max_size_and_pool_timeout(self):
        # Try to get more connections than max_size
        self.pool.get()
        self.pool.get()

        # Start timer
        start = time.time()

        # Make sure getting next connection throws KiddiePoolEmpty
        self.assertRaises(
            KiddiePoolEmpty,
            self.pool.get
        )

        # Make sure it took at least the timeout we set seconds
        self.assertTrue(time.time() - start > self.pool.pool_timeout)

    def test_connect_attempts(self):
        # Make a kiddiepool and mock the connection
        conn = FakeConnection()

        # Make sure it tries each host right number of times
        self.mimic.stub_out_with_mock(conn, 'connect')
        conn.connect('foo', 123).InAnyOrder().AndReturn(False)
        conn.connect('foo', 123).InAnyOrder().AndReturn(False)
        conn.connect('bar', 321).InAnyOrder().AndReturn(False)
        conn.connect('bar', 321).InAnyOrder().AndReturn(False)

        self.mimic.replay_all()

        # Make sure it raises KiddiePoolMaxAttempts as well
        self.assertRaises(
            KiddiePoolMaxAttempts,
            self.pool._connect,
            conn
        )

    def test_connection_options(self):
        # Pass connection options into kiddiepool (in setUp)
        # Make sure they are applied to internal connections
        self.assertFalse(self.pool.connection_pool.get().tcp_keepalives)

    def test_reset_hosts(self):
        # Get connections
        orig_conn1 = self.pool.get()
        orig_conn2 = self.pool.get()

        self.pool.put(orig_conn1)
        self.pool.put(orig_conn2)

        # Change hosts
        self.pool.set_hosts(['baz:666'])

        # Get connections
        conn1 = self.pool.get()
        conn2 = self.pool.get()

        # make sure they changed to the new hosts
        self.assertEqual(conn1.host, 'baz')
        self.assertEqual(conn2.host, 'baz')
        self.assertEqual(conn1.port, 666)
        self.assertEqual(conn2.port, 666)