mirror of
https://github.com/jhillyerd/inbucket.git
synced 2025-12-17 17:47:03 +00:00
239426692e
* lua: update bind_message to use table syntax * lua: update bind_address to use table syntax
116 lines
3.0 KiB
Go
116 lines
3.0 KiB
Go
package luahost_test
|
|
|
|
import (
|
|
"net/mail"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/inbucket/inbucket/pkg/extension"
|
|
"github.com/inbucket/inbucket/pkg/extension/event"
|
|
"github.com/inbucket/inbucket/pkg/extension/luahost"
|
|
"github.com/stretchr/testify/require"
|
|
lua "github.com/yuin/gopher-lua"
|
|
)
|
|
|
|
func TestEmptyScript(t *testing.T) {
|
|
script := ""
|
|
extHost := extension.NewHost()
|
|
|
|
_, err := luahost.NewFromReader(extHost, strings.NewReader(script), "test.lua")
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestAfterMessageStored(t *testing.T) {
|
|
// Register lua event listener, setup notify channel.
|
|
script := `
|
|
local test_ok = true
|
|
|
|
function assert_eq(got, want)
|
|
if got ~= want then
|
|
-- Incorrect value, schedule test to fail.
|
|
print("got '" .. got .. "', wanted '" .. want .. "'")
|
|
test_ok = false
|
|
end
|
|
end
|
|
|
|
function after_message_stored(msg)
|
|
assert_eq(msg.mailbox, "mb1")
|
|
assert_eq(msg.id, "id1")
|
|
assert_eq(msg.subject, "subj1")
|
|
assert_eq(msg.size, 42)
|
|
|
|
assert_eq(msg.from.name, "name1")
|
|
assert_eq(msg.from.address, "addr1")
|
|
|
|
assert_eq(table.getn(msg.to), 1)
|
|
assert_eq(msg.to[1].name, "name2")
|
|
assert_eq(msg.to[1].address, "addr2")
|
|
|
|
assert_eq(msg.date, 981173106)
|
|
|
|
notify:send(test_ok)
|
|
end
|
|
`
|
|
extHost := extension.NewHost()
|
|
luaHost, err := luahost.NewFromReader(extHost, strings.NewReader(script), "test.lua")
|
|
require.NoError(t, err)
|
|
notify := luaHost.CreateChannel("notify")
|
|
|
|
// Send event, check channel response is true.
|
|
msg := &event.MessageMetadata{
|
|
Mailbox: "mb1",
|
|
ID: "id1",
|
|
From: &mail.Address{Name: "name1", Address: "addr1"},
|
|
To: []*mail.Address{{Name: "name2", Address: "addr2"}},
|
|
Date: time.Date(2001, time.February, 3, 4, 5, 6, 0, time.UTC),
|
|
Subject: "subj1",
|
|
Size: 42,
|
|
}
|
|
go extHost.Events.AfterMessageStored.Emit(msg)
|
|
assertNotified(t, notify)
|
|
}
|
|
|
|
func TestBeforeMailAccepted(t *testing.T) {
|
|
// Register lua event listener.
|
|
script := `
|
|
function before_mail_accepted(localpart, domain)
|
|
return localpart == "from" and domain == "test"
|
|
end
|
|
`
|
|
extHost := extension.NewHost()
|
|
_, err := luahost.NewFromReader(extHost, strings.NewReader(script), "test.lua")
|
|
require.NoError(t, err)
|
|
|
|
// Send event to be accepted.
|
|
addr := &event.AddressParts{Local: "from", Domain: "test"}
|
|
got := extHost.Events.BeforeMailAccepted.Emit(addr)
|
|
want := true
|
|
require.NotNil(t, got)
|
|
if *got != want {
|
|
t.Errorf("Got %v, wanted %v for addr %v", *got, want, addr)
|
|
}
|
|
|
|
// Send event to be denied.
|
|
addr = &event.AddressParts{Local: "reject", Domain: "me"}
|
|
got = extHost.Events.BeforeMailAccepted.Emit(addr)
|
|
want = false
|
|
require.NotNil(t, got)
|
|
if *got != want {
|
|
t.Errorf("Got %v, wanted %v for addr %v", *got, want, addr)
|
|
}
|
|
}
|
|
|
|
func assertNotified(t *testing.T, notify chan lua.LValue) {
|
|
t.Helper()
|
|
select {
|
|
case reslv := <-notify:
|
|
// Lua function received event.
|
|
if lua.LVIsFalse(reslv) {
|
|
t.Error("Lua responsed with false, wanted true")
|
|
}
|
|
case <-time.After(2 * time.Second):
|
|
t.Fatal("Lua did not respond to event within timeout")
|
|
}
|
|
}
|