Add a test case for parameterized async tests using IsolatedAsyncioTestCase.

yilei · copybara-github · commit e5f96d9056ef · 2023-08-31T17:09:46.000-07:00
PiperOrigin-RevId: 561801233
diff --git a/absl/testing/tests/parameterized_test.py b/absl/testing/tests/parameterized_test.py
@@ -998,6 +998,35 @@ def test_subclass_inherits_superclass_test_params_reprs(self):
     )
 
 
+# IsolatedAsyncioTestCase is only available in Python 3.8+.
+if sys.version_info[:2] >= (3, 8):
+
+  async def mult(x: float, y: float) -> float:
+    return x * y
+
+  class AsyncTest(unittest.IsolatedAsyncioTestCase, parameterized.TestCase):
+
+    def setUp(self):
+      super().setUp()
+      self.verify_ran = False
+
+    def tearDown(self):
+      super().tearDown()
+
+      # We need the additional check here because originally the test function
+      # would run, but a coroutine results from the execution and is never
+      # awaited, so it looked like a successful test run when in fact the
+      # internal test logic never executed.  If you remove the check for
+      # coroutine and run_until_complete, then set the parameters to fail they
+      # never will.
+      self.assertTrue(self.verify_ran)
+
+    @parameterized.parameters((1, 2, 2), (2, 2, 4), (3, 2, 6))
+    async def test_multiply_expected_matches_actual(self, x, y, expected):
+      self.assertEqual(await mult(x, y), expected)
+      self.verify_ran = True
+
+
 def _decorate_with_side_effects(func, self):
   self.sideeffect = True
   func(self)
