necktie 0.1.1 → 0.2.0

data/rush/spec/process_spec.rb

@@ -0,0 +1,73 @@
+require File.dirname(__FILE__) + '/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/)
+	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/rush/spec/remote_spec.rb

@@ -0,0 +1,140 @@
+require File.dirname(__FILE__) + '/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').and_return('output')
+		@con.bash('cmd', 'user', 'bg').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/rush/spec/rush_spec.rb

@@ -0,0 +1,28 @@
+require File.dirname(__FILE__) + '/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/rush/spec/search_results_spec.rb

@@ -0,0 +1,44 @@
+require File.dirname(__FILE__) + '/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))
+		@results.entries.should == [ @file ]
+	end
+
+	it "only returns each entry once no matter how many line matches it has" do
+		@results.add(@file, %w(a b))
+		@results.entries.should == [ @file ]
+	end
+
+	it "returns its list of matched lines" do
+		@results.add(@file, %w(a b))
+		@results.lines.should == %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))
+		@results.lines.should == %w(a b c d)
+	end
+
+	it "returns a hash of entries_with_lines" do
+		@results.add(@file, %w(a))
+		@results.entries_with_lines.should == { @file => %w(a) }
+	end
+
+	it "mixes in Commands to operate like a dir or entry array" do
+		@results.methods.include?("search").should be_true
+	end
+
+	it "mixes in Enumerable to behave like an array" do
+		@results.add(@file, %w(a))
+		@results.map { |e| e }.should == [ @file ]
+	end
+end

data/rush/spec/shell_spec.rb

@@ -0,0 +1,23 @@
+require File.dirname(__FILE__) + '/base'
+require 'rush/shell'
+
+describe Rush::Shell do
+	before do
+		@shell = Rush::Shell.new
+	end
+
+	it "matches open path commands for readline tab completion" do
+		@shell.path_parts("dir['app").should == [ "dir", "[", "'", "app", "" ]
+		@shell.path_parts('dir/"app').should == [ "dir", "/", '"', "app", "" ]
+	end
+
+	it "matches open path commands on globals for readline tab completion" do
+		@shell.path_parts("$dir['app").should == [ "$dir", "[", "'", "app", "" ]
+		@shell.path_parts('$dir/"app').should == [ "$dir", "/", '"', "app", "" ]
+	end
+
+	it "matches open path commands on instance vars for readline tab completion" do
+		@shell.path_parts("@dir['app").should == [ "@dir", "[", "'", "app", "" ]
+		@shell.path_parts('@dir/"app').should == [ "@dir", "/", '"', "app", "" ]
+	end
+end

data/rush/spec/ssh_tunnel_spec.rb

@@ -0,0 +1,122 @@
+require File.dirname(__FILE__) + '/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.and_return('7771')   # avoid infinite loop
+		@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

data/rush/spec/string_ext_spec.rb

@@ -0,0 +1,23 @@
+require File.dirname(__FILE__) + '/base'
+
+describe String do
+	before do
+		@string = "abc"
+	end
+
+	it "heads from the front of the string" do
+		@string.head(1).should == 'a'
+	end
+
+	it "tails from the back of the string" do
+		@string.tail(1).should == 'c'
+	end
+
+	it "gives the whole string when head exceeds length" do
+		@string.head(999).should == @string
+	end
+
+	it "gives the whole string when tail exceeds length" do
+		@string.tail(999).should == @string
+	end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: necktie
 version: !ruby/object:Gem::Version 
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors: 
 - Assaf Arkin
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-10-09 00:00:00 -07:00
+date: 2009-10-10 00:00:00 -07:00
 default_executable: 
 dependencies: []
 
@@ -23,10 +23,60 @@ extra_rdoc_files: []
 
 files: 
 - bin/necktie
-- lib/necktie/files.rb
 - lib/necktie/gems.rb
+- lib/necktie/rush.rb
 - lib/necktie/services.rb
 - lib/necktie.rb
+- rush/bin/rush
+- rush/bin/rushd
+- rush/lib/rush/access.rb
+- rush/lib/rush/array_ext.rb
+- rush/lib/rush/box.rb
+- rush/lib/rush/commands.rb
+- rush/lib/rush/config.rb
+- rush/lib/rush/dir.rb
+- rush/lib/rush/embeddable_shell.rb
+- rush/lib/rush/entry.rb
+- rush/lib/rush/exceptions.rb
+- rush/lib/rush/file.rb
+- rush/lib/rush/find_by.rb
+- rush/lib/rush/fixnum_ext.rb
+- rush/lib/rush/head_tail.rb
+- rush/lib/rush/local.rb
+- rush/lib/rush/process.rb
+- rush/lib/rush/process_set.rb
+- rush/lib/rush/remote.rb
+- rush/lib/rush/search_results.rb
+- rush/lib/rush/server.rb
+- rush/lib/rush/shell.rb
+- rush/lib/rush/ssh_tunnel.rb
+- rush/lib/rush/string_ext.rb
+- rush/lib/rush.rb
+- rush/Rakefile
+- rush/README.rdoc
+- rush/rush.gemspec
+- rush/spec/access_spec.rb
+- rush/spec/array_ext_spec.rb
+- rush/spec/base.rb
+- rush/spec/box_spec.rb
+- rush/spec/commands_spec.rb
+- rush/spec/config_spec.rb
+- rush/spec/dir_spec.rb
+- rush/spec/embeddable_shell_spec.rb
+- rush/spec/entry_spec.rb
+- rush/spec/file_spec.rb
+- rush/spec/find_by_spec.rb
+- rush/spec/fixnum_ext_spec.rb
+- rush/spec/local_spec.rb
+- rush/spec/process_set_spec.rb
+- rush/spec/process_spec.rb
+- rush/spec/remote_spec.rb
+- rush/spec/rush_spec.rb
+- rush/spec/search_results_spec.rb
+- rush/spec/shell_spec.rb
+- rush/spec/ssh_tunnel_spec.rb
+- rush/spec/string_ext_spec.rb
+- rush/VERSION
 - necktie.gemspec
 has_rdoc: true
 homepage: ""

data/lib/necktie/files.rb

@@ -1,14 +0,0 @@
-def fix_file(file, &block)
-  text = block.call(File.read(file))
-  File.open file, "w" do |out|
-    out.write text
-  end
-end
-
-def touch(name)
-  if File.exist?(name)
-    File.utime time = Time.now, time, "foo"
-  else
-    File.open(name, "a").close
-  end
-end