Skip to content

Commit

Permalink
Add project_records method to liquid assigns service
Browse files Browse the repository at this point in the history
  • Loading branch information
aapomm committed May 30, 2024
1 parent b9f408f commit 571c388
Show file tree
Hide file tree
Showing 2 changed files with 37 additions and 1 deletion.
25 changes: 24 additions & 1 deletion app/services/liquid_assigns_service.rb
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ def initialize(project)
end

def assigns
result = { 'project' => ProjectDrop.new(project) }
result = project_drops
result.merge!(assigns_pro) if defined?(Dradis::Pro)
result
end
Expand All @@ -15,4 +15,27 @@ def assigns

def assigns_pro
end

def project_drops
{
'evidence' => project_records(type: :evidence),
'issues' => project_records(type: :issue),
'nodes' => project_records(type: :node),
'notes' => project_records(type: :note),
'project' => ProjectDrop.new(project),
'tags' => project_records(type: :tag)
}
end

def project_records(type:)
records = project.send(type.to_s.pluralize)
records = records.user_nodes if type == :node

cache_key = "liquid-project-#{type.to_s.pluralize}#{records.maximum(:updated_at)}/#{records.pluck(:id).join('-')}"
drop_class = "#{type.to_s.capitalize}Drop".constantize

Rails.cache.fetch(cache_key) do
records.map { |record| drop_class.new(record) }
end
end
end
13 changes: 13 additions & 0 deletions spec/services/liquid_assigns_service_spec.rb
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,21 @@

let(:liquid_assigns) { described_class.new(project).assigns }

before do
node = create(:node, project: project)
issue = create(:issue, node: project.issue_library)
create(:evidence, issue: issue, node: node)
create(:note, node: node)
create(:tag)
end

it 'builds a hash of liquid assigns' do
expect(liquid_assigns['project'].name).to eq(project.name)
expect(liquid_assigns['issues'].map(&:title)).to eq(project.issues.map(&:title))
expect(liquid_assigns['evidence'].map(&:title)).to eq(project.evidence.map(&:title))
expect(liquid_assigns['nodes'].map(&:label)).to eq(project.nodes.user_nodes.map(&:label))
expect(liquid_assigns['notes'].map(&:title)).to eq(project.notes.map(&:title))
expect(liquid_assigns['tags'].map(&:display_name)).to eq(project.tags.map(&:display_name))
end

context 'with pro records', skip: !defined?(Dradis::Pro) do
Expand Down

0 comments on commit 571c388

Please sign in to comment.