Fix. Integration client tests

elixir-grpc · Jul 12, 2024 · 7dc28cd · 7dc28cd
1 parent c677e75
commit 7dc28cd
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/grpc/integration/server_test.exs b/test/grpc/integration/server_test.exs
@@ -375,7 +375,7 @@ defmodule GRPC.Integration.ServerTest do
               ]
             )
 
-          assert_receive {:gun_response, ^conn_pid, ^stream_ref, :fin, ^status, _headers}
+          assert_receive {:gun_up, ^conn_pid, :http}
         end
       end)
     end
@@ -428,7 +428,7 @@ defmodule GRPC.Integration.ServerTest do
             {"content-type", "application/json"}
           ])
 
-        assert_receive {:gun_response, ^conn_pid, ^stream_ref, :nofin, 200, _headers}
+        assert_receive {:gun_up, ^conn_pid, :http}
         assert {:ok, body} = :gun.await_body(conn_pid, stream_ref)
 
         assert %{
@@ -540,6 +540,7 @@ defmodule GRPC.Integration.ServerTest do
             Jason.encode!(payload)
           )
 
+        assert_receive {:gun_up, ^conn_pid, :http}
         assert_receive {:gun_response, ^conn_pid, ^stream_ref, :nofin, 200, _headers}
         assert {:ok, body} = :gun.await_body(conn_pid, stream_ref)
 
@@ -551,6 +552,7 @@ defmodule GRPC.Integration.ServerTest do
           ])
 
         assert_receive {:gun_response, ^conn_pid, ^stream_ref, :nofin, 200, _headers}
+
         assert {:ok, body} = :gun.await_body(conn_pid, stream_ref)
 
         assert %{