rush 0.1

data/spec/file_spec.rb

@@ -0,0 +1,75 @@
+require File.dirname(__FILE__) + '/base'
+
+describe Rush::File do
+	before do
+		@sandbox_dir = "/tmp/rush_spec.#{Process.pid}"
+		system "rm -rf #{@sandbox_dir}; mkdir -p #{@sandbox_dir}"
+
+		@filename = "#{@sandbox_dir}/test_file"
+		@contents = "1234"
+		system "echo #{@contents} > #{@filename}"
+		@contents += "\n"
+
+		@file = Rush::File.new(@filename)
+	end
+
+	after do
+		system "rm -rf #{@sandbox_dir}"
+	end
+
+	it "is a child of Rush::Entry" do
+		@file.should be_kind_of(Rush::Entry)
+	end
+
+	it "is not a dir" do
+		@file.should_not be_dir
+	end
+
+	it "can create itself as a blank file, and return itself" do
+		create_me = Rush::File.new("#{@sandbox_dir}/create_me")
+		create_me.create.should == create_me
+		File.exists?("#{@sandbox_dir}/create_me").should == true
+	end
+
+	it "knows its size in bytes" do
+		@file.size.should == @contents.length
+	end
+
+	it "can read its contents" do
+		@file.contents.should == @contents
+	end
+
+	it "can write new contents" do
+		@file.write('write test')
+		@file.contents.should == 'write test'
+	end
+
+	it "can count the number of lines it contains" do
+		@file.write("1\n2\n3\n")
+		@file.line_count.should == 3
+	end
+
+	it "searches its contents for matching lines" do
+		@file.write("a\n1\nb\n2\n")
+		@file.search(/\d/).should == %w(1 2)
+	end
+
+	it "search returns nil if no lines match" do
+		@file.write("a\nb\nc\n")
+		@file.search(/\d/).should be_nil
+	end
+
+	it "find-in-file replace" do
+		@file.replace_contents!(/\d/, 'x')
+		@file.contents.should == "xxxx\n"
+	end
+
+	it "can destroy itself" do
+		@file.destroy
+		::File.exists?(@filename).should be_false
+	end
+
+	it "can fetch contents or blank if doesn't exist" do
+		Rush::File.new('/does/not/exist').contents_or_blank.should == ""
+	end
+end

data/spec/fixnum_ext_spec.rb

@@ -0,0 +1,19 @@
+require File.dirname(__FILE__) + '/base'
+
+describe Fixnum do
+	before do
+		@num = 2
+	end
+
+	it "counts kb" do
+		@num.kb.should == 2*1024
+	end
+
+	it "counts mb" do
+		@num.mb.should == 2*1024*1024
+	end
+
+	it "counts gb" do
+		@num.gb.should == 2*1024*1024*1024
+	end
+end

data/spec/local_spec.rb

@@ -0,0 +1,196 @@
+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::Local.new
+	end
+
+	after do
+		system "rm -rf #{@sandbox_dir}"
+	end
+
+	it "receive -> write_file(file, contents)" do
+		@con.should_receive(:write_file).with('file', 'contents')
+		@con.receive(:action => 'write_file', :full_path => 'file', :payload => 'contents')
+	end
+
+	it "receive -> file_contents(file)" do
+		@con.should_receive(:file_contents).with('file').and_return('the contents')
+		@con.receive(:action => 'file_contents', :full_path => 'file').should == 'the contents'
+	end
+
+	it "receive -> destroy(file or dir)" do
+		@con.should_receive(:destroy).with('file')
+		@con.receive(:action => 'destroy', :full_path => 'file')
+	end
+
+	it "receive -> create_dir(path)" do
+		@con.should_receive(:create_dir).with('dir')
+		@con.receive(:action => 'create_dir', :full_path => 'dir')
+	end
+
+	it "receive -> rename(path, name, new_name)" do
+		@con.should_receive(:rename).with('path', 'name', 'new_name')
+		@con.receive(:action => 'rename', :path => 'path', :name => 'name', :new_name => 'new_name')
+	end
+
+	it "receive -> copy(src, dst)" do
+		@con.should_receive(:copy).with('src', 'dst')
+		@con.receive(:action => 'copy', :src => 'src', :dst => 'dst')
+	end
+
+	it "receive -> read_archive(full_path)" do
+		@con.should_receive(:read_archive).with('full_path').and_return('archive data')
+		@con.receive(:action => 'read_archive', :full_path => 'full_path').should == 'archive data'
+	end
+
+	it "receive -> write_archive(archive, dir)" do
+		@con.should_receive(:write_archive).with('archive', 'dir')
+		@con.receive(:action => 'write_archive', :dir => 'dir', :payload => 'archive')
+	end
+
+	it "receive -> index(base_path, glob)" do
+		@con.should_receive(:index).with('base_path', '*').and_return(%w(1 2))
+		@con.receive(:action => 'index', :base_path => 'base_path', :glob => '*').should == "1\n2\n"
+	end
+
+	it "receive -> stat(full_path)" do
+		@con.should_receive(:stat).with('full_path').and_return(1 => 2)
+		@con.receive(:action => 'stat', :full_path => 'full_path').should == YAML.dump(1 => 2)
+	end
+
+	it "receive -> size(full_path)" do
+		@con.should_receive(:size).with('full_path').and_return("1024")
+		@con.receive(:action => 'size', :full_path => 'full_path').should == "1024"
+	end
+
+	it "receive -> processes" do
+		@con.should_receive(:processes).with().and_return([ { :pid => 1 } ])
+		@con.receive(:action => 'processes').should == YAML.dump([ { :pid => 1 } ])
+	end
+
+	it "receive -> process_alive" do
+		@con.should_receive(:process_alive).with(123).and_return(true)
+		@con.receive(:action => 'process_alive', :pid => 123).should == '1'
+	end
+
+	it "receive -> kill_process" do
+		@con.should_receive(:kill_process).with(123).and_return(true)
+		@con.receive(:action => 'kill_process', :pid => 123)
+	end
+
+	it "receive -> unknown action exception" do
+		lambda { @con.receive(:action => 'does_not_exist') }.should raise_error(Rush::Connection::Local::UnknownAction)
+	end
+
+	it "write_file writes contents to a file" do
+		fname = "#{@sandbox_dir}/a_file"
+		data = "some data"
+		@con.write_file(fname, data)
+		File.read(fname).should == data
+	end
+
+	it "file_contents reads a file's contents" do
+		fname = "#{@sandbox_dir}/a_file"
+		system "echo stuff > #{fname}"
+		@con.file_contents(fname).should == "stuff\n"
+	end
+
+	it "destroy to destroy a file or dir" do
+		fname = "#{@sandbox_dir}/delete_me"
+		system "touch #{fname}"
+		@con.destroy(fname)
+		File.exists?(fname).should be_false
+	end
+
+	it "create_dir creates a directory" do
+		fname = "#{@sandbox_dir}/a/b/c/"
+		@con.create_dir(fname)
+		File.directory?(fname).should be_true
+	end
+
+	it "rename to rename entries within a dir" do
+		system "touch #{@sandbox_dir}/a"
+		@con.rename(@sandbox_dir, 'a', 'b')
+		File.exists?("#{@sandbox_dir}/a").should be_false
+		File.exists?("#{@sandbox_dir}/b").should be_true
+	end
+
+	it "copy to copy an entry to another dir on the same box" do
+		system "mkdir #{@sandbox_dir}/subdir"
+		system "touch #{@sandbox_dir}/a"
+		@con.copy("#{@sandbox_dir}/a", "#{@sandbox_dir}/subdir")
+		File.exists?("#{@sandbox_dir}/a").should be_true
+		File.exists?("#{@sandbox_dir}/subdir/a").should be_true
+	end
+
+	it "read_archive to pull an archive of a dir into a byte stream" do
+		system "touch #{@sandbox_dir}/a"
+		@con.read_archive(@sandbox_dir).size.should > 50
+	end
+
+	it "write_archive to turn a byte stream into a dir" do
+		system "cd #{@sandbox_dir}; mkdir -p a; touch a/b; tar cf xfer.tar a; mkdir dst"
+		archive = File.read("#{@sandbox_dir}/xfer.tar")
+		@con.write_archive(archive, "#{@sandbox_dir}/dst")
+		File.directory?("#{@sandbox_dir}/dst/a").should be_true
+		File.exists?("#{@sandbox_dir}/dst/a/b").should be_true
+	end
+
+	it "index fetches list of all files and dirs in a dir when pattern is empty" do
+		system "cd #{@sandbox_dir}; mkdir dir; touch file"
+		@con.index(@sandbox_dir, '').should == [ 'dir/', 'file' ]
+	end
+
+	it "index fetches only files with a certain extension with a flat pattern, *.rb" do
+		system "cd #{@sandbox_dir}; touch a.rb; touch b.txt"
+		@con.index(@sandbox_dir, '*.rb').should == [ 'a.rb' ]
+	end
+
+	it "stat gives file stats like size and timestamps" do
+		@con.stat(@sandbox_dir).should have_key(:ctime)
+		@con.stat(@sandbox_dir).should have_key(:size)
+	end
+
+	if !RUBY_PLATFORM.match(/darwin/)   # doesn't work on OS X 'cause du switches are different
+		it "size gives size of a directory and all its contents recursively" do
+			system "mkdir -p #{@sandbox_dir}/a/b/; echo 1234 > #{@sandbox_dir}/a/b/c"
+			@con.size(@sandbox_dir).should == (4096*3 + 5)
+		end
+	end
+
+	it "parses ps output on os x" do
+		@con.parse_ps("21712   501   1236   0 /usr/bin/vi somefile.rb").should == {
+			:pid => "21712",
+			:uid => "501",
+			:rss => "1236",
+			:cpu => "0",
+			:command => '/usr/bin/vi',
+			:cmdline => '/usr/bin/vi somefile.rb',
+		}
+	end
+
+	it "gets the list of processes on os x via the ps command" do
+		@con.should_receive(:os_x_raw_ps).and_return <<EOPS
+PID UID   RSS  CPU COMMAND
+1     0   1111   0 cmd1 args
+2   501    222   1 cmd2
+EOPS
+		@con.os_x_processes.should == [
+			{ :pid => "1", :uid => "0", :rss => "1111", :cpu => "0", :command => "cmd1", :cmdline => "cmd1 args" },
+			{ :pid => "2", :uid => "501", :rss => "222", :cpu => "1", :command => "cmd2", :cmdline => "cmd2" },
+		]
+	end
+
+	it "checks whether a given process is alive by pid" do
+		@con.process_alive(Process.pid).should == true
+	end
+
+	it "kills a process by pid" do
+		::Process.should_receive(:kill).with('TERM', 123)
+		@con.kill_process('123')
+	end
+end

data/spec/process_spec.rb

@@ -0,0 +1,44 @@
+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
+
+	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 == "ruby"
+	end
+
+	it "knows the command line" do
+		@process.cmdline.should match(/process_spec.rb/)
+	end
+
+	it "can kill itself" do
+		system "sleep 30 &"
+		@process = Rush::Process.all.detect { |p| p.command == "sleep" }
+		@process.kill
+		sleep 0.1
+		@process.alive?.should be_false
+	end
+end

data/spec/remote_spec.rb

@@ -0,0 +1,84 @@
+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 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 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 size" do
+		@con.should_receive(:transmit).with(:action => 'size', :full_path => 'full_path').and_return("")
+		@con.size('full_path')
+	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)
+		@con.kill_process(123)
+	end
+end

data/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