mirror of
https://github.com/ProtonMail/proton-bridge.git
synced 2025-12-15 14:56:42 +00:00
4b6d0d035e
Make sure that we are at least using 16 workers for sync, otherwise multiply the current sync worker count by 2. Finally, this patch also logs the duration of the time it takes to transfer all the messages from the server.
228 lines
6.2 KiB
Go
228 lines
6.2 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 vault_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/Masterminds/semver/v3"
|
|
"github.com/ProtonMail/proton-bridge/v2/internal/updater"
|
|
"github.com/ProtonMail/proton-bridge/v2/internal/vault"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestVault_Settings_IMAP(t *testing.T) {
|
|
// Create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// Check the default IMAP port and SSL setting.
|
|
require.Equal(t, 1143, s.GetIMAPPort())
|
|
require.Equal(t, false, s.GetIMAPSSL())
|
|
|
|
// Modify the IMAP port and SSL setting.
|
|
require.NoError(t, s.SetIMAPPort(1234))
|
|
require.NoError(t, s.SetIMAPSSL(true))
|
|
|
|
// Check the new IMAP port and SSL setting.
|
|
require.Equal(t, 1234, s.GetIMAPPort())
|
|
require.Equal(t, true, s.GetIMAPSSL())
|
|
}
|
|
|
|
func TestVault_Settings_SMTP(t *testing.T) {
|
|
// Create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// Check the default SMTP port and SSL setting.
|
|
require.Equal(t, 1025, s.GetSMTPPort())
|
|
require.Equal(t, false, s.GetSMTPSSL())
|
|
|
|
// Modify the SMTP port and SSL setting.
|
|
require.NoError(t, s.SetSMTPPort(1234))
|
|
require.NoError(t, s.SetSMTPSSL(true))
|
|
|
|
// Check the new SMTP port and SSL setting.
|
|
require.Equal(t, 1234, s.GetSMTPPort())
|
|
require.Equal(t, true, s.GetSMTPSSL())
|
|
}
|
|
|
|
func TestVault_Settings_GluonDir(t *testing.T) {
|
|
// create a new test vault.
|
|
s, corrupt, err := vault.New(t.TempDir(), "/path/to/gluon", []byte("my secret key"))
|
|
require.NoError(t, err)
|
|
require.False(t, corrupt)
|
|
|
|
// Check the default gluon dir.
|
|
require.Equal(t, "/path/to/gluon", s.GetGluonDir())
|
|
|
|
// Modify the gluon dir.
|
|
require.NoError(t, s.SetGluonDir("/tmp/gluon"))
|
|
|
|
// Check the new gluon dir.
|
|
require.Equal(t, "/tmp/gluon", s.GetGluonDir())
|
|
}
|
|
|
|
func TestVault_Settings_UpdateChannel(t *testing.T) {
|
|
// create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// Check the default update channel.
|
|
require.Equal(t, updater.StableChannel, s.GetUpdateChannel())
|
|
|
|
// Modify the update channel.
|
|
require.NoError(t, s.SetUpdateChannel(updater.EarlyChannel))
|
|
|
|
// Check the new update channel.
|
|
require.Equal(t, updater.EarlyChannel, s.GetUpdateChannel())
|
|
}
|
|
|
|
func TestVault_Settings_UpdateRollout(t *testing.T) {
|
|
// create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// Check the default update rollout.
|
|
require.GreaterOrEqual(t, s.GetUpdateRollout(), float64(0))
|
|
require.LessOrEqual(t, s.GetUpdateRollout(), float64(1))
|
|
|
|
// Modify the update rollout.
|
|
require.NoError(t, s.SetUpdateRollout(0.5))
|
|
|
|
// Check the new update rollout.
|
|
require.Equal(t, float64(0.5), s.GetUpdateRollout())
|
|
}
|
|
|
|
func TestVault_Settings_ColorScheme(t *testing.T) {
|
|
// create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// Check the default color scheme.
|
|
require.Equal(t, "", s.GetColorScheme())
|
|
|
|
// Modify the color scheme.
|
|
require.NoError(t, s.SetColorScheme("dark"))
|
|
|
|
// Check the new color scheme.
|
|
require.Equal(t, "dark", s.GetColorScheme())
|
|
}
|
|
|
|
func TestVault_Settings_ProxyAllowed(t *testing.T) {
|
|
// create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// Check the default proxy allowed setting.
|
|
require.Equal(t, true, s.GetProxyAllowed())
|
|
|
|
// Modify the proxy allowed setting.
|
|
require.NoError(t, s.SetProxyAllowed(false))
|
|
|
|
// Check the new proxy allowed setting.
|
|
require.Equal(t, false, s.GetProxyAllowed())
|
|
}
|
|
|
|
func TestVault_Settings_ShowAllMail(t *testing.T) {
|
|
// create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// Check the default show all mail setting.
|
|
require.Equal(t, true, s.GetShowAllMail())
|
|
|
|
// Modify the show all mail setting.
|
|
require.NoError(t, s.SetShowAllMail(false))
|
|
|
|
// Check the new show all mail setting.
|
|
require.Equal(t, false, s.GetShowAllMail())
|
|
}
|
|
|
|
func TestVault_Settings_Autostart(t *testing.T) {
|
|
// create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// Check the default autostart setting.
|
|
require.Equal(t, false, s.GetAutostart())
|
|
|
|
// Modify the autostart setting.
|
|
require.NoError(t, s.SetAutostart(true))
|
|
|
|
// Check the new autostart setting.
|
|
require.Equal(t, true, s.GetAutostart())
|
|
}
|
|
|
|
func TestVault_Settings_AutoUpdate(t *testing.T) {
|
|
// create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// Check the default auto update setting.
|
|
require.Equal(t, true, s.GetAutoUpdate())
|
|
|
|
// Modify the auto update setting.
|
|
require.NoError(t, s.SetAutoUpdate(false))
|
|
|
|
// Check the new auto update setting.
|
|
require.Equal(t, false, s.GetAutoUpdate())
|
|
}
|
|
|
|
func TestVault_Settings_LastVersion(t *testing.T) {
|
|
// create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// Check the default first start value.
|
|
require.True(t, semver.MustParse("0.0.0").Equal(s.GetLastVersion()))
|
|
|
|
// Modify the first start value.
|
|
require.NoError(t, s.SetLastVersion(semver.MustParse("1.2.3")))
|
|
|
|
// Check the new first start value.
|
|
require.True(t, semver.MustParse("1.2.3").Equal(s.GetLastVersion()))
|
|
}
|
|
|
|
func TestVault_Settings_FirstStart(t *testing.T) {
|
|
// create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// Check the default first start value.
|
|
require.Equal(t, true, s.GetFirstStart())
|
|
|
|
// Modify the first start value.
|
|
require.NoError(t, s.SetFirstStart(false))
|
|
|
|
// Check the new first start value.
|
|
require.Equal(t, false, s.GetFirstStart())
|
|
}
|
|
|
|
func TestVault_Settings_FirstStartGUI(t *testing.T) {
|
|
// create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// Check the default first start value.
|
|
require.Equal(t, true, s.GetFirstStartGUI())
|
|
|
|
// Modify the first start value.
|
|
require.NoError(t, s.SetFirstStartGUI(false))
|
|
|
|
// Check the new first start value.
|
|
require.Equal(t, false, s.GetFirstStartGUI())
|
|
}
|
|
|
|
func TestVault_Settings_SyncWorkers(t *testing.T) {
|
|
// create a new test vault.
|
|
s := newVault(t)
|
|
|
|
syncWorkers := vault.GetDefaultSyncWorkerCount()
|
|
require.Equal(t, syncWorkers, s.SyncWorkers())
|
|
require.Equal(t, syncWorkers, s.SyncAttPool())
|
|
}
|