test_task: fixed import error, made all modules from x import * safe

5 files changed, 15 insertions, 1 deletions

diff --git a/lib/git/async/channel.py b/lib/git/async/channel.py
index 9b019707..ae476cda 100644
--- a/lib/git/async/channel.py
+++ b/lib/git/async/channel.py
@@ -13,6 +13,9 @@ from util import (
 from time import time
 import sys
 
+__all__ = ('Channel', 'SerialChannel', 'Writer', 'CallbackWriter', 'Reader',
+			'CallbackReader', 'mkchannel', 'ReadOnly')
+
 #{ Classes 
 class Channel(object):
 	"""A channel is similar to a file like object. It has a write end as well as one or
diff --git a/lib/git/async/graph.py b/lib/git/async/graph.py
index 9ee0e891..4e14c81e 100644
--- a/lib/git/async/graph.py
+++ b/lib/git/async/graph.py
@@ -1,5 +1,7 @@
 """Simplistic implementation of a graph"""
 
+__all__ = ('Node', 'Graph')
+
 class Node(object):
 	"""A Node in the graph. They know their neighbours, and have an id which should 
 	resolve into a string"""
diff --git a/lib/git/async/pool.py b/lib/git/async/pool.py
index 7ee3e8eb..cf14e47b 100644
--- a/lib/git/async/pool.py
+++ b/lib/git/async/pool.py
@@ -30,6 +30,8 @@ from time import sleep
 import new
 
 
+__all__ = ('PoolReader', 'Pool', 'ThreadPool')
+
 class PoolReader(CallbackReader):
 	"""A reader designed to read from channels which take part in pools
 	It acts like a handle to the underlying task in the pool."""
diff --git a/lib/git/async/task.py b/lib/git/async/task.py
index 10b22649..d7f331b7 100644
--- a/lib/git/async/task.py
+++ b/lib/git/async/task.py
@@ -6,8 +6,11 @@ import weakref
 import sys
 import new
 
+__all__ = ('OutputChannelTask', 'ThreadTaskBase', 'InputIteratorTaskBase', 
+			'InputIteratorThreadTask', 'InputChannelTask')
+
 class OutputChannelTask(Node):
-	"""Abstracts a named task as part of a set of interdependent tasks, which contains 
+	"""Abstracts a named task, which contains 
 	additional information on how the task should be queued and processed.
 	
 	Results of the item processing are sent to a write channel, which is to be 
diff --git a/lib/git/async/thread.py b/lib/git/async/thread.py
index afe0d79d..96b4f0c4 100644
--- a/lib/git/async/thread.py
+++ b/lib/git/async/thread.py
@@ -7,6 +7,10 @@ import Queue
 
 import sys
 
+__all__ = ('do_terminate_threads', 'terminate_threads', 'TerminatableThread', 
+			'WorkerThread') 
+		
+
 #{ Decorators
 
 def do_terminate_threads(whitelist=list()):