Import/Export backend prep

internal/transfer/progress_test.go

@@ -0,0 +1,120 @@
+// Copyright (c) 2020 Proton Technologies AG
+//
+// This file is part of ProtonMail Bridge.
+//
+// ProtonMail Bridge is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// ProtonMail Bridge is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with ProtonMail Bridge.  If not, see <https://www.gnu.org/licenses/>.
+
+package transfer
+
+import (
+	"testing"
+
+	"github.com/pkg/errors"
+	a "github.com/stretchr/testify/assert"
+	r "github.com/stretchr/testify/require"
+)
+
+func TestProgressUpdateCount(t *testing.T) {
+	progress := newProgress(log, nil)
+	drainProgressUpdateChannel(&progress)
+
+	progress.start()
+
+	progress.updateCount("inbox", 10)
+	progress.updateCount("archive", 20)
+	progress.updateCount("inbox", 12)
+	progress.updateCount("sent", 5)
+	progress.updateCount("foo", 4)
+	progress.updateCount("foo", 5)
+
+	progress.finish()
+
+	_, _, _, _, total := progress.GetCounts() //nolint[dogsled]
+	r.Equal(t, uint(42), total)
+}
+
+func TestProgressAddingMessages(t *testing.T) {
+	progress := newProgress(log, nil)
+	drainProgressUpdateChannel(&progress)
+
+	progress.start()
+
+	// msg1 has no problem.
+	progress.addMessage("msg1", nil)
+	progress.messageExported("msg1", []byte(""), nil)
+	progress.messageImported("msg1", "", nil)
+
+	// msg2 has an import problem.
+	progress.addMessage("msg2", nil)
+	progress.messageExported("msg2", []byte(""), nil)
+	progress.messageImported("msg2", "", errors.New("failed import"))
+
+	// msg3 has an export problem.
+	progress.addMessage("msg3", nil)
+	progress.messageExported("msg3", []byte(""), errors.New("failed export"))
+
+	// msg4 has an export problem and import is also called.
+	progress.addMessage("msg4", nil)
+	progress.messageExported("msg4", []byte(""), errors.New("failed export"))
+	progress.messageImported("msg4", "", nil)
+
+	progress.finish()
+
+	failed, imported, exported, added, _ := progress.GetCounts()
+	a.Equal(t, uint(4), added)
+	a.Equal(t, uint(4), exported)
+	a.Equal(t, uint(3), imported)
+	a.Equal(t, uint(3), failed)
+
+	errorsMap := map[string]string{}
+	for _, status := range progress.GetFailedMessages() {
+		errorsMap[status.SourceID] = status.GetErrorMessage()
+	}
+	a.Equal(t, map[string]string{
+		"msg2": "failed to import: failed import",
+		"msg3": "failed to export: failed export",
+		"msg4": "failed to export: failed export",
+	}, errorsMap)
+}
+
+func TestProgressFinish(t *testing.T) {
+	progress := newProgress(log, nil)
+	drainProgressUpdateChannel(&progress)
+
+	progress.start()
+	progress.finish()
+	r.Nil(t, progress.updateCh)
+
+	r.Panics(t, func() { progress.addMessage("msg", nil) })
+}
+
+func TestProgressFatalError(t *testing.T) {
+	progress := newProgress(log, nil)
+	drainProgressUpdateChannel(&progress)
+
+	progress.start()
+	progress.fatal(errors.New("fatal error"))
+	r.Nil(t, progress.updateCh)
+
+	r.NotPanics(t, func() { progress.addMessage("msg", nil) })
+}
+
+func drainProgressUpdateChannel(progress *Progress) {
+	// updateCh is not needed to drain under tests - timeout is implemented.
+	// But timeout takes time which would slow down tests.
+	go func() {
+		for range progress.updateCh {
+		}
+	}()
+}