def test_check_uses_custom_csp(self):
assert AutoApprovalSummary.check_uses_custom_csp(self.version) is False
validation_data = {
'messages': [{
'id': ['MANIFEST_CSP'],
}]
}
self.file_validation.update(validation=json.dumps(validation_data))
assert AutoApprovalSummary.check_uses_custom_csp(self.version) is True
def test_check_uses_custom_csp_file_validation_missing(self):
self.file_validation.delete()
del self.version.all_files
with self.assertRaises(AutoApprovalNoValidationResultError):
AutoApprovalSummary.check_uses_custom_csp(self.version)
# Also happens if only one file is missing validation info.
self.file_validation = FileValidation.objects.create(
file=self.version.all_files[0], validation=u'{}')
del self.version.all_files
file_factory(version=self.version, status=amo.STATUS_AWAITING_REVIEW)
with self.assertRaises(AutoApprovalNoValidationResultError):
AutoApprovalSummary.check_uses_custom_csp(self.version)
