Mercurial > prosody-modules
changeset 3357:af824168729a
mod_test_data: New module to generate test data in Prosody's data store
author | Matthew Wild <mwild1@gmail.com> |
---|---|
date | Mon, 15 Oct 2018 14:27:20 +0100 (2018-10-15) |
parents | 31e113823463 |
children | e49660ba3161 |
files | mod_test_data/README.markdown mod_test_data/mod_test_data.lua |
diffstat | 2 files changed, 50 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mod_test_data/README.markdown Mon Oct 15 14:27:20 2018 +0100 @@ -0,0 +1,12 @@ +--- +description: Generate test data +labels: +- 'Stage-Alpha' +--- + +This module is of use to developers who want to fill their Prosody storage with +test data, e.g. for debugging, testing or performance measurements of Prosody or +client code. + +Currently it only supports filling MAM archives, but patches are welcome for other +functionality.
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mod_test_data/mod_test_data.lua Mon Oct 15 14:27:20 2018 +0100 @@ -0,0 +1,38 @@ +local users = { "fezziwig", "badger", "nupkins", "pumblechook", "rouncewell" }; +local host = "localhost"; + +local id = require "util.id"; +local st = require "util.stanza"; +local sm = require "core.storagemanager"; + +-- Return a random number from 1..max excluding n +function random_other(n, max) return ((math.random(1, max-1)+(n-1))%max)+1; end + +local new_time; +do + local _current_time = os.time(); + function new_time() + _current_time = _current_time + math.random(1, 3600); + return _current_time; + end +end + +function module.command(arg) --luacheck: ignore arg + sm.initialize_host(host); + local archive = sm.open(host, "archive", "archive"); + + for _ = 1, 100000 do + local random = math.random(1, #users); + local user, contact = users[random], users[random_other(random, #users)]; + local user_jid, contact_jid = user.."@"..host, contact.."@"..host; + + local stanza = st.message({ to = contact_jid, from = user_jid, type="chat" }) + :tag("body"):text(id.long()); + + archive:append(user, nil, stanza, new_time(), contact_jid) + + local stanza2 = st.clone(stanza); + stanza2.attr.from, stanza2.attr.to = stanza.attr.to, stanza.attr.from; + archive:append(contact, nil, stanza2, new_time(), user_jid) + end +end