astronomer / astro-cli / 8ca19b7f-1af9-4517-92a2-6525047c01dc

39%

main: 45%

Build Type

Pull #2078

circleci

Committed by jlaneve

Commit Message

Switch settings import from CLI commands to Airflow REST API

Replace the heredoc/bash approach with direct HTTP calls to the Airflow
REST API. This is cross-platform (no bash dependency), faster (no CLI
startup overhead), and works identically for Docker and standalone modes.

For each variable, connection, and pool: POST to create, if 409
(already exists) then PATCH to update. This matches the previous
overwrite behavior.

Key changes:
- ConfigSettings now takes airflowURL + authHeader instead of container ID
- AddVariables/AddConnections/AddPools make HTTP calls to the REST API
- Airflow 2: Basic Auth (admin:admin) with /api/v1 endpoints
- Airflow 3: JWT auth via /auth/token with /api/v2 endpoints
- URI-only connections are parsed into individual fields (conn_type,
  host, login, password, port, schema) since the REST API doesn't
  accept a uri field directly
- printStatus/printProxyStatus no longer need container ID or Ps calls
- Tests use httptest.NewServer to mock the Airflow API
- Removed poolImportEntry type (no longer needed)

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Pull Request Pull Request #2078: Batch import Airflow settings to speed up dev start

Coverage Stats

167 of 218 new or added lines in 3 files covered. (76.61%)

21 existing lines in 3 files now uncovered.

24854 of 63174 relevant lines covered (39.34%)

9.52 hits per line