コード例 #1
0
    def test_CellSortKey(self):
        cp = ConnectionPool(None)
        node_uuid_1 = self.getStorageUUID()
        node_uuid_2 = self.getStorageUUID()
        node_uuid_3 = self.getStorageUUID()
        # We are connected to node 1
        cp.connection_dict[node_uuid_1] = None

        def uuid_now(func, uuid, now):
            pool.time = Mock({'time': now})
            try:
                return func(Mock({'getUUID': uuid}))
            finally:
                pool.time = time

        # A connection to node 3 failed, will be forgotten at 5
        uuid_now(cp.notifyFailure, node_uuid_3, 5 - pool.MAX_FAILURE_AGE)

        def getCellSortKey(*args):
            return uuid_now(cp.getCellSortKey, *args)

        # At 0, key values are not ambiguous
        self.assertTrue(
            getCellSortKey(node_uuid_1, 0) < getCellSortKey(node_uuid_2, 0) <
            getCellSortKey(node_uuid_3, 0))
        # At 10, nodes 2 and 3 have the same key value
        self.assertTrue(
            getCellSortKey(node_uuid_1, 10) < getCellSortKey(node_uuid_2, 10))
        self.assertEqual(getCellSortKey(node_uuid_2, 10),
                         getCellSortKey(node_uuid_3, 10))
コード例 #2
0
ファイル: testConnectionPool.py プロジェクト: Nexedi/neoppod
    def test_CellSortKey(self):
        cp = ConnectionPool(None)
        node_uuid_1 = self.getStorageUUID()
        node_uuid_2 = self.getStorageUUID()
        node_uuid_3 = self.getStorageUUID()
        # We are connected to node 1
        cp.connection_dict[node_uuid_1] = None

        def uuid_now(func, uuid, now):
            pool.time = Mock({"time": now})
            try:
                return func(Mock({"getUUID": uuid}))
            finally:
                pool.time = time

        # A connection to node 3 failed, will be forgotten at 5
        uuid_now(cp.notifyFailure, node_uuid_3, 5 - pool.MAX_FAILURE_AGE)

        def getCellSortKey(*args):
            return uuid_now(cp.getCellSortKey, *args)

        # At 0, key values are not ambiguous
        self.assertTrue(
            getCellSortKey(node_uuid_1, 0) < getCellSortKey(node_uuid_2, 0) < getCellSortKey(node_uuid_3, 0)
        )
        # At 10, nodes 2 and 3 have the same key value
        self.assertTrue(getCellSortKey(node_uuid_1, 10) < getCellSortKey(node_uuid_2, 10))
        self.assertEqual(getCellSortKey(node_uuid_2, 10), getCellSortKey(node_uuid_3, 10))
コード例 #3
0
ファイル: testConnectionPool.py プロジェクト: Nexedi/neoppod
 def test_iterateForObject_connectionAccepted(self):
     # connection accepted
     oid = self.getOID(1)
     node = Mock({"__repr__": "node", "isRunning": True})
     cell = Mock({"__repr__": "cell", "getNode": node})
     conn = Mock({"__repr__": "conn"})
     app = Mock()
     app.pt = Mock({"getCellList": [cell]})
     pool = ConnectionPool(app)
     pool.getConnForNode = Mock({"__call__": conn})
     self.assertEqual(list(pool.iterateForObject(oid)), [(node, conn)])
コード例 #4
0
ファイル: testConnectionPool.py プロジェクト: Nexedi/neoppod
 def test_iterateForObject_connectionRefused(self):
     # connection refused at the first try
     oid = self.getOID(1)
     node = Mock({"__repr__": "node", "isRunning": True})
     cell = Mock({"__repr__": "cell", "getNode": node})
     conn = Mock({"__repr__": "conn"})
     app = Mock()
     app.pt = Mock({"getCellList": [cell]})
     pool = ConnectionPool(app)
     pool.getConnForNode = Mock({"__call__": ReturnValues(None, conn)})
     self.assertEqual(list(pool.iterateForObject(oid)), [(node, conn)])
コード例 #5
0
 def test_iterateForObject_connectionRefused(self):
     # connection refused at the first try
     oid = self.getOID(1)
     node = Mock({'__repr__': 'node', 'isRunning': True})
     cell = Mock({'__repr__': 'cell', 'getNode': node})
     conn = Mock({'__repr__': 'conn'})
     app = Mock()
     app.pt = Mock({'getCellList': [cell]})
     pool = ConnectionPool(app)
     pool.getConnForNode = Mock({'__call__': ReturnValues(None, conn)})
     self.assertEqual(list(pool.iterateForObject(oid)), [(node, conn)])
コード例 #6
0
 def test_iterateForObject_connectionAccepted(self):
     # connection accepted
     oid = self.getOID(1)
     node = Mock({'__repr__': 'node', 'isRunning': True})
     cell = Mock({'__repr__': 'cell', 'getNode': node})
     conn = Mock({'__repr__': 'conn'})
     app = Mock()
     app.pt = Mock({'getCellList': [cell]})
     pool = ConnectionPool(app)
     pool.getConnForNode = Mock({'__call__': conn})
     self.assertEqual(list(pool.iterateForObject(oid)), [(node, conn)])
コード例 #7
0
 def test_iterateForObject_noStorageAvailable(self):
     # no node available
     oid = self.getOID(1)
     app = Mock()
     app.pt = Mock({'getCellList': []})
     pool = ConnectionPool(app)
     self.assertRaises(NEOStorageError, pool.iterateForObject(oid).next)
コード例 #8
0
ファイル: testConnectionPool.py プロジェクト: Nexedi/neoppod
 def test_removeConnection(self):
     app = None
     pool = ConnectionPool(app)
     test_node_uuid = self.getStorageUUID()
     other_node_uuid = self.getStorageUUID()
     test_node = Mock({"getUUID": test_node_uuid})
     other_node = Mock({"getUUID": other_node_uuid})
     # Test sanity check
     self.assertEqual(getattr(pool, "connection_dict", None), {})
     # Call must not raise if node is not known
     self.assertEqual(len(pool.connection_dict), 0)
     pool.removeConnection(test_node)
     # Test that removal with another uuid doesn't affect entry
     pool.connection_dict[test_node_uuid] = None
     self.assertEqual(len(pool.connection_dict), 1)
     pool.removeConnection(other_node)
     self.assertEqual(len(pool.connection_dict), 1)
     # Test that removeConnection works
     pool.removeConnection(test_node)
     self.assertEqual(len(pool.connection_dict), 0)
コード例 #9
0
 def test_removeConnection(self):
     app = None
     pool = ConnectionPool(app)
     test_node_uuid = self.getStorageUUID()
     other_node_uuid = self.getStorageUUID()
     test_node = Mock({'getUUID': test_node_uuid})
     other_node = Mock({'getUUID': other_node_uuid})
     # Test sanity check
     self.assertEqual(getattr(pool, 'connection_dict', None), {})
     # Call must not raise if node is not known
     self.assertEqual(len(pool.connection_dict), 0)
     pool.removeConnection(test_node)
     # Test that removal with another uuid doesn't affect entry
     pool.connection_dict[test_node_uuid] = None
     self.assertEqual(len(pool.connection_dict), 1)
     pool.removeConnection(other_node)
     self.assertEqual(len(pool.connection_dict), 1)
     # Test that removeConnection works
     pool.removeConnection(test_node)
     self.assertEqual(len(pool.connection_dict), 0)