Initialize shard correctly in ExecutionMutableStateSuite (#2336)

temporalio · Dec 30, 2021 · 633e456 · 633e456
1 parent 1e02664
commit 633e456
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/common/persistence/tests/execution_mutable_state.go b/common/persistence/tests/execution_mutable_state.go
@@ -100,6 +100,13 @@ func (s *ExecutionMutableStateSuite) SetupTest() {
 		},
 	})
 	s.NoError(err)
+	previousRangeID := resp.ShardInfo.RangeId
+	resp.ShardInfo.RangeId += 1
+	err = s.shardManager.UpdateShard(&p.UpdateShardRequest{
+		ShardInfo:       resp.ShardInfo,
+		PreviousRangeID: previousRangeID,
+	})
+	s.NoError(err)
 	s.rangeID = resp.ShardInfo.RangeId
 
 	s.namespaceID = uuid.New().String()