mirror of
https://github.com/ProtonMail/proton-bridge.git
synced 2025-12-10 12:46:46 +00:00
1b22c32ef9
Rather than returning whether the vault was corrupt or not return the error which caused the vault to be considered as corrupt.
270 lines
7.8 KiB
Go
270 lines
7.8 KiB
Go
// Copyright (c) 2023 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 (
|
|
"math"
|
|
"testing"
|
|
|
|
"github.com/Masterminds/semver/v3"
|
|
"github.com/ProtonMail/gluon/async"
|
|
"github.com/ProtonMail/proton-bridge/v3/internal/updater"
|
|
"github.com/ProtonMail/proton-bridge/v3/internal/useragent"
|
|
"github.com/ProtonMail/proton-bridge/v3/internal/vault"
|
|
"github.com/ProtonMail/proton-bridge/v3/pkg/ports"
|
|
"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, ports.FindFreePortFrom(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, ports.FindFreePortFrom(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"), async.NoopPanicHandler{})
|
|
require.NoError(t, err)
|
|
require.NoError(t, corrupt)
|
|
|
|
// Check the default gluon dir.
|
|
require.Equal(t, "/path/to/gluon", s.GetGluonCacheDir())
|
|
|
|
// Modify the gluon dir.
|
|
require.NoError(t, s.SetGluonDir("/tmp/gluon"))
|
|
|
|
// Check the new gluon dir.
|
|
require.Equal(t, "/tmp/gluon", s.GetGluonCacheDir())
|
|
}
|
|
|
|
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())
|
|
|
|
// Since GODT-2319 0.6046602879796196 is not allowed as a rollout value (RNG was not seeded)
|
|
require.NoError(t, s.SetUpdateRollout(vault.ForbiddenRollout))
|
|
require.GreaterOrEqual(t, math.Abs(s.GetUpdateRollout()-vault.ForbiddenRollout), 0.00000001)
|
|
}
|
|
|
|
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, false, s.GetProxyAllowed())
|
|
|
|
// Modify the proxy allowed setting.
|
|
require.NoError(t, s.SetProxyAllowed(true))
|
|
|
|
// Check the new proxy allowed setting.
|
|
require.Equal(t, true, 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_TelemetryDisabled(t *testing.T) {
|
|
// create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// Check the default show all mail setting.
|
|
require.Equal(t, false, s.GetTelemetryDisabled())
|
|
|
|
// Modify the show all mail setting.
|
|
require.NoError(t, s.SetTelemetryDisabled(true))
|
|
|
|
// Check the new show all mail setting.
|
|
require.Equal(t, true, s.GetTelemetryDisabled())
|
|
}
|
|
|
|
func TestVault_Settings_Autostart(t *testing.T) {
|
|
// create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// Check the default autostart setting.
|
|
require.Equal(t, true, s.GetAutostart())
|
|
|
|
// Modify the autostart setting.
|
|
require.NoError(t, s.SetAutostart(false))
|
|
|
|
// Check the new autostart setting.
|
|
require.Equal(t, false, 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_MaxSyncMemory(t *testing.T) {
|
|
// create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// Check the default first start value.
|
|
require.Equal(t, vault.DefaultMaxSyncMemory, s.GetMaxSyncMemory())
|
|
}
|
|
|
|
func TestVault_Settings_LastUserAgent(t *testing.T) {
|
|
// create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// Check the default first start value.
|
|
require.Equal(t, useragent.DefaultUserAgent, s.GetLastUserAgent())
|
|
}
|
|
|
|
func Test_Settings_PasswordArchive(t *testing.T) {
|
|
// Create a new test vault.
|
|
s := newVault(t)
|
|
|
|
// The store should have no users.
|
|
require.Empty(t, s.GetUserIDs())
|
|
|
|
// Create a new user.
|
|
user, err := s.AddUser("userID1", "username1", "username1@pm.me", "authUID1", "authRef1", []byte("keyPass1"))
|
|
require.NoError(t, err)
|
|
bridgePass := user.BridgePass()
|
|
|
|
// Remove the user.
|
|
require.NoError(t, user.Close())
|
|
require.NoError(t, s.DeleteUser("userID1"))
|
|
|
|
// Add a different user. Another password is generated.
|
|
user, err = s.AddUser("userID2", "username2", "username2@pm.me", "authUID2", "authRef2", []byte("keyPass2"))
|
|
require.NoError(t, err)
|
|
require.NotEqual(t, user.BridgePass(), bridgePass)
|
|
|
|
// Add the first user again. The password is restored.
|
|
user, err = s.AddUser("userID1", "username1", "username1@pm.me", "authUID1", "authRef1", []byte("keyPass1"))
|
|
require.NoError(t, err)
|
|
require.Equal(t, user.BridgePass(), bridgePass)
|
|
}
|