def METHOD_NAME(client: Client):
    # Test set wipe code.
    assert client.features.wipe_code_protection is None
    _ensure_unlocked(client, PIN4)
    assert client.features.wipe_code_protection is False
    with client:
        br_amount = 5 if client.debug.model == "T" else 6
        client.set_expected_responses(
            [messages.ButtonRequest()] * br_amount
            + [messages.Success, messages.Features]
        )
        client.use_pin_sequence([PIN4, WIPE_CODE_MAX, WIPE_CODE_MAX])
        device.change_wipe_code(client)
    client.init_device()
    assert client.features.wipe_code_protection is True
    _check_wipe_code(client, PIN4, WIPE_CODE_MAX)
    # Test change wipe code.
    with client:
        br_amount = 5 if client.debug.model == "T" else 6
        client.set_expected_responses(
            [messages.ButtonRequest()] * br_amount
            + [messages.Success, messages.Features]
        )
        client.use_pin_sequence([PIN4, WIPE_CODE6, WIPE_CODE6])
        device.change_wipe_code(client)
    client.init_device()
    assert client.features.wipe_code_protection is True
    _check_wipe_code(client, PIN4, WIPE_CODE6)
    # Test remove wipe code.
    with client:
        client.set_expected_responses(
            [messages.ButtonRequest()] * 3 + [messages.Success, messages.Features]
        )
        client.use_pin_sequence([PIN4])
        device.change_wipe_code(client, remove=True)
    client.init_device()
    assert client.features.wipe_code_protection is False