fix: put operation waker registration issue with custom priority select combinator

.github/workflows/ci.yml

@@ -43,6 +43,47 @@ jobs:
       - name: Test
         run: cargo test --workspace --no-default-features --features trace,websocket,redb
 
+  ubertest:
+    name: Ubertest
+    needs: test_all
+    # TODO: Re-enable when ubertest is stable - currently failing
+    if: false
+
+    runs-on: freenet-default-runner
+
+    env:
+      FREENET_LOG: error
+      CARGO_TARGET_DIR: ${{ github.workspace }}/target
+      UBERTEST_PEER_COUNT: 6  # Fewer peers for faster CI
+
+    steps:
+      - name: Cancel Previous Runs
+        uses: styfle/cancel-workflow-action@0.12.1
+        with:
+          access_token: ${{ github.token }}
+
+      - uses: actions/checkout@v5
+
+      - uses: dtolnay/rust-toolchain@stable
+        with:
+          toolchain: stable
+          targets: wasm32-unknown-unknown
+
+      - uses: Swatinem/rust-cache@v2
+        if: success() || steps.test.conclusion == 'failure'
+        with:
+          save-if: false
+
+      - name: Install riverctl
+        run: cargo install riverctl
+
+      - name: Build
+        run: cargo build --locked
+
+      - name: Run Ubertest
+        run: cargo test --test ubertest --no-default-features --features trace,websocket,redb
+        working-directory: crates/core
+
   clippy_check:
     name: Clippy
 

crates/core/Cargo.toml

@@ -41,6 +41,7 @@ itertools = "0.14"
 notify = "8"
 pav_regression = "0.5.2"
 parking_lot = "0.12"
+pin-project = "1"
 rand = { features = ["small_rng"], workspace = true }
 redb = { optional = true, version = "3" }
 serde = { features = ["derive", "rc"], workspace = true }
@@ -51,7 +52,7 @@ sqlx = { features = ["runtime-tokio-rustls", "sqlite"], optional = true, version
 stretto = { features = ["async", "sync"], version = "0.8" }
 tar = { version = "0.4" }
 thiserror = "2"
-tokio = { features = ["fs", "macros", "rt-multi-thread", "sync", "process"], version = "1" }
+tokio = { features = ["fs", "macros", "rt-multi-thread", "sync", "process", "tracing"], version = "1" }
 tokio-tungstenite = "0.27.0"
 tower-http = { features = ["fs", "trace"], version = "0.6" }
 ulid = { features = ["serde"], version = "1.1" }
@@ -74,6 +75,7 @@ opentelemetry_sdk = { optional = true, version = "0.31", features = ["rt-tokio"]
 
 # internal deps
 freenet-stdlib = { features = ["net"], workspace = true }
+console-subscriber = { version = "0.4.1", optional = true }
 
 [target.'cfg(windows)'.dependencies]
 winapi = { version = "0.3", features = ["sysinfoapi"] }
@@ -101,3 +103,4 @@ trace = ["tracing-subscriber"]
 trace-ot = ["opentelemetry-jaeger", "trace", "tracing-opentelemetry", "opentelemetry-otlp"]
 websocket = ["axum/ws"]
 testing = ["freenet-stdlib/testing"]
+console-subscriber = ["dep:console-subscriber"]

crates/core/src/bin/freenet.rs

@@ -57,8 +57,8 @@ fn main() -> anyhow::Result<()> {
         .enable_all()
         .build()
         .unwrap();
+    let config = ConfigArgs::parse();
     rt.block_on(async move {
-        let config = ConfigArgs::parse();
         if config.version {
             println!("Freenet version: {}", config.current_version());
             return Ok(());