drbqs 0.0.13 → 0.0.14

data/spec/task/file_transfer_spec.rb

@@ -0,0 +1,107 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+require 'drbqs/utility/transfer/file_transfer'
+
+describe DRbQS::FileTransfer do
+  def create_file(path, str)
+    path = File.join(@tmp, path)
+    open(path, 'w') do |f|
+      f.puts str
+    end
+    path
+  end
+
+  def create_directory(path)
+    path = File.join(@tmp, path)
+    FileUtils.mkdir_p(path)
+    path
+  end
+
+  before(:all) do
+    @tmp = FileName.create('/tmp/drbqs_test', :directory => :self)
+  end
+
+  it "should be empty." do
+    DRbQS::FileTransfer.empty?.should be_true
+  end
+
+  it "should return nil when file queue is empty" do
+    DRbQS::FileTransfer.dequeue_all.should be_nil
+  end
+
+  it "should enqueue path." do
+    path = create_file('file1.txt', 'file1')
+    DRbQS::FileTransfer.enqueue(path)
+    DRbQS::FileTransfer.dequeue.should == path
+  end
+
+  it "should compress and enqueue path." do
+    path = create_file('file2.txt', 'file2')
+    DRbQS::FileTransfer.enqueue(path, :compress => true)
+    DRbQS::FileTransfer.dequeue.should == path + '.gz'
+  end
+
+  it "should rename and enqueue path." do
+    path = create_file('file3.txt', 'file3')
+    rename = 'rename.txt'
+    new_path = File.join(@tmp, rename)
+    DRbQS::FileTransfer.enqueue(path, :rename => rename)
+    DRbQS::FileTransfer.dequeue.should == new_path
+    File.exist?(new_path).should be_true
+  end
+
+  it "should rename to some directory." do
+    path = create_file('file4.txt', 'file4')
+    rename = 'dir/rename.txt'
+    new_path = File.join(@tmp, rename)
+    DRbQS::FileTransfer.enqueue(path, :rename => rename)
+    DRbQS::FileTransfer.dequeue.should == new_path
+    File.exist?(new_path).should be_true
+  end
+
+  it "should rename, compress, and enqueue path." do
+    path = create_file('file5.txt', 'file5')
+    rename = 'dir2/rename.txt'
+    new_path = File.join(@tmp, rename + '.gz')
+    DRbQS::FileTransfer.enqueue(path, :rename => rename, :compress => true)
+    DRbQS::FileTransfer.dequeue.should == new_path
+    File.exist?(new_path).should be_true
+  end
+
+  it "should return array of files" do
+    files = [create_file('file6.txt', 'file6'),
+             create_file('file8.txt', 'file8'),
+             create_file('file8.txt', 'file8')]
+    files.each do |path|
+      DRbQS::FileTransfer.enqueue(path)
+    end
+    DRbQS::FileTransfer.dequeue_all.should == files
+    DRbQS::FileTransfer.empty?.should be_true
+  end
+
+  it "should enqueue path of directory." do
+    path = create_directory('dir/dir1')
+    DRbQS::FileTransfer.enqueue(path)
+    DRbQS::FileTransfer.dequeue.should == path
+  end
+
+  it "should compress and enqueue path of directory." do
+    path = create_directory('dir/dir2')
+    file_path = create_file('dir/dir2/test.txt', 'hello world')
+    DRbQS::FileTransfer.enqueue(path, :compress => true)
+    DRbQS::FileTransfer.dequeue.should == path + '.tar.gz'
+  end
+
+  it "should rename, compress, and enqueue path of directory." do
+    path = create_directory('dir/dir3')
+    file_path = create_file('dir/dir3/test.txt', 'hello world')
+    rename = 'rename_dir'
+    DRbQS::FileTransfer.enqueue(path, :compress => true, :rename => rename)
+    DRbQS::FileTransfer.dequeue.should == File.join(@tmp, 'dir', rename + '.tar.gz')
+  end
+
+  after(:all) do
+    FileUtils.rm_r(@tmp)
+  end
+
+end

data/spec/{task_generator_spec.rb → task/task_generator_spec.rb}

@@ -1,6 +1,6 @@
-require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
 
-require 'drbqs/task_generator'
+require 'drbqs/task/task'
 
 describe DRbQS::TaskGenerator do
   def check_task_ary(tasks, num, cl = DRbQS::Task)

data/spec/{task_spec.rb → task/task_spec.rb}

@@ -1,4 +1,6 @@
-require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+require 'drbqs/task/task'
 
 describe DRbQS::Task do
   it "should not set hook" do

data/spec/utility/argument_spec.rb

@@ -0,0 +1,39 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+require 'drbqs/utility/command_line'
+
+describe DRbQS::CommandLineArgument do
+
+  it "should split arguments" do
+    ary = ['abc', 'def', '--', '123', '45', '6']
+    a1, a2 = DRbQS::CommandLineArgument.split_arguments(ary)
+    a1.should == ['abc', 'def']
+    a2.should == ['123', '45', '6']
+  end
+
+  context "when checking size of array" do
+    it "should return true" do
+      ary = [1, 2, 3]
+      DRbQS::CommandLineArgument.check_argument_size(ary, :>=, 1).should be_true
+    end
+
+    it "should return true" do
+      ary = [1, 2, 3]
+      DRbQS::CommandLineArgument.check_argument_size(ary, :>=, 1, :<=, 4).should be_true
+    end
+
+    it "should raise error" do
+      ary = [1, 2, 3]
+      lambda do
+        DRbQS::CommandLineArgument.check_argument_size(ary, :==, 9)
+      end.should raise_error
+    end
+
+    it "should raise error" do
+      ary = [1, 2, 3]
+      lambda do
+        DRbQS::CommandLineArgument.check_argument_size(ary, :>, 0, :<=, 2)
+      end.should raise_error
+    end
+  end
+end

data/spec/utility/command_line/command_base_spec.rb

@@ -0,0 +1,33 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../spec_helper')
+
+require 'drbqs/utility/command_line'
+
+describe DRbQS::CommandBase do
+  it "should define DRbQS::CommandBase.exec." do
+    argv = [1, 2, 3]
+    obj = mock
+    DRbQS::CommandBase.should_receive(:new).and_return(obj)
+    obj.should_receive(:parse_option)
+    obj.should_receive(:exec)
+    DRbQS::CommandBase.exec(argv)
+  end
+
+  subject do
+    DRbQS::CommandBase.new
+  end
+
+  it "should exit with 0." do
+    Kernel.should_receive(:exit).with(0)
+    subject.__send__(:exit_normally)
+  end
+
+  it "should exit." do
+    Kernel.should_receive(:exit)
+    subject.__send__(:exit_unusually)
+  end
+
+  it "should exit." do
+    Kernel.should_receive(:exit)
+    subject.__send__(:exit_invalid_option)
+  end
+end

data/spec/utility/command_line/commands_spec.rb

@@ -0,0 +1,15 @@
+require File.expand_path(File.dirname(__FILE__) + '/../../spec_helper')
+
+require 'drbqs/utility/command_line'
+
+[DRbQS::CommandServer, DRbQS::CommandNode, DRbQS::CommandManage, DRbQS::CommandSSH].each do |cls|
+  describe cls do
+    it "should have defined parse_option." do
+      cls.method_defined?(:parse_option).should be_true
+    end
+
+    it "should have defined exec." do
+      cls.method_defined?(:exec).should be_true
+    end
+  end
+end

data/spec/utility/misc_spec.rb

@@ -0,0 +1,77 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+require 'drbqs/utility/temporary'
+
+describe DRbQS::LoggerDummy do
+  subject do
+    DRbQS::LoggerDummy.new
+  end
+
+  it "should respond to info." do
+    subject.should respond_to(:info)
+  end
+
+  it "should respond to warn." do
+    subject.should respond_to(:warn)
+  end
+
+  it "should respond to error." do
+    subject.should respond_to(:error)
+  end
+
+  it "should respond to debug." do
+    subject.should respond_to(:debug)
+  end
+end
+
+describe DRbQS::Misc do
+  context "when creating new uri" do
+    it "should return uri of druby." do
+      DRbQS::Misc.create_uri(:port => 10000).should == "druby://:10000"
+    end
+
+    it "should return uri of drbunix." do
+      tmp_path = DRbQS::Temporary.file
+      DRbQS::Misc.create_uri(:unix => tmp_path).should == "drbunix:#{tmp_path}"
+    end
+
+    it "should raise error for non-existing parent directory." do
+      tmp_dir = DRbQS::Temporary.file
+      tmp_path = File.join(tmp_dir, 'tmp_file')
+      lambda do
+        DRbQS::Misc.create_uri(:unix => tmp_path)
+      end.should raise_error
+    end
+
+    it "should raise error for existing path." do
+      tmp_path = DRbQS::Temporary.file
+      open(tmp_path, 'w') do |f|
+        f.puts 'hello world'
+      end
+      lambda do
+        DRbQS::Misc.create_uri(:unix => tmp_path)
+      end.should raise_error
+    end
+
+    after(:all) do
+      DRbQS::Temporary.delete_all
+    end
+  end
+
+  it "should create logger." do
+    logger = DRbQS::Misc.create_logger(File.join(HOME_FOR_SPEC, 'tmp.log'), Logger::INFO)
+    logger.should be_an_instance_of(Logger)
+  end
+
+  it "should return time string for history." do
+    DRbQS::Misc.time_to_history_string(Time.now).should be_an_instance_of String
+  end
+
+  it "should return ramdom key" do
+    a = DRbQS::Misc.random_key
+    b = DRbQS::Misc.random_key
+    a.should be_an_instance_of String
+    b.should be_an_instance_of String
+    a.should_not == b
+  end
+end

data/spec/utility/server_define_spec.rb

@@ -0,0 +1,59 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+describe DRbQS::ServerDefinition do
+  context "when we call class methods" do
+    subject do
+      DRbQS.class_variable_get(:@@server_def)
+    end
+
+    it "should define server." do
+      lambda do
+        DRbQS.define_server do |server, argv, opts|
+          server.add_hook(:finish) do |serv|
+            serv.exit
+          end
+        end
+      end.should change { subject.instance_variable_get(:@default_server_opts) }.from(nil).to({})
+    end
+
+    it "should set parser of options." do
+      lambda do
+        DRbQS.option_parser do |opt, hash|
+          opt.on('--test') do |v|
+            hash[:test] = true
+          end
+        end
+      end.should change { subject.instance_variable_get(:@option_parse) }.from(nil)
+    end
+
+    it "should parse options." do
+      lambda do
+        DRbQS.parse_option(['--test'])
+      end.should change { subject.instance_variable_get(:@argv) }.from(nil)
+    end
+
+    it "should start server." do
+      DRbQS::Server.should_receive(:new)
+      begin
+        # After DRbQS::Server.new returns nil, raise error
+        DRbQS.start_server(:uri => 'druby://localhost:13500')
+      rescue
+      end
+    end
+
+    it "should return empty help message" do
+      DRbQS.clear_definition
+      DRbQS.option_help_message.should be_nil
+    end
+
+    it "should return help message." do
+      DRbQS.option_parser do |opt, hash|
+        opt.on('--test') do |v|
+          hash[:test] = true
+        end
+      end
+      DRbQS.option_help_message.should be_an_instance_of String
+    end
+  end
+
+end

data/spec/utility/temporary_spec.rb

@@ -0,0 +1,39 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+require 'drbqs/utility/temporary'
+
+describe DRbQS::Temporary do
+  before(:all) do
+    DRbQS::Temporary.file
+  end
+
+  it "should create base directory." do
+    File.exist?(DRbQS::Temporary.root).should be_true
+  end
+
+  it "should create empty directory." do
+    dir = DRbQS::Temporary.directory
+    File.directory?(dir).should be_true
+    Dir.entries(dir).should have(2).items
+  end
+
+  it "should return new path of file." do
+    path = DRbQS::Temporary.file
+    File.exist?(path).should_not be_true
+    open(path, 'w') do |f|
+      f.puts 'hello'
+    end
+  end
+
+  it "should make directory empty." do
+    Dir.entries(DRbQS::Temporary.root).size.should > 2
+    DRbQS::Temporary.delete
+    Dir.entries(DRbQS::Temporary.root).should have(2).items
+  end
+
+  it "should all directories." do
+    root = DRbQS::Temporary.root
+    DRbQS::Temporary.delete_all
+    File.exist?(root).should_not be_true
+  end
+end