We build too many walls and not enough bridges

test/accounts/accounts.go

@@ -0,0 +1,79 @@
+// 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 accounts
+
+import (
+	"encoding/json"
+	"io/ioutil"
+
+	"github.com/ProtonMail/proton-bridge/pkg/pmapi"
+	"github.com/pkg/errors"
+)
+
+// BridgePassword is password to be used for IMAP or SMTP under tests.
+const BridgePassword = "bridgepassword"
+
+type TestAccounts struct {
+	Users            map[string]*pmapi.User               // Key is user ID used in BDD.
+	Addresses        map[string]map[string]*pmapi.Address // Key is real user ID, second key is address ID used in BDD.
+	Passwords        map[string]string                    // Key is real user ID.
+	MailboxPasswords map[string]string                    // Key is real user ID.
+	TwoFAs           map[string]bool                      // Key is real user ID.
+}
+
+func Load(path string) (*TestAccounts, error) {
+	data, err := ioutil.ReadFile(path) //nolint[gosec]
+	if err != nil {
+		return nil, errors.Wrap(err, "failed to load JSON")
+	}
+
+	var testAccounts TestAccounts
+	err = json.Unmarshal(data, &testAccounts)
+	if err != nil {
+		return nil, errors.Wrap(err, "failed to unmarshal JSON")
+	}
+
+	return &testAccounts, nil
+}
+
+func (a *TestAccounts) GetTestAccount(username string) *TestAccount {
+	return a.GetTestAccountWithAddress(username, "")
+}
+
+func (a *TestAccounts) GetTestAccountWithAddress(username, addressID string) *TestAccount {
+	// Do lookup by full address and convert to name in tests.
+	// Used by getting real data to ensure correct address or address ID.
+	for key, user := range a.Users {
+		if user.Name == username {
+			username = key
+			break
+		}
+	}
+	user, ok := a.Users[username]
+	if !ok {
+		return nil
+	}
+	return newTestAccount(
+		user,
+		a.Addresses[user.Name],
+		addressID,
+		a.Passwords[user.Name],
+		a.MailboxPasswords[user.Name],
+		a.TwoFAs[user.Name],
+	)
+}