Added AbstractAttractTest.attract property

tests/test_tasks.py

@@ -18,7 +18,7 @@ class TaskWorkflowTest(AbstractAttractTest):
     def setUp(self, **kwargs):
         AbstractAttractTest.setUp(self, **kwargs)
 
-        self.mngr = self.app.pillar_extensions['attract'].task_manager
+        self.mngr = self.attract.task_manager
         self.proj_id, self.project = self.ensure_project_exists()
 
         self.sdk_project = pillarsdk.Project(pillar.tests.mongo_to_sdk(self.project))
@@ -106,7 +106,7 @@ class TaskWorkflowTest(AbstractAttractTest):
                                 _etag=task._etag)
 
         with self.app.test_request_context():
-            acts = self.app.pillar_extensions['attract'].activities_for_node(task['_id'])
+            acts = self.attract.activities_for_node(task['_id'])
         self.assertEqual(2, acts['_meta']['total'])  # Creation + edit
 
         edit_act = acts['_items'][1]