test: stabilize distributed load verification

loadtest/im-pressure/agent_history_test.go

@@ -1,6 +1,9 @@
 package main
 
-import "testing"
+import (
+	"errors"
+	"testing"
+)
 
 func TestHistoryQueriesForMessagesGroupsRangesByTopic(t *testing.T) {
 	p2pClient := &WSClient{}
@@ -25,7 +28,7 @@ func TestHistoryQueriesForMessagesGroupsRangesByTopic(t *testing.T) {
 		switch query.topic {
 		case "p2p-a":
 			p2pQueries++
-			if query.client != p2pClient || query.limit != 200 {
+			if query.client != p2pClient || query.limit != 200 || len(query.expected) == 0 {
 				t.Fatalf("unexpected p2p query: %#v", query)
 			}
 		case "grp-a":
@@ -40,6 +43,14 @@ func TestHistoryQueriesForMessagesGroupsRangesByTopic(t *testing.T) {
 	}
 }
 
+func TestClassifyPublishErrorTreatsClosedConnectionAsUnknownSocketError(t *testing.T) {
+	run := &agentRun{}
+	run.classifyPublishError(errors.New("write tcp: use of closed network connection"))
+	if run.publishSocket.Load() != 1 || run.publishOther.Load() != 0 {
+		t.Fatalf("socket=%d other=%d", run.publishSocket.Load(), run.publishOther.Load())
+	}
+}
+
 func TestMergeMessageLedgersPrefersPeerObservationAndAck(t *testing.T) {
 	senderEcho := AcceptedMessage{ClientMid: "mid-1", SenderMachine: "a", PeerMachine: "a", Seq: 10}
 	peerDelivery := AcceptedMessage{ClientMid: "mid-1", SenderMachine: "a", PeerMachine: "b", PeerIndex: 7, Seq: 10}