From b5bc76cdc77931d616909ad6531f07192682e200 Mon Sep 17 00:00:00 2001
From: Michael Eischer <michael.eischer@fau.de>
Date: Fri, 1 Nov 2024 15:16:24 +0100
Subject: [PATCH] test retry on repo opening

---
 cmd/restic/integration_test.go | 44 ++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/cmd/restic/integration_test.go b/cmd/restic/integration_test.go
index df95031dc..cb4ccba41 100644
--- a/cmd/restic/integration_test.go
+++ b/cmd/restic/integration_test.go
@@ -177,3 +177,47 @@ func TestFindListOnce(t *testing.T) {
 	// the snapshots can only be listed once, if both lists match then the there has been only a single List() call
 	rtest.Equals(t, thirdSnapshot, snapshotIDs)
 }
+
+type failConfigOnceBackend struct {
+	backend.Backend
+	failedOnce bool
+}
+
+func (be *failConfigOnceBackend) Load(ctx context.Context, h backend.Handle,
+	length int, offset int64, fn func(rd io.Reader) error) error {
+
+	if !be.failedOnce && h.Type == restic.ConfigFile {
+		be.failedOnce = true
+		return fmt.Errorf("oops")
+	}
+	return be.Backend.Load(ctx, h, length, offset, fn)
+}
+
+func (be *failConfigOnceBackend) Stat(ctx context.Context, h backend.Handle) (backend.FileInfo, error) {
+	if !be.failedOnce && h.Type == restic.ConfigFile {
+		be.failedOnce = true
+		return backend.FileInfo{}, fmt.Errorf("oops")
+	}
+	return be.Backend.Stat(ctx, h)
+}
+
+func TestBackendRetryConfig(t *testing.T) {
+	env, cleanup := withTestEnvironment(t)
+	defer cleanup()
+
+	var wrappedBackend *failConfigOnceBackend
+	// cause config loading to fail once
+	env.gopts.backendInnerTestHook = func(r backend.Backend) (backend.Backend, error) {
+		wrappedBackend = &failConfigOnceBackend{Backend: r}
+		return wrappedBackend, nil
+	}
+
+	testSetupBackupData(t, env)
+	rtest.Assert(t, wrappedBackend != nil, "backend not wrapped on init")
+	rtest.Assert(t, wrappedBackend != nil && wrappedBackend.failedOnce, "config loading was not retried on init")
+	wrappedBackend = nil
+
+	testRunBackup(t, "", []string{filepath.Join(env.testdata, "0", "0", "9")}, BackupOptions{}, env.gopts)
+	rtest.Assert(t, wrappedBackend != nil, "backend not wrapped on backup")
+	rtest.Assert(t, wrappedBackend != nil && wrappedBackend.failedOnce, "config loading was not retried on init")
+}