@@ -66,14 +66,15 @@ func newMockAdvancerServiceWithContextAndBatchSize(
6666 machineManager : machineManager ,
6767 repository : repo ,
6868 }
69- serviceArgs := & service.CreateInfo {
70- Name : "advancer" ,
71- Impl : s ,
72- Context : ctx ,
73- Cancel : cancelCtx ,
69+ serviceArgs := & service.TickServiceConfigs {
70+ ServiceConfigs : service.ServiceConfigs {
71+ Name : "advancer" ,
72+ Context : ctx ,
73+ Cancel : cancelCtx ,
74+ },
7475 EnableReschedule : true ,
7576 }
76- err := service .NewTickService (serviceArgs , & s .TickService )
77+ err := service .InitTickServiceTemplate (serviceArgs , & s .TickServiceTemplate , s , s )
7778 if err != nil {
7879 return nil , err
7980 }
@@ -1030,8 +1031,8 @@ func (s *AdvancerSuite) TestRemoveSnapshot() {
10301031
10311032 tmpDir := s .T ().TempDir ()
10321033 advancer := & Service {snapshotsDir : tmpDir }
1033- serviceArgs := & service.CreateInfo { Name : "advancer" , Impl : advancer }
1034- require .Nil (service .Create ( context . Background (), serviceArgs , & advancer .Service ))
1034+ serviceArgs := & service.TickServiceConfigs { ServiceConfigs : service. ServiceConfigs { Name : "advancer" } }
1035+ require .Nil (service .InitTickServiceTemplate ( serviceArgs , & advancer .TickServiceTemplate , advancer , advancer ))
10351036
10361037 // Create a snapshot directory
10371038 snapshotPath := filepath .Join (tmpDir , "myapp_epoch0_input0" )
@@ -1049,8 +1050,8 @@ func (s *AdvancerSuite) TestRemoveSnapshot() {
10491050
10501051 tmpDir := s .T ().TempDir ()
10511052 advancer := & Service {snapshotsDir : tmpDir }
1052- serviceArgs := & service.CreateInfo { Name : "advancer" , Impl : advancer }
1053- require .Nil (service .Create ( context . Background (), serviceArgs , & advancer .Service ))
1053+ serviceArgs := & service.TickServiceConfigs { ServiceConfigs : service. ServiceConfigs { Name : "advancer" } }
1054+ require .Nil (service .InitTickServiceTemplate ( serviceArgs , & advancer .TickServiceTemplate , advancer , advancer ))
10541055
10551056 snapshotPath := filepath .Join (tmpDir , "myapp_epoch0_input0" )
10561057 err := advancer .removeSnapshot (snapshotPath , "myapp" )
@@ -1062,8 +1063,8 @@ func (s *AdvancerSuite) TestRemoveSnapshot() {
10621063
10631064 tmpDir := s .T ().TempDir ()
10641065 advancer := & Service {snapshotsDir : tmpDir }
1065- serviceArgs := & service.CreateInfo { Name : "advancer" , Impl : advancer }
1066- require .Nil (service .Create ( context . Background (), serviceArgs , & advancer .Service ))
1066+ serviceArgs := & service.TickServiceConfigs { ServiceConfigs : service. ServiceConfigs { Name : "advancer" } }
1067+ require .Nil (service .InitTickServiceTemplate ( serviceArgs , & advancer .TickServiceTemplate , advancer , advancer ))
10671068
10681069 // Try to traverse outside snapshotsDir
10691070 maliciousPath := filepath .Join (tmpDir , ".." , "outside" , "myapp_evil" )
@@ -1077,8 +1078,8 @@ func (s *AdvancerSuite) TestRemoveSnapshot() {
10771078
10781079 tmpDir := s .T ().TempDir ()
10791080 advancer := & Service {snapshotsDir : tmpDir }
1080- serviceArgs := & service.CreateInfo { Name : "advancer" , Impl : advancer }
1081- require .Nil (service .Create ( context . Background (), serviceArgs , & advancer .Service ))
1081+ serviceArgs := & service.TickServiceConfigs { ServiceConfigs : service. ServiceConfigs { Name : "advancer" } }
1082+ require .Nil (service .InitTickServiceTemplate ( serviceArgs , & advancer .TickServiceTemplate , advancer , advancer ))
10821083
10831084 snapshotPath := filepath .Join (tmpDir , "otherapp_epoch0_input0" )
10841085 err := advancer .removeSnapshot (snapshotPath , "myapp" )
0 commit comments