We build too many walls and not enough bridges

test/context/credentials.go

@@ -0,0 +1,102 @@
+// 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
+
+import (
+	"strings"
+
+	"github.com/ProtonMail/proton-bridge/internal/bridge/credentials"
+)
+
+// bridgePassword is password to be used for IMAP or SMTP under tests.
+const bridgePassword = "bridgepassword"
+
+type fakeCredStore struct {
+	credentials map[string]*credentials.Credentials
+}
+
+// newFakeCredStore returns a fake credentials store (optionally configured with the given credentials).
+func newFakeCredStore(initCreds ...*credentials.Credentials) (c *fakeCredStore) {
+	c = &fakeCredStore{credentials: map[string]*credentials.Credentials{}}
+	for _, creds := range initCreds {
+		if creds == nil {
+			continue
+		}
+		c.credentials[creds.UserID] = &credentials.Credentials{}
+		*c.credentials[creds.UserID] = *creds
+	}
+	return
+}
+
+func (c *fakeCredStore) List() (userIDs []string, err error) {
+	keys := []string{}
+	for key := range c.credentials {
+		keys = append(keys, key)
+	}
+	return keys, nil
+}
+
+func (c *fakeCredStore) Add(userID, userName, apiToken, mailboxPassword string, emails []string) (*credentials.Credentials, error) {
+	bridgePassword := bridgePassword
+	if c, ok := c.credentials[userID]; ok {
+		bridgePassword = c.BridgePassword
+	}
+	c.credentials[userID] = &credentials.Credentials{
+		UserID:                userID,
+		Name:                  userName,
+		Emails:                strings.Join(emails, ";"),
+		APIToken:              apiToken,
+		MailboxPassword:       mailboxPassword,
+		BridgePassword:        bridgePassword,
+		IsCombinedAddressMode: true, // otherwise by default starts in split mode
+	}
+
+	return c.Get(userID)
+}
+
+func (c *fakeCredStore) Get(userID string) (*credentials.Credentials, error) {
+	return c.credentials[userID], nil
+}
+
+func (c *fakeCredStore) SwitchAddressMode(userID string) error {
+	return nil
+}
+
+func (c *fakeCredStore) UpdateEmails(userID string, emails []string) error {
+	return nil
+}
+
+func (c *fakeCredStore) UpdateToken(userID, apiToken string) error {
+	creds, err := c.Get(userID)
+	if err != nil {
+		return err
+	}
+	creds.APIToken = apiToken
+	return nil
+}
+
+func (c *fakeCredStore) Logout(userID string) error {
+	c.credentials[userID].APIToken = ""
+	c.credentials[userID].MailboxPassword = ""
+	return nil
+}
+
+func (c *fakeCredStore) Delete(userID string) error {
+	delete(c.credentials, userID)
+	return nil
+}