forked from Silverfish/proton-bridge
116 lines
4.0 KiB
Go
116 lines
4.0 KiB
Go
// Copyright (c) 2022 Proton AG
|
|
//
|
|
// This file is part of Proton Mail Bridge.
|
|
//
|
|
// Proton Mail 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.
|
|
//
|
|
// Proton Mail 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 Proton Mail Bridge. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package bridge_test
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"fmt"
|
|
"net"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/ProtonMail/go-proton-api"
|
|
"github.com/ProtonMail/go-proton-api/server"
|
|
"github.com/ProtonMail/proton-bridge/v3/internal/bridge"
|
|
"github.com/ProtonMail/proton-bridge/v3/internal/constants"
|
|
"github.com/emersion/go-imap"
|
|
"github.com/emersion/go-imap/client"
|
|
"github.com/emersion/go-sasl"
|
|
"github.com/emersion/go-smtp"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestBridge_Send(t *testing.T) {
|
|
withEnv(t, func(ctx context.Context, s *server.Server, netCtl *proton.NetCtl, locator bridge.Locator, storeKey []byte) {
|
|
_, _, err := s.CreateUser("recipient", password)
|
|
require.NoError(t, err)
|
|
|
|
withBridge(ctx, t, s.GetHostURL(), netCtl, locator, storeKey, func(bridge *bridge.Bridge, mocks *bridge.Mocks) {
|
|
senderUserID, err := bridge.LoginFull(ctx, username, password, nil, nil)
|
|
require.NoError(t, err)
|
|
|
|
recipientUserID, err := bridge.LoginFull(ctx, "recipient", password, nil, nil)
|
|
require.NoError(t, err)
|
|
|
|
senderInfo, err := bridge.GetUserInfo(senderUserID)
|
|
require.NoError(t, err)
|
|
|
|
recipientInfo, err := bridge.GetUserInfo(recipientUserID)
|
|
require.NoError(t, err)
|
|
|
|
for i := 0; i < 10; i++ {
|
|
// Dial the server.
|
|
client, err := smtp.Dial(net.JoinHostPort(constants.Host, fmt.Sprint(bridge.GetSMTPPort())))
|
|
require.NoError(t, err)
|
|
defer client.Close() //nolint:errcheck
|
|
|
|
// Upgrade to TLS.
|
|
require.NoError(t, client.StartTLS(&tls.Config{InsecureSkipVerify: true}))
|
|
|
|
if i%2 == 0 {
|
|
// Authorize with SASL PLAIN.
|
|
require.NoError(t, client.Auth(sasl.NewPlainClient(
|
|
senderInfo.Addresses[0],
|
|
senderInfo.Addresses[0],
|
|
string(senderInfo.BridgePass)),
|
|
))
|
|
} else {
|
|
// Authorize with SASL LOGIN.
|
|
require.NoError(t, client.Auth(sasl.NewLoginClient(
|
|
senderInfo.Addresses[0],
|
|
string(senderInfo.BridgePass)),
|
|
))
|
|
}
|
|
|
|
// Send the message.
|
|
require.NoError(t, client.SendMail(
|
|
senderInfo.Addresses[0],
|
|
[]string{recipientInfo.Addresses[0]},
|
|
strings.NewReader(fmt.Sprintf("Subject: Test %v\r\n\r\nHello world!", i)),
|
|
))
|
|
}
|
|
|
|
// Connect the sender IMAP client.
|
|
senderIMAPClient, err := client.Dial(net.JoinHostPort(constants.Host, fmt.Sprint(bridge.GetIMAPPort())))
|
|
require.NoError(t, err)
|
|
require.NoError(t, senderIMAPClient.Login(senderInfo.Addresses[0], string(senderInfo.BridgePass)))
|
|
defer senderIMAPClient.Logout() //nolint:errcheck
|
|
|
|
// Connect the recipient IMAP client.
|
|
recipientIMAPClient, err := client.Dial(net.JoinHostPort(constants.Host, fmt.Sprint(bridge.GetIMAPPort())))
|
|
require.NoError(t, err)
|
|
require.NoError(t, recipientIMAPClient.Login(recipientInfo.Addresses[0], string(recipientInfo.BridgePass)))
|
|
defer recipientIMAPClient.Logout() //nolint:errcheck
|
|
|
|
// Sender should have 10 messages in the sent folder.
|
|
// Recipient should have 0 messages in inbox.
|
|
require.Eventually(t, func() bool {
|
|
sent, err := senderIMAPClient.Status(`Sent`, []imap.StatusItem{imap.StatusMessages})
|
|
require.NoError(t, err)
|
|
|
|
inbox, err := recipientIMAPClient.Status(`Inbox`, []imap.StatusItem{imap.StatusMessages})
|
|
require.NoError(t, err)
|
|
|
|
return sent.Messages == 10 && inbox.Messages == 10
|
|
}, 10*time.Second, 100*time.Millisecond)
|
|
})
|
|
})
|
|
}
|