proton-bridge/tests/ctx_bridge_test.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 tests

import (
	"context"
	"crypto/tls"
	"fmt"
	"net/http/cookiejar"

	"github.com/ProtonMail/gluon/queue"
	"github.com/ProtonMail/proton-bridge/v2/internal/bridge"
	"github.com/ProtonMail/proton-bridge/v2/internal/cookies"
	"github.com/ProtonMail/proton-bridge/v2/internal/events"
	"github.com/ProtonMail/proton-bridge/v2/internal/useragent"
	"github.com/ProtonMail/proton-bridge/v2/internal/vault"
	"gitlab.protontech.ch/go/liteapi"
)

func (t *testCtx) startBridge() error {
	// Bridge will enable the proxy by default at startup.
	t.mocks.ProxyCtl.EXPECT().AllowProxy()

	// Get the path to the vault.
	vaultDir, err := t.locator.ProvideSettingsPath()
	if err != nil {
		return err
	}

	// Get the default gluon path.
	gluonDir, err := t.locator.ProvideGluonPath()
	if err != nil {
		return err
	}

	// Create the vault.
	vault, corrupt, err := vault.New(vaultDir, gluonDir, t.storeKey)
	if err != nil {
		return err
	} else if corrupt {
		return fmt.Errorf("vault is corrupt")
	}

	// Create the underlying cookie jar.
	jar, err := cookiejar.New(nil)
	if err != nil {
		return err
	}

	// Create the persisting cookie jar.
	persister, err := cookies.NewCookieJar(jar, vault)
	if err != nil {
		return err
	}

	// Create the bridge.
	bridge, eventCh, err := bridge.New(
		// App stuff
		t.locator,
		vault,
		t.mocks.Autostarter,
		t.mocks.Updater,
		t.version,

		// API stuff
		t.api.GetHostURL(),
		persister,
		useragent.New(),
		t.mocks.TLSReporter,
		liteapi.NewDialer(t.netCtl, &tls.Config{InsecureSkipVerify: true}).GetRoundTripper(),
		t.mocks.ProxyCtl,

		// Logging stuff
		false,
		false,
		false,
	)
	if err != nil {
		return err
	}

	// Create the event channels for use in the test.
	t.loginCh = queue.NewQueuedChannel[events.UserLoggedIn](0, 0)
	t.logoutCh = queue.NewQueuedChannel[events.UserLoggedOut](0, 0)
	t.loadedCh = queue.NewQueuedChannel[events.AllUsersLoaded](0, 0)
	t.deletedCh = queue.NewQueuedChannel[events.UserDeleted](0, 0)
	t.deauthCh = queue.NewQueuedChannel[events.UserDeauth](0, 0)
	t.addrCreatedCh = queue.NewQueuedChannel[events.UserAddressCreated](0, 0)
	t.addrDeletedCh = queue.NewQueuedChannel[events.UserAddressDeleted](0, 0)
	t.syncStartedCh = queue.NewQueuedChannel[events.SyncStarted](0, 0)
	t.syncFinishedCh = queue.NewQueuedChannel[events.SyncFinished](0, 0)
	t.forcedUpdateCh = queue.NewQueuedChannel[events.UpdateForced](0, 0)
	t.connStatusCh = queue.NewQueuedChannel[events.Event](0, 0)
	t.updateCh = queue.NewQueuedChannel[events.Event](0, 0)

	// Push the updates to the appropriate channels.
	go func() {
		for event := range eventCh {
			switch event := event.(type) {
			case events.UserLoggedIn:
				t.loginCh.Enqueue(event)
			case events.UserLoggedOut:
				t.logoutCh.Enqueue(event)
			case events.AllUsersLoaded:
				t.loadedCh.Enqueue(event)
			case events.UserDeleted:
				t.deletedCh.Enqueue(event)
			case events.UserDeauth:
				t.deauthCh.Enqueue(event)
			case events.UserAddressCreated:
				t.addrCreatedCh.Enqueue(event)
			case events.UserAddressDeleted:
				t.addrDeletedCh.Enqueue(event)
			case events.SyncStarted:
				t.syncStartedCh.Enqueue(event)
			case events.SyncFinished:
				t.syncFinishedCh.Enqueue(event)
			case events.ConnStatusUp:
				t.connStatusCh.Enqueue(event)
			case events.ConnStatusDown:
				t.connStatusCh.Enqueue(event)
			case events.UpdateAvailable:
				t.updateCh.Enqueue(event)
			case events.UpdateNotAvailable:
				t.updateCh.Enqueue(event)
			case events.UpdateInstalled:
				t.updateCh.Enqueue(event)
			case events.UpdateForced:
				t.forcedUpdateCh.Enqueue(event)
			}
		}
	}()

	// Wait for the users to be loaded.
	<-t.loadedCh.GetChannel()

	// Save the bridge to the context.
	t.bridge = bridge

	return nil
}

func (t *testCtx) stopBridge() error {
	if err := t.bridge.Close(context.Background()); err != nil {
		return err
	}

	t.bridge = nil
	t.loginCh.CloseAndDiscardQueued()
	t.logoutCh.CloseAndDiscardQueued()
	t.loadedCh.CloseAndDiscardQueued()
	t.deletedCh.CloseAndDiscardQueued()
	t.deauthCh.CloseAndDiscardQueued()
	t.addrCreatedCh.CloseAndDiscardQueued()
	t.addrDeletedCh.CloseAndDiscardQueued()
	t.syncStartedCh.CloseAndDiscardQueued()
	t.syncFinishedCh.CloseAndDiscardQueued()
	t.forcedUpdateCh.CloseAndDiscardQueued()
	t.connStatusCh.CloseAndDiscardQueued()
	t.updateCh.CloseAndDiscardQueued()

	return nil
}