admin felület fejlesztése garázs, előfizeztési csomagok

tests/active/test_subscription_feature_flags.py

@@ -0,0 +1,139 @@
+#!/usr/bin/env python3
+"""
+Test script for subscription feature flag endpoints (Gitea #239).
+Tests: GET /subscriptions/my, GET /subscriptions/feature-flags, GET /subscriptions/check/{key}
+"""
+import asyncio
+import httpx
+import sys
+
+BASE_URL = "http://sf_api:8000/api/v1"
+
+async def main():
+    async with httpx.AsyncClient(base_url=BASE_URL) as client:
+        # ── Step 1: Login with OAuth2 form data ──
+        print("=" * 60)
+        print("STEP 1: Login with admin user")
+        print("=" * 60)
+        
+        login_data = {
+            "username": "admin@profibot.hu",
+            "password": "Admin123!",
+            "grant_type": "password",
+        }
+        
+        login_resp = await client.post(
+            "/auth/login",
+            data=login_data,  # OAuth2 form data
+            headers={"Content-Type": "application/x-www-form-urlencoded"},
+        )
+        
+        if login_resp.status_code != 200:
+            print(f"❌ Login failed: {login_resp.status_code}")
+            print(f"   Response: {login_resp.text}")
+            sys.exit(1)
+        
+        token_data = login_resp.json()
+        token = token_data.get("access_token")
+        print(f"✅ Login OK — token: {token[:50]}...")
+        
+        headers = {"Authorization": f"Bearer {token}"}
+        
+        # ── Step 2: GET /subscriptions/my ──
+        print("\n" + "=" * 60)
+        print("STEP 2: GET /subscriptions/my")
+        print("=" * 60)
+        
+        resp = await client.get("/subscriptions/my", headers=headers)
+        if resp.status_code == 200:
+            data = resp.json()
+            print(f"✅ Status: 200")
+            print(f"   Tier: {data.get('tier')}")
+            print(f"   Source: {data.get('source')}")
+            print(f"   Expires: {data.get('expires_at')}")
+            features = data.get('features', {})
+            print(f"   Features count: {len(features)}")
+            # Show first 5 features
+            for i, (k, v) in enumerate(features.items()):
+                if i >= 5:
+                    print(f"   ... and {len(features) - 5} more")
+                    break
+                print(f"   - {k}: {v}")
+            sub = data.get('subscription')
+            if sub:
+                print(f"   Subscription tier_name: {sub.get('tier_name')}")
+                print(f"   Subscription display_name: {sub.get('display_name')}")
+        else:
+            print(f"❌ Failed: {resp.status_code}")
+            print(f"   Response: {resp.text[:500]}")
+        
+        # ── Step 3: GET /subscriptions/feature-flags ──
+        print("\n" + "=" * 60)
+        print("STEP 3: GET /subscriptions/feature-flags")
+        print("=" * 60)
+        
+        resp = await client.get("/subscriptions/feature-flags", headers=headers)
+        if resp.status_code == 200:
+            data = resp.json()
+            print(f"✅ Status: 200")
+            print(f"   Tier: {data.get('tier')}")
+            features = data.get('features', {})
+            print(f"   Features count: {len(features)}")
+            # Show all features
+            for k, v in sorted(features.items()):
+                print(f"   - {k}: {v}")
+        else:
+            print(f"❌ Failed: {resp.status_code}")
+            print(f"   Response: {resp.text[:500]}")
+        
+        # ── Step 4: GET /subscriptions/check/{feature_key} ──
+        print("\n" + "=" * 60)
+        print("STEP 4: GET /subscriptions/check/analytics_tco")
+        print("=" * 60)
+        
+        resp = await client.get("/subscriptions/check/analytics_tco", headers=headers)
+        if resp.status_code == 200:
+            data = resp.json()
+            print(f"✅ Status: 200")
+            print(f"   feature_key: {data.get('feature_key')}")
+            print(f"   granted: {data.get('granted')}")
+            print(f"   tier: {data.get('tier')}")
+            print(f"   required_tier: {data.get('required_tier')}")
+        else:
+            print(f"❌ Failed: {resp.status_code}")
+            print(f"   Response: {resp.text[:500]}")
+        
+        # ── Step 5: GET /subscriptions/check/unknown_feature ──
+        print("\n" + "=" * 60)
+        print("STEP 5: GET /subscriptions/check/unknown_feature (expect 404)")
+        print("=" * 60)
+        
+        resp = await client.get("/subscriptions/check/unknown_feature", headers=headers)
+        if resp.status_code == 404:
+            print(f"✅ Correctly returned 404 for unknown feature")
+        else:
+            print(f"❌ Expected 404, got {resp.status_code}")
+            print(f"   Response: {resp.text[:500]}")
+        
+        # ── Step 6: GET /subscriptions/public ──
+        print("\n" + "=" * 60)
+        print("STEP 6: GET /subscriptions/public")
+        print("=" * 60)
+        
+        resp = await client.get("/subscriptions/public", headers=headers)
+        if resp.status_code == 200:
+            data = resp.json()
+            print(f"✅ Status: 200")
+            print(f"   Public tiers count: {len(data)}")
+            for tier in data[:3]:
+                print(f"   - {tier.get('name')}: {tier.get('resolved_pricing', {}).get('monthly_price', 'N/A')}")
+        else:
+            print(f"❌ Failed: {resp.status_code}")
+            print(f"   Response: {resp.text[:500]}")
+        
+        print("\n" + "=" * 60)
+        print("✅ ALL TESTS COMPLETED")
+        print("=" * 60)
+
+if __name__ == "__main__":
+    asyncio.run(main())