1 files changed, 68 insertions, 0 deletions
diff --git a/src/boot/handlers/boot/test.rs b/src/boot/handlers/boot/test.rs
index a9891eb..7eb4e52 100644
--- a/src/boot/handlers/boot/test.rs
+++ b/src/boot/handlers/boot/test.rs
@@ -81,6 +81,74 @@ async fn includes_messages() {
 }
 
 #[tokio::test]
+async fn includes_vapid_key() {
+    let app = fixtures::scratch_app().await;
+
+    app.vapid()
+        .refresh_key(&fixtures::now())
+        .await
+        .expect("key rotation always succeeds");
+
+    let viewer = fixtures::identity::fictitious();
+    let response = super::handler(State(app), viewer)
+        .await
+        .expect("boot always succeeds");
+
+    response
+        .snapshot
+        .events
+        .into_iter()
+        .filter_map(fixtures::event::vapid)
+        .filter_map(fixtures::event::vapid::changed)
+        .exactly_one()
+        .expect("only one vapid key has been created");
+}
+
+#[tokio::test]
+async fn includes_only_latest_vapid_key() {
+    let app = fixtures::scratch_app().await;
+
+    app.vapid()
+        .refresh_key(&fixtures::ancient())
+        .await
+        .expect("key rotation always succeeds");
+
+    let viewer = fixtures::identity::fictitious();
+    let response = super::handler(State(app.clone()), viewer.clone())
+        .await
+        .expect("boot always succeeds");
+
+    let original_key = response
+        .snapshot
+        .events
+        .into_iter()
+        .filter_map(fixtures::event::vapid)
+        .filter_map(fixtures::event::vapid::changed)
+        .exactly_one()
+        .expect("only one vapid key has been created");
+
+    app.vapid()
+        .refresh_key(&fixtures::now())
+        .await
+        .expect("key rotation always succeeds");
+
+    let response = super::handler(State(app), viewer)
+        .await
+        .expect("boot always succeeds");
+
+    let rotated_key = response
+        .snapshot
+        .events
+        .into_iter()
+        .filter_map(fixtures::event::vapid)
+        .filter_map(fixtures::event::vapid::changed)
+        .exactly_one()
+        .expect("only one vapid key should be returned");
+
+    assert_ne!(original_key, rotated_key);
+}
+
+#[tokio::test]
 async fn includes_expired_messages() {
     let app = fixtures::scratch_app().await;
     let sender = fixtures::user::create(&app, &fixtures::ancient()).await;