protobuf test: don't match on exception error string since it might change

edenhill · edenhill · commit 102c002a6f41 · 2021-01-19T16:15:44.000+01:00
diff --git a/tests/integration/schema_registry/test_proto_serializers.py b/tests/integration/schema_registry/test_proto_serializers.py
@@ -16,7 +16,7 @@
 #
 import pytest
 
-from confluent_kafka import TopicPartition, KafkaException
+from confluent_kafka import TopicPartition, KafkaException, KafkaError
 from confluent_kafka.error import ConsumeError
 from confluent_kafka.schema_registry.protobuf import ProtobufSerializer, ProtobufDeserializer
 from .gen import metadata_proto_pb2
@@ -145,6 +145,6 @@ def dr(err, msg):
                      partition=0)
     producer.flush()
 
-    with pytest.raises(ConsumeError,
-                       match="Error parsing message"):
+    with pytest.raises(ConsumeError) as e:
         consumer.poll()
+    assert e.value.code == KafkaError._KEY_DESERIALIZATION
