hcq/influx-cli
hcq/influx-cli
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: remove racy logic from backup test (#117)

Browse Source
This commit is contained in:
Daniel Moran 2021-06-11 10:01:51 -04:00 committed by GitHub
parent 943e8fba31
commit a1f8e0edf1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
clients/backup/backup_internal_test.go
View File

@ -8,7 +8,6 @@ import (
"fmt"
"io"
"io/ioutil"
"log"
"mime/multipart"
"net/http"
"os"
@ -151,13 +150,8 @@ func TestBackup_DownloadMetadata(t *testing.T) {
return &res, nil
})
stdio := mock.NewMockStdIO(ctrl)
writtenBytes := bytes.Buffer{}
stdio.EXPECT().Write(gomock.Any()).DoAndReturn(writtenBytes.Write).AnyTimes()
log.SetOutput(stdio)
cli := Client{
CLI: clients.CLI{StdIO: stdio},
CLI: clients.CLI{},
BackupApi: backupApi,
baseName: "test",
}
@ -268,13 +262,8 @@ func TestBackup_DownloadShardData(t *testing.T) {
return &res, nil
})
stdio := mock.NewMockStdIO(ctrl)
writtenBytes := bytes.Buffer{}
stdio.EXPECT().Write(gomock.Any()).DoAndReturn(writtenBytes.Write).AnyTimes()
log.SetOutput(stdio)
cli := Client{
CLI: clients.CLI{StdIO: stdio},
CLI: clients.CLI{},
BackupApi: backupApi,
baseName: "test",
}
@ -318,13 +307,8 @@ func TestBackup_DownloadShardData(t *testing.T) {
backupApi.EXPECT().GetBackupShardId(gomock.Any(), gomock.Eq(req.GetShardID())).Return(req)
backupApi.EXPECT().GetBackupShardIdExecute(gomock.Any()).Return(nil, &notFoundErr{})
stdio := mock.NewMockStdIO(ctrl)
writtenBytes := bytes.Buffer{}
stdio.EXPECT().Write(gomock.Any()).DoAndReturn(writtenBytes.Write).AnyTimes()
log.SetOutput(stdio)
cli := Client{
CLI: clients.CLI{StdIO: stdio},
CLI: clients.CLI{},
BackupApi: backupApi,
baseName: "test",
}
@ -340,7 +324,6 @@ func TestBackup_DownloadShardData(t *testing.T) {
metadata, err := cli.downloadShardData(context.Background(), &params, req.GetShardID())
require.NoError(t, err)
require.Nil(t, metadata)
require.Contains(t, writtenBytes.String(), fmt.Sprintf("WARN: Shard %d removed during backup", req.GetShardID()))
})
}