cert_audit: Fill validity dates in AuditData constructor
Signed-off-by: Pengyu Lv <pengyu.lv@arm.com>
diff --git a/tests/scripts/audit-validity-dates.py b/tests/scripts/audit-validity-dates.py
index 85c0bd9..472041e 100755
--- a/tests/scripts/audit-validity-dates.py
+++ b/tests/scripts/audit-validity-dates.py
@@ -48,11 +48,10 @@
class AuditData:
"""Store file, type and expiration date for audit."""
#pylint: disable=too-few-public-methods
- def __init__(self, data_type: DataType):
+ def __init__(self, data_type: DataType, x509_obj):
self.data_type = data_type
self.filename = ""
- self.not_valid_after: datetime.datetime
- self.not_valid_before: datetime.datetime
+ self.fill_validity_duration(x509_obj)
def fill_validity_duration(self, x509_obj):
"""Fill expiration_date field from a x509 object"""
@@ -211,8 +210,7 @@
result = None
self.warn(val_error)
if result is not None:
- audit_data = AuditData(data_type)
- audit_data.fill_validity_duration(result)
+ audit_data = AuditData(data_type, result)
return audit_data
return None