test: add test for UpdateLastSent to peerset_test

pactus-project · Jul 3, 2023 · afcfd8a · afcfd8a
1 parent 50d65e9
commit afcfd8a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/sync/peerset/peer_set_test.go b/sync/peerset/peer_set_test.go
@@ -94,6 +94,14 @@ func TestPeerSet(t *testing.T) {
 		assert.Equal(t, peer1.Status, StatusCodeBanned)
 	})
 
+	t.Run("Testing UpdateLastSent", func(t *testing.T) {
+		now := time.Now()
+		peerSet.UpdateLastSent(pid1)
+
+		peer1 := peerSet.getPeer(pid1)
+		assert.GreaterOrEqual(t, peer1.LastSent, now)
+	})
+
 	t.Run("Testing UpdateLastReceived", func(t *testing.T) {
 		now := time.Now()
 		peerSet.UpdateLastReceived(pid1)