Commit cc63547e authored by Mehmet Emin INAC's avatar Mehmet Emin INAC

Add `checksum` sha_attribute into Vulnerabilities::Remediation model

parent e23cf6a7
...@@ -3,9 +3,12 @@ ...@@ -3,9 +3,12 @@
module Vulnerabilities module Vulnerabilities
class Remediation < ApplicationRecord class Remediation < ApplicationRecord
include FileStoreMounter include FileStoreMounter
include ShaAttribute
self.table_name = 'vulnerability_remediations' self.table_name = 'vulnerability_remediations'
sha_attribute :checksum
has_many :finding_remediations, class_name: 'Vulnerabilities::FindingRemediation', inverse_of: :remediation, foreign_key: 'vulnerability_remediation_id' has_many :finding_remediations, class_name: 'Vulnerabilities::FindingRemediation', inverse_of: :remediation, foreign_key: 'vulnerability_remediation_id'
has_many :findings, through: :finding_remediations has_many :findings, through: :finding_remediations
...@@ -13,5 +16,6 @@ module Vulnerabilities ...@@ -13,5 +16,6 @@ module Vulnerabilities
validates :summary, presence: true, length: { maximum: 200 } validates :summary, presence: true, length: { maximum: 200 }
validates :file, presence: true validates :file, presence: true
validates :checksum, presence: true
end end
end end
...@@ -8,5 +8,6 @@ RSpec.describe Vulnerabilities::Remediation do ...@@ -8,5 +8,6 @@ RSpec.describe Vulnerabilities::Remediation do
it { is_expected.to validate_presence_of(:summary) } it { is_expected.to validate_presence_of(:summary) }
it { is_expected.to validate_presence_of(:file) } it { is_expected.to validate_presence_of(:file) }
it { is_expected.to validate_presence_of(:checksum) }
it { is_expected.to validate_length_of(:summary).is_at_most(200) } it { is_expected.to validate_length_of(:summary).is_at_most(200) }
end end
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment