test: Add tests for wrappers to integration tests

Signed-off-by: Si Beaumont <[email protected]>

Author: simonjbeaumont

Tests/GRPCTests/AsyncAwaitSupport/AsyncIntegrationTests.swift

@@ -73,6 +73,13 @@ final class AsyncIntegrationTests: GRPCTestCase {
     }
   }
 
+  func testUnaryWrapper() {
+    XCTAsyncTest {
+      let response = try await self.echo.get(.with { $0.text = "hello" })
+      XCTAssertEqual(response.text, "Swift echo get: hello")
+    }
+  }
+
   func testClientStreaming() {
     XCTAsyncTest {
       let collect = self.echo.makeCollectCall()
@@ -96,15 +103,28 @@ final class AsyncIntegrationTests: GRPCTestCase {
     }
   }
 
+  func testClientStreamingWrapper() {
+    XCTAsyncTest {
+      let requests: [Echo_EchoRequest] = [
+        .with { $0.text = "boyle" },
+        .with { $0.text = "jeffers" },
+        .with { $0.text = "holt" },
+      ]
+
+      let response = try await self.echo.collect(requests)
+      XCTAssertEqual(response.text, "Swift echo collect: boyle jeffers holt")
+    }
+  }
+
   func testServerStreaming() {
     XCTAsyncTest {
       let expand = self.echo.makeExpandCall(.with { $0.text = "boyle jeffers holt" })
 
       let initialMetadata = try await expand.initialMetadata
       initialMetadata.assertFirst("200", forName: ":status")
 
-      let respones = try await expand.responses.map { $0.text }.collect()
-      XCTAssertEqual(respones, [
+      let responses = try await expand.responses.map { $0.text }.collect()
+      XCTAssertEqual(responses, [
         "Swift echo expand (0): boyle",
         "Swift echo expand (1): jeffers",
         "Swift echo expand (2): holt",
@@ -118,6 +138,18 @@ final class AsyncIntegrationTests: GRPCTestCase {
     }
   }
 
+  func testServerStreamingWrapper() {
+    XCTAsyncTest {
+      let responseStream = self.echo.expand(.with { $0.text = "boyle jeffers holt" })
+      let responses = try await responseStream.map { $0.text }.collect()
+      XCTAssertEqual(responses, [
+        "Swift echo expand (0): boyle",
+        "Swift echo expand (1): jeffers",
+        "Swift echo expand (2): holt",
+      ])
+    }
+  }
+
   func testBidirectionalStreaming() {
     XCTAsyncTest {
       let update = self.echo.makeUpdateCall()
@@ -145,6 +177,24 @@ final class AsyncIntegrationTests: GRPCTestCase {
       XCTAssertTrue(status.isOk)
     }
   }
+
+  func testBidirectionalStreamingWrapper() {
+    XCTAsyncTest {
+      let requests: [Echo_EchoRequest] = [
+        .with { $0.text = "boyle" },
+        .with { $0.text = "jeffers" },
+        .with { $0.text = "holt" },
+      ]
+
+      let responseStream = self.echo.update(requests)
+      let responses = try await responseStream.map { $0.text }.collect()
+      XCTAssertEqual(responses, [
+        "Swift echo update (0): boyle",
+        "Swift echo update (1): jeffers",
+        "Swift echo update (2): holt",
+      ])
+    }
+  }
 }
 
 extension HPACKHeaders {