mirror of
https://github.com/ProtonMail/proton-bridge.git
synced 2025-12-10 04:36:43 +00:00
130 lines
4.0 KiB
Go
130 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 app
|
|
|
|
import (
|
|
"net/http/cookiejar"
|
|
"net/url"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/ProtonMail/proton-bridge/v3/internal/bridge"
|
|
"github.com/ProtonMail/proton-bridge/v3/internal/cookies"
|
|
"github.com/ProtonMail/proton-bridge/v3/internal/locations"
|
|
"github.com/ProtonMail/proton-bridge/v3/internal/updater"
|
|
"github.com/ProtonMail/proton-bridge/v3/internal/vault"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMigratePrefsToVault(t *testing.T) {
|
|
// Create a new vault.
|
|
vault, corrupt, err := vault.New(t.TempDir(), t.TempDir(), []byte("my secret key"))
|
|
require.NoError(t, err)
|
|
require.False(t, corrupt)
|
|
|
|
// load the old prefs file.
|
|
b, err := os.ReadFile(filepath.Join("testdata", "prefs.json"))
|
|
require.NoError(t, err)
|
|
|
|
// Migrate the old prefs file to the new vault.
|
|
require.NoError(t, migratePrefsToVault(vault, b))
|
|
|
|
// Check that the IMAP and SMTP prefs are migrated.
|
|
require.Equal(t, 2143, vault.GetIMAPPort())
|
|
require.Equal(t, 2025, vault.GetSMTPPort())
|
|
require.True(t, vault.GetSMTPSSL())
|
|
|
|
// Check that the update channel is migrated.
|
|
require.True(t, vault.GetAutoUpdate())
|
|
require.Equal(t, updater.EarlyChannel, vault.GetUpdateChannel())
|
|
require.Equal(t, 0.4849529004202015, vault.GetUpdateRollout())
|
|
|
|
// Check that the app settings have been migrated.
|
|
require.False(t, vault.GetFirstStart())
|
|
require.True(t, vault.GetFirstStartGUI())
|
|
require.Equal(t, "blablabla", vault.GetColorScheme())
|
|
require.Equal(t, "2.3.0+git", vault.GetLastVersion().String())
|
|
require.True(t, vault.GetAutostart())
|
|
|
|
// Check that the other app settings have been migrated.
|
|
require.Equal(t, 16, vault.SyncWorkers())
|
|
require.Equal(t, 16, vault.SyncAttPool())
|
|
require.False(t, vault.GetProxyAllowed())
|
|
require.False(t, vault.GetShowAllMail())
|
|
|
|
// Check that the cookies have been migrated.
|
|
jar, err := cookiejar.New(nil)
|
|
require.NoError(t, err)
|
|
|
|
cookies, err := cookies.NewCookieJar(jar, vault)
|
|
require.NoError(t, err)
|
|
|
|
url, err := url.Parse("https://api.protonmail.ch")
|
|
require.NoError(t, err)
|
|
|
|
// There should be a cookie for the API.
|
|
require.NotEmpty(t, cookies.Cookies(url))
|
|
}
|
|
|
|
func TestKeychainMigration(t *testing.T) {
|
|
// migration needed only for linux
|
|
if runtime.GOOS != "linux" {
|
|
return
|
|
}
|
|
|
|
tmpDir := t.TempDir()
|
|
os.Setenv("XDG_CONFIG_HOME", tmpDir)
|
|
|
|
oldCacheDir := filepath.Join(tmpDir, "protonmail", "bridge")
|
|
require.NoError(t, os.MkdirAll(oldCacheDir, 0o700))
|
|
|
|
oldPrefs, err := os.ReadFile(filepath.Join("testdata", "prefs.json"))
|
|
require.NoError(t, err)
|
|
|
|
require.NoError(t, os.WriteFile(
|
|
filepath.Join(oldCacheDir, "prefs.json"),
|
|
oldPrefs, 0o600,
|
|
))
|
|
|
|
locations := locations.New(bridge.NewTestLocationsProvider(tmpDir), "config-name")
|
|
|
|
settingsFolder, err := locations.ProvideSettingsPath()
|
|
require.NoError(t, err)
|
|
|
|
keychainName, err := vault.GetHelper(settingsFolder)
|
|
require.NoError(t, err)
|
|
require.Equal(t, "", keychainName)
|
|
|
|
require.NoError(t, migrateKeychainHelper(locations))
|
|
|
|
keychainName, err = vault.GetHelper(settingsFolder)
|
|
require.NoError(t, err)
|
|
require.Equal(t, "secret-service", keychainName)
|
|
|
|
require.NoError(t, vault.SetHelper(settingsFolder, "different"))
|
|
|
|
// Calling migration again will not overwrite
|
|
require.NoError(t, migrateKeychainHelper(locations))
|
|
keychainName, err = vault.GetHelper(settingsFolder)
|
|
require.NoError(t, err)
|
|
require.Equal(t, "different", keychainName)
|
|
|
|
}
|