Fixed a few more encoding issues

Bytes should always be returned from the streams, so the tests
should be checking against byte strings instead of text strings.

This also fixes the `sha_iter` as it relied on the Python 2
`iterkeys` which has been renamed to `keys` in Python 3.

Author: kevin-brown

gitdb/db/loose.py

@@ -52,6 +52,7 @@
 )
 
 from gitdb.utils.compat import MAXSIZE
+from gitdb.utils.encoding import force_bytes
 
 import tempfile
 import mmap
@@ -116,7 +117,7 @@ def partial_to_complete_sha_hex(self, partial_hexsha):
         :raise BadObject: """
         candidate = None
         for binsha in self.sha_iter():
-            if bin_to_hex(binsha).startswith(partial_hexsha):
+            if bin_to_hex(binsha).startswith(force_bytes(partial_hexsha)):
                 # it can't ever find the same object twice
                 if candidate is not None:
                     raise AmbiguousObjectName(partial_hexsha)

gitdb/db/mem.py

@@ -86,7 +86,10 @@ def size(self):
         return len(self._cache)
 
     def sha_iter(self):
-        return self._cache.iterkeys()
+        try:
+            return self._cache.iterkeys()
+        except AttributeError:
+            return self._cache.keys()
 
 
     #{ Interface

gitdb/test/db/lib.py

@@ -21,6 +21,7 @@
 
 from gitdb.exc import BadObject
 from gitdb.typ import str_blob_type
+from gitdb.utils.encoding import force_bytes
 
 from async import IteratorReader
 
@@ -97,7 +98,7 @@ def _assert_object_writing(self, db):
                     assert info.size == len(data)
 
                     ostream = db.stream(sha)
-                    assert ostream.read() == data
+                    assert ostream.read() == force_bytes(data)
                     assert ostream.type == str_blob_type
                     assert ostream.size == len(data)
                 else: