From e653bebf51a60a68c1cdf759666e22d23c962287 Mon Sep 17 00:00:00 2001
From: Tomasz Godzik <tgodzik@virtuslab.com>
Date: Thu, 7 Mar 2024 15:27:15 +0100
Subject: [PATCH] improvement: Add method to test dependency modules

---
 bsp-testkit/client/TestClient.kt | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/bsp-testkit/client/TestClient.kt b/bsp-testkit/client/TestClient.kt
index 72f7d33..e5a1b4e 100644
--- a/bsp-testkit/client/TestClient.kt
+++ b/bsp-testkit/client/TestClient.kt
@@ -209,6 +209,18 @@ open class TestClient(
     }
   }
 
+  fun testDependencyModule(
+      timeout: Duration,
+      params: DependencyModulesParams,
+      expectedResult: DependencyModulesResult,
+  ) {
+    val transformedParams = applyJsonTransform(params)
+    test(timeout) { session, _ ->
+      val result = session.server.buildTargetDependencyModules(transformedParams).await()
+      assertJsonEquals(expectedResult, result)
+    }
+  }
+
   /**
    * Simulates a typical workflow
    */