train 0.28.0 → 0.29.0

data/lib/train/transports/winrm_connection.rb

@@ -51,7 +51,7 @@ class Train::Transports::WinRM
     end
 
     def file(path)
-      @files[path] ||= WindowsFile.new(self, path)
+      @files[path] ||= Train::File::Remote::Windows.new(self, path)
     end
 
     def run_command(command)
@@ -84,7 +84,13 @@ class Train::Transports::WinRM
 
     # (see Base::Connection#upload)
     def upload(locals, remote)
-      file_transporter.upload(locals, remote)
+      file_manager.upload(locals, remote)
+    end
+
+    def download(remotes, local)
+      Array(remotes).each do |remote|
+        file_manager.download(remote, local)
+      end
     end
 
     # (see Base::Connection#wait_until_ready)
@@ -124,10 +130,10 @@ class Train::Transports::WinRM
       content
     end
 
-    # @return [Winrm::FileTransporter] a file transporter
+    # @return [Winrm::FileManager] a file transporter
     # @api private
-    def file_transporter
-      @file_transporter ||= WinRM::FS::Core::FileTransporter.new(session)
+    def file_manager
+      @file_manager ||= WinRM::FS::FileManager.new(session)
     end
 
     # Builds a `LoginCommand` for use by Linux-based platforms.

data/lib/train/version.rb

@@ -3,5 +3,5 @@
 # Author:: Dominik Richter (<dominik.richter@gmail.com>)
 
 module Train
-  VERSION = '0.28.0'.freeze
+  VERSION = '0.29.0'.freeze
 end

data/test/integration/tests/path_block_device_test.rb

@@ -64,11 +64,11 @@ describe 'file interface' do
     end
 
     it 'has no product_version' do
-      file.product_version.must_equal(nil)
+      file.product_version.must_be_nil
     end
 
     it 'has no file_version' do
-      file.file_version.must_equal(nil)
+      file.file_version.must_be_nil
     end
   end
 end

data/test/integration/tests/path_character_device_test.rb

@@ -64,11 +64,11 @@ describe 'file interface' do
     end
 
     it 'has no product_version' do
-      file.product_version.must_equal(nil)
+      file.product_version.must_be_nil
     end
 
     it 'has no file_version' do
-      file.file_version.must_equal(nil)
+      file.file_version.must_be_nil
     end
   end
 end

data/test/integration/tests/path_file_test.rb

@@ -69,11 +69,11 @@ describe 'file interface' do
     end
 
     it 'has no product_version' do
-      file.product_version.must_equal(nil)
+      file.product_version.must_be_nil
     end
 
     it 'has no file_version' do
-      file.file_version.must_equal(nil)
+      file.file_version.must_be_nil
     end
 
     it 'provides a json representation' do

data/test/integration/tests/path_folder_test.rb

@@ -34,15 +34,15 @@ describe 'file interface' do
 
     else
       it 'has no content' do
-        file.content.must_equal(nil)
+        file.content.must_be_nil
       end
 
       it 'has an md5sum' do
-        file.md5sum.must_equal(nil)
+        file.md5sum.must_be_nil
       end
 
       it 'has an sha256sum' do
-        file.sha256sum.must_equal(nil)
+        file.sha256sum.must_be_nil
       end
     end
 
@@ -80,11 +80,11 @@ describe 'file interface' do
     end
 
     it 'has no product_version' do
-      file.product_version.must_equal(nil)
+      file.product_version.must_be_nil
     end
 
     it 'has no file_version' do
-      file.file_version.must_equal(nil)
+      file.file_version.must_be_nil
     end
   end
 end

data/test/integration/tests/path_missing_test.rb

@@ -72,6 +72,5 @@ describe 'file interface' do
     it 'has no file_version' do
       file.file_version.must_be_nil
     end
-
   end
 end

data/test/integration/tests/path_pipe_test.rb

@@ -66,13 +66,12 @@ describe 'file interface' do
       file.selinux_label.must_equal(res)
     end
 
-
     it 'has no product_version' do
-      file.product_version.must_equal(nil)
+      file.product_version.must_be_nil
     end
 
     it 'has no file_version' do
-      file.file_version.must_equal(nil)
+      file.file_version.must_be_nil
     end
   end
 end

data/test/integration/tests/path_symlink_test.rb

@@ -85,11 +85,11 @@ describe 'file interface' do
     end
 
     it 'has no product_version' do
-      file.product_version.must_equal(nil)
+      file.product_version.must_be_nil
     end
 
     it 'has no file_version' do
-      file.file_version.must_equal(nil)
+      file.file_version.must_be_nil
     end
   end
 end

data/test/unit/extras/os_detect_linux_test.rb

@@ -125,8 +125,8 @@ describe 'os_detect_linux' do
           detector.stubs(:fetch_os_release).returns({ 'ID_LIKE' => 'something_else' })
 
           detector.detect_linux_via_config.must_equal(false)
-          detector.platform[:family].must_equal(nil)
-          detector.platform[:release].must_equal(nil)
+          detector.platform[:family].must_be_nil
+          detector.platform[:release].must_be_nil
         end
       end
 
@@ -154,7 +154,7 @@ describe 'os_detect_linux' do
     describe 'when no os-release data is available' do
       it 'returns nil' do
         detector.expects(:get_config).with('/etc/os-release').returns(nil)
-        detector.fetch_os_release.must_equal(nil)
+        detector.fetch_os_release.must_be_nil
       end
     end
 

data/test/unit/extras/os_detect_windows_test.rb

@@ -97,7 +97,7 @@ describe 'os_detect_windows' do
       detector.detect_windows
       detector.platform[:family].must_equal('windows')
       detector.platform[:name].must_equal('Windows 4.10.1998')
-      detector.platform[:arch].must_equal(nil)
+      detector.platform[:arch].must_be_nil
       detector.platform[:release].must_equal('4.10.1998')
     end
   end

data/test/unit/file/local/unix_test.rb

@@ -0,0 +1,112 @@
+require 'helper'
+require 'train/file/local/unix'
+require 'train/transports/mock'
+require 'train/transports/local'
+
+describe Train::File::Local::Unix do
+  let(:cls) { Train::File::Local::Unix }
+  
+  let(:backend) {
+    backend = Train::Transports::Mock.new.connection
+    backend.mock_os({ family: 'linux' })
+    backend
+  }
+
+  it 'checks a mounted path' do
+    backend.mock_command("mount | grep -- ' on /mount/path '", rand.to_s)
+    cls.new(backend, '/mount/path').mounted?.must_equal true
+  end
+
+  describe 'file metadata' do
+    let(:transport) { Train::Transports::Local.new }
+    let(:connection) { transport.connection }
+
+    let(:stat) { Struct.new(:mode, :size, :mtime, :uid, :gid) }
+    let(:uid) { rand }
+    let(:gid) { rand }
+    let(:statres) { stat.new(00140755, rand, (rand*100).to_i, uid, gid) }
+
+    def meta_stub(method, param, &block)
+      pwres = Struct.new(:name)
+      Etc.stub :getpwuid, pwres.new('owner') do
+        Etc.stub :getgrgid, pwres.new('group') do
+          File.stub method, param do; yield; end
+        end
+      end
+    end
+
+    it 'recognizes type' do
+      meta_stub :stat, statres do
+        connection.file(rand.to_s).stat[:type].must_equal :socket
+      end
+    end
+
+    it 'recognizes mode' do
+      meta_stub :stat, statres do
+        connection.file(rand.to_s).stat[:mode].must_equal 00755
+      end
+    end
+
+    it 'recognizes mtime' do
+      meta_stub :stat, statres do
+        connection.file(rand.to_s).stat[:mtime].must_equal statres.mtime
+      end
+    end
+
+    it 'recognizes size' do
+      meta_stub :stat, statres do
+        connection.file(rand.to_s).stat[:size].must_equal statres.size
+      end
+    end
+
+    it 'recognizes uid' do
+      meta_stub :stat, statres do
+        connection.file(rand.to_s).stat[:uid].must_equal uid
+      end
+    end
+
+    it 'recognizes gid' do
+      meta_stub :stat, statres do
+        connection.file(rand.to_s).stat[:gid].must_equal gid
+      end
+    end
+
+    it 'recognizes owner' do
+      meta_stub :stat, statres do
+        connection.file(rand.to_s).owner.must_equal 'owner'
+      end
+    end
+
+    it 'recognizes group' do
+      meta_stub :stat, statres do
+        connection.file(rand.to_s).group.must_equal 'group'
+      end
+    end
+
+    it 'grouped_into' do
+      meta_stub :stat, statres do
+        connection.file(rand.to_s).grouped_into?('group').must_equal true
+      end  
+    end  
+
+    it 'recognizes selinux label' do
+      meta_stub :stat, statres do
+        label = rand.to_s
+        res = Train::Extras::CommandResult.new(label, nil, 0)
+        connection.stub :run_command, res do
+          connection.file(rand.to_s).selinux_label.must_equal label
+        end
+      end
+    end
+
+    it 'recognizes source selinux label' do
+      meta_stub :lstat, statres do
+        label = rand.to_s
+        res = Train::Extras::CommandResult.new(label, nil, 0)
+        connection.stub :run_command, res do
+          connection.file(rand.to_s).source.selinux_label.must_equal label
+        end
+      end
+    end
+  end
+end

data/test/unit/file/local/windows_test.rb

@@ -0,0 +1,41 @@
+# encoding: utf-8
+
+require 'helper'
+require 'train/transports/mock'
+require 'train/file/local/windows'
+
+describe 'file common' do
+  let(:cls) { Train::File::Local::Windows }
+  let(:backend) {
+    backend = Train::Transports::Mock.new.connection
+    backend.mock_os({ family: 'windows' })
+    backend
+  }
+
+  it 'check escaping of invalid chars in path' do
+    wf = cls.new(nil, nil)
+    wf.sanitize_filename('c:/test') .must_equal 'c:/test'
+    wf.sanitize_filename('c:/test directory') .must_equal 'c:/test directory'
+    %w{ < > " * ?}.each do |char|
+      wf.sanitize_filename("c:/test#{char}directory") .must_equal 'c:/testdirectory'
+    end
+  end
+
+  it 'returns file version' do
+    out = rand.to_s
+    backend.mock_command('[System.Diagnostics.FileVersionInfo]::GetVersionInfo("path").FileVersion', out)
+    cls.new(backend, 'path').file_version.must_equal out
+  end
+
+  it 'returns product version' do
+    out = rand.to_s
+    backend.mock_command('[System.Diagnostics.FileVersionInfo]::GetVersionInfo("path").FileVersion', out)
+    cls.new(backend, 'path').file_version.must_equal out
+  end
+
+  it 'returns owner of file' do
+    out = rand.to_s
+    backend.mock_command('Get-Acl "path" | select -expand Owner', out)
+    cls.new(backend, 'path').owner.must_equal out
+  end
+end

data/test/unit/file/local_test.rb

@@ -0,0 +1,110 @@
+require 'helper'
+require 'train/transports/local'
+
+describe Train::File::Local do
+  let(:transport) { Train::Transports::Local.new }
+  let(:connection) { transport.connection }
+
+  it 'gets file contents' do
+    res = rand.to_s
+    File.stub :read, res do
+      connection.file(rand.to_s).content.must_equal(res)
+    end
+  end 
+
+  {
+    exist?:            :exist?,
+    file?:             :file?,
+    socket?:           :socket?,
+    directory?:        :directory?,
+    symlink?:          :symlink?,
+    pipe?:             :pipe?,
+    character_device?: :chardev?,
+    block_device?:     :blockdev?,
+  }.each do |method, file_method|
+    it "checks if file is a #{method}" do
+      File.stub file_method.to_sym, true do
+        connection.file(rand.to_s).method(method.to_sym).call.must_equal(true)
+      end
+    end
+  end
+
+  describe '#type' do
+    it 'returns the type block_device if it is block device' do
+      File.stub :ftype, "blockSpecial" do
+        connection.file(rand.to_s).type.must_equal :block_device
+      end  
+    end
+
+    it 'returns the type character_device if it is character device' do
+      File.stub :ftype, "characterSpecial" do
+        connection.file(rand.to_s).type.must_equal :character_device
+      end  
+    end
+
+    it 'returns the type symlink if it is symlink' do
+      File.stub :ftype, "link" do
+        connection.file(rand.to_s).type.must_equal :symlink
+      end
+    end
+
+    it 'returns the type file if it is file' do
+      File.stub :ftype, "file" do
+        connection.file(rand.to_s).type.must_equal :file
+      end
+    end
+
+    it 'returns the type directory if it is block directory' do
+      File.stub :ftype, "directory" do
+        connection.file(rand.to_s).type.must_equal :directory
+      end
+    end
+
+    it 'returns the type pipe if it is pipe' do
+      File.stub :ftype, "fifo" do
+        connection.file(rand.to_s).type.must_equal :pipe
+      end
+    end
+
+    it 'returns the type socket if it is socket' do
+      File.stub :ftype, "socket" do
+        connection.file(rand.to_s).type.must_equal :socket
+      end
+    end
+
+    it 'returns the unknown if not known' do
+      File.stub :ftype, "unknown" do
+        connection.file(rand.to_s).type.must_equal :unknown
+      end
+    end
+  end  
+
+  describe '#path' do
+    it 'returns the path if it is not a symlink' do
+      File.stub :symlink?, false do
+        filename = rand.to_s
+        connection.file(filename).path.must_equal filename
+      end
+    end
+
+    it 'returns the link_path if it is a symlink' do
+      File.stub :symlink?, true do
+        file_obj = connection.file(rand.to_s)
+        file_obj.stub :link_path, '/path/to/resolved_link' do
+          file_obj.path.must_equal '/path/to/resolved_link'
+        end
+      end
+    end
+  end
+
+  describe '#link_path' do
+    it 'returns file\'s link path' do
+      out = rand.to_s
+      File.stub :realpath, out do
+        File.stub :symlink?, true do
+          connection.file(rand.to_s).link_path.must_equal out
+        end
+      end
+    end
+  end
+end