diff --git a/test/vtgatev3_test.py b/test/vtgatev3_test.py index ec6879a75c4..6c352d1abb8 100755 --- a/test/vtgatev3_test.py +++ b/test/vtgatev3_test.py @@ -25,6 +25,7 @@ import environment import keyspace_util import utils +from protocols_flavor import protocols_flavor from vtdb import dbexceptions from vtdb import vtgate_cursor @@ -669,7 +670,7 @@ def test_user(self): self.fail('Execute went through') except dbexceptions.DatabaseError as e: s = str(e) - self.assertIn('DeadlineExceeded', s) + self.assertIn(protocols_flavor().rpc_timeout_message(), s) # test directive timeout longer than the query time cursor.execute('SELECT /*vt+ QUERY_TIMEOUT_MS=2000 */ SLEEP(1)', {})