Fix HLS tests

test/membrane_rtc_engine/hls_endpoint_test.exs

@@ -350,29 +350,20 @@ defmodule Membrane.RTC.HLSEndpointTest do
       case input_type do
         :single ->
           fn
-            %{segments: %{video_segments: 2}} = state ->
-              %{state | stopped?: true}
-
-            state ->
-              state
+            %{video_segments: video} ->
+              video > 1
           end
 
         :muxed ->
           fn
-            %{segments: %{muxed_segments: 2}} = state ->
-              %{state | stopped?: true}
-
-            state ->
-              state
+            %{muxed_segments: muxed} ->
+              muxed > 1
           end
 
         :multiple ->
           fn
-            %{segments: %{video_segments: 2, audio_segments: 2}} = state ->
-              %{state | stopped?: true}
-
-            state ->
-              state
+            %{video_segments: video, audio_segments: audio} ->
+              video > 1 and audio > 1
           end
       end
 

test/support/hls_storage.ex

@@ -36,12 +36,14 @@ defmodule Membrane.RTC.Engine.Support.HLSStorage do
     {result, file_storage} =
       FileStorage.store(parent_id, resource_name, content, metadata, context, state.file_storage)
 
-    if context.type == :segment do
-      send(state.pid, {context.type, resource_name})
-    end
+    state = %{state | file_storage: file_storage}
+
+    case context.type do
+      :segment ->
+        send(state.pid, {context.type, resource_name})
+        {result, state}
 
-    state =
-      if context.type == :manifest do
+      :manifest ->
         content = String.split(content, "\n")
 
         video_segments = Enum.count(content, &String.match?(&1, ~r/^video.*\.m4s/))
@@ -59,12 +61,11 @@ defmodule Membrane.RTC.Engine.Support.HLSStorage do
           {context.type, segments}
         )
 
-        update_segments(segments, state)
-      else
-        state
-      end
+        {result, update_segments(segments, state)}
 
-    {result, %{state | file_storage: file_storage}}
+      _other ->
+        {result, state}
+    end
   end
 
   defp update_segments(
@@ -85,7 +86,7 @@ defmodule Membrane.RTC.Engine.Support.HLSStorage do
       muxed_segments: muxed_segments
     }
 
-    %{state | segments: segments, stopped?: state.should_stop?.(state)}
+    %{state | segments: segments, stopped?: state.should_stop?.(segments)}
   end
 
   @impl true