File tree Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Original file line number Diff line number Diff line change 2323import struct
2424
2525import 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
2831from tests .avro import data_gen
2932from confluent_kafka .avro .serializer .message_serializer import MessageSerializer
3033from tests .avro .mock_schema_registry_client import MockSchemaRegistryClient
3134from confluent_kafka import avro
35+ from pyflakes .test .harness import skipIf
3236
3337
3438class 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 )
You can’t perform that action at this time.
0 commit comments