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
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