fix flaky test on mysqlshell backup engine

Signed-off-by: Renan Rangel <[email protected]>

go/vt/mysqlctl/mysqlshellbackupengine.go

@@ -56,8 +56,6 @@ var (
 	// disable redo logging and double write buffer
 	mysqlShellSpeedUpRestore = false
 
-	mysqlShellBackupBinaryName = "mysqlsh"
-
 	// use when checking if we need to create the directory on the local filesystem or not.
 	knownObjectStoreParams = []string{"s3BucketName", "osBucketName", "azureContainerName"}
 
@@ -87,7 +85,9 @@ type MySQLShellBackupManifest struct {
 }
 
 func init() {
-	BackupRestoreEngineMap[mysqlShellBackupEngineName] = &MySQLShellBackupEngine{}
+	BackupRestoreEngineMap[mysqlShellBackupEngineName] = &MySQLShellBackupEngine{
+		binaryName: "mysqlsh",
+	}
 
 	for _, cmd := range []string{"vtcombo", "vttablet", "vtbackup", "vttestserver", "vtctldclient"} {
 		servenv.OnParseFor(cmd, registerMysqlShellBackupEngineFlags)
@@ -106,6 +106,7 @@ func registerMysqlShellBackupEngineFlags(fs *pflag.FlagSet) {
 // MySQLShellBackupEngine encapsulates the logic to implement the restoration
 // of a mysql-shell based backup.
 type MySQLShellBackupEngine struct {
+	binaryName string
 }
 
 const (
@@ -164,7 +165,7 @@ func (be *MySQLShellBackupEngine) ExecuteBackup(ctx context.Context, params Back
 		return BackupUnusable, vterrors.Wrap(err, "failed to fetch position")
 	}
 
-	cmd := exec.CommandContext(ctx, mysqlShellBackupBinaryName, args...)
+	cmd := exec.CommandContext(ctx, be.binaryName, args...)
 
 	params.Logger.Infof("running %s", cmd.String())
 
@@ -371,7 +372,7 @@ func (be *MySQLShellBackupEngine) ExecuteRestore(ctx context.Context, params Res
 	if err := cmd.Wait(); err != nil {
 		return nil, vterrors.Wrap(err, mysqlShellBackupEngineName+" failed")
 	}
-	params.Logger.Infof("%s completed successfully", mysqlShellBackupBinaryName)
+	params.Logger.Infof("%s completed successfully", be.binaryName)
 
 	// disable local_infile now that the restore is done.
 	err = params.Mysqld.ExecuteSuperQuery(ctx, "SET GLOBAL LOCAL_INFILE=0")

go/vt/mysqlctl/mysqlshellbackupengine_test.go

@@ -325,13 +325,10 @@ func generateTestFile(t *testing.T, name, contents string) {
 // during ExecuteBackup(), even if the backup didn't succeed.
 func TestMySQLShellBackupEngine_ExecuteBackup_ReleaseLock(t *testing.T) {
 	originalLocation := mysqlShellBackupLocation
-	originalBinary := mysqlShellBackupBinaryName
 	mysqlShellBackupLocation = "logical"
-	mysqlShellBackupBinaryName = path.Join(t.TempDir(), "test.sh")
 
-	defer func() { // restore the original values.
+	defer func() { // restore the original value.
 		mysqlShellBackupLocation = originalLocation
-		mysqlShellBackupBinaryName = originalBinary
 	}()
 
 	logger := logutil.NewMemoryLogger()
@@ -340,7 +337,6 @@ func TestMySQLShellBackupEngine_ExecuteBackup_ReleaseLock(t *testing.T) {
 	mysql := NewFakeMysqlDaemon(fakedb)
 	defer mysql.Close()
 
-	be := &MySQLShellBackupEngine{}
 	params := BackupParams{
 		TabletAlias: "test",
 		Logger:      logger,
@@ -351,6 +347,7 @@ func TestMySQLShellBackupEngine_ExecuteBackup_ReleaseLock(t *testing.T) {
 	}
 
 	t.Run("lock released if we see the mysqlsh lock being acquired", func(t *testing.T) {
+		be := &MySQLShellBackupEngine{binaryName: path.Join(t.TempDir(), "mysqlsh.sh")}
 		logger.Clear()
 		manifestBuffer := ioutil.NewBytesBufferWriter()
 		bs.StartBackupReturn.BackupHandle = &FakeBackupHandle{
@@ -359,7 +356,7 @@ func TestMySQLShellBackupEngine_ExecuteBackup_ReleaseLock(t *testing.T) {
 		}
 
 		// this simulates mysql shell completing without any issues.
-		generateTestFile(t, mysqlShellBackupBinaryName, fmt.Sprintf("#!/bin/bash\n>&2 echo %s", mysqlShellLockMessage))
+		generateTestFile(t, be.binaryName, fmt.Sprintf("#!/bin/bash\n>&2 echo %s", mysqlShellLockMessage))
 
 		bh, err := bs.StartBackup(context.Background(), t.TempDir(), t.Name())
 		require.NoError(t, err)
@@ -381,6 +378,7 @@ func TestMySQLShellBackupEngine_ExecuteBackup_ReleaseLock(t *testing.T) {
 	})
 
 	t.Run("lock released if when we don't see mysqlsh released it", func(t *testing.T) {
+		be := &MySQLShellBackupEngine{binaryName: path.Join(t.TempDir(), "mysqlsh.sh")}
 		mysql.GlobalReadLock = false // clear lock status.
 		logger.Clear()
 		manifestBuffer := ioutil.NewBytesBufferWriter()
@@ -390,7 +388,7 @@ func TestMySQLShellBackupEngine_ExecuteBackup_ReleaseLock(t *testing.T) {
 		}
 
 		// this simulates mysqlshell completing, but we don't see the message that is released its lock.
-		generateTestFile(t, mysqlShellBackupBinaryName, "#!/bin/bash\nexit 0")
+		generateTestFile(t, be.binaryName, "#!/bin/bash\nexit 0")
 
 		bh, err := bs.StartBackup(context.Background(), t.TempDir(), t.Name())
 		require.NoError(t, err)
@@ -407,6 +405,7 @@ func TestMySQLShellBackupEngine_ExecuteBackup_ReleaseLock(t *testing.T) {
 	})
 
 	t.Run("lock released when backup fails", func(t *testing.T) {
+		be := &MySQLShellBackupEngine{binaryName: path.Join(t.TempDir(), "mysqlsh.sh")}
 		mysql.GlobalReadLock = false // clear lock status.
 		logger.Clear()
 		manifestBuffer := ioutil.NewBytesBufferWriter()
@@ -416,7 +415,7 @@ func TestMySQLShellBackupEngine_ExecuteBackup_ReleaseLock(t *testing.T) {
 		}
 
 		// this simulates the backup process failing.
-		generateTestFile(t, mysqlShellBackupBinaryName, "#!/bin/bash\nexit 1")
+		generateTestFile(t, be.binaryName, "#!/bin/bash\nexit 1")
 
 		bh, err := bs.StartBackup(context.Background(), t.TempDir(), t.Name())
 		require.NoError(t, err)