Add testing for websocket notifications.

Signed-off-by: David Venhoek <david@tweedegolf.com>

Author: davidv1992

Cargo.lock

@@ -29,8 +29,10 @@ serde_with = { version = "3.17.0", features = ["macros"] }
 reqwest = { version = "0.13.2", default-features = false, features = ["http2", "charset", "rustls", "json", "query", "form"] }
 tokio = { version = "1.50.0", features = ["full", "test-util"] }
 tokio-test = "0.4.5"
+futures = "0.3.32"
 axum = { version = "0.8.8", features = ["macros"] }
 axum-extra = { version = "0.12.5", features = ["query", "typed-header"] }
+tokio-tungstenite = "0.29.0"
 tower = { version = "0.5.3", features = ["util"] }
 
 tracing = "0.1.44"

Cargo.toml

@@ -49,12 +49,14 @@ mdns-sd = { workspace = true, optional = true }
 
 [dev-dependencies]
 tokio = { workspace = true, features = ["full", "test-util"] }
+futures.workspace = true
+tokio-tungstenite.workspace = true
 serial_test.workspace = true
 serde_html_form = "0.4.0"
 openleadr-client = { path = "../openleadr-client" }
 
 [features]
-default = ["postgres", "compression-br", "compression-deflate", "compression-gzip", "compression-zstd"]
+default = ["postgres", "compression-br", "compression-deflate", "compression-gzip", "compression-zstd", "experimental-websockets"]
 live-db-test = ["postgres", "internal-oauth"]
 postgres = ["sqlx/postgres", "dep:dotenvy", "dep:argon2"]
 internal-oauth = []

openleadr-vtn/Cargo.toml

@@ -115,7 +115,12 @@ pub mod test {
     use reqwest::Method;
     use serde::de::DeserializeOwned;
     use sqlx::PgPool;
-    use std::fmt::Display;
+    use std::{
+        fmt::Display,
+        future::IntoFuture,
+        net::{Ipv4Addr, SocketAddr},
+    };
+    use tokio::{net::TcpListener, task::JoinHandle};
     use tower::ServiceExt;
 
     pub(crate) struct ApiTest {
@@ -142,6 +147,18 @@ pub mod test {
             Self { router, token }
         }
 
+        #[cfg(feature = "experimental-websockets")]
+        pub(crate) async fn run(
+            &self,
+        ) -> (String, SocketAddr, JoinHandle<Result<(), std::io::Error>>) {
+            let listener = TcpListener::bind(SocketAddr::new(Ipv4Addr::LOCALHOST.into(), 0))
+                .await
+                .unwrap();
+            let listen_addr = listener.local_addr().unwrap();
+            let handle = tokio::spawn(axum::serve(listener, self.router.clone()).into_future());
+            (self.token.clone(), listen_addr, handle)
+        }
+
         pub(crate) async fn request<T: DeserializeOwned>(
             &self,
             method: Method,