mirror of
https://github.com/KumoCorp/kumomta.git
synced 2026-09-09 08:01:32 +00:00
Wez Furlong
2a657f68fb
and also in resolve-shaping-domain, for consistency. We can use the message rate from the scheduled queue config to refine the computed ceilings/constraints too, which is nice.
89 lines
2.2 KiB
Lua
89 lines
2.2 KiB
Lua
local kumo = require 'kumo'
|
|
package.path = '../../assets/?.lua;' .. package.path
|
|
|
|
local TEST_DIR = os.getenv 'KUMOD_TEST_DIR'
|
|
|
|
kumo.on('init', function()
|
|
kumo.configure_accounting_db_path(TEST_DIR .. '/accounting.db')
|
|
kumo.aaa.configure_acct_log {
|
|
log_dir = TEST_DIR .. '/acct',
|
|
max_segment_duration = '1s',
|
|
}
|
|
|
|
kumo.start_esmtp_listener {
|
|
listen = '127.0.0.1:0',
|
|
relay_hosts = { '0.0.0.0/0' },
|
|
}
|
|
|
|
kumo.start_http_listener {
|
|
listen = '127.0.0.1:0',
|
|
}
|
|
|
|
kumo.configure_local_logs {
|
|
log_dir = TEST_DIR .. '/logs',
|
|
max_segment_duration = '1s',
|
|
}
|
|
|
|
kumo.define_spool {
|
|
name = 'data',
|
|
path = TEST_DIR .. '/data-spool',
|
|
}
|
|
|
|
kumo.define_spool {
|
|
name = 'meta',
|
|
path = TEST_DIR .. '/meta-spool',
|
|
}
|
|
end)
|
|
|
|
kumo.on('get_listener_domain', function(_domain, _listener, _conn_meta)
|
|
return kumo.make_listener_domain {
|
|
relay_to = true,
|
|
}
|
|
end)
|
|
|
|
-- Custom Lua dispatcher whose send method sleeps long enough for
|
|
-- the inspect/abort integration test to observe and act on it, but
|
|
-- with the watchdog turned off so the test, not the watchdog, drives
|
|
-- the abort.
|
|
kumo.on('make.slow_send', function(_domain, _tenant, _campaign)
|
|
local sender = {}
|
|
function sender:send(_message)
|
|
kumo.sleep(60)
|
|
end
|
|
return sender
|
|
end)
|
|
|
|
kumo.on(
|
|
'get_queue_config',
|
|
function(domain, _tenant, _campaign, _routing_domain)
|
|
local cfg = {
|
|
protocol = {
|
|
custom_lua = {
|
|
constructor = 'make.slow_send',
|
|
},
|
|
},
|
|
}
|
|
-- For the scheduled-queue-constraint integration test.
|
|
if domain == 'sched-rate.example.com' then
|
|
cfg.max_message_rate = '100/s'
|
|
end
|
|
return kumo.make_queue_config(cfg)
|
|
end
|
|
)
|
|
|
|
kumo.on('get_egress_path_config', function(domain, _source_name, _site_name)
|
|
local cfg = {
|
|
connection_limit = 1,
|
|
-- A large value so the watchdog cannot interfere; the test
|
|
-- aborts the dispatcher itself via the admin API.
|
|
dispatcher_progress_watchdog_timeout = '1h',
|
|
}
|
|
-- For the scheduled-queue-constraint integration test: the path
|
|
-- declares a max_message_rate of 1000/s, which the scheduled
|
|
-- queue's 100/s rate then shadows.
|
|
if domain == 'sched-rate.example.com' then
|
|
cfg.max_message_rate = '1000/s'
|
|
end
|
|
return kumo.make_egress_path(cfg)
|
|
end)
|