60
serializer serialization.Serializer,
61
logger log.Logger,
63
>
return &ExecutionMutableStateSuite{
64
>
Assertions: require.New(t),
65
>
ProtoAssertions: protorequire.New(t),
66
>
ShardManager: p.NewShardManager(
67
>
shardStore,
68
>
serializer,
69
>
),
70
>
ExecutionManager: p.NewExecutionManager(
71
>
executionStore,
72
>
serializer,
73
>
nil,
74
>
logger,
75
>
dynamicconfig.GetIntPropertyFn(4*1024*1024),
76
>
dynamicconfig.GetBoolPropertyFn(false),
77
>
),
78
>
HistoryBranchUtil: p.NewHistoryBranchUtil(serializer),
79
>
Logger: logger,
80
>
}
81
>
}
82
84
>
}
85
87
>
}
88
90
>
s.Assertions = require.New(s.T())
91
>
s.Ctx, s.Cancel = context.WithTimeout(context.Background(), 30*time.Second*debug.TimeoutMultiplier)
92
>
93
>
s.ShardID++
94
>
resp, err := s.ShardManager.GetOrCreateShard(s.Ctx, &p.GetOrCreateShardRequest{
95
>
ShardID: s.ShardID,
96
>
InitialShardInfo: &persistencespb.ShardInfo{
97
>
ShardId: s.ShardID,
98
>
RangeId: 1,
99
>
},
100
>
})
101
>
s.NoError(err)
102
>
previousRangeID := resp.ShardInfo.RangeId
103
>
resp.ShardInfo.RangeId++
104
>
err = s.ShardManager.UpdateShard(s.Ctx, &p.UpdateShardRequest{
105
>
ShardInfo: resp.ShardInfo,
106
>
PreviousRangeID: previousRangeID,
107
>
})
108
>
s.NoError(err)
109
>
s.RangeID = resp.ShardInfo.RangeId
110
>
111
>
s.NamespaceID = uuid.New().String()
112
>
s.WorkflowID = uuid.New().String()
113
>
s.RunID = uuid.New().String()
114
>
}
115
117
>
s.Cancel()
118
>
}
119
120
func (s *ExecutionMutableStateSuite) TestCreate_BrandNew() {