commit 2b1402257f165779db5c22f082def0425f1cc71f
author Pat Ferate <pferate@gmail.com> Tue Mar 03 18:05:11 2015 -0800
committer Pat Ferate <pferate+github@gmail.com> Mon Mar 09 10:34:04 2015 -0700
tree 481696831c0cfed24d79ecf27a047a147656c48f
parent c9abbbdecaa84114aa9b3689702690e507a9fe03

Updating string buffers to byte buffers

tests/test_discovery.py

diff --git a/tests/test_discovery.py b/tests/test_discovery.py
index 0a71cfc..458e29d 100644
--- a/tests/test_discovery.py
+++ b/tests/test_discovery.py
@@ -596,7 +596,7 @@
     zoo = build('zoo', 'v1', http=self.http)
     request = zoo.animals().crossbreed(media_body=datafile('small.png'))
     self.assertEquals('image/png', request.headers['content-type'])
-    self.assertEquals('PNG', request.body[1:4])
+    self.assertEquals(b'PNG', request.body[1:4])
 
   def test_simple_media_raise_correct_exceptions(self):
     self.http = HttpMock(datafile('zoo.json'), {'status': '200'})
@@ -620,7 +620,7 @@
 
     request = zoo.animals().insert(media_body=datafile('small.png'))
     self.assertEquals('image/png', request.headers['content-type'])
-    self.assertEquals('PNG', request.body[1:4])
+    self.assertEquals(b'PNG', request.body[1:4])
     assertUrisEqual(self,
         'https://www.googleapis.com/upload/zoo/v1/animals?uploadType=media&alt=json',
         request.uri)
@@ -852,7 +852,7 @@
     zoo = build('zoo', 'v1', http=self.http)
 
     # Set up a seekable stream and try to upload in single chunk.
-    fd = BytesIO('01234"56789"')
+    fd = BytesIO(b'01234"56789"')
     media_upload = MediaIoBaseUpload(
         fd=fd, mimetype='text/plain', chunksize=-1, resumable=True)
 
@@ -876,7 +876,7 @@
     zoo = build('zoo', 'v1', http=self.http)
 
     # Set up a seekable stream and try to upload in chunks.
-    fd = BytesIO('0123456789')
+    fd = BytesIO(b'0123456789')
     media_upload = MediaIoBaseUpload(
         fd=fd, mimetype='text/plain', chunksize=5, resumable=True)
 
@@ -892,7 +892,7 @@
     try:
       body = request.execute(http=http)
     except HttpError as e:
-      self.assertEqual('01234', e.content)
+      self.assertEqual(b'01234', e.content)
 
   def test_resumable_media_handle_uploads_of_unknown_size(self):
     http = HttpMockSequence([
@@ -1001,7 +1001,7 @@
     self.http = HttpMock(datafile('zoo.json'), {'status': '200'})
     zoo = build('zoo', 'v1', http=self.http)
 
-    fd = BytesIO('data goes here')
+    fd = BytesIO(b'data goes here')
 
     # Create an upload that doesn't know the full size of the media.
     upload = MediaIoBaseUpload(
@@ -1025,7 +1025,7 @@
     zoo = build('zoo', 'v1', http=self.http)
 
     # Create an upload that doesn't know the full size of the media.
-    fd = BytesIO('data goes here')
+    fd = BytesIO(b'data goes here')
 
     upload = MediaIoBaseUpload(
         fd=fd, mimetype='image/png', chunksize=500, resumable=True)