GODT-1779: Remove go-imap

tests/smtp_test.go

@@ -0,0 +1,107 @@
+package tests
+
+import (
+	"fmt"
+	"net/smtp"
+
+	"github.com/ProtonMail/proton-bridge/v2/internal/constants"
+)
+
+func (s *scenario) userConnectsSMTPClient(username, clientID string) error {
+	return s.t.newSMTPClient(s.t.getUserID(username), clientID)
+}
+
+func (s *scenario) userConnectsSMTPClientOnPort(username, clientID string, port int) error {
+	return s.t.newSMTPClientOnPort(s.t.getUserID(username), clientID, port)
+}
+
+func (s *scenario) userConnectsAndAuthenticatesSMTPClient(username, clientID string) error {
+	if err := s.t.newSMTPClient(s.t.getUserID(username), clientID); err != nil {
+		return err
+	}
+
+	userID, client := s.t.getSMTPClient(clientID)
+
+	s.t.pushError(client.Auth(smtp.PlainAuth("", s.t.getUserAddr(userID), s.t.getUserPass(userID), constants.Host)))
+
+	return nil
+}
+
+func (s *scenario) smtpClientCanAuthenticate(clientID string) error {
+	userID, client := s.t.getSMTPClient(clientID)
+
+	if err := client.Auth(smtp.PlainAuth("", s.t.getUserAddr(userID), s.t.getUserPass(userID), constants.Host)); err != nil {
+		return fmt.Errorf("expected no error, got %v", err)
+	}
+
+	return nil
+}
+
+func (s *scenario) smtpClientCannotAuthenticate(clientID string) error {
+	userID, client := s.t.getSMTPClient(clientID)
+
+	if err := client.Auth(smtp.PlainAuth("", s.t.getUserAddr(userID), s.t.getUserPass(userID), constants.Host)); err == nil {
+		return fmt.Errorf("expected error, got nil")
+	}
+
+	return nil
+}
+
+func (s *scenario) smtpClientCannotAuthenticateWithIncorrectUsername(clientID string) error {
+	userID, client := s.t.getSMTPClient(clientID)
+
+	if err := client.Auth(smtp.PlainAuth("", s.t.getUserAddr(userID)+"bad", s.t.getUserPass(userID), constants.Host)); err == nil {
+		return fmt.Errorf("expected error, got nil")
+	}
+
+	return nil
+}
+
+func (s *scenario) smtpClientCannotAuthenticateWithIncorrectPassword(clientID string) error {
+	userID, client := s.t.getSMTPClient(clientID)
+
+	if err := client.Auth(smtp.PlainAuth("", s.t.getUserAddr(userID), s.t.getUserPass(userID)+"bad", constants.Host)); err == nil {
+		return fmt.Errorf("expected error, got nil")
+	}
+
+	return nil
+}
+
+func (s *scenario) smtpClientSendsMailFrom(clientID, from string) error {
+	_, client := s.t.getSMTPClient(clientID)
+
+	s.t.pushError(client.Mail(from))
+
+	return nil
+}
+
+func (s *scenario) smtpClientSendsRcptTo(clientID, to string) error {
+	_, client := s.t.getSMTPClient(clientID)
+
+	s.t.pushError(client.Rcpt(to))
+
+	return nil
+}
+
+func (s *scenario) smtpClientSendsData(clientID, data string) error {
+	_, client := s.t.getSMTPClient(clientID)
+
+	rc, err := client.Data()
+	if err != nil {
+		s.t.pushError(err)
+	} else if _, err := rc.Write([]byte(data)); err != nil {
+		s.t.pushError(err)
+	} else if err := rc.Close(); err != nil {
+		s.t.pushError(err)
+	}
+
+	return nil
+}
+
+func (s *scenario) smtpClientSendsReset(clientID string) error {
+	_, client := s.t.getSMTPClient(clientID)
+
+	s.t.pushError(client.Reset())
+
+	return nil
+}