diff --git a/tuskarclient/tests/v2/test_plans.py b/tuskarclient/tests/v2/test_plans.py index 8e8dac0..aeeed97 100644 --- a/tuskarclient/tests/v2/test_plans.py +++ b/tuskarclient/tests/v2/test_plans.py @@ -122,8 +122,7 @@ class PlanManagerTest(tutils.TestCase): self.pm._get.return_value.to_dict.return_value = 'fake_templates_dict' self.assertEqual(self.pm.templates('fake_plan'), 'fake_templates_dict') - self.pm._get.assert_called_with('/v2/plans/fake_plan/templates', - obj_class=plans.Templates) + self.pm._get.assert_called_with('/v2/plans/fake_plan/templates') def test_roles_subresource(self): self.pm._get = mock.Mock( diff --git a/tuskarclient/v2/plans.py b/tuskarclient/v2/plans.py index 70cc14b..45309c3 100644 --- a/tuskarclient/v2/plans.py +++ b/tuskarclient/v2/plans.py @@ -164,5 +164,7 @@ class PlanManager(base.BaseManager): :rtype: dict """ - return self._get(self._templates_path(plan_uuid), - obj_class=Templates).to_dict() + self.resource_class = Templates + res = self._get(self._templates_path(plan_uuid)).to_dict() + self.resource_class = Plan + return res