Fix python3 compatibility in dulwich.greenthreads, enable travis testing. Fixes:...

[jelmer/dulwich.git] / dulwich / tests / test_greenthreads.py

diff --git a/dulwich/tests/test_greenthreads.py b/dulwich/tests/test_greenthreads.py
index 49681db5eb99b219d8acd7e0a5c879008a64c6d4..33ae0599c634ce91aa1ab73017ecb71455a0c69e 100644 (file)
--- a/dulwich/tests/test_greenthreads.py
+++ b/dulwich/tests/test_greenthreads.py
@@ -22,6 +22,7 @@
 import time
 
 from dulwich.tests import (
+    skipIf,
     TestCase,
     )
 from dulwich.object_store import (
@@ -35,8 +36,6 @@ from dulwich.objects import (
     parse_timezone,
     )
 
-from unittest import skipIf
-
 try:
     import gevent
     gevent_support = True
@@ -52,14 +51,14 @@ if gevent_support:
 skipmsg = "Gevent library is not installed"
 
 def create_commit(marker=None):
-    blob = Blob.from_string('The blob content %s' % marker)
+    blob = Blob.from_string(b'The blob content %s' % marker)
     tree = Tree()
-    tree.add("thefile %s" % marker, 0100644, blob.id)
+    tree.add(b"thefile %s" % marker, 0o100644, blob.id)
     cmt = Commit()
     cmt.tree = tree.id
-    cmt.author = cmt.committer = "John Doe <john@doe.net>"
-    cmt.message = "%s" % marker
-    tz = parse_timezone('-0200')[0]
+    cmt.author = cmt.committer = b"John Doe <john@doe.net>"
+    cmt.message = marker
+    tz = parse_timezone(b'-0200')[0]
     cmt.commit_time = cmt.author_time = int(time.time())
     cmt.commit_timezone = cmt.author_timezone = tz
     return cmt, tree, blob
@@ -67,8 +66,8 @@ def create_commit(marker=None):
 
 def init_store(store, count=1):
     ret = []
-    for i in xrange(0, count):
-        objs = create_commit(marker=i)
+    for i in range(0, count):
+        objs = create_commit(marker=("%d" % i).encode('ascii'))
         for obj in objs:
             ret.append(obj)
             store.add_object(obj)
@@ -128,7 +127,7 @@ class TestGreenThreadsMissingObjectFinder(TestCase):
         self.assertEqual(len(finder.objects_to_send), self.cmt_amount)
 
         finder = GreenThreadsMissingObjectFinder(self.store,
-                                             wants[0:self.cmt_amount/2],
+                                             wants[0:int(self.cmt_amount/2)],
                                              wants)
         # sha_done will contains commit id and sha of blob refered in tree
         self.assertEqual(len(finder.sha_done), (self.cmt_amount/2)*2)