From 4403e8a41a0ab970e61f3814bc7f3e89aff5e33d Mon Sep 17 00:00:00 2001
From: Jonathan Lennox <jonathan.lennox@8x8.com>
Date: Tue, 15 Oct 2024 17:08:39 -0400
Subject: [PATCH] Command-line args for SocketPoolTest main().

---
 .../kotlin/org/ice4j/socket/SocketPoolTest.kt | 30 +++++++++++++++----
 1 file changed, 25 insertions(+), 5 deletions(-)

diff --git a/src/test/kotlin/org/ice4j/socket/SocketPoolTest.kt b/src/test/kotlin/org/ice4j/socket/SocketPoolTest.kt
index cf0fb961..f60b8756 100644
--- a/src/test/kotlin/org/ice4j/socket/SocketPoolTest.kt
+++ b/src/test/kotlin/org/ice4j/socket/SocketPoolTest.kt
@@ -177,11 +177,15 @@ class SocketPoolTest : ShouldSpec() {
     }
 
     companion object {
-        private fun sendTimeOnAllSockets(pool: SocketPool, numThreads: Int = pool.numSockets): Duration {
+        private fun sendTimeOnAllSockets(
+            pool: SocketPool,
+            numThreads: Int = pool.numSockets,
+            numPackets: Int = Sender.NUM_PACKETS
+        ): Duration {
             val threads = mutableListOf<Thread>()
             Sender.reset(numThreads)
             repeat(numThreads) {
-                val thread = Thread(Sender(Sender.NUM_PACKETS / numThreads, pool, loopbackDiscard))
+                val thread = Thread(Sender(numPackets / numThreads, pool, loopbackDiscard))
                 threads.add(thread)
                 thread.start()
             }
@@ -189,9 +193,14 @@ class SocketPoolTest : ShouldSpec() {
             return Sender.elapsed
         }
 
-        fun testSendingOnce(numSockets: Int, numThreads: Int, warmup: Boolean = false) {
+        private fun testSendingOnce(
+            numSockets: Int,
+            numThreads: Int,
+            numPackets: Int = Sender.NUM_PACKETS,
+            warmup: Boolean = false
+        ) {
             val pool = SocketPool(loopbackAny, numSockets)
-            val elapsed = sendTimeOnAllSockets(pool, numThreads)
+            val elapsed = sendTimeOnAllSockets(pool, numThreads, numPackets)
             if (!warmup) {
                 println(
                     "Send ${Sender.NUM_PACKETS} packets on $numSockets sockets on $numThreads threads " +
@@ -229,7 +238,18 @@ class SocketPoolTest : ShouldSpec() {
 
         @JvmStatic
         fun main(args: Array<String>) {
-            testSending()
+            if (args.size >= 2) {
+                val numThreads = args[0].toInt()
+                val numSockets = args[1].toInt()
+                val numPackets = if (args.size > 2) {
+                    args[2].toInt()
+                } else {
+                    Sender.NUM_PACKETS
+                }
+                testSendingOnce(numThreads = numThreads, numSockets = numSockets, numPackets = numPackets)
+            } else {
+                testSending()
+            }
         }
     }
 }