MetaWalletGen-CLI/test_cli.py at main · JustineDevs/MetaWalletGen-CLI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#!/usr/bin/env python3
"""
Test script for MetaWalletGen CLI
"""

import sys
import os

# Add the current directory to Python path
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

def test_cli():
    """Test the CLI functionality"""
    try:
        print("Testing MetaWalletGen CLI...")
        print("=" * 50)

        # Test importing the package
        print("1. Testing package import...")
        import metawalletgen
        print("   ✓ Package imported successfully")

        # Test importing core modules
        print("2. Testing core modules...")
        from metawalletgen.core.wallet_generator import WalletGenerator
        from metawalletgen.core.storage_manager import StorageManager
        from metawalletgen.core.encryption import EncryptionManager
        print("   ✓ Core modules imported successfully")

        # Test CLI modules
        print("3. Testing CLI modules...")
        from metawalletgen.cli.main import main
        print("   ✓ CLI modules imported successfully")

        # Test wallet generation
        print("4. Testing wallet generation...")
        generator = WalletGenerator()
        wallet = generator.generate_new_wallet()
        print(f"   ✓ Generated wallet: {wallet.address[:10]}...")

        # Test storage
        print("5. Testing storage...")
        storage = StorageManager()
        print("   ✓ Storage manager initialized")

        # Test encryption
        print("6. Testing encryption...")
        encryption = EncryptionManager()
        test_data = "test data"
        encrypted = encryption.encrypt_data(test_data, "test_password")
        decrypted = encryption.decrypt_data(encrypted, "test_password")
        assert decrypted == test_data
        print("   ✓ Encryption/decryption working")

        print("\n" + "=" * 50)
        print("🎉 All tests passed! CLI is working correctly.")
        print("\nTo run the CLI, use:")
        print("  python -m metawalletgen.cli.main --help")
        print("  python -m metawalletgen.cli.main generate --count 1")

    except ImportError as e:
        print(f"❌ Import error: {e}")
        print("Try installing dependencies: pip install -r requirements.txt")
    except Exception as e:
        print(f"❌ Error: {e}")
        import traceback
        traceback.print_exc()

if __name__ == "__main__":
    test_cli()