From 052395f917fe55cfa2e1ab5c42539ed3c07df5bc Mon Sep 17 00:00:00 2001
From: James Houlahan <james.houlahan@protonmail.com>
Date: Wed, 4 Nov 2020 14:58:45 +0100
Subject: [PATCH] test: add benchmarks for rfc5322 address/date parser

---
 pkg/message/rfc5322/benchmark_test.go | 157 ++++++++++++++++++++++++++
 1 file changed, 157 insertions(+)
 create mode 100644 pkg/message/rfc5322/benchmark_test.go

diff --git a/pkg/message/rfc5322/benchmark_test.go b/pkg/message/rfc5322/benchmark_test.go
new file mode 100644
index 00000000..ad075c8e
--- /dev/null
+++ b/pkg/message/rfc5322/benchmark_test.go
@@ -0,0 +1,157 @@
+// Copyright (c) 2020 Proton Technologies AG
+//
+// This file is part of ProtonMail 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 rfc5322
+
+import (
+	"net/mail"
+	"testing"
+)
+
+func BenchmarkStandardSet(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		parseSet(benchStandardSet)
+	}
+}
+
+func BenchmarkStandardSetGolang(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		parseSetGolang(benchStandardSet)
+	}
+}
+
+func BenchmarkEncodedSet(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		parseSet(benchEncodedSet)
+	}
+}
+
+func BenchmarkEncodedSetGolang(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		parseSetGolang(benchEncodedSet)
+	}
+}
+
+func BenchmarkAddressListSet(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		parseSet(benchAddressListSet)
+	}
+}
+
+func BenchmarkAddressListSetGolang(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		parseSetGolang(benchAddressListSet)
+	}
+}
+
+func BenchmarkGroupSet(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		parseSet(benchGroupSet)
+	}
+}
+
+func BenchmarkGroupSetGolang(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		parseSetGolang(benchGroupSet)
+	}
+}
+
+func parseSet(set []string) {
+	for _, addr := range set {
+		_, _ = ParseAddressList(addr)
+	}
+}
+
+func parseSetGolang(set []string) {
+	for _, addr := range set {
+		_, _ = mail.ParseAddressList(addr)
+	}
+}
+
+var benchStandardSet = []string{
+	`user@example.com`,
+	`John Doe <jdoe@machine.example>`,
+	`Mary Smith <mary@example.net>`,
+	`"Joe Q. Public" <john.q.public@example.com>`,
+	`Mary Smith <mary@x.test>`,
+	`jdoe@example.org`,
+	`Who? <one@y.test>`,
+	`<boss@nil.test>`,
+	`"Giant; \"Big\" Box" <sysservices@example.net>`,
+	`Pete <pete@silly.example>`,
+	`"Mary Smith: Personal Account" <smith@home.example>`,
+	`Pete(A nice \) chap) <pete(his account)@silly.test(his host)>`,
+	`Gogh Fir <gf@example.com>`,
+	`normal name  <username@server.com>`,
+	`"comma, name"  <username@server.com>`,
+	`name  <username@server.com> (ignore comment)`,
+	`"Mail Robot" <>`,
+	`Michal Hořejšek <hořejšek@mail.com>`,
+	`First Last <user@domain.com >`,
+	`First Last <user@domain.com. >`,
+	`First Last <user@domain.com.>`,
+	`First Last <user@domain.com:25>`,
+	`First Last <user@[10.0.0.1]>`,
+	`<postmaster@[10.10.10.10]>`,
+	`user@domain <user@domain.com>`,
+	`First Last < user@domain.com>`,
+	`First Middle @ Last <user@domain.com>`,
+	`user@domain.com,`,
+	`First Middle "Last" <user@domain.com>`,
+	`First Middle Last <user@domain.com>`,
+	`First Middle"Last" <user@domain.com>`,
+	`First Middle "Last"<user@domain.com>`,
+	`First "Middle" "Last" <user@domain.com>`,
+	`First "Middle""Last" <user@domain.com>`,
+}
+
+var benchEncodedSet = []string{
+	`=?US-ASCII?Q?Keith_Moore?= <moore@cs.utk.edu>`,
+	`=?ISO-8859-1?Q?Keld_J=F8rn_Simonsen?= <keld@dkuug.dk>`,
+	`=?ISO-8859-1?Q?Andr=E9?= Pirard <PIRARD@vm1.ulg.ac.be>`,
+	`=?ISO-8859-1?Q?Olle_J=E4rnefors?= <ojarnef@admin.kth.se>`,
+	`=?ISO-8859-1?Q?Patrik_F=E4ltstr=F6m?= <paf@nada.kth.se>`,
+	`Nathaniel Borenstein <nsb@thumper.bellcore.com> (=?iso-8859-8?b?7eXs+SDv4SDp7Oj08A==?=)`,
+	`=?UTF-8?B?PEJlemUgam3DqW5hPg==?= <user@domain.com>`,
+	`First Middle =?utf-8?Q?Last?= <user@domain.com>`,
+	`First Middle =?utf-8?Q?Last?=<user@domain.com>`,
+	`First =?utf-8?Q?Middle?= =?utf-8?Q?Last?= <user@domain.com>`,
+	`First =?utf-8?Q?Middle?==?utf-8?Q?Last?= <user@domain.com>`,
+	`First "Middle"=?utf-8?Q?Last?= <user@domain.com>`,
+	`First "Middle" =?utf-8?Q?Last?= <user@domain.com>`,
+	`First "Middle" =?utf-8?Q?Last?=<user@domain.com>`,
+	`=?UTF-8?B?PEJlemUgam3DqW5hPg==?= <user@domain.com>`,
+	`=?utf-8?B?6YCZ5piv5ryi5a2X55qE5LiA5YCL5L6L5a2Q?= <proton.testqa@gmail.com>`,
+	`=?utf-8?B?8J+MmfCfjbc=?= <user@domain.com>`,
+	`=?utf-8?B?8J+MmfCfjbc=?= <user@domain.com>`,
+}
+
+var benchAddressListSet = []string{
+	`Alice <alice@example.com>, Bob <bob@example.com>, Eve <eve@example.com>`,
+	`Ed Jones <c@a.test>,joe@where.test,John <jdoe@one.test>`,
+	`name (ignore comment)  <username@server.com>,  (Comment as name) username2@server.com`,
+	`"normal name"  <username@server.com>, "comma, name" <address@server.com>`,
+	`"comma, one"  <username@server.com>, "comma, two" <address@server.com>`,
+	`normal name  <username@server.com>, (comment)All.(around)address@(the)server.com`,
+	`normal name  <username@server.com>, All.("comma, in comment")address@(the)server.com`,
+}
+
+var benchGroupSet = []string{
+	`A Group:Ed Jones <c@a.test>,joe@where.test,John <jdoe@one.test>;`,
+	`Undisclosed recipients:;`,
+	`(Empty list)(start)Hidden recipients  :(nobody(that I know))  ;`,
+}