Add test case for handle batch

upmaru · Oct 29, 2024 · 4d0f688 · 4d0f688
1 parent afc0487
commit 4d0f688
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/test/uplink/metrics/pipeline_test.exs b/test/uplink/metrics/pipeline_test.exs
@@ -116,4 +116,16 @@ defmodule Uplink.Metrics.PipelineTest do
 
     assert not is_nil(load_15)
   end
+
+  test "handle batch", %{
+    message_with_previous_cpu_metric: message1,
+    message_with_previous_network_metric: message2
+  } do
+    ref = Broadway.test_batch(Uplink.Metrics.Pipeline, [message1, message2])
+
+    assert_receive {:ack, ^ref, successful, failed}, 10_000
+
+    assert length(successful) == 2
+    assert length(failed) == 0
+  end
 end