import ./make-test-python.nix ( { pkgs, lib, ... }: { name = "mediamtx"; meta.maintainers = with lib.maintainers; [ fpletz ]; nodes = { machine = { services.mediamtx = { enable = true; settings = { metrics = true; paths.all.source = "publisher"; }; }; systemd.services.rtmp-publish = { description = "Publish an RTMP stream to mediamtx"; after = [ "mediamtx.service" ]; bindsTo = [ "mediamtx.service" ]; wantedBy = [ "multi-user.target" ]; serviceConfig = { DynamicUser = true; Restart = "on-failure"; RestartSec = "1s"; TimeoutStartSec = "10s"; ExecStart = "${lib.getBin pkgs.ffmpeg-headless}/bin/ffmpeg -re -f lavfi -i smptebars=size=800x600:rate=10 -c libx264 -f flv rtmp://localhost:1935/test"; }; }; systemd.services.rtmp-receive = { description = "Receive an RTMP stream from mediamtx"; after = [ "rtmp-publish.service" ]; bindsTo = [ "rtmp-publish.service" ]; wantedBy = [ "multi-user.target" ]; serviceConfig = { DynamicUser = true; Restart = "on-failure"; RestartSec = "1s"; TimeoutStartSec = "10s"; ExecStart = "${lib.getBin pkgs.ffmpeg-headless}/bin/ffmpeg -y -re -i rtmp://localhost:1935/test -f flv /dev/null"; }; }; }; }; testScript = '' start_all() machine.wait_for_unit("mediamtx.service") machine.wait_for_unit("rtmp-publish.service") machine.sleep(10) machine.wait_for_unit("rtmp-receive.service") machine.wait_for_open_port(9998) machine.succeed("curl http://localhost:9998/metrics | grep '^rtmp_conns.*state=\"publish\".*1$'") machine.succeed("curl http://localhost:9998/metrics | grep '^rtmp_conns.*state=\"read\".*1$'") ''; } )