package courier

import (
	"bytes"
	"io/ioutil"
	"os"
	"testing"
	"time"
)

func TestProcmail(t *testing.T) {
	dir, err := ioutil.TempDir("", "test-chasquid-courier")
	if err != nil {
		t.Fatalf("Failed to create temp dir: %v", err)
	}
	defer os.RemoveAll(dir)

	MailDeliveryAgentBin = "tee"
	MailDeliveryAgentArgs = []string{dir + "/%user%"}

	p := Procmail{}

	err = p.Deliver("from@x", "to@local", []byte("data"))
	if err != nil {
		t.Fatalf("Deliver: %v", err)
	}

	data, err := ioutil.ReadFile(dir + "/to")
	if err != nil || !bytes.Equal(data, []byte("data")) {
		t.Errorf("Invalid data: %q - %v", string(data), err)
	}
}

func TestProcmailTimeout(t *testing.T) {
	MailDeliveryAgentBin = "/bin/sleep"
	MailDeliveryAgentArgs = []string{"1"}
	procmailTimeout = 100 * time.Millisecond

	p := Procmail{}

	err := p.Deliver("from", "to@local", []byte("data"))
	if err != timeoutError {
		t.Errorf("Unexpected error: %v", err)
	}

	procmailTimeout = 1 * time.Second
}

func TestProcmailBadCommandLine(t *testing.T) {
	p := Procmail{}

	// Non-existent binary.
	MailDeliveryAgentBin = "thisdoesnotexist"
	err := p.Deliver("from", "to", []byte("data"))
	if err == nil {
		t.Errorf("Unexpected success: %q %v",
			MailDeliveryAgentBin, MailDeliveryAgentArgs)
	}

	// Incorrect arguments.
	MailDeliveryAgentBin = "cat"
	MailDeliveryAgentArgs = []string{"--fail_unknown_option"}

	err = p.Deliver("from", "to", []byte("data"))
	if err == nil {
		t.Errorf("Unexpected success: %q %v",
			MailDeliveryAgentBin, MailDeliveryAgentArgs)
	}
}

func TestSanitize(t *testing.T) {
	cases := []struct{ v, expected string }{
		// These are the same.
		{"thisisfine", "thisisfine"},
		{"ñaca", "ñaca"},
		{"123-456_789", "123-456_789"},
		{"123+456~789", "123+456~789"},

		// These have problematic characters that get dropped.
		{"with spaces", "withspaces"},
		{"with/slash", "withslash"},
		{"quote';andsemicolon", "quoteandsemicolon"},
		{"a;b", "ab"},
		{`"test"`, "test"},

		// Interesting cases taken from
		// http://www.user.uni-hannover.de/nhtcapri/bidirectional-text.html
		// We allow them, they're the same on both sides.
		{"١٩٩٩–١٢–٣١", "١٩٩٩–١٢–٣١"},
		{"موزه‌ها", "موزه\u200cها"},
	}
	for _, c := range cases {
		out := sanitizeForProcmail(c.v)
		if out != c.expected {
			t.Errorf("%q: expected %q, got %q", c.v, c.expected, out)
		}
	}
}