asset_manager.rb

module GdsApi
  module TestHelpers
    module AssetManager
      ASSET_MANAGER_ENDPOINT = Plek.find("asset-manager")

      def stub_any_asset_manager_call
        stub_request(:any, %r{\A#{ASSET_MANAGER_ENDPOINT}}).to_return(status: 200)
      end

      def stub_asset_manager_isnt_available
        stub_request(:any, %r{\A#{ASSET_MANAGER_ENDPOINT}}).to_return(status: 503)
      end

      def stub_asset_manager_updates_any_asset(body = {})
        stub_request(:put, %r{\A#{ASSET_MANAGER_ENDPOINT}/assets})
          .to_return(body: body.to_json, status: 200)
      end

      def stub_asset_manager_deletes_any_asset(body = {})
        stub_request(:delete, %r{\A#{ASSET_MANAGER_ENDPOINT}/assets})
          .to_return(body: body.to_json, status: 200)
      end

      def stub_asset_manager_has_an_asset(id, atts, filename = "")
        response = atts.merge("_response_info" => { "status" => "ok" })

        stub_request(:get, "#{ASSET_MANAGER_ENDPOINT}/assets/#{id}")
          .to_return(body: response.to_json, status: 200)

        stub_request(:get, "#{ASSET_MANAGER_ENDPOINT}/media/#{id}/#{filename}")
          .to_return(body: "Some file content", status: 200)
      end

      def stub_asset_manager_has_a_whitehall_media_asset(legacy_url_path, content)
        stub_request(:get, "#{ASSET_MANAGER_ENDPOINT}/#{legacy_url_path}")
          .to_return(body: content, status: 200)
      end

      def stub_asset_manager_does_not_have_an_asset(id)
        response = {
          "_response_info" => { "status" => "not found" },
        }

        stub_request(:any, "#{ASSET_MANAGER_ENDPOINT}/assets/#{id}")
          .to_return(body: response.to_json, status: 404)
      end

      # This can take a string of an exact url or a hash of options
      #
      # with a string:
      # `stub_asset_manager_receives_an_asset("https://asset-manager/media/619ce797-b415-42e5-b2b1-2ffa0df52302/file.jpg")`
      #
      # with a hash:
      # `stub_asset_manager_receives_an_asset(id: "20d04259-e3ae-4f71-8157-e6c843096e96", filename: "file.jpg")`
      # which would return a file url of "https://asset-manager/media/20d04259-e3ae-4f71-8157-e6c843096e96/file.jpg"
      #
      # with no argument
      #
      # `stub_asset_manager_receives_an_asset`
      # which would return a file url of "https://asset-manager/media/0053adbf-0737-4923-9d8a-8180f2c723af/0d19136c4a94f07"
      def stub_asset_manager_receives_an_asset(response_url = {})
        stub_request(:post, "#{ASSET_MANAGER_ENDPOINT}/assets").to_return do
          if response_url.is_a?(String)
            file_url = response_url
          else
            options = {
              id: SecureRandom.uuid,
              filename: SecureRandom.hex(8),
            }.merge(response_url)

            file_url = "#{ASSET_MANAGER_ENDPOINT}/media/#{options[:id]}/#{options[:filename]}"
          end
          { body: { file_url: }.to_json, status: 200 }
        end
      end

      def stub_asset_manager_upload_failure
        stub_request(:post, "#{ASSET_MANAGER_ENDPOINT}/assets").to_return(status: 500)
      end

      def stub_asset_manager_update_asset(asset_id, body = {})
        stub_request(:put, "#{ASSET_MANAGER_ENDPOINT}/assets/#{asset_id}")
          .to_return(body: body.to_json, status: 200)
      end

      def stub_asset_manager_update_asset_failure(asset_id)
        stub_request(:put, "#{ASSET_MANAGER_ENDPOINT}/assets/#{asset_id}").to_return(status: 500)
      end

      def stub_asset_manager_delete_asset(asset_id, body = {})
        stub_request(:delete, "#{ASSET_MANAGER_ENDPOINT}/assets/#{asset_id}")
          .to_return(body: body.to_json, status: 200)
      end

      def stub_asset_manager_delete_asset_missing(asset_id)
        stub_request(:delete, "#{ASSET_MANAGER_ENDPOINT}/assets/#{asset_id}")
          .to_return(status: 404)
      end

      def stub_asset_manager_delete_asset_failure(asset_id)
        stub_request(:delete, "#{ASSET_MANAGER_ENDPOINT}/assets/#{asset_id}").to_return(status: 500)
      end
    end
  end
end