rush2 0.7.0

data/spec/process_set_spec.rb

@@ -0,0 +1,50 @@
+require_relative 'base'
+
+describe Rush::ProcessSet do
+	before do
+		@process = double('process')
+		@set = Rush::ProcessSet.new([ @process ])
+	end
+
+	it "is Enumerable" do
+		@set.select { |s| s == @process }.should == [ @process ]
+	end
+
+	it "defines size" do
+		@set.size.should == 1
+	end
+
+	it "defines first" do
+		@set.first.should == @process
+	end
+
+	it "is equal to sets with the same contents" do
+		@set.should == Rush::ProcessSet.new([ @process ])
+	end
+
+	it "is equal to arrays with the same contents" do
+		@set.should == [ @process ]
+	end
+
+	it "kills all processes in the set" do
+		@process.should_receive(:kill)
+		@set.kill
+	end
+
+	it "checks the alive? state of all processes in the set" do
+		@process.should_receive(:alive?).and_return(true)
+		@set.alive?.should == [ true ]
+	end
+
+	it "filters the set from a conditions hash and returns the filtered set" do
+		@process.stub(:pid).and_return(123)
+		@set.filter(:pid => 123).first.should == @process
+		@set.filter(:pid => 456).size.should == 0
+	end
+
+	it "filters with regexps if provided in the conditions" do
+		@process.stub(:command).and_return('foobaz')
+		@set.filter(:command => /baz/).first.should == @process
+		@set.filter(:command => /blerg/).size.should == 0
+	end
+end

data/spec/process_spec.rb

@@ -0,0 +1,73 @@
+require_relative 'base'
+
+describe Rush::Process do
+	before do
+		@pid = fork do
+			sleep 999
+		end
+		@process = Rush::Process.all.detect { |p| p.pid == @pid }
+	end
+
+	after do
+		system "kill -9 #{@pid}"
+	end
+
+	if !RUBY_PLATFORM.match(/darwin/)   # OS x reports pids weird
+		it "knows all its child processes" do
+			parent = Rush::Process.all.detect { |p| p.pid == Process.pid }
+			parent.children.should == [ @process ]
+		end
+	end
+
+	it "gets the list of all processes" do
+		list = Rush::Process.all
+		list.size.should > 5
+		list.first.should be_kind_of(Rush::Process)
+	end
+
+	it "knows the pid" do
+		@process.pid.should == @pid
+	end
+
+	it "knows the uid" do
+		@process.uid.should == ::Process.uid
+	end
+
+	it "knows the executed binary" do
+		@process.command.should match(/(ruby|rbx)/)
+	end
+
+	it "knows the command line" do
+		@process.cmdline.should match(/process_spec.rb/)
+	end
+
+	it "knows the memory used" do
+		@process.mem.should > 0
+	end
+
+	it "knows the cpu used" do
+		@process.cpu.should >= 0
+	end
+
+	it "knows the parent process pid" do
+		@process.parent_pid.should == Process.pid
+	end
+
+	it "knows the parent process" do
+		this = Rush::Box.new.processes.select { |p| p.pid == Process.pid }.first
+		@process.parent.should == this
+	end
+
+	it "can kill itself" do
+		process = Rush.bash("sleep 30", :background => true)
+		process.alive?.should be_true
+		process.kill
+		sleep 0.1
+		process.alive?.should be_false
+	end
+
+	it "if box and pid are the same, process is equal" do
+		other = Rush::Process.new({ :pid => @process.pid }, @process.box)
+		@process.should == other
+	end
+end

data/spec/remote_spec.rb

@@ -0,0 +1,140 @@
+require_relative 'base'
+
+describe Rush::Connection::Local do
+	before do
+		@sandbox_dir = "/tmp/rush_spec.#{Process.pid}"
+		system "rm -rf #{@sandbox_dir}; mkdir -p #{@sandbox_dir}"
+
+		@con = Rush::Connection::Remote.new('spec.example.com')
+	end
+
+	after do
+		system "rm -rf #{@sandbox_dir}"
+	end
+
+	it "transmits write_file" do
+		@con.should_receive(:transmit).with(:action => 'write_file', :full_path => 'file', :payload => 'contents')
+		@con.write_file('file', 'contents')
+	end
+
+	it "transmits append_to_file" do
+		@con.should_receive(:transmit).with(:action => 'append_to_file', :full_path => 'file', :payload => 'contents')
+		@con.append_to_file('file', 'contents')
+	end
+
+	it "transmits file_contents" do
+		@con.should_receive(:transmit).with(:action => 'file_contents', :full_path => 'file').and_return('contents')
+		@con.file_contents('file').should == 'contents'
+	end
+
+	it "transmits destroy" do
+		@con.should_receive(:transmit).with(:action => 'destroy', :full_path => 'file')
+		@con.destroy('file')
+	end
+
+	it "transmits purge" do
+		@con.should_receive(:transmit).with(:action => 'purge', :full_path => 'dir')
+		@con.purge('dir')
+	end
+
+	it "transmits create_dir" do
+		@con.should_receive(:transmit).with(:action => 'create_dir', :full_path => 'file')
+		@con.create_dir('file')
+	end
+
+	it "transmits rename" do
+		@con.should_receive(:transmit).with(:action => 'rename', :path => 'path', :name => 'name', :new_name => 'new_name')
+		@con.rename('path', 'name', 'new_name')
+	end
+
+	it "transmits copy" do
+		@con.should_receive(:transmit).with(:action => 'copy', :src => 'src', :dst => 'dst')
+		@con.copy('src', 'dst')
+	end
+
+	it "transmits read_archive" do
+		@con.should_receive(:transmit).with(:action => 'read_archive', :full_path => 'full_path').and_return('archive data')
+		@con.read_archive('full_path').should == 'archive data'
+	end
+
+	it "transmits write_archive" do
+		@con.should_receive(:transmit).with(:action => 'write_archive', :dir => 'dir', :payload => 'archive')
+		@con.write_archive('archive', 'dir')
+	end
+
+	it "transmits index" do
+		@con.should_receive(:transmit).with(:action => 'index', :base_path => 'base_path', :glob => '*').and_return("1\n2\n")
+		@con.index('base_path', '*').should == %w(1 2)
+	end
+
+	it "transmits stat" do
+		@con.should_receive(:transmit).with(:action => 'stat', :full_path => 'full_path').and_return(YAML.dump(1 => 2))
+		@con.stat('full_path').should == { 1 => 2 }
+	end
+
+	it "transmits set_access" do
+		@con.should_receive(:transmit).with(:action => 'set_access', :full_path => 'full_path', :user => 'joe', :user_read => 1)
+		@con.set_access('full_path', :user => 'joe', :user_read => 1)
+	end
+
+	it "transmits size" do
+		@con.should_receive(:transmit).with(:action => 'size', :full_path => 'full_path').and_return("123")
+		@con.size('full_path').should == 123
+	end
+
+	it "transmits processes" do
+		@con.should_receive(:transmit).with(:action => 'processes').and_return(YAML.dump([ { :pid => 1 } ]))
+		@con.processes.should == [ { :pid => 1 } ]
+	end
+
+	it "transmits process_alive" do
+		@con.should_receive(:transmit).with(:action => 'process_alive', :pid => 123).and_return(true)
+		@con.process_alive(123).should == true
+	end
+
+	it "transmits kill_process" do
+		@con.should_receive(:transmit).with(:action => 'kill_process', :pid => 123, :payload => YAML.dump(:wait => 10))
+		@con.kill_process(123, :wait => 10)
+	end
+
+	it "transmits bash" do
+		@con.should_receive(:transmit).with(:action => 'bash', :payload => 'cmd', :user => 'user', :background => 'bg', :reset_environment => false).and_return('output')
+		@con.bash('cmd', 'user', 'bg', false).should == 'output'
+	end
+
+	it "an http result code of 401 raises NotAuthorized" do
+		lambda { @con.process_result("401", "") }.should raise_error(Rush::NotAuthorized)
+	end
+
+	it "an http result code of 400 raises the exception passed in the result body" do
+		@con.stub(:parse_exception).and_return(Rush::DoesNotExist, "message")
+		lambda { @con.process_result("400", "") }.should raise_error(Rush::DoesNotExist)
+	end
+
+	it "an http result code of 501 (or anything other than the other defined codes) raises FailedTransmit" do
+		lambda { @con.process_result("501", "") }.should raise_error(Rush::FailedTransmit)
+	end
+
+	it "parse_exception takes the class from the first line and the message from the second" do
+		@con.parse_exception("Rush::DoesNotExist\nthe message\n").should == [ Rush::DoesNotExist, "the message" ]
+	end
+
+	it "parse_exception rejects unrecognized exceptions" do
+		lambda { @con.parse_exception("NotARushException\n") }.should raise_error
+	end
+
+	it "passes through ensure_tunnel" do
+		@con.tunnel.should_receive(:ensure_tunnel)
+		@con.ensure_tunnel
+	end
+
+	it "is alive if the box is responding to commands" do
+		@con.should_receive(:index).and_return(:dummy)
+		@con.should be_alive
+	end
+
+	it "not alive if an attempted command throws an exception" do
+		@con.should_receive(:index).and_raise(RuntimeError)
+		@con.should_not be_alive
+	end
+end

data/spec/rush_spec.rb

@@ -0,0 +1,28 @@
+require_relative 'base'
+
+describe Rush do
+	it "fetches a local file path" do
+		Rush['/etc/hosts'].full_path.should == '/etc/hosts'
+	end
+
+	it "fetches the dir of __FILE__" do
+		Rush.dir(__FILE__).name.should == 'spec'
+	end
+
+	it "fetches the launch dir (aka current working directory or pwd)" do
+		Dir.stub(:pwd).and_return('/tmp')
+		Rush.launch_dir.should == Rush::Box.new['/tmp/']
+	end
+
+	it "runs a bash command" do
+		Rush.bash('echo hi').should == "hi\n"
+	end
+
+	it "gets the list of local processes" do
+		Rush.processes.should be_kind_of(Rush::ProcessSet)
+	end
+
+	it "gets my process" do
+		Rush.my_process.pid.should == Process.pid
+	end
+end

data/spec/search_results_spec.rb

@@ -0,0 +1,44 @@
+require_relative 'base'
+
+describe Rush::SearchResults do
+	before do
+		@results = Rush::SearchResults.new(/pat/)
+		@file = Rush::File.new("file")
+	end
+
+	it "returns its list of entries" do
+		@results.add(@file, %w(a))
+		expect(@results.entries).to eq [ @file ]
+	end
+
+	it "only returns each entry once no matter how many line matches it has" do
+		@results.add(@file, %w(a b))
+		expect(@results.entries).to eq [ @file ]
+	end
+
+	it "returns its list of matched lines" do
+		@results.add(@file, %w(a b))
+		expect(@results.lines).to eq %w(a b)
+	end
+
+	it "returns all lines for each entry in a flattened form" do
+		file2 = Rush::File.new("another file")
+		@results.add(@file, %w(a b))
+		@results.add(file2, %w(c d))
+		expect(@results.lines).to eq %w(a b c d)
+	end
+
+	it "returns a hash of entries_with_lines" do
+		@results.add(@file, %w(a))
+		expect(@results.entries_with_lines).to eq({ @file => %w(a) })
+	end
+
+	it "mixes in Commands to operate like a dir or entry array" do
+		expect(@results.methods.include?(:search)).to eq(true)
+	end
+
+	it "mixes in Enumerable to behave like an array" do
+		@results.add(@file, %w(a))
+		expect(@results.map { |e| e }).to eq [ @file ]
+	end
+end

data/spec/shell_spec.rb

@@ -0,0 +1,35 @@
+require_relative 'base'
+require_relative '../lib/rush/shell'
+
+describe Rush::Shell do
+	before do
+		@shell = Rush::Shell.new
+	end
+
+ it 'Complete constants' do
+   expect(@shell.complete('Obj')).to eq(["Object", "ObjectSpace"])
+   expect(@shell.complete('Rush::Sh')).to eq(["Rush::Shell"])
+ end
+
+ it 'Complete executables' do
+   expect(@shell.complete('rub')).to include 'ruby'
+ end
+
+ it 'Complete global method names' do
+   eval('def wakawaka; p "hi"; end', @shell.pure_binding)
+   expect(@shell.complete('waka')).to eq ["wakawaka"]
+ end
+
+ it 'Complete method names' do
+   expect(@shell.complete('Rush.meth')).
+     to eq(["Rush.method_part", "Rush.method_defined?", "Rush.methods", "Rush.method"])
+   expect(@shell.complete('Rush.methods.inc')).to eq ["Rush.methods.include?"]
+ end
+
+ it 'Complete paths' do
+   expect(@shell.complete('root["bin/ba')).to  eq ["root[\"bin/bash"]
+   expect(@shell.complete('root[\'bin/ba')).to eq ["root['bin/bash"]
+   expect(@shell.complete('root/"bin/ba')).to  eq ["root/\"bin/bash"]
+   expect(@shell.complete('root/\'bin/ba')).to eq ["root/'bin/bash"]
+ end
+end

data/spec/ssh_tunnel_spec.rb

@@ -0,0 +1,122 @@
+require_relative 'base'
+
+describe Rush::SshTunnel do
+	before do
+		@tunnel = Rush::SshTunnel.new('spec.example.com')
+		@tunnel.stub(:config).and_return(mock_config_start)
+		@tunnel.stub(:display)
+	end
+
+	after do
+		mock_config_cleanup
+	end
+
+	it "ensure_tunnel sets everything up for the tunnel when one does not already exist" do
+		@tunnel.should_receive(:push_credentials)
+		@tunnel.should_receive(:launch_rushd)
+		@tunnel.should_receive(:establish_tunnel)
+		@tunnel.ensure_tunnel
+	end
+
+	it "ensure_tunnel uses the existing port as long as the tunnel is still alive" do
+		@tunnel.should_receive(:tunnel_alive?).and_return(true)
+		@tunnel.instance_eval("@port = 2345")
+		@tunnel.ensure_tunnel
+		@tunnel.port.should == 2345
+	end
+
+	it "existing tunnel is used when it is specified in the tunnels file" do
+		@tunnel.config.tunnels_file.write "spec.example.com:4567\n"
+		@tunnel.should_receive(:tunnel_alive?).and_return(true)
+		@tunnel.should_not_receive(:setup_everything)
+		@tunnel.ensure_tunnel
+		@tunnel.port.should == 4567
+	end
+
+	it "tunnel host is always local" do
+		@tunnel.host.should == 'localhost'
+	end
+
+	it "picks the first port number when there are no tunnels yet" do
+		@tunnel.next_available_port.should == 7771
+	end
+
+	it "picks the next port number when there is already a tunnel" do
+		@tunnel.config.tunnels_file.write("#{@tunnel.host}:7771")
+		@tunnel.next_available_port.should == 7772
+	end
+
+	it "establishes a tunnel and saves it to ~/.rush/tunnels" do
+		@tunnel.should_receive(:make_ssh_tunnel)
+		@tunnel.should_receive(:port).exactly(0).times
+		@tunnel.establish_tunnel
+		@tunnel.config.tunnels_file.contents.should == "spec.example.com:7771\n"
+	end
+
+	it "converts instance vars to options hash for ssh_tunnel_command" do
+		@tunnel.instance_eval("@port = 1234")
+		@tunnel.tunnel_options.should == {
+			:local_port => 1234,
+			:remote_port => 7770,
+			:ssh_host => 'spec.example.com'
+		}
+	end
+
+	it "ssh_stall_command uses an infinite loop for :timeout => :infinite" do
+		@tunnel.ssh_stall_command(:timeout => :infinite).should match(/while .* sleep .* done/)
+	end
+
+	it "ssh_stall_command sleeps for the number of seconds given as the :timeout option" do
+		@tunnel.ssh_stall_command(:timeout => 123).should == "sleep 123"
+	end
+
+	it "ssh_stall_command uses the default timeout when no options are given" do
+		@tunnel.ssh_stall_command.should == "sleep 9000"
+	end
+
+	it "constructs the ssh tunnel command (everything but stall) from the options hash" do
+		@tunnel.should_receive(:tunnel_options).at_least(:once).and_return(
+			:local_port => 123,
+			:remote_port => 456,
+			:ssh_host => 'example.com'
+		)
+		@tunnel.ssh_tunnel_command_without_stall.should == "ssh -f -L 123:127.0.0.1:456 example.com"
+	end
+
+	it "combines the tunnel command without stall and the stall command into the final command" do
+		@tunnel.should_receive(:ssh_tunnel_command_without_stall).and_return('ssh command')
+		@tunnel.should_receive(:ssh_stall_command).and_return('sleep 123')
+		@tunnel.ssh_tunnel_command.should == 'ssh command "sleep 123"'
+	end
+
+	it "ssh_tunnel_command request that the port be set" do
+		@tunnel.should_receive(:tunnel_options).at_least(:once).and_return(:local_port => nil)
+		lambda { @tunnel.ssh_tunnel_command }.should raise_error(Rush::SshTunnel::NoPortSelectedYet)
+	end
+
+
+	it "push_credentials uses ssh to append to remote host's passwords file" do
+		@tunnel.should_receive(:ssh_append_to_credentials).and_return(true)
+		@tunnel.push_credentials
+	end
+
+	it "launches rushd on the remote host via ssh" do
+		@tunnel.should_receive(:ssh) do |cmd|
+			cmd.should match(/rushd/)
+		end
+		@tunnel.launch_rushd
+	end
+
+	it "tunnel_alive? checks whether a tunnel is still up" do
+		@tunnel.should_receive(:tunnel_count_command).and_return("echo 1")
+		@tunnel.tunnel_alive?.should be_true
+	end
+
+	it "tunnel_count_command greps ps to find the ssh tunnel" do
+		@tunnel.should_receive(:ssh_tunnel_command_without_stall).and_return('ssh command')
+		command = @tunnel.tunnel_count_command
+		command.should match(/ps/)
+		command.should match(/grep/)
+		command.should match(/ssh command/)
+	end
+end