Allow injecting tracer factory to the Query class

Trying to allow sphinx autodoc the query module on readthedocs.org
FrostyX · Mar 10, 2015 · 2e104a2 · 2e104a2
1 parent 2c22255
commit 2e104a2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_query.py b/tests/test_query.py
@@ -5,8 +5,7 @@
 class TestQuery(unittest.TestCase):
 
 	def setUp(self):
-		self.query = Query()
-		self.query._tracer = TracerMock()
+		self.query = Query(tracer=TracerMock)
 
 	def test_affected_applications(self):
 		apps_query = self.query.affected_applications()

diff --git a/tracer/query.py b/tracer/query.py
@@ -40,7 +40,8 @@ class Query(object):
 
 	"""
 
-	_tracer = Tracer(System.package_manager(), Rules, Applications, dump_memory)
+	def __init__(self, tracer=Tracer):
+		self._tracer = tracer(System.package_manager(), Rules, Applications, dump_memory)
 
 	def affected_applications(self, user=None):
 		"""