From 87090139f63b3e0f94c9cd23f66321f57eb0967b Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Mon, 31 Mar 2025 08:42:03 +1300 Subject: [PATCH] test: Refactor onboarding test imports --- tests/basic/test_onboarding.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/tests/basic/test_onboarding.py b/tests/basic/test_onboarding.py index 5dd461efc..6a8b8bc75 100644 --- a/tests/basic/test_onboarding.py +++ b/tests/basic/test_onboarding.py @@ -2,13 +2,23 @@ import argparse import base64 import hashlib import os -import secrets -import socketserver import unittest from unittest.mock import MagicMock, mock_open, patch import requests +# Import the functions to be tested +from aider.onboarding import ( + check_openrouter_tier, + exchange_code_for_key, + find_available_port, + generate_pkce_codes, + offer_openrouter_oauth, + select_default_model, + start_openrouter_oauth_flow, + try_to_select_default_model, +) + # Mock the Analytics class as it's used in some functions class DummyAnalytics: