Finalize CSV export for ComplianceCheckMessage and Import<essage

2 files changed, 4 insertions, 4 deletions

diff --git a/app/models/compliance_check_message_export.rb b/app/models/compliance_check_message_export.rb
index b3e3f80cf..04e1a9caa 100644
--- a/app/models/compliance_check_message_export.rb
+++ b/app/models/compliance_check_message_export.rb
@@ -22,14 +22,14 @@ class ComplianceCheckMessageExport
   end
 
   def column_names
-    ["criticity", "message key", "message", "resource objectid"]
+    ["criticity", "message key", "resource objectid", "link", "message"]
   end
 
   def to_csv(options = {})
-    CSV.generate(options) do |csv|
+    CSV.generate(options.slice(:col_sep, :quote_char, :force_quotes)) do |csv|
       csv << column_names
       compliance_check_messages.each do |compliance_check_message|
-        csv << [compliance_check_message.compliance_check.criticity, compliance_check_message.message_attributes[:test_id], I18n.t("compliance_check_messages.#{compliance_check_message.message_key}", compliance_check_message.message_attributes.deep_symbolize_keys), compliance_check_message.resource_attributes[:objectid] ]
+        csv << [compliance_check_message.compliance_check.criticity, *compliance_check_message.message_attributes.values_at('test_id', 'source_objectid'), options[:server_url] + compliance_check_message.message_attributes['source_object_path'], I18n.t("compliance_check_messages.#{compliance_check_message.message_key}", compliance_check_message.message_attributes.deep_symbolize_keys)]
       end
     end
   end
diff --git a/app/models/import_message_export.rb b/app/models/import_message_export.rb
index eeb6ebb18..05f8a2cc7 100644
--- a/app/models/import_message_export.rb
+++ b/app/models/import_message_export.rb
@@ -29,7 +29,7 @@ class ImportMessageExport
     CSV.generate(options) do |csv|
       csv << column_names
       import_messages.each do |import_message|
-        csv << [import_message.criticity, import_message.message_key, I18n.t("import_messages.compliance_check_messages.#{import_message.message_key}", import_message.message_attributes.deep_symbolize_keys), *import_message.resource_attributes.values_at("filename", "line_number", "column_number")  ]
+        csv << [import_message.criticity, import_message.message_key, I18n.t("import_messages.#{import_message.message_key}", import_message.message_attributes.deep_symbolize_keys), *import_message.resource_attributes.values_at("filename", "line_number", "column_number")  ]
       end
     end
   end