Small test change

PiperOrigin-RevId: 752387212

Author: lukebaumann

pathwaysutils/test/profiling_test.py

@@ -32,13 +32,14 @@ def setUp(self):
 
   @parameterized.parameters(8000, 1234)
   def test_collect_profile_port(self, port):
-    profiling.collect_profile(
+    result = profiling.collect_profile(
         port=port,
         duration_ms=1000,
         host="127.0.0.1",
         log_dir="gs://test_bucket/test_dir",
     )
 
+    self.assertTrue(result)
     self.mock_post.assert_called_once_with(
         f"http://127.0.0.1:{port}/profiling",
         json={
@@ -49,13 +50,14 @@ def test_collect_profile_port(self, port):
 
   @parameterized.parameters(1000, 1234)
   def test_collect_profile_duration_ms(self, duration_ms):
-    profiling.collect_profile(
+    result = profiling.collect_profile(
         port=8000,
         duration_ms=duration_ms,
         host="127.0.0.1",
         log_dir="gs://test_bucket/test_dir",
     )
 
+    self.assertTrue(result)
     self.mock_post.assert_called_once_with(
         "http://127.0.0.1:8000/profiling",
         json={
@@ -66,13 +68,14 @@ def test_collect_profile_duration_ms(self, duration_ms):
 
   @parameterized.parameters("127.0.0.1", "localhost", "192.168.1.1")
   def test_collect_profile_host(self, host):
-    profiling.collect_profile(
+    result = profiling.collect_profile(
         port=8000,
         duration_ms=1000,
         host=host,
         log_dir="gs://test_bucket/test_dir",
     )
 
+    self.assertTrue(result)
     self.mock_post.assert_called_once_with(
         f"http://{host}:8000/profiling",
         json={
@@ -87,10 +90,11 @@ def test_collect_profile_host(self, host):
       "gs://test_bucket3/test/log/dir",
   )
   def test_collect_profile_log_dir(self, log_dir):
-    profiling.collect_profile(
+    result = profiling.collect_profile(
         port=8000, duration_ms=1000, host="127.0.0.1", log_dir=log_dir
     )
 
+    self.assertTrue(result)
     self.mock_post.assert_called_once_with(
         "http://127.0.0.1:8000/profiling",
         json={
@@ -102,7 +106,7 @@ def test_collect_profile_log_dir(self, log_dir):
   @parameterized.parameters("/logs/test_log_dir", "relative_path/my_log_dir")
   def test_collect_profile_log_dir_error(self, log_dir):
     with self.assertRaises(ValueError):
-      profiling.collect_profile(
+      result = profiling.collect_profile(
           port=8000, duration_ms=1000, host="127.0.0.1", log_dir=log_dir
       )