ms2k/remora
SHA256
Watch
0
Code Releases Activity

Centralize durable schema migration ownership

Browse source
This commit is contained in:
ms2k 2026-07-19 01:20:18 +09:00
commit 069f3813ae
1 changed files with 354 additions and 350 deletions
Download patch file Download diff file Expand all files Collapse all files

21
crates/remora-worker/tests/runtime.rs
View file

@ -16,13 +16,11 @@ use rcgen::{
BasicConstraints, Certificate, CertificateParams, ExtendedKeyUsagePurpose, IsCa, Issuer,
KeyPair, KeyUsagePurpose,
};
use remora_admission::{
AdmissionConfig, AdmissionDecision, AdmissionService, migrate as admission_migrate,
};
use remora_admission::{AdmissionConfig, AdmissionDecision, AdmissionService};
use remora_api::{WorkerApiState, authenticate_rustls_worker, configure_worker_api};
use remora_attempt::{
ActivationOutcome, AttemptConfig, AttemptService, CheckpointRecord, DurableDescriptor,
RecoveryConfig, StagingObject, migrate as attempt_migrate,
RecoveryConfig, StagingObject,
};
use remora_contracts::{
CheckpointDescriptorRefDto, HttpSourceSubmissionDto, HuggingFaceRepositoryTypeDto,
@ -34,19 +32,19 @@ use remora_contracts::{
use remora_coordination::{CoordinationStore, LeaseOwner, StoreConfig};
use remora_credentials::{
CreateCredential, CredentialRegistry, GrantBoundResolver, SecretMaterial, VaultClient,
VaultConfig, migrate as credentials_migrate,
VaultConfig,
};
use remora_domain::{BucketName, JobPhase, OpaqueId, StorageScope, TargetPrefix, TargetStoreId};
use remora_durable::{PoolConfig, connect, migrate as durable_migrate};
use remora_durable::{PoolConfig, connect};
use remora_identity::{DefaultTarget, Principal};
use remora_outbound::OutboundPolicyBuilder;
use remora_result::{ResultInventory, ResultMember, migrate as result_migrate};
use remora_result::{ResultInventory, ResultMember};
use remora_schema::migrate as migrate_schema;
use remora_stream::{CancelToken, CopyConfig, RetryTiming};
use remora_target::{CreateObjectRequest, TargetStore, TargetStoreConfig, attempt_manifest_key_in};
use remora_worker::{
AdapterExecutor, AttemptContext, AttemptExecution, AttemptExecutor, DescriptorStore,
ObjectSourceEndpoint, WorkerApiClient, WorkerConfig, WorkerRunOutcome, WorkerRuntime,
migrate as worker_migrate,
};
use reqwest::{Certificate as ClientCertificate, Client, Identity};
use rustls::{
@ -80,12 +78,7 @@ impl Harness {
)
.await
.unwrap();
durable_migrate(&pool).await.unwrap();
credentials_migrate(&pool).await.unwrap();
admission_migrate(&pool).await.unwrap();
attempt_migrate(&pool).await.unwrap();
result_migrate(&pool).await.unwrap();
worker_migrate(&pool).await.unwrap();
migrate_schema(&pool).await.unwrap();
sqlx::query("TRUNCATE TABLE jobs CASCADE")
.execute(&pool)
.await