Skip to content

chore(dwctl): remove unused get_realtime_api_key test helper #1707

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
detail-app wants to merge 1 commit into
base: main
Choose a base branch
from
Open

chore(dwctl): remove unused get_realtime_api_key test helper #1707

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 1 addition & 13 deletions dwctl/src/test/sla.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,16 +31,6 @@ async fn get_batch_api_key(pool: &PgPool, user_id: Uuid) -> String {
.expect("Failed to get batch API key")
}

/// Helper to get or create a realtime API key for a user (for actual inference requests)
async fn get_realtime_api_key(pool: &PgPool, user_id: Uuid) -> String {
let mut conn = pool.acquire().await.expect("Failed to acquire connection");
let mut api_keys_repo = ApiKeys::new(&mut conn);
api_keys_repo
.get_or_create_hidden_key(user_id, ApiKeyPurpose::Realtime, user_id)
.await
.expect("Failed to get realtime API key")
}

/// Test route-at-claim-time escalation when batch is near expiry
///
/// This test verifies that when a batch is within `escalation_threshold_seconds`
Expand All @@ -58,9 +48,8 @@ async fn test_route_at_claim_time_escalation(pool: PgPool) {

// Create batch API key for the user
let user_batch_api_key = get_batch_api_key(&pool, user.id).await;
let _user_realtime_api_key = get_realtime_api_key(&pool, user.id).await;

tracing::info!("✅ Created batch and realtime API keys for user");
tracing::info!("✅ Created batch API key for user");

// Setup mock server that always returns 200 OK
let mock_server = MockServer::start().await;
Expand Down Expand Up @@ -354,7 +343,6 @@ async fn test_no_escalation_when_not_near_expiry(pool: PgPool) {
add_user_to_group(&pool, user.id, uuid::Uuid::nil()).await;

let user_batch_api_key = get_batch_api_key(&pool, user.id).await;
let _user_realtime_api_key = get_realtime_api_key(&pool, user.id).await;

// Setup mock server
let mock_server = MockServer::start().await;
Expand Down
Loading