mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-10-28 09:03:46 +08:00
firestar5683
36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
from dataclasses import fields
|
|
import pytest
|
|
from openpilot.starpilot.flm import effect_trace
|
|
from openpilot.starpilot.flm.tests.test_selected_effect_owner import fixture, NOW
|
|
from openpilot.starpilot.flm.torque_adapter import Admission
|
|
|
|
|
|
def original():
|
|
owner, p, v, context = fixture()
|
|
owner.prepare(Admission(p, NOW, True, False, True, v), active=True, current_context=context)
|
|
return owner.trace
|
|
|
|
|
|
def test_virtual_validation_subclass_cannot_mint_sender_cache_proof():
|
|
class Forged(effect_trace.FullEffectTrace):
|
|
def validate(self):
|
|
return self
|
|
|
|
real = original()
|
|
forged = Forged(**{field.name: getattr(real, field.name) for field in fields(real)})
|
|
with pytest.raises(ValueError):
|
|
effect_trace.retain_sender_verification(forged)
|
|
assert id(forged) not in effect_trace._SENDER_PROOFS
|
|
|
|
|
|
@pytest.mark.parametrize('kind', ['subclass', 'short', 'nonhex'])
|
|
def test_noncanonical_digest_cannot_mint_sender_cache_proof(kind):
|
|
class MutableString(str):
|
|
pass
|
|
|
|
value = original()
|
|
digest = MutableString(value.applied_full_profile_sha256) if kind == 'subclass' else 'a' * 63 if kind == 'short' else 'z' * 64
|
|
object.__setattr__(value, 'applied_full_profile_sha256', digest)
|
|
with pytest.raises(ValueError):
|
|
effect_trace.retain_sender_verification(value)
|