fix: refresh last state even no proved state, again

nervosnetwork · Jan 8, 2024 · 030c0e4 · 030c0e4
1 parent 7837fa2
commit 030c0e4
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/src/protocols/light_client/peers.rs b/src/protocols/light_client/peers.rs
@@ -910,6 +910,7 @@ impl PeerState {
                 let new_state = Self::RequestFirstLastState { when_sent };
                 Ok(new_state)
             }
+            Self::OnlyHasLastState { .. } => Ok(self),
             Self::Ready {
                 last_state,
                 prove_state,

diff --git a/src/tests/protocols/light_client/mod.rs b/src/tests/protocols/light_client/mod.rs
@@ -304,7 +304,20 @@ async fn refresh_all_peers() {
 
         protocol.notify(nc.context(), REFRESH_PEERS_TOKEN).await;
 
-        // TODO FIXME A request `GetLastState` should be sent.
-        assert!(nc.sent_messages().borrow().is_empty());
+        let content = packed::GetLastState::new_builder()
+            .subscribe(true.pack())
+            .build();
+        let get_last_state_message = packed::LightClientMessage::new_builder()
+            .set(content)
+            .build()
+            .as_bytes();
+        assert_eq!(
+            nc.sent_messages().borrow().clone(),
+            vec![(
+                SupportProtocols::LightClient.protocol_id(),
+                peer_index,
+                get_last_state_message
+            )]
+        );
     }
 }