mirror of
https://github.com/PR0M3TH3AN/SeedPass.git
synced 2026-11-02 02:42:04 +00:00
google-labs-jules[bot]andPR0M3TH3AN
d729466513
- Updated `InMemorySecret.get_bytes()` in `src/utils/memory_protection.py` to raise a `RuntimeError` if the secret has been wiped, replacing the previous `AttributeError`. - Added `test_access_after_wipe_raises_error` to `src/tests/test_memory_protection.py` to verify this behavior. Co-authored-by: PR0M3TH3AN <[email protected]>
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
import sys
|
|
from pathlib import Path
|
|
|
|
sys.path.append(str(Path(__file__).resolve().parents[1]))
|
|
|
|
import pytest
|
|
|
|
from utils.memory_protection import InMemorySecret
|
|
|
|
|
|
def test_inmemory_secret_round_trip_bytes_and_str():
|
|
plaintext = b"super secret"
|
|
secret = InMemorySecret(plaintext)
|
|
assert secret.get_bytes() == plaintext
|
|
assert secret.get_str() == plaintext.decode("utf-8")
|
|
|
|
|
|
def test_inmemory_secret_invalid_type():
|
|
with pytest.raises(TypeError):
|
|
InMemorySecret("not bytes")
|
|
|
|
|
|
def test_inmemory_secret_wipe_clears_attributes():
|
|
secret = InMemorySecret(b"wipe me")
|
|
secret.wipe()
|
|
assert secret._key is None
|
|
assert secret._nonce is None
|
|
assert secret._cipher is None
|
|
assert secret._encrypted is None
|
|
|
|
|
|
def test_access_after_wipe_raises_error():
|
|
secret = InMemorySecret(b"secret data")
|
|
secret.wipe()
|
|
|
|
with pytest.raises(RuntimeError) as excinfo:
|
|
secret.get_bytes()
|
|
assert str(excinfo.value) == "Secret has been wiped"
|
|
|
|
with pytest.raises(RuntimeError) as excinfo:
|
|
secret.get_str()
|
|
assert str(excinfo.value) == "Secret has been wiped"
|