padreug
57394c15dd
- Added Nostr private key storage to the accounts table via a new migration. - Updated Account model to include Nostr private key field. - Modified user creation process to automatically generate Nostr keypairs. - Introduced new API endpoints for retrieving Nostr public keys and user information including private keys. - Implemented tests to verify Nostr keypair generation and account model updates.
96 lines
No EOL
2.7 KiB
Python
96 lines
No EOL
2.7 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Test script to verify Nostr keypair integration with LNBits User model
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
|
|
# Add the lnbits directory to the path
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'lnbits'))
|
|
|
|
def test_nostr_keypair_generation():
|
|
"""Test that we can generate Nostr keypairs"""
|
|
try:
|
|
from lnbits.utils.nostr import generate_keypair
|
|
private_key, public_key = generate_keypair()
|
|
print(f"✓ Nostr keypair generation works")
|
|
print(f" Private key: {private_key[:16]}...")
|
|
print(f" Public key: {public_key}")
|
|
return True
|
|
except Exception as e:
|
|
print(f"✗ Nostr keypair generation failed: {e}")
|
|
return False
|
|
|
|
def test_account_model():
|
|
"""Test that the Account model includes prvkey field"""
|
|
try:
|
|
from lnbits.core.models.users import Account
|
|
account = Account(
|
|
id="test123",
|
|
username="testuser",
|
|
)
|
|
print(f"✓ Account model includes prvkey field")
|
|
print(f" prvkey: {account.prvkey}")
|
|
print(f" pubkey: {account.pubkey}")
|
|
return True
|
|
except Exception as e:
|
|
print(f"✗ Account model test failed: {e}")
|
|
return False
|
|
|
|
def test_user_model():
|
|
"""Test that the User model includes pubkey field"""
|
|
try:
|
|
from lnbits.core.models.users import User
|
|
user = User(
|
|
id="test123",
|
|
created_at=None,
|
|
updated_at=None,
|
|
pubkey="test_public_key"
|
|
)
|
|
print(f"✓ User model includes pubkey field")
|
|
print(f" pubkey: {user.pubkey}")
|
|
return True
|
|
except Exception as e:
|
|
print(f"✗ User model test failed: {e}")
|
|
return False
|
|
|
|
def test_migration():
|
|
"""Test that the migration function exists"""
|
|
try:
|
|
from lnbits.core.migrations import m034_add_nostr_private_key_to_accounts
|
|
print(f"✓ Migration function exists")
|
|
return True
|
|
except Exception as e:
|
|
print(f"✗ Migration test failed: {e}")
|
|
return False
|
|
|
|
def main():
|
|
print("Testing Nostr integration with LNBits...")
|
|
print("=" * 50)
|
|
|
|
tests = [
|
|
test_nostr_keypair_generation,
|
|
test_account_model,
|
|
test_user_model,
|
|
test_migration,
|
|
]
|
|
|
|
passed = 0
|
|
total = len(tests)
|
|
|
|
for test in tests:
|
|
if test():
|
|
passed += 1
|
|
print()
|
|
|
|
print("=" * 50)
|
|
print(f"Tests passed: {passed}/{total}")
|
|
|
|
if passed == total:
|
|
print("✓ All tests passed! Nostr integration is ready.")
|
|
else:
|
|
print("✗ Some tests failed. Please check the implementation.")
|
|
|
|
if __name__ == "__main__":
|
|
main() |