thePR0M3TH3AN/SeedPass
1
0 Fork 0
mirror of https://github.com/PR0M3TH3AN/SeedPass.git synced 2026-11-02 02:42:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(utils): Add test and error handling for wiped InMemorySecret

Browse Source 
- 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]>
This commit is contained in:
google-labs-jules[bot]andPR0M3TH3AN committed 2026-02-16 17:21:45 +00:00
1 parent 335de7abe3
commit d729466513
2 files changed
+15
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/tests/test_memory_protection.py
+13
View File
@@ -27,3 +27,16 @@ def test_inmemory_secret_wipe_clears_attributes():
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"