policy.py: Fix incorrect policy key

Signed-off-by: Niels Thykier <niels@thykier.net>
8 years ago · 2b5d23f69f
parent 2fadfe3724
commit 2b5d23f69f
1 changed files with 3 additions and 3 deletions
--- a/policies/policy.py
+++ b/policies/policy.py
@ -87,10 +87,10 @@ class BasePolicy(object):


    def apply_policy(self, general_policy_info, suite, source_name, source_data_tdist, source_data_srcdist):
-        if 'age' not in general_policy_info:
-            general_policy_info['age'] = pinfo = {}
+        if self.policy_id not in general_policy_info:
+            general_policy_info[self.policy_id] = pinfo = {}
        else:
-            pinfo = general_policy_info['age']
+            pinfo = general_policy_info[self.policy_id]
        return self.apply_policy_impl(pinfo, suite, source_name, source_data_tdist, source_data_srcdist)

    @abstractmethod