192
193
// Setup sets up the test base, must be called as part of SetupSuite
195
>
var err error
196
>
shardID := int32(10)
197
>
if clusterMetadataConfig == nil {
198
clusterMetadataConfig = cluster.NewTestClusterMetadataConfig(false, false)
199
}
201
>
s.PersistenceHealthSignals = persistence.NoopHealthSignalAggregator
202
>
}
203
205
>
206
>
s.DefaultTestCluster.SetupTestDatabase()
207
>
208
>
cfg := s.DefaultTestCluster.Config()
209
>
serializer := serialization.NewSerializer()
210
>
dataStoreFactory := client.DataStoreFactoryProvider(
211
>
client.ClusterName(clusterName),
212
>
resolver.NewNoopResolver(),
213
>
&cfg,
214
>
s.AbstractDataStoreFactory,
215
>
s.Logger,
216
>
metrics.NoopMetricsHandler,
217
>
s.TracerProvider,
218
>
serializer,
219
>
)
220
>
factory := client.NewFactory(
221
>
dataStoreFactory,
222
>
&cfg,
223
>
s.PersistenceRateLimiter,
224
>
quotas.NoopRequestRateLimiter,
225
>
quotas.NoopRequestRateLimiter,
226
>
serializer,
227
>
nil,
228
>
clusterName,
229
>
metrics.NoopMetricsHandler,
230
>
s.Logger,
231
>
s.PersistenceHealthSignals,
232
>
func() bool { return false },
233
func() bool { return false },
234
)
235
237
>
s.fatalOnError("NewTaskManager", err)
238
>
239
>
s.FairTaskMgr, err = factory.NewFairTaskManager()
240
>
// TODO: re-enable error check after FairTaskManager is implemented for sql
241
>
// s.fatalOnError("NewFairTaskManager", err)
242
>
_ = err
243
>
244
>
s.ClusterMetadataManager, err = factory.NewClusterMetadataManager()
245
>
s.fatalOnError("NewClusterMetadataManager", err)
246
>
247
>
s.ClusterMetadata = cluster.NewMetadataFromConfig(clusterMetadataConfig, s.ClusterMetadataManager, dynamicconfig.NewNoopCollection(), s.Logger)
248
>
s.SearchAttributesManager = searchattribute.NewManager(
249
>
clock.NewRealTimeSource(),
250
>
s.ClusterMetadataManager,
251
>
s.Logger,
252
>
dynamicconfig.GetBoolPropertyFn(true),
253
>
)
254
>
255
>
s.MetadataManager, err = factory.NewMetadataManager()
256
>
s.fatalOnError("NewMetadataManager", err)
257
>
258
>
s.ShardMgr, err = factory.NewShardManager()
259
>
s.fatalOnError("NewShardManager", err)
260
>
261
>
s.ExecutionManager, err = factory.NewExecutionManager()
262
>
s.fatalOnError("NewExecutionManager", err)
263
>
264
>
s.NexusEndpointManager, err = factory.NewNexusEndpointManager()
265
>
s.fatalOnError("NewNexusEndpointManager", err)
266
>
267
>
s.Factory = factory
268
>
269
>
s.ReadLevel = 0
270
>
s.ReplicationReadLevel = 0
271
>
s.ShardInfo = &persistencespb.ShardInfo{
272
>
ShardId: shardID,
273
>
RangeId: 0,
274
>
}
275
>
276
>
s.TaskIDGenerator = &TestTransferTaskIDGenerator{}
277
>
_, err = s.ShardMgr.GetOrCreateShard(context.Background(), &persistence.GetOrCreateShardRequest{
278
>
ShardID: shardID,
279
>
InitialShardInfo: s.ShardInfo,
280
>
})
281
>
s.fatalOnError("CreateShard", err)
282
>
283
>
queue, err := factory.NewNamespaceReplicationQueue()
284
>
s.fatalOnError("Create NamespaceReplicationQueue", err)
285
>
s.NamespaceReplicationQueue = queue
286
}
287
289
>
if err != nil {
290
s.Logger.Fatal(msg, tag.Error(err))
291
}