diff options
Diffstat (limited to 'test/test_protocol.py')
| -rw-r--r-- | test/test_protocol.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/test/test_protocol.py b/test/test_protocol.py index ac7bea6..368c2d0 100644 --- a/test/test_protocol.py +++ b/test/test_protocol.py @@ -780,11 +780,11 @@ class TestProtocol(unittest.TestCase): @contextmanager def mock_create_message_fns(self): import kafka.protocol - with patch.object(kafka.protocol, "create_message", + with patch.object(kafka.protocol.legacy, "create_message", return_value=sentinel.message): - with patch.object(kafka.protocol, "create_gzip_message", + with patch.object(kafka.protocol.legacy, "create_gzip_message", return_value=sentinel.gzip_message): - with patch.object(kafka.protocol, "create_snappy_message", + with patch.object(kafka.protocol.legacy, "create_snappy_message", return_value=sentinel.snappy_message): yield |