Upload folder using huggingface_hub

tests/test_subscription.py

@@ -1,12 +1,20 @@
-"""Test subscription manager — $1/month subscription with free trial."""
+"""Test subscription manager — $1/month subscription with free trial.
+
+Payment integration with Soulmate OS wallet API.
+"""
 
 import sys
 import os
 import tempfile
 import time
+import json
+from unittest.mock import patch, MagicMock
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
 
-from splitbit_llm.subscription import SubscriptionManager, MONTHLY_PRICE
+from splitbit_llm.subscription import (
+    SubscriptionManager, MONTHLY_PRICE, SOULMATE_API_URL,
+    FOUNDER_EMAIL, ACCEPTED_TOKENS, NETWORK,
+)
 
 
 def test_no_subscription_blocks_access():
@@ -50,7 +58,7 @@ def test_subscribe():
         sub = SubscriptionManager(data_dir=tmpdir)
 
         # Subscribe
-        result = sub.subscribe()
+        result = sub.subscribe(create_deposit=False)
         assert result["success"]
         assert result["status"] == "active"
         assert result["price"] == MONTHLY_PRICE
@@ -78,12 +86,12 @@ def test_subscribe_extends():
         sub = SubscriptionManager(data_dir=tmpdir)
 
         # First subscription
-        result1 = sub.subscribe()
+        result1 = sub.subscribe(create_deposit=False)
         assert result1["success"]
         assert result1["days_remaining"] >= 29
 
         # Second subscription should extend
-        result2 = sub.subscribe()
+        result2 = sub.subscribe(create_deposit=False)
         assert result2["success"]
         assert result2["days_remaining"] >= 59  # 30 + 30
         print(f"  Extended: {result2['days_remaining']} days (30 + 30)")
@@ -98,7 +106,7 @@ def test_unsubscribe():
     """Test cancelling subscription."""
     with tempfile.TemporaryDirectory() as tmpdir:
         sub = SubscriptionManager(data_dir=tmpdir)
-        sub.subscribe()
+        sub.subscribe(create_deposit=False)
 
         # Unsubscribe
         result = sub.unsubscribe()
@@ -118,7 +126,7 @@ def test_expired_blocks_access():
         sub = SubscriptionManager(data_dir=tmpdir)
 
         # Subscribe
-        sub.subscribe()
+        sub.subscribe(create_deposit=False)
 
         # Manually expire the subscription
         sub._state["expiration_date"] = time.time() - 1  # expired 1 second ago
@@ -145,7 +153,7 @@ def test_auto_renew():
     """Test auto-renewal when subscription expires."""
     with tempfile.TemporaryDirectory() as tmpdir:
         sub = SubscriptionManager(data_dir=tmpdir)
-        sub.subscribe()
+        sub.subscribe(create_deposit=False)
 
         months_before = sub._state["months_subscribed"]
 
@@ -168,7 +176,7 @@ def test_trial_persistence():
     """Test that subscription state persists across restarts."""
     with tempfile.TemporaryDirectory() as tmpdir:
         sub1 = SubscriptionManager(data_dir=tmpdir)
-        sub1.subscribe()
+        sub1.subscribe(create_deposit=False)
 
         # Create new instance — should load state
         sub2 = SubscriptionManager(data_dir=tmpdir)
@@ -193,7 +201,7 @@ def test_trial_then_subscribe():
         print(f"  Trial: {trial['days_remaining']} days")
 
         # Subscribe while on trial
-        result = sub.subscribe()
+        result = sub.subscribe(create_deposit=False)
         assert result["success"]
         assert result["status"] == "active"
         print(f"  Subscribed during trial: {result['days_remaining']} days")
@@ -210,9 +218,9 @@ def test_payment_history():
         sub = SubscriptionManager(data_dir=tmpdir)
 
         # Multiple payments
-        sub.subscribe(payment_reference="stripe-001")
-        sub.subscribe(payment_reference="stripe-002")
-        sub.subscribe(payment_reference="stripe-003")
+        sub.subscribe(payment_reference="stripe-001", create_deposit=False)
+        sub.subscribe(payment_reference="stripe-002", create_deposit=False)
+        sub.subscribe(payment_reference="stripe-003", create_deposit=False)
 
         stats = sub.get_stats()
         assert stats["months_subscribed"] == 3
@@ -222,6 +230,117 @@ def test_payment_history():
         print(f"  History: {len(stats['payment_history'])} records")
 
 
+def test_payment_instructions():
+    """Test payment instructions are returned correctly."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub = SubscriptionManager(data_dir=tmpdir)
+        # Set a fake wallet in state so we don't hit the API
+        sub._state["founder_wallet"] = "0xABC123DEF456"
+        sub._founder_wallet = "0xABC123DEF456"
+
+        instructions = sub.get_payment_instructions()
+        assert instructions["amount"] == MONTHLY_PRICE
+        assert instructions["currency"] == "USD"
+        assert instructions["period"] == "monthly"
+        assert instructions["founder_wallet"] == "0xABC123DEF456"
+        assert instructions["network"] == NETWORK
+        assert "USDT" in instructions["accepted_tokens"]
+        assert "USDC" in instructions["accepted_tokens"]
+        assert "BNB" in instructions["accepted_tokens"]
+        assert "INC" in instructions["accepted_tokens"]
+        assert "soulmate_wallet_url" in instructions
+        assert SOULMATE_API_URL in instructions["soulmate_wallet_url"]
+        print(f"  Amount: ${instructions['amount']:.2f}")
+        print(f"  Wallet: {instructions['founder_wallet']}")
+        print(f"  Network: {instructions['network']}")
+        print(f"  Tokens: {instructions['accepted_tokens']}")
+        print(f"  Google Pay URL: {instructions['soulmate_wallet_url']}")
+
+
+def test_payment_instructions_with_wallet():
+    """Test payment instructions include wallet from state."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub = SubscriptionManager(data_dir=tmpdir)
+        sub._state["founder_wallet"] = "0xDEADBEEF1234"
+        sub._founder_wallet = "0xDEADBEEF1234"
+
+        instructions = sub.get_payment_instructions()
+        assert instructions["founder_wallet"] == "0xDEADBEEF1234"
+        assert "0xDEADBEEF1234" in instructions["instructions"]
+        print(f"  Wallet in instructions: {instructions['founder_wallet']}")
+
+
+def test_create_deposit():
+    """Test deposit creation (mocked API)."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub = SubscriptionManager(data_dir=tmpdir)
+        sub._state["founder_wallet"] = "0xTEST123"
+        sub._founder_wallet = "0xTEST123"
+
+        # Mock the API call — deposit creation will fall back to pending
+        result = sub.create_deposit("test-user", "USDT")
+        assert "deposit_id" in result
+        assert result["founder_wallet"] == "0xTEST123"
+        assert result["token"] == "USDT"
+        assert result["amount"] == MONTHLY_PRICE
+        print(f"  Deposit ID: {result['deposit_id'][:16]}...")
+        print(f"  Status: {result.get('status', 'pending')}")
+        print(f"  Wallet: {result['founder_wallet']}")
+
+
+def test_verify_payment():
+    """Test payment verification (mocked API — will return error since no real API)."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub = SubscriptionManager(data_dir=tmpdir)
+
+        # Verify a non-existent deposit — should return some status
+        result = sub.verify_payment("fake-deposit-123")
+        assert isinstance(result, dict)
+        # Without a real API, it'll be "error", "not_found", or have some other status
+        print(f"  Verify result: {result.get('status', 'unknown')} — '{result.get('message', '')}'")
+
+
+def test_subscribe_with_deposit():
+    """Test subscribing with deposit creation."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub = SubscriptionManager(data_dir=tmpdir)
+        sub._state["founder_wallet"] = "0xWALLET456"
+        sub._founder_wallet = "0xWALLET456"
+
+        result = sub.subscribe(user_id="test-user", token="USDT")
+        assert result["success"]
+        assert result["status"] == "active"
+        assert "deposit_id" in result
+        assert result["founder_wallet"] == "0xWALLET456"
+        assert "payment" in result
+        assert result["payment"]["founder_wallet"] == "0xWALLET456"
+        print(f"  Subscribed with deposit: {result['deposit_id'][:16]}...")
+        print(f"  Payment instructions included: {bool(result.get('payment'))}")
+
+        # Check payment history has deposit info
+        stats = sub.get_stats()
+        last_payment = stats["payment_history"][-1]
+        assert last_payment["token"] == "USDT"
+        assert last_payment["deposit_id"] == result["deposit_id"]
+        assert "0xWALLET456" in last_payment["founder_wallet"] or last_payment["founder_wallet"].startswith("0xWALLET")
+        print(f"  Payment history: token={last_payment['token']}, deposit={last_payment['deposit_id'][:16]}...")
+
+
+def test_soulmate_config():
+    """Test Soulmate OS config constants are set correctly."""
+    assert SOULMATE_API_URL == "https://191.44.121.29.sslip.io"
+    assert FOUNDER_EMAIL == "hawpetossjustin25@gmail.com"
+    assert "USDT" in ACCEPTED_TOKENS
+    assert "USDC" in ACCEPTED_TOKENS
+    assert "BNB" in ACCEPTED_TOKENS
+    assert "INC" in ACCEPTED_TOKENS
+    assert NETWORK == "BSC (Binance Smart Chain)"
+    print(f"  API URL: {SOULMATE_API_URL}")
+    print(f"  Founder: {FOUNDER_EMAIL}")
+    print(f"  Tokens: {ACCEPTED_TOKENS}")
+    print(f"  Network: {NETWORK}")
+
+
 if __name__ == "__main__":
     print("Running subscription manager tests...")
     test_no_subscription_blocks_access()
@@ -244,4 +363,16 @@ if __name__ == "__main__":
     print("  ✓ test_trial_then_subscribe")
     test_payment_history()
     print("  ✓ test_payment_history")
+    test_payment_instructions()
+    print("  ✓ test_payment_instructions")
+    test_payment_instructions_with_wallet()
+    print("  ✓ test_payment_instructions_with_wallet")
+    test_create_deposit()
+    print("  ✓ test_create_deposit")
+    test_verify_payment()
+    print("  ✓ test_verify_payment")
+    test_subscribe_with_deposit()
+    print("  ✓ test_subscribe_with_deposit")
+    test_soulmate_config()
+    print("  ✓ test_soulmate_config")
     print("\nAll subscription tests passed!")