Add API test script

Author: B-Hartley

custom_components/tunnelflight/__init__.py

@@ -1,8 +1,11 @@
 import logging
-from homeassistant.helpers import config_validation as cv
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import ConfigEntryNotReady
+from homeassistant.helpers import config_validation as cv
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
 
+from .api import TunnelflightApi
 from .const import DOMAIN
 from .logbook_service import async_setup_services, async_unload_services
 
@@ -36,7 +39,21 @@ async def async_setup(hass, config):
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Set up IBA Tunnelflight from a config entry."""
     hass.data.setdefault(DOMAIN, {})
-    hass.data[DOMAIN][entry.entry_id] = entry.data
+
+    session = async_get_clientsession(hass)
+    api = TunnelflightApi(
+        entry.data.get("username"), entry.data.get("password"), session
+    )
+
+    try:
+        login_success = await api.login()
+    except Exception as err:
+        raise ConfigEntryNotReady("Unable to connect to Tunnelflight API") from err
+
+    if not login_success:
+        raise ConfigEntryNotReady("Unable to authenticate with Tunnelflight API")
+
+    hass.data[DOMAIN][entry.entry_id] = {"api": api, **entry.data}
 
     # Set up platforms
     await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)

custom_components/tunnelflight/manifest.json

@@ -8,5 +8,5 @@
   "iot_class": "cloud_polling",
   "issue_tracker": "https://github.com/B-Hartley/tunnelflight/issues",
   "requirements": [],
-  "version": "1.3.5"
+  "version": "1.3.6"
 }

custom_components/tunnelflight/sensor.py

@@ -36,8 +36,11 @@ async def async_setup_entry(
     password = config[CONF_PASSWORD]
     name = config.get(CONF_NAME, DEFAULT_NAME)
 
-    session = async_get_clientsession(hass)
-    api = TunnelflightApi(username, password, session)
+    stored = hass.data.get(DOMAIN, {}).get(entry.entry_id, {})
+    api = stored.get("api")
+    if not api:
+        session = async_get_clientsession(hass)
+        api = TunnelflightApi(username, password, session)
 
     # Create a data coordinator to handle updates
     coordinator = TunnelflightCoordinator(hass, api)

test_api_calls.ps1

@@ -0,0 +1,60 @@
+param(
+    [string]$Username,
+    [string]$Password,
+    [string]$LogFile = "tunnelflight_api_test.log"
+)
+
+function Write-Log {
+    param([string]$Message)
+    $timestamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss"
+    "$timestamp`t$Message" | Out-File -FilePath $LogFile -Append
+}
+
+$headers = @{
+    "User-Agent" = "Mozilla/5.0 (Windows NT; Win64; x64)"
+    "Content-Type" = "application/json"
+}
+
+try {
+    Write-Log "Starting login request"
+    $loginBody = @{
+        username = $Username.ToLower()
+        password = $Password
+        passcode = ""
+        enable2fa = $false
+        checkTwoFactor = $true
+        passcodeOption = "email"
+    } | ConvertTo-Json
+    $loginResponse = Invoke-RestMethod -Uri "https://www.tunnelflight.com/login" -Method Post -Headers $headers -Body $loginBody
+    Write-Log "Login response: $($loginResponse | ConvertTo-Json -Compress)"
+    $token = $loginResponse.token
+} catch {
+    Write-Log "Login error: $_"
+    return
+}
+
+if (-not $token) {
+    Write-Log "No token received. Aborting further tests."
+    return
+}
+
+$authHeaders = $headers.Clone()
+$authHeaders["Authorization"] = "Bearer $token"
+
+$endpoints = @(
+    "https://www.tunnelflight.com/user/module-type/flyer-card/",
+    "https://www.tunnelflight.com/user/module-type/flyer-charts/",
+    "https://www.tunnelflight.com/account/logbook/tunnels/"
+)
+
+foreach ($endpoint in $endpoints) {
+    try {
+        Write-Log "Requesting $endpoint"
+        $response = Invoke-RestMethod -Uri $endpoint -Method Get -Headers $authHeaders
+        Write-Log "Response from $endpoint: $($response | ConvertTo-Json -Compress)"
+    } catch {
+        Write-Log "Error calling $endpoint: $_"
+    }
+}
+
+Write-Log "API test script finished"