248 lines
9.2 KiB
Nix
248 lines
9.2 KiB
Nix
|
import ../make-test-python.nix ({ pkgs, ... }:
|
||
|
|
let
|
||
|
|
homeserverDomain = "server";
|
||
|
|
homeserverUrl = "http://server:8008";
|
||
|
|
username = "alice";
|
||
|
|
instagramBotUsername = "instagrambot";
|
||
|
|
facebookBotUsername = "facebookbot";
|
||
|
|
in
|
||
|
|
{
|
||
|
|
name = "mautrix-meta-sqlite";
|
||
|
|
meta.maintainers = pkgs.mautrix-meta.meta.maintainers;
|
||
|
|
|
||
|
|
nodes = {
|
||
|
|
server = { config, pkgs, ... }: {
|
||
|
|
services.matrix-synapse = {
|
||
|
|
enable = true;
|
||
|
|
settings = {
|
||
|
|
database.name = "sqlite3";
|
||
|
|
|
||
|
|
enable_registration = true;
|
||
|
|
|
||
|
|
# don't use this in production, always use some form of verification
|
||
|
|
enable_registration_without_verification = true;
|
||
|
|
|
||
|
|
listeners = [ {
|
||
|
|
# The default but tls=false
|
||
|
|
bind_addresses = [
|
||
|
|
"0.0.0.0"
|
||
|
|
];
|
||
|
|
port = 8008;
|
||
|
|
resources = [ {
|
||
|
|
"compress" = true;
|
||
|
|
"names" = [ "client" ];
|
||
|
|
} {
|
||
|
|
"compress" = false;
|
||
|
|
"names" = [ "federation" ];
|
||
|
|
} ];
|
||
|
|
tls = false;
|
||
|
|
type = "http";
|
||
|
|
} ];
|
||
|
|
};
|
||
|
|
};
|
||
|
|
|
||
|
|
services.mautrix-meta.instances.facebook = {
|
||
|
|
enable = true;
|
||
|
|
|
||
|
|
settings = {
|
||
|
|
homeserver = {
|
||
|
|
address = homeserverUrl;
|
||
|
|
domain = homeserverDomain;
|
||
|
|
};
|
||
|
|
|
||
|
|
appservice = {
|
||
|
|
port = 8009;
|
||
|
|
|
||
|
|
bot.username = facebookBotUsername;
|
||
|
|
};
|
||
|
|
|
||
|
|
bridge.permissions."@${username}:server" = "user";
|
||
|
|
};
|
||
|
|
};
|
||
|
|
|
||
|
|
services.mautrix-meta.instances.instagram = {
|
||
|
|
enable = true;
|
||
|
|
|
||
|
|
settings = {
|
||
|
|
homeserver = {
|
||
|
|
address = homeserverUrl;
|
||
|
|
domain = homeserverDomain;
|
||
|
|
};
|
||
|
|
|
||
|
|
appservice = {
|
||
|
|
port = 8010;
|
||
|
|
|
||
|
|
bot.username = instagramBotUsername;
|
||
|
|
};
|
||
|
|
|
||
|
|
bridge.permissions."@${username}:server" = "user";
|
||
|
|
};
|
||
|
|
};
|
||
|
|
|
||
|
|
networking.firewall.allowedTCPPorts = [ 8008 ];
|
||
|
|
};
|
||
|
|
|
||
|
|
client = { pkgs, ... }: {
|
||
|
|
environment.systemPackages = [
|
||
|
|
(pkgs.writers.writePython3Bin "register_user"
|
||
|
|
{
|
||
|
|
libraries = [ pkgs.python3Packages.matrix-nio ];
|
||
|
|
flakeIgnore = [
|
||
|
|
# We don't live in the dark ages anymore.
|
||
|
|
# Languages like Python that are whitespace heavy will overrun
|
||
|
|
# 79 characters..
|
||
|
|
"E501"
|
||
|
|
];
|
||
|
|
} ''
|
||
|
|
import sys
|
||
|
|
import asyncio
|
||
|
|
|
||
|
|
from nio import AsyncClient
|
||
|
|
|
||
|
|
|
||
|
|
async def run(username: str, homeserver: str):
|
||
|
|
matrix = AsyncClient(homeserver)
|
||
|
|
|
||
|
|
response = await matrix.register(username, "foobar")
|
||
|
|
print("Matrix register response: ", response)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
asyncio.run(run(sys.argv[1], sys.argv[2]))
|
||
|
|
''
|
||
|
|
)
|
||
|
|
(pkgs.writers.writePython3Bin "do_test"
|
||
|
|
{
|
||
|
|
libraries = [ pkgs.python3Packages.matrix-nio ];
|
||
|
|
flakeIgnore = [
|
||
|
|
# We don't live in the dark ages anymore.
|
||
|
|
# Languages like Python that are whitespace heavy will overrun
|
||
|
|
# 79 characters..
|
||
|
|
"E501"
|
||
|
|
];
|
||
|
|
} ''
|
||
|
|
import sys
|
||
|
|
import functools
|
||
|
|
import asyncio
|
||
|
|
|
||
|
|
from nio import AsyncClient, RoomMessageNotice, RoomCreateResponse, RoomInviteResponse
|
||
|
|
|
||
|
|
|
||
|
|
async def message_callback(matrix: AsyncClient, msg: str, _r, e):
|
||
|
|
print("Received matrix text message: ", e)
|
||
|
|
assert msg in e.body
|
||
|
|
exit(0) # Success!
|
||
|
|
|
||
|
|
|
||
|
|
async def run(username: str, bot_username: str, homeserver: str):
|
||
|
|
matrix = AsyncClient(homeserver, f"@{username}:${homeserverDomain}")
|
||
|
|
|
||
|
|
response = await matrix.login("foobar")
|
||
|
|
print("Matrix login response: ", response)
|
||
|
|
|
||
|
|
# Open a DM with the bridge bot
|
||
|
|
response = await matrix.room_create()
|
||
|
|
print("Matrix create room response:", response)
|
||
|
|
assert isinstance(response, RoomCreateResponse)
|
||
|
|
room_id = response.room_id
|
||
|
|
|
||
|
|
response = await matrix.room_invite(room_id, f"@{bot_username}:${homeserverDomain}")
|
||
|
|
assert isinstance(response, RoomInviteResponse)
|
||
|
|
|
||
|
|
callback = functools.partial(
|
||
|
|
message_callback, matrix, "Hello, I'm an Instagram bridge bot."
|
||
|
|
)
|
||
|
|
matrix.add_event_callback(callback, RoomMessageNotice)
|
||
|
|
|
||
|
|
print("Waiting for matrix message...")
|
||
|
|
await matrix.sync_forever(timeout=30000)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
asyncio.run(run(sys.argv[1], sys.argv[2], sys.argv[3]))
|
||
|
|
''
|
||
|
|
)
|
||
|
|
];
|
||
|
|
};
|
||
|
|
};
|
||
|
|
|
||
|
|
testScript = ''
|
||
|
|
def extract_token(data):
|
||
|
|
stdout = data[1]
|
||
|
|
stdout = stdout.strip()
|
||
|
|
line = stdout.split('\n')[-1]
|
||
|
|
return line.split(':')[-1].strip("\" '\n")
|
||
|
|
|
||
|
|
def get_token_from(token, file):
|
||
|
|
data = server.execute(f"cat {file} | grep {token}")
|
||
|
|
return extract_token(data)
|
||
|
|
|
||
|
|
def get_as_token_from(file):
|
||
|
|
return get_token_from("as_token", file)
|
||
|
|
|
||
|
|
def get_hs_token_from(file):
|
||
|
|
return get_token_from("hs_token", file)
|
||
|
|
|
||
|
|
config_yaml = "/var/lib/mautrix-meta-facebook/config.yaml"
|
||
|
|
registration_yaml = "/var/lib/mautrix-meta-facebook/meta-registration.yaml"
|
||
|
|
|
||
|
|
start_all()
|
||
|
|
|
||
|
|
with subtest("wait for bridges and homeserver"):
|
||
|
|
# bridge
|
||
|
|
server.wait_for_unit("mautrix-meta-facebook.service")
|
||
|
|
server.wait_for_unit("mautrix-meta-instagram.service")
|
||
|
|
|
||
|
|
# homeserver
|
||
|
|
server.wait_for_unit("matrix-synapse.service")
|
||
|
|
|
||
|
|
server.wait_for_open_port(8008)
|
||
|
|
# Bridges only open the port after they contact the homeserver
|
||
|
|
server.wait_for_open_port(8009)
|
||
|
|
server.wait_for_open_port(8010)
|
||
|
|
|
||
|
|
with subtest("register user"):
|
||
|
|
client.succeed("register_user ${username} ${homeserverUrl} >&2")
|
||
|
|
|
||
|
|
with subtest("ensure messages can be exchanged"):
|
||
|
|
client.succeed("do_test ${username} ${facebookBotUsername} ${homeserverUrl} >&2")
|
||
|
|
client.succeed("do_test ${username} ${instagramBotUsername} ${homeserverUrl} >&2")
|
||
|
|
|
||
|
|
with subtest("ensure as_token and hs_token stays same after restart"):
|
||
|
|
generated_as_token_facebook = get_as_token_from(config_yaml)
|
||
|
|
generated_hs_token_facebook = get_hs_token_from(config_yaml)
|
||
|
|
|
||
|
|
generated_as_token_facebook_registration = get_as_token_from(registration_yaml)
|
||
|
|
generated_hs_token_facebook_registration = get_hs_token_from(registration_yaml)
|
||
|
|
|
||
|
|
# Indirectly checks the as token is not set to something like empty string or "null"
|
||
|
|
assert len(generated_as_token_facebook) > 20, f"as_token ({generated_as_token_facebook}) is too short, something went wrong"
|
||
|
|
assert len(generated_hs_token_facebook) > 20, f"hs_token ({generated_hs_token_facebook}) is too short, something went wrong"
|
||
|
|
|
||
|
|
assert generated_as_token_facebook == generated_as_token_facebook_registration, f"as_token should be the same in registration ({generated_as_token_facebook_registration}) and configuration ({generated_as_token_facebook}) files"
|
||
|
|
assert generated_hs_token_facebook == generated_hs_token_facebook_registration, f"hs_token should be the same in registration ({generated_hs_token_facebook_registration}) and configuration ({generated_hs_token_facebook}) files"
|
||
|
|
|
||
|
|
server.systemctl("restart mautrix-meta-facebook")
|
||
|
|
server.systemctl("restart mautrix-meta-instagram")
|
||
|
|
|
||
|
|
server.wait_for_open_port(8009)
|
||
|
|
server.wait_for_open_port(8010)
|
||
|
|
|
||
|
|
new_as_token_facebook = get_as_token_from(config_yaml)
|
||
|
|
new_hs_token_facebook = get_hs_token_from(config_yaml)
|
||
|
|
|
||
|
|
assert generated_as_token_facebook == new_as_token_facebook, f"as_token should stay the same after restart inside the configuration file (is: {new_as_token_facebook}, was: {generated_as_token_facebook})"
|
||
|
|
assert generated_hs_token_facebook == new_hs_token_facebook, f"hs_token should stay the same after restart inside the configuration file (is: {new_hs_token_facebook}, was: {generated_hs_token_facebook})"
|
||
|
|
|
||
|
|
new_as_token_facebook = get_as_token_from(registration_yaml)
|
||
|
|
new_hs_token_facebook = get_hs_token_from(registration_yaml)
|
||
|
|
|
||
|
|
assert generated_as_token_facebook == new_as_token_facebook, f"as_token should stay the same after restart inside the registration file (is: {new_as_token_facebook}, was: {generated_as_token_facebook})"
|
||
|
|
assert generated_hs_token_facebook == new_hs_token_facebook, f"hs_token should stay the same after restart inside the registration file (is: {new_hs_token_facebook}, was: {generated_hs_token_facebook})"
|
||
|
|
|
||
|
|
with subtest("ensure messages can be exchanged after restart"):
|
||
|
|
client.succeed("do_test ${username} ${instagramBotUsername} ${homeserverUrl} >&2")
|
||
|
|
client.succeed("do_test ${username} ${facebookBotUsername} ${homeserverUrl} >&2")
|
||
|
|
'';
|
||
|
|
})
|