[py3] fix TypeError: a bytes-like object is required, not 'str'

Use `six.PY3` to force non-unicode strings in Python 2.

Author: MrSaints

jaeger_client/constants.py

@@ -22,6 +22,7 @@
 
 from . import __version__
 
+import six
 
 # Max number of bits to use when generating random ID
 MAX_ID_BITS = 64
@@ -33,10 +34,10 @@
 DEFAULT_FLUSH_INTERVAL = 1
 
 # Name of the HTTP header used to encode trace ID
-TRACE_ID_HEADER = b'uber-trace-id'
+TRACE_ID_HEADER = 'uber-trace-id' if six.PY3 else b'uber-trace-id'
 
 # Prefix for HTTP headers used to record baggage items
-BAGGAGE_HEADER_PREFIX = b'uberctx-'
+BAGGAGE_HEADER_PREFIX = 'uberctx-' if six.PY3 else b'uberctx-'
 
 # The name of HTTP header or a TextMap carrier key which, if found in the
 # carrier, forces the trace to be sampled as "debug" trace. The value of the

tests/test_TUDPTransport.py

@@ -39,7 +39,7 @@ def test_constructor_nonblocking(self):
         assert t.transport_sock.gettimeout() == 0
 
     def test_write(self):
-        self.t.write('hello')
+        self.t.write(b'hello')
 
     def test_isopen_when_open(self):
         assert self.t.isOpen() == True
@@ -52,4 +52,4 @@ def test_close(self):
         self.t.close()
         with self.assertRaises(Exception):
             # Something bad should happen if we send on a closed socket..
-            self.t.write('hello')
+            self.t.write(b'hello')