From 8f15262c794b16ad330194f062091d0b42ac5a41 Mon Sep 17 00:00:00 2001
From: Dana Powers <dana.powers@rd.io>
Date: Sun, 3 Jan 2016 17:12:15 -0800
Subject: Assert Message value and (optional) key are bytes

---
 kafka/protocol/message.py | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'kafka/protocol')

diff --git a/kafka/protocol/message.py b/kafka/protocol/message.py
index f893912..430ecad 100644
--- a/kafka/protocol/message.py
+++ b/kafka/protocol/message.py
@@ -22,6 +22,8 @@ class Message(Struct):
     CODEC_SNAPPY = 0x02
 
     def __init__(self, value, key=None, magic=0, attributes=0, crc=0):
+        assert isinstance(value, bytes), 'value must be bytes'
+        assert key is None or isinstance(key, bytes), 'key must be bytes'
         self.crc = crc
         self.magic = magic
         self.attributes = attributes
-- 
cgit v1.2.1