Import/Export backend prep

internal/transfer/provider_test.go

@@ -0,0 +1,111 @@
+// 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 (
+	"fmt"
+	"testing"
+	"time"
+
+	a "github.com/stretchr/testify/assert"
+	r "github.com/stretchr/testify/require"
+)
+
+func getTestMsgBody(subject string) []byte {
+	return []byte(fmt.Sprintf(`Subject: %s
+From: Bridge Test <bridgetest@pm.test>
+To: Bridge Test <bridgetest@protonmail.com>
+Content-Type: multipart/mixed; boundary=c672b8d1ef56ed28ab87c3622c5114069bdd3ad7b8f9737498d0c01ecef0967a
+
+--c672b8d1ef56ed28ab87c3622c5114069bdd3ad7b8f9737498d0c01ecef0967a
+Content-Disposition: inline
+Content-Transfer-Encoding: 7bit
+Content-Type: text/plain; charset=utf-8
+
+hello
+
+--c672b8d1ef56ed28ab87c3622c5114069bdd3ad7b8f9737498d0c01ecef0967a--
+`, subject))
+}
+
+func testTransferTo(t *testing.T, rules transferRules, provider SourceProvider, expectedMessageIDs []string) {
+	progress := newProgress(log, nil)
+	drainProgressUpdateChannel(&progress)
+
+	ch := make(chan Message)
+	go func() {
+		provider.TransferTo(rules, &progress, ch)
+		close(ch)
+	}()
+
+	gotMessageIDs := []string{}
+	for msg := range ch {
+		gotMessageIDs = append(gotMessageIDs, msg.ID)
+	}
+	r.ElementsMatch(t, expectedMessageIDs, gotMessageIDs)
+
+	r.Empty(t, progress.GetFailedMessages())
+}
+
+func testTransferFrom(t *testing.T, rules transferRules, provider TargetProvider, messages []Message) {
+	progress := newProgress(log, nil)
+	drainProgressUpdateChannel(&progress)
+
+	ch := make(chan Message)
+	go func() {
+		for _, message := range messages {
+			progress.addMessage(message.ID, nil)
+			progress.messageExported(message.ID, []byte(""), nil)
+			ch <- message
+		}
+		close(ch)
+	}()
+
+	go func() {
+		provider.TransferFrom(rules, &progress, ch)
+		progress.finish()
+	}()
+
+	maxWait := time.Duration(len(messages)) * time.Second
+	a.Eventually(t, func() bool {
+		return progress.updateCh == nil
+	}, maxWait, 10*time.Millisecond, "Waiting for imported messages timed out")
+
+	r.Empty(t, progress.GetFailedMessages())
+}
+
+func testTransferFromTo(t *testing.T, rules transferRules, source SourceProvider, target TargetProvider, maxWait time.Duration) {
+	progress := newProgress(log, nil)
+	drainProgressUpdateChannel(&progress)
+
+	ch := make(chan Message)
+	go func() {
+		source.TransferTo(rules, &progress, ch)
+		close(ch)
+	}()
+	go func() {
+		target.TransferFrom(rules, &progress, ch)
+		progress.finish()
+	}()
+
+	a.Eventually(t, func() bool {
+		return progress.updateCh == nil
+	}, maxWait, 10*time.Millisecond, "Waiting for export and import timed out")
+
+	r.Empty(t, progress.GetFailedMessages())
+}