update reader coercion in client tests

inhabitedtype · Jun 3, 2021 · 203f06c · 203f06c
1 parent 90de89e
commit 203f06c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib_test/test_client_connection.ml b/lib_test/test_client_connection.ml
@@ -41,7 +41,7 @@ let read_response t r =
 
 let reader_ready t =
   Alcotest.check read_operation "Reader is ready"
-    `Read (next_read_operation t :> [`Close | `Read | `Yield | `Upgrade]);
+    `Read (next_read_operation t :> Read_operation.t);
 ;;
 
 let write_string ?(msg="output written") t str =
@@ -69,7 +69,7 @@ let writer_closed t =
 
 let connection_is_shutdown t =
   Alcotest.check read_operation "Reader is closed"
-    `Close (next_read_operation t :> [`Close | `Read | `Yield | `Upgrade]);
+    `Close (next_read_operation t :> Read_operation.t);
   writer_closed t;
 ;;