mailhook/hook_test.go at master · ushis/mailhook · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
package main

import (
	"gopkg.in/yaml.v3"
	"testing"
)

const hooksYaml = `
---
- hook: https://example.com/v1/mails
  emails:
    - 'alice@example.com'
    - '@example.net'

- hook: https://example.com/v2/mails
  emails:
    - '@example.org'
    - 'bob@example.io'

- hook: https://example.com/v3/mails
  emails:
    - '@'
`

func parseHooks() (Hooks, error) {
	hooks := Hooks{}

	if err := yaml.Unmarshal([]byte(hooksYaml), &hooks); err != nil {
		return nil, err
	}
	return hooks, nil
}

func TestFindByAddress(t *testing.T) {
	hooks, err := parseHooks()

	if err != nil {
		t.Fatal(err)
	}
	hook, ok := hooks.Find("alice@example.com")

	if !ok {
		t.Fatal("expected to find hook")
	}

	if hook.Hook != "https://example.com/v1/mails" {
		t.Fatalf("\"https://example.com/v1/mails\" != \"%s\"", hook.Hook)
	}
	hook, ok = hooks.Find("bob@example.io")

	if !ok {
		t.Fatal("expected to find hook")
	}

	if hook.Hook != "https://example.com/v2/mails" {
		t.Fatalf("\"https://example.com/v2/mails\" != \"%s\"", hook.Hook)
	}
}

func TestFindByTaggedAddress(t *testing.T) {
	hooks, err := parseHooks()

	if err != nil {
		t.Fatal(err)
	}
	hook, ok := hooks.Find("alice+249c01d587797d3451b03f90a2ecbc7a@example.com")

	if !ok {
		t.Fatal("expected to find hook")
	}

	if hook.Hook != "https://example.com/v1/mails" {
		t.Fatalf("\"https://example.com/v1/mails\" != \"%s\"", hook.Hook)
	}
}

func TestFindByDomain(t *testing.T) {
	hooks, err := parseHooks()

	if err != nil {
		t.Fatal(err)
	}
	hook, ok := hooks.Find("42b71135b144c5c5a1c803c9dedd6ceb@example.net")

	if !ok {
		t.Fatal("expected to find hook")
	}

	if hook.Hook != "https://example.com/v1/mails" {
		t.Fatalf("\"https://example.com/v1/mails\" != \"%s\"", hook.Hook)
	}
	hook, ok = hooks.Find("a73bd5e153e6f1321ebaf62ad01a93c8@example.org")

	if !ok {
		t.Fatal("expected to find hook")
	}

	if hook.Hook != "https://example.com/v2/mails" {
		t.Fatalf("\"https://example.com/v1/mails\" != \"%s\"", hook.Hook)
	}
}

func TestFindByCatchAll(t *testing.T) {
	hooks, err := parseHooks()

	if err != nil {
		t.Fatal(err)
	}
	hook, ok := hooks.Find("32cf34012668@41248827644a.e1e86c9fa707")

	if !ok {
		t.Fatal("expected to find hook")
	}

	if hook.Hook != "https://example.com/v3/mails" {
		t.Fatalf("\"https://example.com/v3/mails\" != \"%s\"", hook.Hook)
	}
}

func TestFindNothing(t *testing.T) {
	hooks, err := parseHooks()

	if err != nil {
		t.Fatal(err)
	}
	hooks = hooks[:len(hooks)-1]

	if _, ok := hooks.Find("32cf34012668@41248827644a.e1e86c9fa707"); ok {
		t.Fatal("expected not to find anything")
	}
}