This is the v1 version, had the v2 before.

search_test.go

@@ -0,0 +1,141 @@
+package imap
+
+import (
+	"bytes"
+	"io"
+	"net/textproto"
+	"reflect"
+	"strings"
+	"testing"
+	"time"
+)
+
+// Note to myself: writing these boring tests actually fixed 2 bugs :P
+
+var searchSeqSet1, _ = ParseSeqSet("1:42")
+var searchSeqSet2, _ = ParseSeqSet("743:938")
+var searchDate1 = time.Date(1997, 11, 21, 0, 0, 0, 0, time.UTC)
+var searchDate2 = time.Date(1984, 11, 5, 0, 0, 0, 0, time.UTC)
+
+var searchCriteriaTests = []struct {
+	expected string
+	criteria *SearchCriteria
+}{
+	{
+		expected: `(1:42 UID 743:938 ` +
+			`SINCE "5-Nov-1984" BEFORE "21-Nov-1997" SENTSINCE "5-Nov-1984" SENTBEFORE "21-Nov-1997" ` +
+			`FROM "root@protonmail.com" BODY "hey there" TEXT "DILLE" ` +
+			`ANSWERED DELETED KEYWORD cc UNKEYWORD microsoft ` +
+			`LARGER 4242 SMALLER 4342 ` +
+			`NOT (SENTON "21-Nov-1997" HEADER "Content-Type" "text/csv") ` +
+			`OR (ON "5-Nov-1984" DRAFT FLAGGED UNANSWERED UNDELETED OLD) (UNDRAFT UNFLAGGED UNSEEN))`,
+		criteria: &SearchCriteria{
+			SeqNum:     searchSeqSet1,
+			Uid:        searchSeqSet2,
+			Since:      searchDate2,
+			Before:     searchDate1,
+			SentSince:  searchDate2,
+			SentBefore: searchDate1,
+			Header: textproto.MIMEHeader{
+				"From": {"root@protonmail.com"},
+			},
+			Body:         []string{"hey there"},
+			Text:         []string{"DILLE"},
+			WithFlags:    []string{AnsweredFlag, DeletedFlag, "cc"},
+			WithoutFlags: []string{"microsoft"},
+			Larger:       4242,
+			Smaller:      4342,
+			Not: []*SearchCriteria{{
+				SentSince:  searchDate1,
+				SentBefore: searchDate1.Add(24 * time.Hour),
+				Header: textproto.MIMEHeader{
+					"Content-Type": {"text/csv"},
+				},
+			}},
+			Or: [][2]*SearchCriteria{{
+				{
+					Since:        searchDate2,
+					Before:       searchDate2.Add(24 * time.Hour),
+					WithFlags:    []string{DraftFlag, FlaggedFlag},
+					WithoutFlags: []string{AnsweredFlag, DeletedFlag, RecentFlag},
+				},
+				{
+					WithoutFlags: []string{DraftFlag, FlaggedFlag, SeenFlag},
+				},
+			}},
+		},
+	},
+	{
+		expected: "(ALL)",
+		criteria: &SearchCriteria{},
+	},
+}
+
+func TestSearchCriteria_Format(t *testing.T) {
+	for i, test := range searchCriteriaTests {
+		fields := test.criteria.Format()
+
+		got, err := formatFields(fields)
+		if err != nil {
+			t.Fatal("Unexpected no error while formatting fields, got:", err)
+		}
+
+		if got != test.expected {
+			t.Errorf("Invalid search criteria fields for #%v: got \n%v\n instead of \n%v", i+1, got, test.expected)
+		}
+	}
+}
+
+func TestSearchCriteria_Parse(t *testing.T) {
+	for i, test := range searchCriteriaTests {
+		criteria := new(SearchCriteria)
+
+		b := bytes.NewBuffer([]byte(test.expected))
+		r := NewReader(b)
+		fields, _ := r.ReadFields()
+
+		if err := criteria.ParseWithCharset(fields[0].([]interface{}), nil); err != nil {
+			t.Errorf("Cannot parse search criteria for #%v: %v", i+1, err)
+		} else if !reflect.DeepEqual(criteria, test.criteria) {
+			t.Errorf("Invalid search criteria for #%v: got \n%+v\n instead of \n%+v", i+1, criteria, test.criteria)
+		}
+	}
+}
+
+var searchCriteriaParseTests = []struct {
+	fields   []interface{}
+	criteria *SearchCriteria
+	charset  func(io.Reader) io.Reader
+}{
+	{
+		fields:   []interface{}{"ALL"},
+		criteria: &SearchCriteria{},
+	},
+	{
+		fields: []interface{}{"NEW"},
+		criteria: &SearchCriteria{
+			WithFlags:    []string{RecentFlag},
+			WithoutFlags: []string{SeenFlag},
+		},
+	},
+	{
+		fields: []interface{}{"SUBJECT", strings.NewReader("café")},
+		criteria: &SearchCriteria{
+			Header: textproto.MIMEHeader{"Subject": {"café"}},
+		},
+		charset: func(r io.Reader) io.Reader {
+			return r
+		},
+	},
+}
+
+func TestSearchCriteria_Parse_others(t *testing.T) {
+	for i, test := range searchCriteriaParseTests {
+		criteria := new(SearchCriteria)
+		if err := criteria.ParseWithCharset(test.fields, test.charset); err != nil {
+			t.Errorf("Cannot parse search criteria for #%v: %v", i+1, err)
+		} else if !reflect.DeepEqual(criteria, test.criteria) {
+			t.Errorf("Invalid search criteria for #%v: got \n%+v\n instead of \n%+v", i+1, criteria, test.criteria)
+		}
+	}
+}