tddium-old 0.4.7

data/rails/init.rb

@@ -0,0 +1 @@
+require "tddium_loader.rb"

data/tddium-old.gemspec

@@ -0,0 +1,110 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{tddium-old}
+  s.version = "0.4.7"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Jay Moorthi"]
+  s.date = %q{2011-12-26}
+  s.default_executable = %q{tddium-old}
+  s.description = %q{tddium gets your rspec+selenium tests into the cloud by running them on your VMs}
+  s.email = %q{info@tddium.com}
+  s.executables = ["tddium-old"]
+  s.extra_rdoc_files = [
+    "LICENSE.txt",
+    "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+    "CHANGELOG",
+    "Gemfile",
+    "Gemfile.lock",
+    "LICENSE.txt",
+    "README.rdoc",
+    "Rakefile",
+    "VERSION",
+    "doc/aws-keypair-example.tiff",
+    "doc/aws-secgroup-example.tiff",
+    "lib/spec_storm/action_controller_ext.rb",
+    "lib/spec_storm/action_view_ext.rb",
+    "lib/spec_storm/active_record_ext.rb",
+    "lib/tddium.rb",
+    "lib/tddium/config.rb",
+    "lib/tddium/instance.rb",
+    "lib/tddium/parallel.rb",
+    "lib/tddium/rails.rb",
+    "lib/tddium/reporting.rb",
+    "lib/tddium/ssh.rb",
+    "lib/tddium/taskalias.rb",
+    "lib/tddium/tasks.rb",
+    "lib/tddium_helper.rb",
+    "lib/tddium_loader.rb",
+    "parallelrun",
+    "rails/init.rb",
+    "tddium-old.gemspec",
+    "test/helper.rb",
+    "test/test_config.rb",
+    "test/test_parallel.rb",
+    "test/test_tddium.rb"
+  ]
+  s.homepage = %q{http://www.tddium.com/}
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.6.2}
+  s.summary = %q{tddium Cloud Test Runner}
+  s.test_files = [
+    "test/helper.rb",
+    "test/test_config.rb",
+    "test/test_parallel.rb",
+    "test/test_tddium.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<highline>, [">= 0"])
+      s.add_runtime_dependency(%q<rspec>, ["< 2.0.0", "> 1.2.6"])
+      s.add_runtime_dependency(%q<parallel>, [">= 0"])
+      s.add_runtime_dependency(%q<selenium-client>, [">= 1.2.18"])
+      s.add_runtime_dependency(%q<fog>, ["= 0.4.0"])
+      s.add_development_dependency(%q<shoulda>, [">= 0"])
+      s.add_development_dependency(%q<bundler>, ["~> 1.0.0"])
+      s.add_development_dependency(%q<jeweler>, ["~> 1.5.1"])
+      s.add_development_dependency(%q<rcov>, [">= 0"])
+      s.add_development_dependency(%q<mocha>, [">= 0"])
+      s.add_development_dependency(%q<fakefs>, [">= 0.3.1"])
+      s.add_development_dependency(%q<rails>, ["= 2.3.8"])
+    else
+      s.add_dependency(%q<highline>, [">= 0"])
+      s.add_dependency(%q<rspec>, ["< 2.0.0", "> 1.2.6"])
+      s.add_dependency(%q<parallel>, [">= 0"])
+      s.add_dependency(%q<selenium-client>, [">= 1.2.18"])
+      s.add_dependency(%q<fog>, ["= 0.4.0"])
+      s.add_dependency(%q<shoulda>, [">= 0"])
+      s.add_dependency(%q<bundler>, ["~> 1.0.0"])
+      s.add_dependency(%q<jeweler>, ["~> 1.5.1"])
+      s.add_dependency(%q<rcov>, [">= 0"])
+      s.add_dependency(%q<mocha>, [">= 0"])
+      s.add_dependency(%q<fakefs>, [">= 0.3.1"])
+      s.add_dependency(%q<rails>, ["= 2.3.8"])
+    end
+  else
+    s.add_dependency(%q<highline>, [">= 0"])
+    s.add_dependency(%q<rspec>, ["< 2.0.0", "> 1.2.6"])
+    s.add_dependency(%q<parallel>, [">= 0"])
+    s.add_dependency(%q<selenium-client>, [">= 1.2.18"])
+    s.add_dependency(%q<fog>, ["= 0.4.0"])
+    s.add_dependency(%q<shoulda>, [">= 0"])
+    s.add_dependency(%q<bundler>, ["~> 1.0.0"])
+    s.add_dependency(%q<jeweler>, ["~> 1.5.1"])
+    s.add_dependency(%q<rcov>, [">= 0"])
+    s.add_dependency(%q<mocha>, [">= 0"])
+    s.add_dependency(%q<fakefs>, [">= 0.3.1"])
+    s.add_dependency(%q<rails>, ["= 2.3.8"])
+  end
+end
+

data/test/helper.rb

@@ -0,0 +1,31 @@
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'tddium'
+
+class Test::Unit::TestCase
+end
+
+require 'fakefs'
+class FakeFS::File::Stat
+  @@themode = 0600
+  def mode
+    return @@themode
+  end
+
+  def self.mode=(newmode)
+    @@themode = newmode
+  end
+end
+

data/test/test_config.rb

@@ -0,0 +1,194 @@
+=begin
+Copyright (c) 2010 tddium.com All Rights Reserved
+=end
+
+require 'helper'
+require 'fakefs'
+require 'fileutils'
+require 'mocha'
+
+class TestFileops < Test::Unit::TestCase
+  context "init task" do
+    setup do
+      @path = File.expand_path(get_config_paths[0])
+    end
+
+    context "when ~/.tddium doesn't exist" do
+      setup do
+        FakeFS::FileSystem.clear
+        HighLine.any_instance.stubs(:ask).returns('abc', 'def', 'ghi')
+        @keys = %w(aws_key aws_secret test_pattern key_directory key_name result_directory server_tag ssh_tunnel)
+        init_task
+      end
+    
+      should "write ~/.tddium" do
+        assert File.exists?(@path), "#{@path} doesn't exist"
+      end
+
+      should "have the right contents" do
+        f = FakeFS::FileSystem.find(@path)
+        assert f.content.include?('abc'), "Should contain magic string"
+        @keys.each do |field|
+          assert f.content.include?(field), "Should contain #{field}"
+        end
+      end
+
+      should "write a YAML file with the right fields" do
+        result = YAML::load_file(@path)
+        @keys.each do |field|
+          assert result.has_key?(field.to_sym), "Should contain #{field}"
+        end
+      end 
+    end
+
+    context "when ~/.tddium exists" do
+      setup do
+        File.new(@path, 'w').write('a')
+      end
+
+      should "print a warning message" do
+        HighLine.expects(:ask).never
+        init_task
+      end
+    end
+  end
+end
+
+class TestConfigRead < Test::Unit::TestCase
+  context "read config" do
+    context "file exists" do
+      setup do
+        @path = File.expand_path(get_config_paths[0])
+      end
+
+      should "read config file values into a dict" do
+        File.open(@path, 'w') do |f|
+          f.write <<EOF
+---
+:aws_secret: abc
+:aws_key: abx
+:test_pattern: **/*_spec.rb
+EOF
+        end
+        conf = read_config
+        assert_equal conf[:aws_key], 'abx'
+        assert_equal conf[:aws_secret], 'abc'
+        assert_equal conf[:test_pattern], '**/*_spec.rb'
+      end
+    end
+
+    context "file doesn't exist" do
+      should "return nils" do
+        conf = read_config
+        assert_nil conf[:aws_key]
+        assert_nil conf[:aws_secret]
+      end
+    end
+  end
+
+  context "find_config" do
+    should "check rails root if it's set" do
+      oldroot = ENV['RAILS_ROOT']
+      ENV['RAILS_ROOT'] = '/home/rails'
+      File.expects(:exists?).with('/home/rails/.tddium')
+      File.expects(:exists?).with(File.expand_path('~/.tddium'))
+      File.expects(:exists?).with('.tddium')
+      assert_nil find_config
+      ENV['RAILS_ROOT'] = oldroot
+    end
+
+    should "find rails root file if it's there" do
+      oldroot = ENV['RAILS_ROOT']
+      ENV['RAILS_ROOT'] = '/home/rails'
+      path = get_config_paths[0]
+      assert_equal '/home/rails/.tddium', path
+      File.open(path, 'w') {|f| f.write('a')}
+      assert_equal path, find_config
+      ENV['RAILS_ROOT'] = oldroot
+    end
+  end
+end
+
+class TestKeyFile < Test::Unit::TestCase
+  context "get_keyfile" do
+    setup do
+      @config = {:key_directory => 'dir', :key_name => 'key'}
+      stubs(:read_config => @config).once
+      FakeFS::FileSystem.clear
+    end
+
+    context "no keyfile present" do
+      should "return nil" do
+        STDERR.expects(:puts).once
+        assert_nil get_keyfile
+      end
+    end
+
+    context "keyfile wrong perms" do
+      should "return nil" do
+        FileUtils.mkdir_p @config[:key_directory]
+        File.open(key_file_name(@config), 'w') do |f|
+          f.write 'abc'
+        end
+        FakeFS::File::Stat.mode=0644
+        STDERR.expects(:puts).once
+        assert_nil get_keyfile
+      end
+    end
+
+    context "keyfile OK" do
+      should "return keyfile name" do
+        @name = key_file_name(@config)
+        FileUtils.mkdir_p @config[:key_directory]
+        File.open(@name, 'w') do |f|
+          f.write 'abc'
+        end
+        FakeFS::File::Stat.mode=0600
+        STDERR.expects(:puts).never
+        assert_equal @name, get_keyfile
+      end
+    end
+  end
+end
+
+class TestFiles < Test::Unit::TestCase
+  context "find_test_files" do
+    setup do
+      @config = {:test_pattern => 'a'}
+      stubs(:read_config => @config)
+    end
+    
+    should "be callable" do
+      assert find_test_files
+    end
+  end
+end
+
+class TestSpecOpts < Test::Unit::TestCase
+  context "spec_opts" do
+    should "return a list" do
+      stubs(:read_config => CONFIG_DEFAULTS)
+      opts = spec_opts('a')
+      assert opts.is_a?(Array)
+      assert_equal 1, opts.select{|x| /require/.match(x)}.size
+    end
+
+    should "include require_files" do
+      stubs(:read_config => {:require_files => 'a.rb'})
+      opts = spec_opts('a')
+      assert opts.is_a?(Array)
+      assert_equal 2, opts.select{|x| /require/.match(x)}.size
+    end
+
+    should "read TDDIUM_SPEC_OPTS environment" do
+      stubs(:read_config => CONFIG_DEFAULTS)
+      specopts = 'abcdefgh'
+      oldenv = ENV['TDDIUM_SPEC_OPTS']
+      ENV['TDDIUM_SPEC_OPTS'] = specopts
+      opts = spec_opts('a')
+      ENV['TDDIUM_SPEC_OPTS'] = oldenv
+      assert opts.is_a?(Array)
+      assert_equal 1, opts.select{|x| /#{specopts}/.match(x)}.size
+    end
+  end
+end

data/test/test_parallel.rb

@@ -0,0 +1,47 @@
+require 'helper'
+require 'mocha'
+require 'fakefs'
+require 'tddium/parallel'
+
+class TestParallel < Test::Unit::TestCase
+  context "make_spec_cmd" do
+    should "raise if no result_path specified" do
+      assert_raise RuntimeError do
+        make_spec_cmd([1], 'a', nil)
+      end
+    end
+    should "suppress RSPEC_COLOR if stdout is not tty" do
+      $stdout.stubs(:tty? => false)
+      assert_no_match /RSPEC_COLOR/, make_spec_cmd(['a'], 'a', 'a')
+    end
+  end
+
+  context "test_batches" do
+    setup do
+      @config = {:test_pattern => '*.rb'}
+      stubs(:read_config => @config)
+      @files = %w{a b c d e}
+      @files.each { |f| File.open("#{f}.rb", 'w') { |o| o.write('a') } }
+    end
+    should "find batches" do
+      [1,2,3,4,5,6].each do |n|
+        b = test_batches(n)
+        assert_equal n, b.size
+      end
+    end
+  end
+
+  context "parallel_task" do
+    setup do
+      @batches = [['a', 'b'], ['c']]
+      stubs(:test_batches => @batches)
+      stubs(:execute_command => 'a')
+    end
+    should "run with defaults" do
+      args = mock()
+      args.expects(:with_defaults)
+      args.stubs(:threads => 5, :inspect => 'args', :environment => 'selenium', :result_directory => '.')
+      parallel_task(args)
+    end
+  end
+end

data/test/test_tddium.rb

@@ -0,0 +1,372 @@
+=begin
+Copyright (c) 2010 tddium.com All Rights Reserved
+=end
+
+require 'helper'
+require 'fakefs'
+require 'fileutils'
+require 'mocha'
+
+class FakeFS::File::Stat
+  @@themode = 0600
+  def mode
+    return @@themode
+  end
+
+  def self.mode=(newmode)
+    @@themode = newmode
+  end
+end
+
+class TestEC2 < Test::Unit::TestCase
+  context "start instances" do
+    setup do
+      Fog.mock!
+      @config = {:aws_key => 'abc', :aws_secret => 'def'}
+      stubs(:read_config => @config)
+      stubs(:system => 0)
+      mockstart
+    end
+
+    should "create new EC2 instance with configured key" do
+      server = start_instance
+      assert_equal server.image_id, AMI_NAME
+      assert_equal nil, $tunnel_pid
+    end
+
+    should "not crash if keyfile is not provided"
+
+    should "set SELENIUM_RC_HOST environment variable" do
+      server = start_instance
+      assert_equal server.dns_name, ENV['SELENIUM_RC_HOST']
+    end
+
+    should "set TDDIUM environment variable" do
+      server = start_instance
+      assert_equal '1', ENV['TDDIUM']
+    end
+
+    context "if ssh_tunnel is needed" do
+      setup do
+        @config[:ssh_tunnel] = 1
+        @config[:key_name] = 'sg-keypair'
+        @config[:key_directory] = '.'
+        FakeFS::FileSystem.clear
+        File.open(key_file_name(@config), 'w', 0600) do |f|
+          f.write('foo')
+        end
+        @testpid = 10000
+        Process.stubs(:fork => @testpid)
+      end
+
+      should "set up an ssh tunnel and save its pid" do
+        server = start_instance
+        assert_equal @testpid,  $tunnel_pid
+        $tunnel_pid = nil
+      end
+
+      should "set the SELENIUM_RC_HOST environment variable to localhost" do
+        server = start_instance
+        assert_equal 'localhost', ENV['SELENIUM_RC_HOST']
+        $tunnel_pid = nil
+      end
+    end
+  end
+  
+  context "stop instance" do
+    setup do
+      Fog.mock!
+      config = {:aws_key => 'abc', :aws_secret => 'def'}
+      stub(:read_config => config)
+      mockstart
+    end
+
+    should "stop instances" do
+      server = start_instance
+      assert server.ready?
+      stop_instance
+      assert server.terminated?
+    end
+  end
+
+  context "find_instances" do
+    setup do
+      Fog.mock!
+      config = {:aws_key => 'abc', :aws_secret => 'def'}
+      stub(:read_config => config)
+    end
+
+    should "exist as a method" do
+      find_instances
+    end
+
+    should "not find any instances" do
+      result = find_instances
+      assert_nil result
+    end
+
+    should "find an instance" do
+      mockstart
+      server = start_instance
+      result = find_instances
+      assert_equal server.id, result[0].id
+    end
+
+    should "not filter by tag"
+  end
+
+  context "session_instances" do
+    setup do
+      Fog.mock!
+      config = {:aws_key => 'abc', :aws_secret => 'def'}
+      stub(:read_config => config)
+      stop_all_instances
+    end
+
+    should "exist as a method" do
+      session_instances('a')
+    end
+
+    should "not find any instances" do
+      result = session_instances('b')
+      assert_nil result
+    end
+
+    should "find an instance" do
+      mockstart
+      server = start_instance
+      result = session_instances(@tddium_session)
+      assert_equal server.id, result[0].id
+    end
+  end
+
+  context "stopall instances" do
+    setup do
+      Fog.mock!
+      config = {:aws_key => 'abc', :aws_secret => 'def'}
+      stub(:read_config => config)
+    end
+
+    should "exist as a method" do
+      stop_all_instances
+    end
+
+    should "destroy all instances" do
+      instancemock1 = mock().expects(:destroy)
+      instancemock2 = mock().expects(:destroy)
+      instances = [instancemock1, instancemock2]
+      stub(:find_instances => instances)
+      stop_all_instances
+    end
+  end
+
+  context "checkstart dev instance" do
+    setup do
+      Fog.mock!
+      config = {:aws_key => 'abc', :aws_secret => 'def'}
+      stub(:read_config => config)
+      mockstart
+    end
+
+    should "exist as a method" do
+      checkstart_dev_instance
+    end
+
+    should "only start 1 instance" do
+      server1 = checkstart_dev_instance
+      server2 = checkstart_dev_instance
+      assert_equal server1.id, server2.id
+    end
+  end
+
+  private
+
+    def mockstart
+      httpmock = mock()
+      httpmock.expects(:open_timeout=)
+      httpmock.expects(:read_timeout=)
+      httpmock.expects(:request)
+      Net::HTTP.stubs(:new).returns(httpmock)
+      getmock = mock()
+      Net::HTTP::Get.stubs(:new).returns(getmock)
+    end
+end
+
+class TestSshTunnel < Test::Unit::TestCase
+  should "have ssh tunnel method" do
+    stubs(:system => true)
+    ssh_tunnel('abc')
+  end
+
+  context "kill_tunnel" do
+    should "kill tunnel" do
+      @testpid = 10000
+      $tunnel_pid = @testpid
+      Process.expects(:kill).with("TERM", @testpid)
+      Process.expects(:waitpid).with(@testpid)
+      kill_tunnel
+    end
+
+    should "not kill if no tunnel" do
+      $tunnel_pid = nil
+      Process.expects(:kill).never
+      kill_tunnel
+    end
+  end
+end
+
+class TestEnvSet < Test::Unit::TestCase
+  context "setup_environment" do
+    setup do
+      @smock = mock()
+      @smock.stubs(:dns_name => 'a')
+    end
+
+    should "always set TDDIUM environment" do
+      setup_environment(@smock)
+      assert_equal ENV['TDDIUM'], '1'
+    end
+
+    should "set RC_HOST to remote if no tunnel" do
+      setup_environment(@smock)
+      assert_equal ENV['SELENIUM_RC_HOST'], @smock.dns_name
+    end
+
+    should "set RC_HOST to localhost if tunnel" do
+      $tunnel_pid = 1
+      setup_environment(@smock)
+      assert_equal ENV['SELENIUM_RC_HOST'], 'localhost'
+      $tunnel_pid = nil
+    end
+  end
+end
+
+class TestLogRotate < Test::Unit::TestCase
+  context "calling result_directory" do
+    setup do
+      @config = {:result_directory => 'results'}
+      @latest = File.join(@config[:result_directory], 'latest')
+      stubs(:read_config => @config)
+    end
+    context "regardless" do
+      should "return directory name" do
+        x = result_directory
+        assert_equal @latest, x
+      end
+    end
+    context "with no results" do
+      setup do
+        FakeFS::FileSystem.clear
+      end
+      should "create results/latest/" do
+        result_directory
+        assert File.directory?(@latest)
+      end
+    end
+    
+    context "with existing <results>/latest directory" do
+      setup do
+        FileUtils.mkdir_p @latest
+      end
+
+      should "rotate latest to date-extended directory name" do
+        result_directory
+        files = Dir.glob("#{@config[:result_directory]}/*")
+        assert_equal 2, files.length
+      end
+
+      should "preserve contents of rotated report" do
+        fname = File.join(@latest, 'report.html')
+        File.open(fname, 'w') do |f|
+          f.write('foo')
+        end
+        result_directory
+        assert !File.exists?(fname)
+        files = Dir.glob('**/report.html')
+        assert_equal 1, files.length
+        assert_equal 'foo', File.open(files[0]).read
+      end
+    end
+  end
+
+  context "default result path" do
+    setup do
+      @config = {:result_directory => "results"}
+      stubs(:read_config => @config)
+    end
+    should "return the right path" do
+      x = default_report_path
+      expected = File.join(@config[:result_directory], 'latest', REPORT_FILENAME)
+      assert_equal expected, x
+    end
+  end
+end
+
+class TestCollectLogs < Test::Unit::TestCase
+  context "no keyfile" do
+    setup do
+      stubs(:get_keyfile => nil)
+    end
+    should "raise an error" do
+      assert_raise RuntimeError do
+        collect_syslog
+      end
+    end
+  end
+
+  context "normal mode" do
+    context "no host started" do
+      setup do
+        stubs(:system => true).never
+        stubs(:session_instances => [])
+        stubs(:get_keyfile => 'a')
+      end
+      should "fail gracefully" do
+        collect_syslog
+      end
+    end
+
+    context "host started" do
+      setup do
+        @host = 'somehost'
+        mockinst = mock()
+        mockinst.stubs(:dns_name => @host)
+        stubs(:session_instances => [mockinst])
+        @keyfile = 'keyfile'
+        stubs(:get_keyfile => @keyfile)
+      end
+      should "scp from host" do
+        stubs(:system).twice.with() do |cmd| 
+          (cmd =~ /#{@host}/ && 
+           cmd =~ /^scp/ && 
+           cmd =~ /#{@keyfile}/ && 
+           cmd =~ /\.\/selenium-rc.#{@host}/) ||
+          (cmd =~ /#{@host}/ && 
+           cmd =~ /^scp/ && 
+           cmd =~ /#{@keyfile}/ && 
+           cmd =~ /\.\/selenium-hub.#{@host}/)
+        end
+        collect_syslog
+      end
+
+      should "let target directory be specified" do
+        target = 'targetdir'
+        stubs(:system).twice.with() do |cmd| 
+          (cmd =~ /#{@host}/ && 
+           cmd =~ /^scp/ && 
+           cmd =~ /#{@keyfile}/ && 
+           cmd =~ /#{target}\/selenium-rc.#{@host}/) ||
+          (cmd =~ /#{@host}/ && 
+           cmd =~ /^scp/ && 
+           cmd =~ /#{@keyfile}/ && 
+           cmd =~ /#{target}\/selenium-hub.#{@host}/)
+        end
+        collect_syslog(target)
+      end
+    end
+  end
+
+  context "dev mode" do
+    should "use dev mode host"
+  end
+end