@@ -31,6 +31,8 @@ import (
3131 "testing"
3232 "time"
3333
34+ "go.temporal.io/server/common/convert"
35+
3436 "go.temporal.io/server/api/persistenceblobs/v1"
3537
3638 "github.com/golang/mock/gomock"
@@ -117,7 +119,7 @@ func (s *shardControllerSuite) TestAcquireShardSuccess() {
117119 if hostID == 0 {
118120 myShards = append (myShards , int (shardID ))
119121 s .mockHistoryEngine .EXPECT ().Start ().Return ().Times (1 )
120- s .mockServiceResolver .EXPECT ().Lookup (string (shardID )).Return (s .hostInfo , nil ).Times (2 )
122+ s .mockServiceResolver .EXPECT ().Lookup (convert . Int32ToString (shardID )).Return (s .hostInfo , nil ).Times (2 )
121123 s .mockEngineFactory .On ("CreateEngine" , mock .Anything ).Return (s .mockHistoryEngine ).Once ()
122124 s .mockShardManager .On ("GetShard" , & persistence.GetShardRequest {ShardID : shardID }).Return (
123125 & persistence.GetShardResponse {
@@ -162,7 +164,7 @@ func (s *shardControllerSuite) TestAcquireShardSuccess() {
162164 }).Return (nil ).Once ()
163165 } else {
164166 ownerHost := fmt .Sprintf ("test-acquire-shard-host-%v" , hostID )
165- s .mockServiceResolver .EXPECT ().Lookup (string (shardID )).Return (membership .NewHostInfo (ownerHost , nil ), nil ).Times (1 )
167+ s .mockServiceResolver .EXPECT ().Lookup (convert . Int32ToString (shardID )).Return (membership .NewHostInfo (ownerHost , nil ), nil ).Times (1 )
166168 }
167169 }
168170
@@ -197,7 +199,7 @@ func (s *shardControllerSuite) TestAcquireShardsConcurrently() {
197199 if hostID == 0 {
198200 myShards = append (myShards , int (shardID ))
199201 s .mockHistoryEngine .EXPECT ().Start ().Return ().Times (1 )
200- s .mockServiceResolver .EXPECT ().Lookup (string (shardID )).Return (s .hostInfo , nil ).Times (2 )
202+ s .mockServiceResolver .EXPECT ().Lookup (convert . Int32ToString (shardID )).Return (s .hostInfo , nil ).Times (2 )
201203 s .mockEngineFactory .On ("CreateEngine" , mock .Anything ).Return (s .mockHistoryEngine ).Once ()
202204 s .mockShardManager .On ("GetShard" , & persistence.GetShardRequest {ShardID : shardID }).Return (
203205 & persistence.GetShardResponse {
@@ -242,7 +244,7 @@ func (s *shardControllerSuite) TestAcquireShardsConcurrently() {
242244 }).Return (nil ).Once ()
243245 } else {
244246 ownerHost := fmt .Sprintf ("test-acquire-shard-host-%v" , hostID )
245- s .mockServiceResolver .EXPECT ().Lookup (string (shardID )).Return (membership .NewHostInfo (ownerHost , nil ), nil ).Times (1 )
247+ s .mockServiceResolver .EXPECT ().Lookup (convert . Int32ToString (shardID )).Return (membership .NewHostInfo (ownerHost , nil ), nil ).Times (1 )
246248 }
247249 }
248250
@@ -262,12 +264,12 @@ func (s *shardControllerSuite) TestAcquireShardLookupFailure() {
262264 numShards := 2
263265 s .config .NumberOfShards = numShards
264266 for shardID := 1 ; shardID <= numShards ; shardID ++ {
265- s .mockServiceResolver .EXPECT ().Lookup (string (shardID )).Return (nil , errors .New ("ring failure" )).Times (1 )
267+ s .mockServiceResolver .EXPECT ().Lookup (convert . IntToString (shardID )).Return (nil , errors .New ("ring failure" )).Times (1 )
266268 }
267269
268270 s .shardController .acquireShards ()
269271 for shardID := 1 ; shardID <= numShards ; shardID ++ {
270- s .mockServiceResolver .EXPECT ().Lookup (string (shardID )).Return (nil , errors .New ("ring failure" )).Times (1 )
272+ s .mockServiceResolver .EXPECT ().Lookup (convert . IntToString (shardID )).Return (nil , errors .New ("ring failure" )).Times (1 )
271273 s .Nil (s .shardController .getEngineForShard (shardID ))
272274 }
273275}
@@ -284,7 +286,7 @@ func (s *shardControllerSuite) TestAcquireShardRenewSuccess() {
284286
285287 for shardID := int32 (1 ); shardID <= int32 (numShards ); shardID ++ {
286288 s .mockHistoryEngine .EXPECT ().Start ().Return ().Times (1 )
287- s .mockServiceResolver .EXPECT ().Lookup (string (shardID )).Return (s .hostInfo , nil ).Times (2 )
289+ s .mockServiceResolver .EXPECT ().Lookup (convert . Int32ToString (shardID )).Return (s .hostInfo , nil ).Times (2 )
288290 s .mockEngineFactory .On ("CreateEngine" , mock .Anything ).Return (s .mockHistoryEngine ).Once ()
289291 s .mockShardManager .On ("GetShard" , & persistence.GetShardRequest {ShardID : shardID }).Return (
290292 & persistence.GetShardResponse {
@@ -335,7 +337,7 @@ func (s *shardControllerSuite) TestAcquireShardRenewSuccess() {
335337 s .shardController .acquireShards ()
336338
337339 for shardID := 1 ; shardID <= numShards ; shardID ++ {
338- s .mockServiceResolver .EXPECT ().Lookup (string (shardID )).Return (s .hostInfo , nil ).Times (1 )
340+ s .mockServiceResolver .EXPECT ().Lookup (convert . IntToString (shardID )).Return (s .hostInfo , nil ).Times (1 )
339341 }
340342 s .shardController .acquireShards ()
341343
@@ -356,7 +358,7 @@ func (s *shardControllerSuite) TestAcquireShardRenewLookupFailed() {
356358
357359 for shardID := int32 (1 ); shardID <= int32 (numShards ); shardID ++ {
358360 s .mockHistoryEngine .EXPECT ().Start ().Return ().Times (1 )
359- s .mockServiceResolver .EXPECT ().Lookup (string (shardID )).Return (s .hostInfo , nil ).Times (2 )
361+ s .mockServiceResolver .EXPECT ().Lookup (convert . Int32ToString (shardID )).Return (s .hostInfo , nil ).Times (2 )
360362 s .mockEngineFactory .On ("CreateEngine" , mock .Anything ).Return (s .mockHistoryEngine ).Once ()
361363 s .mockShardManager .On ("GetShard" , & persistence.GetShardRequest {ShardID : shardID }).Return (
362364 & persistence.GetShardResponse {
@@ -407,7 +409,7 @@ func (s *shardControllerSuite) TestAcquireShardRenewLookupFailed() {
407409 s .shardController .acquireShards ()
408410
409411 for shardID := 1 ; shardID <= numShards ; shardID ++ {
410- s .mockServiceResolver .EXPECT ().Lookup (string (shardID )).Return (nil , errors .New ("ring failure" )).Times (1 )
412+ s .mockServiceResolver .EXPECT ().Lookup (convert . IntToString (shardID )).Return (nil , errors .New ("ring failure" )).Times (1 )
411413 }
412414 s .shardController .acquireShards ()
413415
@@ -454,7 +456,7 @@ func (s *shardControllerSuite) TestHistoryEngineClosed() {
454456 for shardID := 1 ; shardID <= 2 ; shardID ++ {
455457 mockEngine := historyEngines [shardID ]
456458 mockEngine .EXPECT ().Stop ().Return ().Times (1 )
457- s .mockServiceResolver .EXPECT ().Lookup (string (shardID )).Return (differentHostInfo , nil ).AnyTimes ()
459+ s .mockServiceResolver .EXPECT ().Lookup (convert . IntToString (shardID )).Return (differentHostInfo , nil ).AnyTimes ()
458460 s .shardController .shardClosedCallback (shardID , nil )
459461 }
460462
@@ -499,7 +501,7 @@ func (s *shardControllerSuite) TestHistoryEngineClosed() {
499501 for shardID := 3 ; shardID <= numShards ; shardID ++ {
500502 mockEngine := historyEngines [shardID ]
501503 mockEngine .EXPECT ().Stop ().Return ().Times (1 )
502- s .mockServiceResolver .EXPECT ().Lookup (string (shardID )).Return (s .hostInfo , nil ).AnyTimes ()
504+ s .mockServiceResolver .EXPECT ().Lookup (convert . IntToString (shardID )).Return (s .hostInfo , nil ).AnyTimes ()
503505 }
504506 s .shardController .Stop ()
505507}
@@ -546,7 +548,7 @@ func (s *shardControllerSuite) TestShardControllerClosed() {
546548 for shardID := 1 ; shardID <= numShards ; shardID ++ {
547549 mockEngine := historyEngines [shardID ]
548550 mockEngine .EXPECT ().Stop ().Times (1 )
549- s .mockServiceResolver .EXPECT ().Lookup (string (shardID )).Return (s .hostInfo , nil ).AnyTimes ()
551+ s .mockServiceResolver .EXPECT ().Lookup (convert . IntToString (shardID )).Return (s .hostInfo , nil ).AnyTimes ()
550552 }
551553 s .shardController .Stop ()
552554 workerWG .Wait ()
@@ -563,7 +565,7 @@ func (s *shardControllerSuite) setupMocksForAcquireShard(shardID int, mockEngine
563565
564566 // s.mockResource.ExecutionMgr.On("Close").Return()
565567 mockEngine .EXPECT ().Start ().Times (1 )
566- s .mockServiceResolver .EXPECT ().Lookup (string (shardID )).Return (s .hostInfo , nil ).Times (2 )
568+ s .mockServiceResolver .EXPECT ().Lookup (convert . IntToString (shardID )).Return (s .hostInfo , nil ).Times (2 )
567569 s .mockEngineFactory .On ("CreateEngine" , mock .Anything ).Return (mockEngine ).Once ()
568570 s .mockShardManager .On ("GetShard" , & persistence.GetShardRequest {ShardID : int32 (shardID )}).Return (
569571 & persistence.GetShardResponse {
0 commit comments