#!/usr/bin/env python3
"""Test session cookie handling"""
import requests
import urllib3
import base64
import pickle
urllib3.disable_warnings()

BASE = "https://localhost:5000"

# First, get a session cookie
print("[*] Getting a session from welcome app...")
s = requests.Session()
r = s.get(f"{BASE}/welcome/default/index", verify=False)
print(f"    Cookies: {s.cookies}")

# Check if there's a session_data cookie (cookie-based sessions)
for cookie in s.cookies:
    print(f"    Cookie: {cookie.name} = {cookie.value[:100]}")

# Try to register a user to get a session
print("\n[*] Trying to create a session via registration...")
r = s.get(f"{BASE}/welcome/default/user/register", verify=False)
print(f"    Status: {r.status_code}")
print(f"    Cookies after registration page: {s.cookies}")

for cookie in s.cookies:
    print(f"    Cookie: {cookie.name} = {cookie.value[:100] if len(cookie.value) > 100 else cookie.value}")
