test_client.py 文件源码

def test_handle_message_iam():
    """Test if a gateway is created and discover_devices is called with gateway IP
    after message "iam" is received"""

    src_addr = "10.10.10.10"
    msg_iam = {
        "cmd": "iam",
        "ip": src_addr,
        "sid": "123456"
    }

    mock_handler = MagicMock()
    client = AqaraClient()
    client.subscribe(mock_handler)

    client.handle_message(msg_iam, src_addr)

    assert len(client.gateways.keys()) == 1
    new_gateway = client.gateways["123456"]
    mock_handler.assert_called_once_with(
        sender=client,
        gateway=new_gateway,
        signal=AQARA_EVENT_NEW_GATEWAY
    )