如何对协议及其扩展进行单元测试

import json

def test_message_format():
    message = {
        "type": "request",
        "id": 123,
        "data": {"key": "value"}
    }
    serialized = json.dumps(message)
    deserialized = json.loads(serialized)
    assert deserialized == message

from transitions import Machine

class ProtocolStateMachine:
    states = ['init', 'ready', 'processing', 'completed']

    def __init__(self):
        self.machine = Machine(model=self, states=ProtocolStateMachine.states, initial='init')
        self.machine.add_transition(trigger='start', source='init', dest='ready')
        self.machine.add_transition(trigger='process', source='ready', dest='processing')
        self.machine.add_transition(trigger='finish', source='processing', dest='completed')

def test_state_transitions():
    state_machine = ProtocolStateMachine()
    state_machine.start()
    assert state_machine.state == 'ready'
    state_machine.process()
    assert state_machine.state == 'processing'
    state_machine.finish()
    assert state_machine.state == 'completed'

import time

def test_performance():
    start_time = time.time()
    # 模拟协议处理过程
    for i in range(1000000):
        pass
    end_time = time.time()
    assert end_time - start_time < 1.0  # 确保处理时间小于1秒