From 32c7c20364e29456e7abd2fd5e044401922a29c4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Karel=20Pi=C4=8Dman?= <karel.picman@kontron.com>
Date: Mon, 17 Dec 2018 14:12:58 +0100
Subject: [PATCH] Post releasing errors

---
 app/controllers/dmsf_context_menus_controller.rb |  2 +-
 app/models/dmsf_folder.rb                        |  3 ++-
 test/unit/dmsf_folder_test.rb                    | 12 ++++++++++++
 3 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/app/controllers/dmsf_context_menus_controller.rb b/app/controllers/dmsf_context_menus_controller.rb
index 87f554ec..b4b29788 100644
--- a/app/controllers/dmsf_context_menus_controller.rb
+++ b/app/controllers/dmsf_context_menus_controller.rb
@@ -54,7 +54,7 @@ class DmsfContextMenusController < ApplicationController
       selected_files = params[:ids].select{ |x| x =~ /file-\d+/ }.map{ |x| $1.to_i if x =~ /file-(\d+)/ }
       selected_file_links = params[:ids].select{ |x| x =~ /file-link-\d+/ }.map{ |x| $1.to_i if x =~ /file-link-(\d+)/ }
       selected_file_links.each do |id|
-        target_id = DmsfLink.find_by(id: id).pluck(:target_id).first
+        target_id = DmsfLink.where(id: id).pluck(:target_id).first
         selected_files << target_id if target_id && !selected_files.include?(target_id)
       end
       if (selected_files.size == 1) && (params[:ids].size == 1)
diff --git a/app/models/dmsf_folder.rb b/app/models/dmsf_folder.rb
index 21ca2fc5..bcaec122 100644
--- a/app/models/dmsf_folder.rb
+++ b/app/models/dmsf_folder.rb
@@ -202,7 +202,8 @@ class DmsfFolder < ActiveRecord::Base
 
   def self.directory_tree(project, current_folder = nil)
     tree = [[l(:link_documents), nil]]
-    folders = DmsfFolder.where(project_id: project.id).visible(false).to_a
+    project_id = (project.is_a?(Project)) ? project.id : project
+    folders = DmsfFolder.where(project_id: project_id).visible(false).to_a
     folders.delete(current_folder)
     folders = folders.delete_if{ |f| f.locked_for_user? }
     folders.each do |folder|
diff --git a/test/unit/dmsf_folder_test.rb b/test/unit/dmsf_folder_test.rb
index 0c9b94df..0f363612 100644
--- a/test/unit/dmsf_folder_test.rb
+++ b/test/unit/dmsf_folder_test.rb
@@ -161,6 +161,18 @@ class DmsfFolderTest < RedmineDmsf::Test::UnitTest
     assert !tree.to_s.include?('......folder2'), "'......folder2' string in the folder tree not expected."
   end
 
+  def test_directory_tree_id
+    tree = DmsfFolder.directory_tree(@project.id)
+    assert tree
+    # [["Documents", nil],
+    #  ["...folder7", 7],
+    #  ["...folder1", 1],
+    #  ["......folder2", 2] - locked
+    #  ["...folder6", 6]]
+    assert tree.to_s.include?('...folder1'), "'...folder3' string in the folder tree expected."
+    assert !tree.to_s.include?('......folder2'), "'......folder2' string in the folder tree not expected."
+  end
+
   def test_folder_tree
     tree = @folder1.folder_tree
     assert tree