Upload folder using huggingface_hub

tests/test_subscription.py

@@ -13,7 +13,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, ACCEPTED_TOKENS, NETWORK,
+    FOUNDER_EMAIL, FOUNDER_PASSWORD, ACCEPTED_TOKENS, NETWORK,
 )
 
 
@@ -341,6 +341,77 @@ def test_soulmate_config():
     print(f"  Network: {NETWORK}")
 
 
+def test_founder_unlock_correct_password():
+    """Test founder unlock with correct password grants free forever access."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub = SubscriptionManager(data_dir=tmpdir)
+
+        # No access before unlock
+        access = sub.check_access()
+        assert not access["has_access"]
+
+        # Unlock with correct password
+        result = sub.founder_unlock(FOUNDER_PASSWORD)
+        assert result["success"]
+        assert result["status"] == "founder"
+        print(f"  Unlock: {result['message']}")
+
+        # Now has access — forever
+        access = sub.check_access()
+        assert access["has_access"]
+        assert access["status"] == "founder"
+        assert access["days_remaining"] == -1  # forever
+        print(f"  Access: {access['status']} — {access['message']}")
+
+
+def test_founder_unlock_wrong_password():
+    """Test founder unlock with wrong password fails."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub = SubscriptionManager(data_dir=tmpdir)
+
+        result = sub.founder_unlock("wrongpassword123")
+        assert not result["success"]
+        assert "Invalid" in result["error"]
+        print(f"  Wrong password: {result['error']}")
+
+        # Still no access
+        access = sub.check_access()
+        assert not access["has_access"]
+        print(f"  Access after wrong password: blocked")
+
+
+def test_founder_unlock_persists():
+    """Test founder unlock persists across restarts."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub1 = SubscriptionManager(data_dir=tmpdir)
+        sub1.founder_unlock(FOUNDER_PASSWORD)
+
+        # New instance should load founder status
+        sub2 = SubscriptionManager(data_dir=tmpdir)
+        access = sub2.check_access()
+        assert access["has_access"]
+        assert access["status"] == "founder"
+        print(f"  Persisted: status={access['status']}, forever={access['days_remaining'] == -1}")
+
+
+def test_founder_unlock_bypasses_payment():
+    """Test founder doesn't need to pay — access is free forever."""
+    with tempfile.TemporaryDirectory() as tmpdir:
+        sub = SubscriptionManager(data_dir=tmpdir)
+        sub.founder_unlock(FOUNDER_PASSWORD)
+
+        # Founder should have no payment history
+        stats = sub.get_stats()
+        assert stats["months_subscribed"] == 0
+        assert stats["total_paid"] == 0.0
+        print(f"  Founder payments: ${stats['total_paid']:.2f} (free)")
+
+        # But has full access
+        access = sub.check_access()
+        assert access["has_access"]
+        print(f"  Founder access: free forever")
+
+
 if __name__ == "__main__":
     print("Running subscription manager tests...")
     test_no_subscription_blocks_access()
@@ -375,4 +446,12 @@ if __name__ == "__main__":
     print("  ✓ test_subscribe_with_deposit")
     test_soulmate_config()
     print("  ✓ test_soulmate_config")
+    test_founder_unlock_correct_password()
+    print("  ✓ test_founder_unlock_correct_password")
+    test_founder_unlock_wrong_password()
+    print("  ✓ test_founder_unlock_wrong_password")
+    test_founder_unlock_persists()
+    print("  ✓ test_founder_unlock_persists")
+    test_founder_unlock_bypasses_payment()
+    print("  ✓ test_founder_unlock_bypasses_payment")
     print("\nAll subscription tests passed!")