From 99d42bb05e003c6f6aa1ea51a470191a812904b7 Mon Sep 17 00:00:00 2001
From: "dom.song" <dom.song@amperecomputing.com>
Date: Thu, 6 Jul 2023 16:41:05 +0800
Subject: [PATCH] tests: Correctly parse number of host cpus > 255

test_cpu_affinity needs the number of host CPUs. Since it is possible
for the host to have more than 255 CPUs; increase the size of the
datatype used for parsing the string to accomodate this.

Signed-off-by: dom.song <dom.song@amperecomputing.com>
---
 tests/integration.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/integration.rs b/tests/integration.rs
index 18ead1135..d637786e2 100644
--- a/tests/integration.rs
+++ b/tests/integration.rs
@@ -2439,7 +2439,7 @@ mod common_parallel {
         assert!(
             String::from_utf8_lossy(&host_cpus_count.stdout)
                 .trim()
-                .parse::<u8>()
+                .parse::<u16>()
                 .unwrap_or(0)
                 >= 4
         );