We build too many walls and not enough bridges

test/imap_actions_messages_test.go

@@ -0,0 +1,158 @@
+// 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 tests
+
+import (
+	"github.com/cucumber/godog"
+)
+
+func IMAPActionsMessagesFeatureContext(s *godog.Suite) {
+	s.Step(`^IMAP client fetches "([^"]*)"$`, imapClientFetches)
+	s.Step(`^IMAP client fetches by UID "([^"]*)"$`, imapClientFetchesByUID)
+	s.Step(`^IMAP client searches for "([^"]*)"$`, imapClientSearchesFor)
+	s.Step(`^IMAP client deletes messages "([^"]*)"$`, imapClientDeletesMessages)
+	s.Step(`^IMAP client "([^"]*)" deletes messages "([^"]*)"$`, imapClientNamedDeletesMessages)
+	s.Step(`^IMAP client copies messages "([^"]*)" to "([^"]*)"$`, imapClientCopiesMessagesTo)
+	s.Step(`^IMAP client moves messages "([^"]*)" to "([^"]*)"$`, imapClientMovesMessagesTo)
+	s.Step(`^IMAP client creates message "([^"]*)" from "([^"]*)" to "([^"]*)" with body "([^"]*)" in "([^"]*)"$`, imapClientCreatesMessageFromToWithBody)
+	s.Step(`^IMAP client creates message "([^"]*)" from "([^"]*)" to address "([^"]*)" of "([^"]*)" with body "([^"]*)" in "([^"]*)"$`, imapClientCreatesMessageFromToAddressOfUserWithBody)
+	s.Step(`^IMAP client creates message "([^"]*)" from address "([^"]*)" of "([^"]*)" to "([^"]*)" with body "([^"]*)" in "([^"]*)"$`, imapClientCreatesMessageFromAddressOfUserToWithBody)
+	s.Step(`^IMAP client marks message "([^"]*)" as read$`, imapClientMarksMessageAsRead)
+	s.Step(`^IMAP client "([^"]*)" marks message "([^"]*)" as read$`, imapClientNamedMarksMessageAsRead)
+	s.Step(`^IMAP client marks message "([^"]*)" as unread$`, imapClientMarksMessageAsUnread)
+	s.Step(`^IMAP client "([^"]*)" marks message "([^"]*)" as unread$`, imapClientNamedMarksMessageAsUnread)
+	s.Step(`^IMAP client marks message "([^"]*)" as starred$`, imapClientMarksMessageAsStarred)
+	s.Step(`^IMAP client "([^"]*)" marks message "([^"]*)" as starred$`, imapClientNamedMarksMessageAsStarred)
+	s.Step(`^IMAP client marks message "([^"]*)" as unstarred$`, imapClientMarksMessageAsUnstarred)
+	s.Step(`^IMAP client "([^"]*)" marks message "([^"]*)" as unstarred$`, imapClientNamedMarksMessageAsUnstarred)
+	s.Step(`^IMAP client starts IDLE-ing$`, imapClientStartsIDLEing)
+	s.Step(`^IMAP client "([^"]*)" starts IDLE-ing$`, imapClientNamedStartsIDLEing)
+}
+
+func imapClientFetches(fetchRange string) error {
+	res := ctx.GetIMAPClient("imap").Fetch(fetchRange, "UID")
+	ctx.SetIMAPLastResponse("imap", res)
+	return nil
+}
+
+func imapClientFetchesByUID(fetchRange string) error {
+	res := ctx.GetIMAPClient("imap").FetchUID(fetchRange, "UID")
+	ctx.SetIMAPLastResponse("imap", res)
+	return nil
+}
+
+func imapClientSearchesFor(query string) error {
+	res := ctx.GetIMAPClient("imap").Search(query)
+	ctx.SetIMAPLastResponse("imap", res)
+	return nil
+}
+
+func imapClientDeletesMessages(messageRange string) error {
+	return imapClientNamedDeletesMessages("imap", messageRange)
+}
+
+func imapClientNamedDeletesMessages(imapClient, messageRange string) error {
+	res := ctx.GetIMAPClient(imapClient).Delete(messageRange)
+	ctx.SetIMAPLastResponse(imapClient, res)
+	return nil
+}
+
+func imapClientCopiesMessagesTo(messageRange, newMailboxName string) error {
+	res := ctx.GetIMAPClient("imap").Copy(messageRange, newMailboxName)
+	ctx.SetIMAPLastResponse("imap", res)
+	return nil
+}
+
+func imapClientMovesMessagesTo(messageRange, newMailboxName string) error {
+	res := ctx.GetIMAPClient("imap").Move(messageRange, newMailboxName)
+	ctx.SetIMAPLastResponse("imap", res)
+	return nil
+}
+
+func imapClientCreatesMessageFromToWithBody(subject, from, to, body, mailboxName string) error {
+	res := ctx.GetIMAPClient("imap").Append(mailboxName, subject, from, to, body)
+	ctx.SetIMAPLastResponse("imap", res)
+	return nil
+}
+
+func imapClientCreatesMessageFromToAddressOfUserWithBody(subject, from, bddAddressID, bddUserID, body, mailboxName string) error {
+	account := ctx.GetTestAccountWithAddress(bddUserID, bddAddressID)
+	if account == nil {
+		return godog.ErrPending
+	}
+	return imapClientCreatesMessageFromToWithBody(subject, from, account.Address(), body, mailboxName)
+}
+
+func imapClientCreatesMessageFromAddressOfUserToWithBody(subject, bddAddressID, bddUserID, to, body, mailboxName string) error {
+	account := ctx.GetTestAccountWithAddress(bddUserID, bddAddressID)
+	if account == nil {
+		return godog.ErrPending
+	}
+
+	return imapClientCreatesMessageFromToWithBody(subject, account.Address(), to, body, mailboxName)
+}
+
+func imapClientMarksMessageAsRead(messageRange string) error {
+	return imapClientNamedMarksMessageAsRead("imap", messageRange)
+}
+
+func imapClientNamedMarksMessageAsRead(imapClient, messageRange string) error {
+	res := ctx.GetIMAPClient(imapClient).MarkAsRead(messageRange)
+	ctx.SetIMAPLastResponse(imapClient, res)
+	return nil
+}
+
+func imapClientMarksMessageAsUnread(messageRange string) error {
+	return imapClientNamedMarksMessageAsUnread("imap", messageRange)
+}
+
+func imapClientNamedMarksMessageAsUnread(imapClient, messageRange string) error {
+	res := ctx.GetIMAPClient(imapClient).MarkAsUnread(messageRange)
+	ctx.SetIMAPLastResponse(imapClient, res)
+	return nil
+}
+
+func imapClientMarksMessageAsStarred(messageRange string) error {
+	return imapClientNamedMarksMessageAsStarred("imap", messageRange)
+}
+
+func imapClientNamedMarksMessageAsStarred(imapClient, messageRange string) error {
+	res := ctx.GetIMAPClient(imapClient).MarkAsStarred(messageRange)
+	ctx.SetIMAPLastResponse(imapClient, res)
+	return nil
+}
+
+func imapClientMarksMessageAsUnstarred(messageRange string) error {
+	return imapClientNamedMarksMessageAsUnstarred("imap", messageRange)
+}
+
+func imapClientNamedMarksMessageAsUnstarred(imapClient, messageRange string) error {
+	res := ctx.GetIMAPClient(imapClient).MarkAsUnstarred(messageRange)
+	ctx.SetIMAPLastResponse(imapClient, res)
+	return nil
+}
+
+func imapClientStartsIDLEing() error {
+	return imapClientNamedStartsIDLEing("imap")
+}
+
+func imapClientNamedStartsIDLEing(imapClient string) error {
+	res := ctx.GetIMAPClient(imapClient).StartIDLE()
+	ctx.SetIMAPLastResponse(imapClient, res)
+	return nil
+}