You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
115 lines
3.2 KiB
115 lines
3.2 KiB
// Copyright (c) 2021 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) []Message { |
|
progress := newProgress(log, nil) |
|
drainProgressUpdateChannel(&progress) |
|
|
|
ch := make(chan Message) |
|
go func() { |
|
provider.TransferTo(rules, &progress, ch) |
|
close(ch) |
|
}() |
|
|
|
msgs := []Message{} |
|
gotMessageIDs := []string{} |
|
for msg := range ch { |
|
msgs = append(msgs, msg) |
|
gotMessageIDs = append(gotMessageIDs, msg.ID) |
|
} |
|
r.ElementsMatch(t, expectedMessageIDs, gotMessageIDs) |
|
|
|
r.Empty(t, progress.GetFailedMessages()) |
|
|
|
return msgs |
|
} |
|
|
|
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, []string{}, []string{}) |
|
progress.messageExported(message.ID, []byte(""), nil) |
|
ch <- message |
|
} |
|
close(ch) |
|
}() |
|
|
|
go func() { |
|
provider.TransferFrom(rules, &progress, ch) |
|
progress.finish() |
|
}() |
|
|
|
maxWait := time.Duration(len(messages)) * 2 * 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()) |
|
}
|
|
|