check_task: ping db when apply dbconfig (pingcap#769)

GMHDBJD · Jul 2, 2020 · 3b752b8 · 3b752b8
1 parent 0c10af0
commit 3b752b8
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/pkg/conn/basedb.go b/pkg/conn/basedb.go
@@ -69,6 +69,10 @@ func (d *defaultDBProvider) Apply(config config.DBConfig) (*BaseDB, error) {
 		return nil, terror.DBErrorAdapt(err, terror.ErrDBDriverError)
 	}
 
+	if err = db.Ping(); err != nil {
+		return nil, terror.DBErrorAdapt(err, terror.ErrDBDriverError)
+	}
+
 	db.SetMaxIdleConns(maxIdleConns)
 
 	return NewBaseDB(db), nil

diff --git a/tests/dmctl_basic/check_list/check_task.sh b/tests/dmctl_basic/check_list/check_task.sh
@@ -34,3 +34,10 @@ function check_task_not_pass() {
         "check-task $task_conf" \
         "\"result\": false" 1
 }
+
+function check_task_error_database_config() {
+    task_conf=$1
+    run_dm_ctl $WORK_DIR "127.0.0.1:$MASTER_PORT" \
+        "check-task $task_conf" \
+        "database driver error" 1
+}
diff --git a/tests/dmctl_basic/run.sh b/tests/dmctl_basic/run.sh
@@ -114,6 +114,10 @@ function run() {
     check_task_pass $TASK_CONF
     check_task_not_pass $cur/conf/dm-task2.yaml
 
+    cp $cur/conf/dm-task.yaml $WORK_DIR/dm-task-error-database-config.yaml
+    sed -i "s/password: \"\"/password: \"wrond password\"/g" $WORK_DIR/dm-task-error-database-config.yaml
+    check_task_error_database_config $WORK_DIR/dm-task-error-database-config.yaml
+
     dmctl_start_task
     check_sync_diff $WORK_DIR $cur/conf/diff_config.toml
     update_task_not_paused $TASK_CONF