nixos/tests/mediamtx.nix

   1 import ./make-test-python.nix ({ pkgs, lib, ...} :
   2
   3 {
   4   name = "mediamtx";
   5   meta.maintainers = with lib.maintainers; [ fpletz ];
   6
   7   nodes = {
   8     machine = { config, ... }: {
   9       services.mediamtx = {
  10         enable = true;
  11         settings = {
  12           metrics = true;
  13           paths.all.source = "publisher";
  14         };
  15       };
  16
  17       systemd.services.rtmp-publish = {
  18         description = "Publish an RTMP stream to mediamtx";
  19         after = [ "mediamtx.service" ];
  20         bindsTo = [ "mediamtx.service" ];
  21         wantedBy = [ "multi-user.target" ];
  22         serviceConfig = {
  23           DynamicUser = true;
  24           Restart = "on-failure";
  25           RestartSec = "1s";
  26           TimeoutStartSec = "10s";
  27           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";
  28         };
  29       };
  30
  31       systemd.services.rtmp-receive = {
  32         description = "Receive an RTMP stream from mediamtx";
  33         after = [ "rtmp-publish.service" ];
  34         bindsTo = [ "rtmp-publish.service" ];
  35         wantedBy = [ "multi-user.target" ];
  36         serviceConfig = {
  37           DynamicUser = true;
  38           Restart = "on-failure";
  39           RestartSec = "1s";
  40           TimeoutStartSec = "10s";
  41           ExecStart = "${lib.getBin pkgs.ffmpeg-headless}/bin/ffmpeg -y -re -i rtmp://localhost:1935/test -f flv /dev/null";
  42         };
  43       };
  44     };
  45   };
  46
  47   testScript = ''
  48     start_all()
  49
  50     machine.wait_for_unit("mediamtx.service")
  51     machine.wait_for_unit("rtmp-publish.service")
  52     machine.wait_for_unit("rtmp-receive.service")
  53     machine.wait_for_open_port(9998)
  54     machine.succeed("curl http://localhost:9998/metrics | grep '^rtmp_conns.*state=\"publish\".*1$'")
  55     machine.succeed("curl http://localhost:9998/metrics | grep '^rtmp_conns.*state=\"read\".*1$'")
  56   '';
  57 })