54 lines
2.1 KiB
Go
54 lines
2.1 KiB
Go
|
package server
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"scm.bstein.dev/bstein/soteria/internal/config"
|
||
|
|
"scm.bstein.dev/bstein/soteria/internal/k8s"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestWaitForLonghornBackupCoversLookupErrorAndTimeout(t *testing.T) {
|
||
|
|
originalTimeout := longhornBackupWaitTimeout
|
||
|
|
originalPoll := longhornBackupPollInterval
|
||
|
|
longhornBackupWaitTimeout = 20 * time.Millisecond
|
||
|
|
longhornBackupPollInterval = 5 * time.Millisecond
|
||
|
|
defer func() {
|
||
|
|
longhornBackupWaitTimeout = originalTimeout
|
||
|
|
longhornBackupPollInterval = originalPoll
|
||
|
|
}()
|
||
|
|
|
||
|
|
srv := newBackupTestServer(
|
||
|
|
&config.Config{AuthRequired: false, BackupDriver: "longhorn"},
|
||
|
|
&backupTestKubeClient{restoreTestKubeClient: &restoreTestKubeClient{fakeKubeClient: &fakeKubeClient{
|
||
|
|
longhornBackupLookupErr: errors.New("lookup exploded"),
|
||
|
|
}}},
|
||
|
|
&backupTestLonghornClient{restoreTestLonghornClient: &restoreTestLonghornClient{fakeLonghornClient: &fakeLonghornClient{}}},
|
||
|
|
)
|
||
|
|
if _, result, err := srv.waitForLonghornBackup(context.Background(), "snap-a"); result != "backend_error" || err == nil || !strings.Contains(err.Error(), "lookup exploded") {
|
||
|
|
t.Fatalf("expected lookup backend error, result=%q err=%v", result, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
srv.client = &backupTestKubeClient{restoreTestKubeClient: &restoreTestKubeClient{fakeKubeClient: &fakeKubeClient{
|
||
|
|
longhornBackups: []k8s.LonghornBackupSummary{{
|
||
|
|
Name: "backup-running",
|
||
|
|
SnapshotName: "snap-a",
|
||
|
|
State: "InProgress",
|
||
|
|
Progress: 42,
|
||
|
|
}},
|
||
|
|
}}}
|
||
|
|
if _, result, err := srv.waitForLonghornBackup(context.Background(), "snap-a"); result != "in_progress" || err == nil || !strings.Contains(err.Error(), "backup-running") {
|
||
|
|
t.Fatalf("expected in-progress timeout with backup name, result=%q err=%v", result, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
srv.client = &backupTestKubeClient{restoreTestKubeClient: &restoreTestKubeClient{fakeKubeClient: &fakeKubeClient{
|
||
|
|
longhornBackups: []k8s.LonghornBackupSummary{},
|
||
|
|
}}}
|
||
|
|
if _, result, err := srv.waitForLonghornBackup(context.Background(), "snap-missing"); result != "in_progress" || err == nil || !strings.Contains(err.Error(), "did not appear") {
|
||
|
|
t.Fatalf("expected missing backup timeout, result=%q err=%v", result, err)
|
||
|
|
}
|
||
|
|
}
|