peroxide/test/context/context.go

// Copyright (c) 2020 Proton Technologies AG
//
// This file is part of ProtonMail Bridge.Bridge.
//
// ProtonMail 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.
//
// ProtonMail 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 ProtonMail Bridge.  If not, see <https://www.gnu.org/licenses/>.

// Package context allows integration tests to be written in a fluent, english-like way.
package context

import (
	"github.com/ProtonMail/proton-bridge/internal/bridge"
	"github.com/ProtonMail/proton-bridge/internal/importexport"
	"github.com/ProtonMail/proton-bridge/internal/transfer"
	"github.com/ProtonMail/proton-bridge/internal/users"
	"github.com/ProtonMail/proton-bridge/pkg/listener"
	"github.com/ProtonMail/proton-bridge/pkg/pmapi"
	"github.com/ProtonMail/proton-bridge/test/accounts"
	"github.com/ProtonMail/proton-bridge/test/mocks"
	"github.com/sirupsen/logrus"
)

type server interface {
	ListenAndServe()
	Close()
}

// TestContext manages a bridge test (mocked API, bridge instance, IMAP/SMTP servers, setup steps).
type TestContext struct {
	// Base setup for the whole bridge (core & imap & smtp).
	t            *bddT
	cfg          *fakeConfig
	listener     listener.Listener
	testAccounts *accounts.TestAccounts

	// pmapiController is used to control real or fake pmapi clients.
	// The clients are created by the clientManager.
	pmapiController PMAPIController
	clientManager   *pmapi.ClientManager

	// Core related variables.
	bridge       *bridge.Bridge
	importExport *importexport.ImportExport
	users        *users.Users
	credStore    users.CredentialsStorer
	lastError    error

	// IMAP related variables.
	imapAddr          string
	imapServer        server
	imapClients       map[string]*mocks.IMAPClient
	imapLastResponses map[string]*mocks.IMAPResponse

	// SMTP related variables.
	smtpAddr          string
	smtpServer        server
	smtpClients       map[string]*mocks.SMTPClient
	smtpLastResponses map[string]*mocks.SMTPResponse

	// Transfer related variables.
	transferLocalRootForImport string
	transferLocalRootForExport string
	transferRemoteIMAPServer   *mocks.IMAPServer
	transferProgress           *transfer.Progress

	// These are the cleanup steps executed when Cleanup() is called.
	cleanupSteps []*Cleaner

	// logger allows logging of test labels to be handled differently (silenced/diverted/whatever).
	logger logrus.FieldLogger
}

// New returns a new test TestContext.
func New(app string) *TestContext {
	setLogrusVerbosityFromEnv()

	cfg := newFakeConfig()

	cm := pmapi.NewClientManager(cfg.GetAPIConfig())

	ctx := &TestContext{
		t:                 &bddT{},
		cfg:               cfg,
		listener:          listener.New(),
		pmapiController:   newPMAPIController(cm),
		clientManager:     cm,
		testAccounts:      newTestAccounts(),
		credStore:         newFakeCredStore(),
		imapClients:       make(map[string]*mocks.IMAPClient),
		imapLastResponses: make(map[string]*mocks.IMAPResponse),
		smtpClients:       make(map[string]*mocks.SMTPClient),
		smtpLastResponses: make(map[string]*mocks.SMTPResponse),
		logger:            logrus.StandardLogger(),
	}

	// Ensure that the config is cleaned up after the test is over.
	ctx.addCleanupChecked(cfg.ClearData, "Cleaning bridge config data")

	// Create bridge or import-export instance under test.
	switch app {
	case "bridge":
		ctx.withBridgeInstance()
	case "ie":
		ctx.withImportExportInstance()
	default:
		panic("unknown app: " + app)
	}

	return ctx
}

// Cleanup runs through all cleanup steps.
// This can be a deferred call so that it is run even if the test steps failed the test.
func (ctx *TestContext) Cleanup() *TestContext {
	for _, cleanup := range ctx.cleanupSteps {
		cleanup.Execute()
	}
	return ctx
}

// GetPMAPIController returns API controller, either fake or live.
func (ctx *TestContext) GetPMAPIController() PMAPIController {
	return ctx.pmapiController
}

// GetTestingT returns testing.T compatible struct.
func (ctx *TestContext) GetTestingT() *bddT { //nolint[golint]
	return ctx.t
}

// GetTestingError returns error if test failed by using custom TestingT.
func (ctx *TestContext) GetTestingError() error {
	return ctx.t.getErrors()
}

// SetLastError sets the last error that occurred while executing an action.
func (ctx *TestContext) SetLastError(err error) {
	ctx.lastError = err
}

// GetLastError returns the last error that occurred while executing an action.
func (ctx *TestContext) GetLastError() error {
	return ctx.lastError
}