mirror of
https://github.com/axllent/mailpit.git
synced 2026-06-28 06:56:06 +00:00
198 lines
8.3 KiB
Go
198 lines
8.3 KiB
Go
package tools
|
|
|
|
import (
|
|
"net"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestIsInternalIP(t *testing.T) {
|
|
internal := []string{
|
|
"127.0.0.1", // loopback
|
|
"::1", // IPv6 loopback
|
|
"10.0.0.1", // private
|
|
"172.16.0.1", // private
|
|
"192.168.1.1", // private
|
|
"169.254.1.1", // link-local unicast
|
|
"fe80::1", // IPv6 link-local
|
|
"0.0.0.0", // unspecified
|
|
"224.0.0.1", // multicast
|
|
"100.64.0.1", // CGNAT start
|
|
"100.127.255.255", // CGNAT end
|
|
// IPv6 transition forms embedding an internal IPv4 destination — golang/go#79925.
|
|
"64:ff9b::a9fe:a9fe", // NAT64 well-known (RFC 6052) wrapping 169.254.169.254
|
|
"64:ff9b:1::a9fe:a9fe", // NAT64 local-use (RFC 8215) wrapping 169.254.169.254
|
|
"2002:a9fe:a9fe::", // 6to4 (RFC 3056) wrapping 169.254.169.254
|
|
"::a9fe:a9fe", // IPv4-compatible IPv6 (RFC 4291) wrapping 169.254.169.254
|
|
"64:ff9b::7f00:1", // NAT64 wrapping 127.0.0.1
|
|
"2002:0a00:0001::", // 6to4 wrapping 10.0.0.1
|
|
"::ffff:169.254.169.254", // IPv4-mapped (also caught by stdlib via To4)
|
|
"::5efe:a9fe:a9fe", // ISATAP (RFC 5214) wrapping 169.254.169.254
|
|
"2001:db8::5efe:7f00:1", // ISATAP under a documentation prefix wrapping 127.0.0.1
|
|
"fec0::1", // deprecated site-local (RFC 3879 / RFC 4291 §2.5.7)
|
|
"2001:db8::1", // documentation prefix (RFC 3849)
|
|
"2001:db8::5efe:0808:0808", // documentation prefix (blocked regardless of embedded IPv4)
|
|
}
|
|
external := []string{
|
|
"8.8.8.8",
|
|
"1.1.1.1",
|
|
"100.128.0.1", // just outside CGNAT range
|
|
"2001:4860:4860::8888", // Google public DNS over IPv6
|
|
"2002:0808:0808::", // 6to4 wrapping 8.8.8.8 (public IPv4)
|
|
"64:ff9b::0808:0808", // NAT64 wrapping 8.8.8.8 (public IPv4)
|
|
}
|
|
|
|
for _, s := range internal {
|
|
ip := net.ParseIP(s)
|
|
if !IsInternalIP(ip) {
|
|
t.Errorf("expected %s to be internal", s)
|
|
}
|
|
}
|
|
for _, s := range external {
|
|
ip := net.ParseIP(s)
|
|
if IsInternalIP(ip) {
|
|
t.Errorf("expected %s to be external", s)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIsValidLinkURL(t *testing.T) {
|
|
valid := []string{
|
|
"http://example.com",
|
|
"https://example.com",
|
|
"https://example.com/path?q=1#anchor",
|
|
}
|
|
invalid := []string{
|
|
"",
|
|
"ftp://example.com",
|
|
"example.com",
|
|
"//example.com",
|
|
"https://",
|
|
}
|
|
|
|
for _, s := range valid {
|
|
if !IsValidLinkURL(s) {
|
|
t.Errorf("expected %q to be a valid link URL", s)
|
|
}
|
|
}
|
|
for _, s := range invalid {
|
|
if IsValidLinkURL(s) {
|
|
t.Errorf("expected %q to be an invalid link URL", s)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestArgsParser(t *testing.T) {
|
|
tests := map[string][]string{}
|
|
tests["this is a test"] = []string{"this", "is", "a", "test"}
|
|
tests["\"this is\" a test"] = []string{"this is", "a", "test"}
|
|
tests["!\"this is\" a test"] = []string{"!this is", "a", "test"}
|
|
tests["subject:this is a test"] = []string{"subject:this", "is", "a", "test"}
|
|
tests["subject:\"this is\" a test"] = []string{"subject:this is", "a", "test"}
|
|
tests["subject:\"this is\" \"a test\""] = []string{"subject:this is", "a test"}
|
|
tests["subject:\"this 'is\" \"a test\""] = []string{"subject:this 'is", "a test"}
|
|
tests["subject:\"this 'is a test"] = []string{"subject:this 'is a test"}
|
|
tests["\"this is a test\"=\"this is a test\""] = []string{"this is a test=this is a test"}
|
|
|
|
for search, expected := range tests {
|
|
res := ArgsParser(search)
|
|
if !reflect.DeepEqual(res, expected) {
|
|
t.Log("Args parser error:", res, "!=", expected)
|
|
t.Fail()
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestCleanTag(t *testing.T) {
|
|
tests := map[string]string{}
|
|
tests["this is a test"] = "this is a test"
|
|
tests["thiS IS a Test"] = "thiS IS a Test"
|
|
tests["thiS IS a Test :-)"] = "thiS IS a Test -"
|
|
tests[" thiS 99 IS a Test :-)"] = "thiS 99 IS a Test -"
|
|
tests["this_is-a test "] = "this_is-a test"
|
|
tests["this_is-a&^%%(*)@ test"] = "this_is-a @ test"
|
|
tests["this is a long tag title with more than 100 characters, which should get automatically truncated to 100 characters"] = "this is a long tag title with more than 100 characters which should get automatically truncated to 1"
|
|
|
|
for search, expected := range tests {
|
|
res := CleanTag(search)
|
|
if res != expected {
|
|
t.Log("CleanTags error:", res, "!=", expected)
|
|
t.Fail()
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSnippets(t *testing.T) {
|
|
tests := map[string]string{}
|
|
tests["this is a test"] = "this is a test"
|
|
tests["thiS IS a Test"] = "thiS IS a Test"
|
|
tests["thiS IS a Test :-)"] = "thiS IS a Test :-)"
|
|
tests["<h1>This is a test.</h1> "] = "This is a test."
|
|
tests["this_is-a test "] = "this_is-a test"
|
|
tests["this_is-a&^%%(*)@ test"] = "this_is-a&^%%(*)@ test"
|
|
tests["<h1>Heading</h1><p>Paragraph</p>"] = "Heading Paragraph"
|
|
tests[`<h1>Heading</h1>
|
|
<p>Paragraph</p>`] = "Heading Paragraph"
|
|
tests[`<h1>Heading</h1><p> <a href="https://github.com">linked text</a></p>`] = "Heading linked text"
|
|
// broken html
|
|
tests[`<h1>Heading</h3><p> <a href="https://github.com">linked text.`] = "Heading linked text."
|
|
// truncation to 200 chars + ...
|
|
tests["abcdefghijklmnopqrstuvwxyx0123456789 abcdefghijklmnopqrstuvwxyx0123456789 abcdefghijklmnopqrstuvwxyx0123456789 abcdefghijklmnopqrstuvwxyx0123456789 abcdefghijklmnopqrstuvwxyx0123456789 abcdefghijklmnopqrstuvwxyx0123456789 abcdefghijklmnopqrstuvwxyx0123456789 abcdefghijklmnopqrstuvwxyx0123456789 abcdefghijklmnopqrstuvwxyx0123456789"] = "abcdefghijklmnopqrstuvwxyx0123456789 abcdefghijklmnopqrstuvwxyx0123456789 abcdefghijklmnopqrstuvwxyx0123456789 abcdefghijklmnopqrstuvwxyx0123456789 abcdefghijklmnopqrstuvwxyx0123456789 abcdefghijklmno..."
|
|
|
|
for str, expected := range tests {
|
|
res := CreateSnippet(str, str)
|
|
if res != expected {
|
|
t.Log("CreateSnippet error:", res, "!=", expected)
|
|
t.Fail()
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestListUnsubscribeParser(t *testing.T) {
|
|
tests := map[string]bool{}
|
|
|
|
// should pass
|
|
tests["<mailto:unsubscribe@example.com>"] = true
|
|
tests["<https://example.com>"] = true
|
|
tests["<HTTPS://EXAMPLE.COM>"] = true
|
|
tests["<mailto:unsubscribe@example.com>, <http://example.com>"] = true
|
|
tests["<mailto:unsubscribe@example.com>, <https://example.com>"] = true
|
|
tests["<https://example.com>, <mailto:unsubscribe@example.com>"] = true
|
|
tests["<https://example.com> , <mailto:unsubscribe@example.com>"] = true
|
|
tests["<https://example.com> ,<mailto:unsubscribe@example.com>"] = true
|
|
tests["<mailto:unsubscribe@example.com>,<https://example.com>"] = true
|
|
tests[`<https://example.com> ,
|
|
<mailto:unsubscribe@example.com>`] = true
|
|
tests["<mailto:unsubscribe@example.com?subject=unsubscribe%20me>"] = true
|
|
tests["(Use this command to get off the list) <mailto:unsubscribe@example.com?subject=unsubscribe%20me>"] = true
|
|
tests["<mailto:unsubscribe@example.com> (Use this command to get off the list)"] = true
|
|
tests["(Use this command to get off the list) <mailto:unsubscribe@example.com>, (Click this link to unsubscribe) <http://example.com>"] = true
|
|
|
|
// should fail
|
|
tests["mailto:unsubscribe@example.com"] = false // no <>
|
|
tests["<mailto::unsubscribe@example.com>"] = false // ::
|
|
tests["https://example.com/"] = false // no <>
|
|
tests["mailto:unsubscribe@example.com, <https://example.com/>"] = false // no <>
|
|
tests["<MAILTO:unsubscribe@example.com>"] = false // capitals
|
|
tests["<mailto:unsubscribe@example.com>, <mailto:test2@example.com>"] = false // two emails
|
|
tests["<http://exampl\\e2.com>, <http://example2.com>"] = false // two links
|
|
tests["<http://example.com>, <mailto:unsubscribe@example.com>, <http://example2.com>"] = false // two links
|
|
tests["<mailto:unsubscribe@example.com>, <example.com>"] = false // no mailto || http(s)
|
|
tests["<mailto: unsubscribe@example.com>, <unsubscribe@lol.com>"] = false // space
|
|
tests["<mailto:unsubscribe@example.com?subject=unsubscribe me>"] = false // space
|
|
tests["<http:///example.com>"] = false // http:///
|
|
|
|
for search, expected := range tests {
|
|
_, err := ListUnsubscribeParser(search)
|
|
hasError := err != nil
|
|
if expected == hasError {
|
|
if err != nil {
|
|
t.Logf("ListUnsubscribeParser: %v", err)
|
|
} else {
|
|
t.Logf("ListUnsubscribeParser: \"%s\" expected: %v", search, expected)
|
|
}
|
|
t.Fail()
|
|
}
|
|
}
|
|
}
|