nixos/tests/mediamtx.nix

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