Refix test

6 years ago · 6ece2af4f1
parent 72a736fe07
commit 6ece2af4f1
1 changed files with 1 additions and 1 deletions
--- a/src/python/grpcio_tests/tests/unit/_metadata_flags_test.py
+++ b/src/python/grpcio_tests/tests/unit/_metadata_flags_test.py
@ -201,7 +201,7 @@ class MetadataFlagsTest(unittest.TestCase):
            fn(channel, wait_for_ready)
            self.fail("The Call should fail")
        except BaseException as e:  # pylint: disable=broad-except
-            self.assertIn('StatusCode.UNAVAILABLE', str(e))
+            self.assertIs(grpc.StatusCode.UNAVAILABLE, e.code())

    def test_call_wait_for_ready_default(self):
        for perform_call in _ALL_CALL_CASES: