From 2f9a997a8d31c4ccc30cbecd8a08961347f890f5 Mon Sep 17 00:00:00 2001
From: Ingo Oppermann <ingo@datarhei.com>
Date: Tue, 25 Apr 2023 21:59:22 +0200
Subject: [PATCH] Add test for process limits

---
 restream/restream_test.go | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/restream/restream_test.go b/restream/restream_test.go
index bd615f15..c1134bc8 100644
--- a/restream/restream_test.go
+++ b/restream/restream_test.go
@@ -1217,3 +1217,26 @@ func TestProcessLogPattern(t *testing.T) {
 	err = rs.StopProcess("process")
 	require.NoError(t, err)
 }
+
+func TestProcessLimit(t *testing.T) {
+	rsi, err := getDummyRestreamer(nil, nil, nil, nil)
+	require.NoError(t, err)
+
+	process := getDummyProcess()
+	process.LimitCPU = 61
+	process.LimitMemory = 42
+	process.Autostart = false
+
+	err = rsi.AddProcess(process)
+	require.NoError(t, err)
+
+	rs := rsi.(*restream)
+
+	task, ok := rs.tasks[process.ID]
+	require.True(t, ok)
+
+	status := task.ffmpeg.Status()
+
+	require.Equal(t, float64(61), status.CPU.Limit)
+	require.Equal(t, uint64(42), status.Memory.Limit)
+}