@@ -459,18 +459,11 @@ func TestConvertBetweenOpenAPITaskAndTaskConfig(t *testing.T) {
459459 sourceSchema := task .TableMigrateRule [0 ].Source .Schema
460460 targetSchema := * task .TableMigrateRule [0 ].Target .Schema
461461 // only route schema
462- task .TableMigrateRule [0 ].Source = struct {
463- Schema string `json:"schema"`
464- SourceName string `json:"source_name"`
465- Table string `json:"table"`
466- }{
462+ task .TableMigrateRule [0 ].Source = openapi.TaskTableMigrateRuleSource {
467463 SourceName : source1Name ,
468464 Schema : sourceSchema ,
469465 }
470- task .TableMigrateRule [0 ].Target = & struct {
471- Schema * string `json:"schema,omitempty"`
472- Table * string `json:"table,omitempty"`
473- }{
466+ task .TableMigrateRule [0 ].Target = & openapi.TaskTableMigrateRuleTarget {
474467 Schema : & targetSchema ,
475468 }
476469 taskCfg , err = OpenAPITaskToTaskConfig (& task , sourceCfgMap )
@@ -504,38 +497,24 @@ func TestConvertBetweenOpenAPITaskAndTaskConfig(t *testing.T) {
504497 // only route table will meet error
505498 sourceTable := "tb"
506499 targetTable := "tb1"
507- task .TableMigrateRule [0 ].Source = struct {
508- Schema string `json:"schema"`
509- SourceName string `json:"source_name"`
510- Table string `json:"table"`
511- }{
500+ task .TableMigrateRule [0 ].Source = openapi.TaskTableMigrateRuleSource {
512501 SourceName : source1Name ,
513502 Schema : sourceSchema ,
514503 Table : sourceTable ,
515504 }
516- task .TableMigrateRule [0 ].Target = & struct {
517- Schema * string `json:"schema,omitempty"`
518- Table * string `json:"table,omitempty"`
519- }{
505+ task .TableMigrateRule [0 ].Target = & openapi.TaskTableMigrateRuleTarget {
520506 Table : & targetTable ,
521507 }
522508 _ , err = OpenAPITaskToTaskConfig (& task , sourceCfgMap )
523509 require .True (t , terror .ErrConfigGenTableRouter .Equal (err ))
524510
525511 // route both
526- task .TableMigrateRule [0 ].Source = struct {
527- Schema string `json:"schema"`
528- SourceName string `json:"source_name"`
529- Table string `json:"table"`
530- }{
512+ task .TableMigrateRule [0 ].Source = openapi.TaskTableMigrateRuleSource {
531513 SourceName : source1Name ,
532514 Schema : sourceSchema ,
533515 Table : sourceTable ,
534516 }
535- task .TableMigrateRule [0 ].Target = & struct {
536- Schema * string `json:"schema,omitempty"`
537- Table * string `json:"table,omitempty"`
538- }{
517+ task .TableMigrateRule [0 ].Target = & openapi.TaskTableMigrateRuleTarget {
539518 Schema : & targetSchema ,
540519 Table : & targetTable ,
541520 }
@@ -578,11 +557,7 @@ func TestConvertBetweenOpenAPITaskAndTaskConfig(t *testing.T) {
578557 require .EqualValues (t , taskAfterConvert , & task )
579558
580559 // no route and sync one schema
581- task .TableMigrateRule [0 ].Source = struct {
582- Schema string `json:"schema"`
583- SourceName string `json:"source_name"`
584- Table string `json:"table"`
585- }{
560+ task .TableMigrateRule [0 ].Source = openapi.TaskTableMigrateRuleSource {
586561 SourceName : source1Name ,
587562 Schema : sourceSchema ,
588563 Table : "" ,
0 commit comments