update specs

gojek · Dec 4, 2023 · d1a422a · d1a422a
1 parent bd26961
commit d1a422a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/client_publish_test.go b/client_publish_test.go
@@ -5,6 +5,7 @@ import (
 	"context"
 	"encoding/json"
 	"errors"
+	"fmt"
 	"math"
 	"sync"
 	"testing"
@@ -260,7 +261,7 @@ func (s *ClientPublishSuite) TestPublishWithMultiConnectionMode() {
 		tba := testBrokerAddress
 		tba.Port = uint16(1883 + i)
 
-		clients[tba.String()] = mc
+		clients[fmt.Sprintf("%s-%d", tba.String(), ii)] = mc
 
 		mcks = append(mcks, mc, mt)
 	}

diff --git a/client_resolver_test.go b/client_resolver_test.go
@@ -663,7 +663,7 @@ func TestClient_AddressUpdates(t *testing.T) {
 
 		go func() { ch <- []TCPAddress{testBrokerAddress, testBrokerAddress2} }()
 
-		wantAddrs := []string{testBrokerAddress.String(), testBrokerAddress2.String()}
+		wantAddrs := []string{testBrokerAddress.String() + "-0", testBrokerAddress2.String() + "-1"}
 		assert.Eventually(t, func() bool {
 			c.clientMu.RLock()
 			defer c.clientMu.RUnlock()
@@ -767,7 +767,7 @@ func TestClient_AddressUpdates(t *testing.T) {
 
 		go func() { ch <- []TCPAddress{testBrokerAddress, testBrokerAddress2} }()
 
-		wantAddrs := []string{testBrokerAddress.String(), testBrokerAddress2.String()}
+		wantAddrs := []string{testBrokerAddress.String() + "-0", testBrokerAddress2.String() + "-1"}
 		assert.Eventually(t, func() bool {
 			c.clientMu.RLock()
 			defer c.clientMu.RUnlock()