Added AbstractAttractTest.attract property

tests/test_shots.py

@@ -18,8 +18,8 @@ class AbstractShotTest(AbstractAttractTest):
     def setUp(self, **kwargs):
         AbstractAttractTest.setUp(self, **kwargs)
 
-        self.tmngr = self.app.pillar_extensions['attract'].task_manager
-        self.smngr = self.app.pillar_extensions['attract'].shot_manager
+        self.tmngr = self.attract.task_manager
+        self.smngr = self.attract.shot_manager
 
         self.proj_id, self.project = self.ensure_project_exists()
 
@@ -190,7 +190,7 @@ class PatchShotTest(AbstractShotTest):
         self.patch(url, json=patch, auth_token='token')
 
         with self.app.test_request_context():
-            acts = self.app.pillar_extensions['attract'].activities_for_node(shot._id)
+            acts = self.attract.activities_for_node(shot._id)
         self.assertEqual(2, acts['_meta']['total'])  # Creation + edit
         edit_act = acts['_items'][1]
         self.assertIn(old_name, edit_act['verb'])