Python CircularBuffer.write方法代码示例

# 需要导入模块: from circular_buffer import CircularBuffer [as 别名]
# 或者: from circular_buffer.CircularBuffer import write [as 别名]
 def test_overwrite_replaces_oldest_item(self):
     buf = CircularBuffer(2)
     buf.write('1')
     buf.write('2')
     buf.overwrite('3')
     self.assertEqual(buf.read(), '2')
     self.assertEqual(buf.read(), '3')

# 需要导入模块: from circular_buffer import CircularBuffer [as 别名]
# 或者: from circular_buffer.CircularBuffer import write [as 别名]
class Peer(object):
    def __init__(self, peer_id, socket):
        self.peer_id = peer_id
        self.socket = socket
        self.pending_requests = {}  # req_id -> ValueEvent
        self.incoming_buffer = msgpack.Unpacker()
        self.outgoing_buffer = CircularBuffer(OUTGOING_PEER_BUFFER_SIZE)
        self.lock = threading.Lock()

    def close(self):
        try:
            self.lock.release()
        except threading.ThreadError:
            pass  # lock may not have been acquired.
        self.socket.close()

    def add_request(self, req_id, request_bytes, value_event):
        """Add ValueEvent to pending requests and write request_bytes to
        outgoing_buffer"""
        with self.lock:
            self.pending_requests[req_id] = value_event
            self.outgoing_buffer.write(request_bytes)

    def handle_response(self, response):
        """Extract req_id from response and check if there is a pending request 
        for the req_id. If so set the value.
        """
        with self.lock:
            req_id, status, message = response
            if req_id in self.pending_requests:  # request may have timed out
                self.pending_requests[req_id].set((status, message))

    def remove_request(self, req_id):
        with self.lock:
            del self.pending_requests[req_id]

# 需要导入模块: from circular_buffer import CircularBuffer [as 别名]
# 或者: from circular_buffer.CircularBuffer import write [as 别名]
 def test_write_and_read_back_multiple_items(self):
     buf = CircularBuffer(2)
     buf.write('1')
     buf.write('2')
     self.assertEqual('1', buf.read())
     self.assertEqual('2', buf.read())
     with self.assertRaises(BufferEmptyException):
         buf.read()

# 需要导入模块: from circular_buffer import CircularBuffer [as 别名]
# 或者: from circular_buffer.CircularBuffer import write [as 别名]
 def test_read_back_oldest_item(self):
     buf = CircularBuffer(3)
     buf.write('1')
     buf.write('2')
     buf.read()
     buf.write('3')
     buf.read()
     self.assertEqual('3', buf.read())

# 需要导入模块: from circular_buffer import CircularBuffer [as 别名]
# 或者: from circular_buffer.CircularBuffer import write [as 别名]
 def test_overwrite_full_buffer(self):
     buf = CircularBuffer(2)
     buf.write('1')
     buf.write('2')
     buf.overwrite('A')
     self.assertEqual('2', buf.read())
     self.assertEqual('A', buf.read())
     with self.assertRaises(BufferEmptyException):
         buf.read()

# 需要导入模块: from circular_buffer import CircularBuffer [as 别名]
# 或者: from circular_buffer.CircularBuffer import write [as 别名]
 def test_overwrite_replaces_oldest_remaining_item(self):
     buf = CircularBuffer(3)
     buf.write('1')
     buf.write('2')
     buf.write('3')
     self.assertEqual(buf.read(), '1')
     buf.write('4')
     buf.overwrite('5')
     self.assertEqual(buf.read(), '3')
     self.assertEqual(buf.read(), '4')
     self.assertEqual(buf.read(), '5')

# 需要导入模块: from circular_buffer import CircularBuffer [as 别名]
# 或者: from circular_buffer.CircularBuffer import write [as 别名]
 def test_clearing_buffer(self):
     buf = CircularBuffer(3)
     for c in '123':
         buf.write(c)
     buf.clear()
     with self.assertRaises(BufferEmptyException):
         buf.read()
     buf.write('1')
     buf.write('2')
     self.assertEqual('1', buf.read())
     buf.write('3')
     self.assertEqual('2', buf.read())

# 需要导入模块: from circular_buffer import CircularBuffer [as 别名]
# 或者: from circular_buffer.CircularBuffer import write [as 别名]
 def test_alternate_read_and_overwrite(self):
     buf = CircularBuffer(5)
     for c in '123':
         buf.write(c)
     buf.read()
     buf.read()
     buf.write('4')
     buf.read()
     for c in '5678':
         buf.write(c)
     buf.overwrite('A')
     buf.overwrite('B')
     self.assertEqual('6', buf.read())
     self.assertEqual('7', buf.read())
     self.assertEqual('8', buf.read())
     self.assertEqual('A', buf.read())
     self.assertEqual('B', buf.read())
     with self.assertRaises(BufferEmptyException):
         buf.read()

# 需要导入模块: from circular_buffer import CircularBuffer [as 别名]
# 或者: from circular_buffer.CircularBuffer import write [as 别名]
 def test_write_and_read_back_multiple_items_ordered(self):
     buf = CircularBuffer(2)
     buf.write('1')
     buf.write('2')
     self.assertEqual(buf.read(), '1')
     self.assertEqual(buf.read(), '2')

# 需要导入模块: from circular_buffer import CircularBuffer [as 别名]
# 或者: from circular_buffer.CircularBuffer import write [as 别名]
 def test_write_full_buffer(self):
     buf = CircularBuffer(2)
     buf.write('1')
     buf.write('2')
     with self.assertRaises(BufferFullException):
         buf.write('A')

# 需要导入模块: from circular_buffer import CircularBuffer [as 别名]
# 或者: from circular_buffer.CircularBuffer import write [as 别名]
 def test_cant_write_to_full_buffer(self):
     buf = CircularBuffer(1)
     buf.write('1')
     with self.assertRaisesWithMessage(BufferFullException):
         buf.write('2')

# 需要导入模块: from circular_buffer import CircularBuffer [as 别名]
# 或者: from circular_buffer.CircularBuffer import write [as 别名]
 def test_alternate_write_and_read(self):
     buf = CircularBuffer(2)
     buf.write('1')
     self.assertEqual('1', buf.read())
     buf.write('2')
     self.assertEqual('2', buf.read())

# 需要导入模块: from circular_buffer import CircularBuffer [as 别名]
# 或者: from circular_buffer.CircularBuffer import write [as 别名]
 def test_clearing_buffer(self):
     buf = CircularBuffer(1)
     buf.write('1')
     buf.clear()
     with self.assertRaisesWithMessage(BufferEmptyException):
         buf.read()

# 需要导入模块: from circular_buffer import CircularBuffer [as 别名]
# 或者: from circular_buffer.CircularBuffer import write [as 别名]
 def test_write_and_read_back_one_item(self):
     buf = CircularBuffer(1)
     buf.write('1')
     self.assertEqual('1', buf.read())
     with self.assertRaises(BufferEmptyException):
         buf.read()

# 需要导入模块: from circular_buffer import CircularBuffer [as 别名]
# 或者: from circular_buffer.CircularBuffer import write [as 别名]
 def test_read_just_written_item(self):
     buf = CircularBuffer(1)
     buf.write('1')
     self.assertEqual(buf.read(), '1')