DMP Assistant Feature - allow phase-only download - Ready

app/controllers/plan_exports_controller.rb

@@ -36,12 +36,18 @@ def show
 
     @hash           = @plan.as_pdf(current_user, @show_coversheet)
     @formatting     = export_params[:formatting] || @plan.settings(:export).formatting
-    @selected_phase = if params.key?(:phase_id)
-                        @plan.phases.find(params[:phase_id])
-                      else
-                        @plan.phases.order('phases.updated_at DESC')
+    if params.key?(:phase_id) && params[:phase_id].length.positive?
+      # order phases by phase number asc
+      @hash[:phases] = @hash[:phases].sort_by { |phase| phase[:number] }
+      if params[:phase_id] == 'All'
+        @hash[:all_phases] = true
+      else
+        @selected_phase = @plan.phases.find(params[:phase_id])
+      end
+    else
+      @selected_phase = @plan.phases.order('phases.updated_at DESC')
                              .detect { |p| p.visibility_allowed?(@plan) }
-                      end
+    end
 
     # Added contributors to coverage of plans.
     # Users will see both roles and contributor names if the role is filled
@@ -101,7 +107,7 @@ def show_pdf
                             date: l(@plan.updated_at.to_date, format: :readable)),
              font_size: 8,
              spacing: (Integer(@formatting[:margin][:bottom]) / 2) - 4,
-             right: '[page] of [topage]',
+             right: _('[page] of [topage]'),
              encoding: 'utf8'
            }
   end

app/controllers/plans_controller.rb

@@ -376,6 +376,7 @@ def download
     @plan = Plan.find(params[:id])
     authorize @plan
     @phase_options = @plan.phases.order(:number).pluck(:title, :id)
+    @phase_options.insert(0, ['All phases', 'All']) if @phase_options.length > 1
     @export_settings = @plan.settings(:export)
     render 'download'
   end

app/javascript/src/plans/download.js

@@ -21,5 +21,15 @@ $(() => {
     } else {
       $('#download-settings').show();
     }
-  });
+
+    if (frmt === 'csv') {
+      $('#phase_id').find('option[value="All"').hide();
+      $('#phase_id option:eq(1)').attr('selected', 'selected');
+      $('#phase_id').val($('#phase_id option:eq(1)').val());
+    } else if (frmt === 'pdf' || frmt === 'html' || frmt === 'docx' || frmt === 'text') {
+      $('#phase_id').find('option[value="All"').show();
+      $('#phase_id').val($('#phase_id option:first').val());
+      $('#phase_id option:first').attr('selected', 'selected');
+    }
+  }).trigger('change');
 });

app/views/shared/export/_plan.erb

@@ -18,7 +18,7 @@
 
     <% @hash[:phases].each do |phase| %>
       <%# Only render selected phase %>
-      <% if phase[:title] == @selected_phase.title %>
+      <% if @hash[:all_phases] || (@selected_phase.present? && phase[:title] == @selected_phase.title) %>
         <%# Page break before each phase %>
         <div style="page-break-before:always;"></div>
         <h1><%= download_plan_page_title(@plan, phase, @hash) %></h1>

app/views/shared/export/_plan_txt.erb

@@ -38,7 +38,7 @@
 
 <% @hash[:phases].each do |phase| %>
 <%# Only render selected phase %>
-<% if phase[:title] == @selected_phase.title %>
+<% if @hash[:all_phases] || (@selected_phase.present? && phase[:title] == @selected_phase.title) %>
 <%= (@hash[:phases].length > 1 ? "#{phase[:title]}" : "") %>
   <% phase[:sections].each do |section| %>
     <% if display_section?(@hash[:customization], section, @show_custom_sections) && num_section_questions(@plan, section, phase) > 0 %>

db/schema.rb

@@ -666,4 +666,4 @@
   add_foreign_key "users", "departments"
   add_foreign_key "users", "languages"
   add_foreign_key "users", "orgs"
-end
+end

spec/features/plans/exports_spec.rb

@@ -78,15 +78,35 @@
     expect(page).not_to have_text(new_plan.title)
   end
 
+  # Separate code to test all-phase-download for html since it requires operation in new window
   scenario 'User downloads their plan as HTML' do
     within("#plan_#{plan.id}") do
       click_button('Actions')
       click_link 'Download'
     end
     select('html')
-    new_window = window_opened_by { click_button 'Download Plan' }
-    within_window new_window do
-      expect(page.source).to have_text(plan.title)
+    if plan.phases.present?
+      new_window = window_opened_by do
+        _select_option('phase_id', 'All')
+        click_button 'Download Plan'
+      end
+      within_window new_window do
+        expect(page.source).to have_text(plan.title)
+        plan.phases.each do |phase|
+          expect(page.source).to have_text(phase.title)
+        end
+      end
+      new_window = window_opened_by do
+        _select_option('phase_id', plan.phases[1].id)
+        click_button 'Download Plan'
+      end
+      within_window new_window do
+        expect(page.source).to have_text(plan.title)
+        expect(page.source).to have_text(plan.phases[1].title)
+        expect(page.source).not_to have_text(plan.phases[2].title) if plan.phases.length > 2
+      end
+    else
+      _regular_download('html')
     end
   end
 
@@ -96,8 +116,12 @@
       click_link 'Download'
     end
     select('pdf')
-    click_button 'Download Plan'
-    expect(page.source).to have_text(plan.title)
+    if plan.phases.present?
+      _all_phase_download
+      _single_phase_download
+    else
+      _regular_download('pdf')
+    end
   end
 
   scenario 'User downloads their plan as CSV' do
@@ -106,8 +130,7 @@
       click_link 'Download'
     end
     select('csv')
-    click_button 'Download Plan'
-    expect(page.source).to have_text(plan.title)
+    _regular_download('csv')
   end
 
   scenario 'User downloads their plan as text' do
@@ -116,8 +139,12 @@
       click_link 'Download'
     end
     select('text')
-    click_button 'Download Plan'
-    expect(page.source).to have_text(plan.title)
+    if plan.phases.present?
+      _all_phase_download
+      _single_phase_download
+    else
+      _regular_download('text')
+    end
   end
 
   scenario 'User downloads their plan as docx' do
@@ -126,7 +153,53 @@
       click_link 'Download'
     end
     select('docx')
+    if plan.phases.present?
+      _all_phase_download
+      _single_phase_download
+    else
+      _regular_download('docx')
+    end
+  end
+
+  # ===========================
+  # = Helper methods =
+  # ===========================
+
+  # rubocop:disable Metrics/AbcSize
+  # disable Rubocup metrics check to confirm both plan title and phase title on downloaded file
+  def _regular_download(format)
+    if format == 'html'
+      new_window = window_opened_by do
+        click_button 'Download Plan'
+      end
+      within_window new_window do
+        expect(page.source).to have_text(plan.title)
+      end
+    else
+      click_button 'Download Plan'
+      expect(page.source).to have_text(plan.title)
+    end
+  end
+
+  def _all_phase_download
+    _select_option('phase_id', 'All')
     click_button 'Download Plan'
     expect(page.source).to have_text(plan.title)
+    plan.phases.each do |phase| # All phase titles should be included in output
+      expect(page.source).to have_text(phase.title)
+    end
+  end
+
+  def _single_phase_download
+    _select_option('phase_id', plan.phases[1].id)
+    click_button 'Download Plan'
+    expect(page.source).to have_text(plan.title)
+    expect(page.source).to have_text(plan.phases[1].title)
+    expect(page.source).not_to have_text(plan.phases[2].title) if plan.phases.length > 2
+  end
+
+  # rubocop:enable Metrics/AbcSize
+  def _select_option(select_id, option_value)
+    find(:id, select_id).find("option[value='#{option_value}']").select_option
   end
 end

spec/spec_helper.rb

@@ -117,7 +117,7 @@
   # Enable Capybara webmocks if we are testing a feature
   config.before(:each) do |example|
     if example.metadata[:type] == :feature
-      Capybara::Webmock.start
+      # Capybara::Webmock.start
 
       # Allow Capybara to make localhost requests and also contact the
       # google api chromedriver store