Upload folder using huggingface_hub

tests/test_subscription.py

@@ -14,6 +14,7 @@ sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
 from splitbit_llm.subscription import (
     SubscriptionManager, MONTHLY_PRICE, SOULMATE_API_URL,
     FOUNDER_EMAIL, FOUNDER_PASSWORD, ACCEPTED_TOKENS, NETWORK,
+    BANK_ROUTING, BANK_ACCOUNT, AUTO_TRANSFER_ENABLED,
 )
 
 
@@ -412,6 +413,153 @@ def test_founder_unlock_bypasses_payment():
         print(f"  Founder access: free forever")
 
 
+def test_auto_transfer_init():
+    """Test auto-transfer state initializes correctly."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub = SubscriptionManager(data_dir=tmpdir)
+        sub._init_auto_transfer()
+        assert "auto_transfer_history" in sub._state
+        assert "transferred_payments" in sub._state
+        assert "auto_transfer_enabled" in sub._state
+        print(f"  Auto-transfer enabled: {sub._state['auto_transfer_enabled']}")
+
+
+def test_auto_transfer_stats_empty():
+    """Test auto-transfer stats with no transfers."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub = SubscriptionManager(data_dir=tmpdir)
+        stats = sub.get_auto_transfer_stats()
+        assert stats["transfer_count"] == 0
+        assert stats["total_transferred"] == 0.0
+        assert "auto_transfer_enabled" in stats
+        print(f"  Stats: {stats['transfer_count']} transfers, ${stats['total_transferred']:.2f}")
+
+
+def test_auto_transfer_process_no_payments():
+    """Test auto-transfer when there are no payments to transfer."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub = SubscriptionManager(data_dir=tmpdir)
+        result = sub.process_auto_transfers()
+        assert result["status"] == "ok"
+        assert result["transfers_made"] == 0
+        print(f"  No payments: {result['message']}")
+
+
+def test_auto_transfer_process_with_payment():
+    """Test auto-transfer processes a confirmed $1 payment."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub = SubscriptionManager(data_dir=tmpdir)
+
+        # Simulate a confirmed payment
+        sub.subscribe(payment_reference="test-pay-001", create_deposit=False)
+
+        # Process auto-transfers
+        result = sub.process_auto_transfers()
+        assert result["status"] == "ok"
+        assert result["transfers_made"] == 1
+        assert result["total_amount"] == MONTHLY_PRICE
+        print(f"  Processed: {result['transfers_made']} transfer(s), ${result['total_amount']:.2f}")
+
+        # Verify transfer recorded
+        stats = sub.get_auto_transfer_stats()
+        assert stats["transfer_count"] == 1
+        assert stats["total_transferred"] == MONTHLY_PRICE
+        print(f"  Recorded: {stats['transfer_count']} transfer(s), ${stats['total_transferred']:.2f}")
+
+
+def test_auto_transfer_no_duplicates():
+    """Test auto-transfer doesn't transfer the same payment twice."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub = SubscriptionManager(data_dir=tmpdir)
+        sub.subscribe(payment_reference="test-pay-002", create_deposit=False)
+
+        # First transfer
+        result1 = sub.process_auto_transfers()
+        assert result1["transfers_made"] == 1
+
+        # Second call should not re-transfer
+        result2 = sub.process_auto_transfers()
+        assert result2["transfers_made"] == 0
+        assert "No pending transfers" in result2["message"]
+        print(f"  No duplicates: first={result1['transfers_made']}, second={result2['transfers_made']}")
+
+
+def test_auto_transfer_toggle():
+    """Test enabling/disabling auto-transfer."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub = SubscriptionManager(data_dir=tmpdir)
+
+        # Disable
+        result = sub.set_auto_transfer(False)
+        assert result["success"]
+        assert not result["auto_transfer_enabled"]
+
+        # Process should be skipped
+        sub.subscribe(payment_reference="test-pay-003", create_deposit=False)
+        process_result = sub.process_auto_transfers()
+        assert process_result["status"] == "disabled"
+        print(f"  Disabled: {process_result['message']}")
+
+        # Re-enable
+        result = sub.set_auto_transfer(True)
+        assert result["success"]
+        assert result["auto_transfer_enabled"]
+        print(f"  Re-enabled: {result['message']}")
+
+
+def test_auto_transfer_multiple_payments():
+    """Test auto-transfer processes multiple payments at once."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub = SubscriptionManager(data_dir=tmpdir)
+
+        # Three payments
+        sub.subscribe(payment_reference="pay-001", create_deposit=False)
+        sub.subscribe(payment_reference="pay-002", create_deposit=False)
+        sub.subscribe(payment_reference="pay-003", create_deposit=False)
+
+        # Process all
+        result = sub.process_auto_transfers()
+        assert result["transfers_made"] == 3
+        assert result["total_amount"] == MONTHLY_PRICE * 3
+        print(f"  Batch: {result['transfers_made']} transfers, ${result['total_amount']:.2f}")
+
+        stats = sub.get_auto_transfer_stats()
+        assert stats["transfer_count"] == 3
+        assert stats["total_transferred"] == MONTHLY_PRICE * 3
+        print(f"  Total: {stats['transfer_count']} transfers, ${stats['total_transferred']:.2f}")
+
+
+def test_bank_info():
+    """Test bank info returns masked account details."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub = SubscriptionManager(data_dir=tmpdir)
+        info = sub.get_bank_info()
+        assert "routing_number" in info
+        assert "account_number" in info
+        assert "auto_transfer_enabled" in info
+        assert "total_transferred" in info
+        # Should be masked or "Not set"
+        assert info["routing_number"] == "Not set" or "****" in info["routing_number"]
+        assert info["account_number"] == "Not set" or "****" in info["account_number"]
+        print(f"  Routing: {info['routing_number']}")
+        print(f"  Account: {info['account_number']}")
+
+
+def test_auto_transfer_persists():
+    """Test auto-transfer history persists across restarts."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub1 = SubscriptionManager(data_dir=tmpdir)
+        sub1.subscribe(payment_reference="persist-pay-001", create_deposit=False)
+        sub1.process_auto_transfers()
+
+        # New instance
+        sub2 = SubscriptionManager(data_dir=tmpdir)
+        stats = sub2.get_auto_transfer_stats()
+        assert stats["transfer_count"] == 1
+        assert stats["total_transferred"] == MONTHLY_PRICE
+        print(f"  Persisted: {stats['transfer_count']} transfers, ${stats['total_transferred']:.2f}")
+
+
 if __name__ == "__main__":
     print("Running subscription manager tests...")
     test_no_subscription_blocks_access()
@@ -454,4 +602,22 @@ if __name__ == "__main__":
     print("  ✓ test_founder_unlock_persists")
     test_founder_unlock_bypasses_payment()
     print("  ✓ test_founder_unlock_bypasses_payment")
+    test_auto_transfer_init()
+    print("  ✓ test_auto_transfer_init")
+    test_auto_transfer_stats_empty()
+    print("  ✓ test_auto_transfer_stats_empty")
+    test_auto_transfer_process_no_payments()
+    print("  ✓ test_auto_transfer_process_no_payments")
+    test_auto_transfer_process_with_payment()
+    print("  ✓ test_auto_transfer_process_with_payment")
+    test_auto_transfer_no_duplicates()
+    print("  ✓ test_auto_transfer_no_duplicates")
+    test_auto_transfer_toggle()
+    print("  ✓ test_auto_transfer_toggle")
+    test_auto_transfer_multiple_payments()
+    print("  ✓ test_auto_transfer_multiple_payments")
+    test_bank_info()
+    print("  ✓ test_bank_info")
+    test_auto_transfer_persists()
+    print("  ✓ test_auto_transfer_persists")
     print("\nAll subscription tests passed!")