Fix single use links for Valkyrie.

app/controllers/concerns/hyrax/valkyrie_downloads_controller_behavior.rb

@@ -25,7 +25,7 @@ def send_file_contents_valkyrie(file_set)
         file.rewind
         send_data send_range_valkyrie(file: file), data_options(file_metadata)
       else
-        send_file file.disk_path
+        send_file file.disk_path, data_options(file_metadata).except(:status)
       end
     end
 

app/controllers/hyrax/single_use_links_viewer_controller.rb

@@ -37,6 +37,14 @@ def show
 
     private
 
+    def send_content
+      if @asset.is_a?(Hyrax::FileSet)
+        send_file_contents_valkyrie(@asset)
+      else
+        super
+      end
+    end
+
     def curation_concern
       response, _document_list = search_service.search_results
       response.documents.first
@@ -52,6 +60,12 @@ def content_options
       end
     end
 
+    def data_options(file_metadata)
+      super.tap do |options|
+        options[:disposition] = 'attachment' if action_name == 'download'
+      end
+    end
+
     # This is called in a before filter. It causes @asset to be set.
     def authorize_download!
       authorize! :read, asset
@@ -66,7 +80,7 @@ def not_found_exception
     end
 
     def asset
-      @asset ||= Hyrax.query_service.find_by_alternate_identifier(alternate_identifier: single_use_link.item_id, use_valkyrie: false)
+      @asset ||= Hyrax.query_service.find_by(id: single_use_link.item_id)
     end
 
     def current_ability
@@ -94,7 +108,7 @@ def initialize(user, single_use_link)
         return unless single_use_link
 
         @single_use_link = single_use_link
-        can :read, [ActiveFedora::Base, ::SolrDocument] do |obj|
+        can :read, [ActiveFedora::Base, ::SolrDocument, Hyrax::Resource] do |obj|
           single_use_link.valid? && single_use_link.item_id == obj.id && single_use_link.destroy!
         end
       end

spec/controllers/hyrax/single_use_links_viewer_controller_spec.rb

@@ -3,7 +3,12 @@
   routes { Hyrax::Engine.routes }
   let(:user) { build(:user) }
   let(:file) do
-    create(:file_set, label: 'world.png', user: user)
+    Hyrax.query_service.find_by(id: work.member_ids.first)
+  end
+  let(:file_metadata) { Hyrax.custom_queries.find_file_metadata_by(id: file.original_file_id) }
+  let(:disk_file) { Hyrax.storage_adapter.find_by(id: file_metadata.file_identifier) }
+  let(:work) do
+    FactoryBot.valkyrie_create(:hyrax_work, uploaded_files: [FactoryBot.create(:uploaded_file)])
   end
 
   describe "retrieval links" do
@@ -12,17 +17,16 @@
     end
 
     let :download_link do
-      Hydra::Works::AddFileToFileSet.call(file, File.open(fixture_path + '/world.png'), :original_file)
-      SingleUseLink.create item_id: file.id, path: Hyrax::Engine.routes.url_helpers.download_path(id: file, locale: 'en')
+      SingleUseLink.create item_id: file.id, path: Hyrax::Engine.routes.url_helpers.download_path(id: file.id.to_s, locale: 'en')
     end
 
     let(:show_link_hash) { show_link.download_key }
     let(:download_link_hash) { download_link.download_key }
 
     describe "GET 'download'" do
-      let(:expected_content) { Hyrax.query_service.find_by_alternate_identifier(alternate_identifier: file.id, use_valkyrie: false).original_file.content }
+      let(:expected_content) { disk_file.read }
       it "downloads the file and deletes the link from the database" do
-        expect(controller).to receive(:send_file_headers!).with({ filename: 'world.png', disposition: 'attachment', type: 'image/png' })
+        expect(controller).to receive(:send_file_headers!).with({ filename: 'image.jp2', disposition: 'attachment', type: file_metadata.mime_type })
         get :download, params: { id: download_link_hash }
         expect(response.body).to eq expected_content
         expect(response).to be_successful

spec/factories/hyrax_work.rb

@@ -19,6 +19,7 @@
       members            { nil }
       visibility_setting { nil }
       with_index         { true }
+      uploaded_files { [] }
     end
 
     after(:build) do |work, evaluator|
@@ -41,6 +42,15 @@
           .new(resource: work)
           .assign_access_for(visibility: evaluator.visibility_setting)
       end
+      if evaluator.uploaded_files.present?
+        Hyrax::WorkUploadsHandler.new(work: work).add(files: evaluator.uploaded_files).attach
+        evaluator.uploaded_files.each do |file|
+          allow(Hyrax.config.characterization_service).to receive(:run).and_return(true)
+          # I don't love this - we might want to just run background jobs so
+          # this is more real, but we'd have to stub some things.
+          ValkyrieIngestJob.perform_now(file)
+        end
+      end
 
       work.permission_manager.edit_groups = evaluator.edit_groups
       work.permission_manager.edit_users  = evaluator.edit_users

spec/spec_helper.rb

@@ -86,6 +86,7 @@ def ci_build?
 FactoryBot.register_strategy(:json, JsonStrategy)
 FactoryBot.definition_file_paths = [File.expand_path("../factories", __FILE__)]
 FactoryBot.find_definitions
+require 'rspec/mocks'
 
 require 'shoulda/matchers'
 require 'shoulda/callback/matchers'
@@ -142,6 +143,7 @@ def clean_active_fedora_repository
   config.use_transactional_fixtures = false
 
   config.before :suite do
+    FactoryBot::SyntaxRunner.include RSpec::Mocks::ExampleMethods
     Hyrax::RedisEventStore.instance.then(&:flushdb)
     DatabaseCleaner.clean_with(:truncation)
     # Noid minting causes extra LDP requests which slow the test suite.