Skip to content

Commit 1fa9c1f

Browse files
author
Matt Howlett
authored
fix test_message_serializer.py (#1061)
* fix test_message_serializer.py * use version_info etc * flake8 wants a blank line
1 parent b7f8dce commit 1fa9c1f

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

tests/avro/test_message_serializer.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,12 +23,16 @@
2323
import struct
2424

2525
import unittest
26-
from unittest.mock import patch
26+
from sys import version_info
27+
28+
if version_info >= (3,):
29+
from unittest.mock import patch
2730

2831
from tests.avro import data_gen
2932
from confluent_kafka.avro.serializer.message_serializer import MessageSerializer
3033
from tests.avro.mock_schema_registry_client import MockSchemaRegistryClient
3134
from confluent_kafka import avro
35+
from pyflakes.test.harness import skipIf
3236

3337

3438
class TestMessageSerializer(unittest.TestCase):
@@ -76,6 +80,8 @@ def test_encode_record_with_schema(self):
7680
message = self.ms.encode_record_with_schema(topic, basic, record)
7781
self.assertMessageIsSame(message, record, schema_id)
7882

83+
@skipIf(version_info < (3,),
84+
'unittest.mock.patch not available in Python 2')
7985
def test_encode_record_with_schema_sets_writers_cache_once(self):
8086
topic = 'test'
8187
basic = avro.loads(data_gen.BASIC_SCHEMA)

0 commit comments

Comments
 (0)