puppet-testing 0.0.1

data/.gitignore

@@ -0,0 +1,6 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*
+*.swp
+*.swo

data/.rvmrc

@@ -0,0 +1 @@
+rvm use ruby-1.8.7-p334

data/Gemfile

@@ -0,0 +1,16 @@
+source "http://rubygems.org"
+gem 'ruby-libvirt' ,'0.3'
+gem 'cucumber' ,'0.3'
+gem 'rspec' , '1.3.1'
+gem 'net-ssh'
+gem 'fire_poll'
+gem 'differ'
+
+group :development do
+	gem 'irbtools'
+	gem 'minitest'
+	gem 'guard'
+	gem 'guard-minitest'
+	gem 'libnotify'
+	gem 'rb-inotify'
+end

data/Guardfile

@@ -0,0 +1,14 @@
+# A sample Guardfile
+# More info at https://github.com/guard/guard#readme
+
+guard 'minitest' do
+  # with Minitest::Unit
+  watch(%r|^test/(.*)/test_(.*)\.rb|)
+  watch(%r|^lib/(.*)/(.*)\.rb|)     { |m| "test/#{m[1]}/test_#{m[2]}.rb" }
+  watch(%r|^test/test_helper\.rb|)    { "test" }
+
+  # with Minitest::Spec
+  # watch(%r|^spec/(.*)_spec\.rb|)
+  # watch(%r|^lib/(.*)\.rb|)            { |m| "spec/#{m[1]}_spec.rb" }
+  # watch(%r|^spec/spec_helper\.rb|)    { "spec" }
+end

data/Rakefile

@@ -0,0 +1,2 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks

data/features/PuppetRun.feature

@@ -0,0 +1,9 @@
+Feature: Puppet run  
+    Running puppet on a simple node
+    
+    Scenario: Basic puppet run on puppet-client 
+	  Given the following vms are started:
+		| vm_name      |
+		| puppet       |
+		| puppet-client|
+	  And puppetd run on "puppet-client"

data/features/Sanity.feature

@@ -0,0 +1,10 @@
+Feature: Sanity 
+    Vm lifecycle 
+
+    Scenario: Vm manipulation and cleanup
+	  When I start "puppet-client"
+	  Then I should be able to ssh into "puppet-client"
+	  Given I save into "/tmp/puppet-client.state"
+	  And I touch "/tmp/sanity" on "puppet-client"
+	  When I restore back "/tmp/puppet-client.state"
+	  Then "/tmp/sanity" shouldn't exist on "puppet-client"

data/features/step_defenitions/puppet_run.rb

@@ -0,0 +1,23 @@
+require 'libvirt'
+require 'net/ssh'
+require 'lib/ssh/port'
+
+Before do 
+    @conn = Libvirt::open("qemu:///system")
+end
+
+
+Given /^the following vms are started:$/ do |table|
+    table.hashes.each do |hash|
+	  @domain = @conn.lookup_domain_by_name(hash[:vm_name])
+	  @domain.create if not @domain.active? 
+    end
+end
+
+And /puppetd run on "(.*)"/ do |host|
+    ssh_after_boot(host,{:timeout => 60,:user => 'ronen'}) do |ssh|
+	 ssh.exec!('puppetd --test --report').chomp
+    end
+end
+
+

data/features/step_defenitions/sanity.rb

@@ -0,0 +1,48 @@
+require 'libvirt'
+require 'net/ssh'
+require 'lib/ssh/port'
+begin require 'rspec'; rescue LoadError; require 'rspec/expectations'; end
+
+Before do 
+    @conn = Libvirt::open("qemu:///system")
+end
+
+
+When /I start "(.*)"/ do |name|
+    @domain = @conn.lookup_domain_by_name(name)
+    @domain.create if not @domain.active? 
+end
+
+Then /I should be able to ssh into "(.*)"/ do |host|
+    result = ""
+    ssh_after_boot(host,{:timeout => 120,:user => 'ronen'}) do |ssh|
+	  result = ssh.exec!('pwd').chomp
+    end
+    result.should == '/home/ronen'
+end
+
+Given /^I save into "(.*)"$/ do |file|
+    @domain.save(file)
+    @domain.create
+end
+
+When /I restore back "(.*)"/ do |file|
+    @domain.destroy
+    Libvirt::Domain::restore(@conn, file)
+end
+
+And /I touch "(.*)" on "(.*)"/ do |file,host|
+    ssh_to(host,{:timeout => 120,:user => 'ronen'}) do |ssh|
+	  ssh.exec!("touch #{file}")
+    end
+end
+
+
+Then /"(.*)" shouldn't exist on "(.*)"/ do |file,host|
+    result = ""
+    ssh_after_boot(host,{:timeout => 60,:user => 'ronen'}) do |ssh|
+	  result = ssh.exec!("ls #{file}").chomp
+    end
+    result.should == 'ls: cannot access /tmp/sanity: No such file or directory'
+
+end

data/lib/console.rb

@@ -0,0 +1,8 @@
+require 'console/libvirt'
+require 'fire_poll'
+require 'differ'
+require 'console/expectancies'
+require 'console/conversation'
+require 'console/kvm_machine'
+
+

data/lib/console/conversation.rb

@@ -0,0 +1,65 @@
+require 'logger'
+
+module Console
+    class Conversation
+	  include Console::Expectancies
+	  def initialize(domain)
+		shell = PTY.spawn "virsh console #{domain}"	
+		@o,@i,@pid = shell[0],shell[1],shell[2]
+	  end
+
+	  def self.talk_with(domain,&block)
+		conv = Conversation.new(domain)
+		begin
+		    conv.instance_eval(&block)
+		ensure 
+		    conv.end_talk
+		end
+	  end   	
+
+	  def enter(input)
+		@i.puts(input)
+		@i.flush
+	  end
+
+	  def exepecting(expected)
+		output =''
+		begin
+		    FirePoll.poll(nil,5) do
+			  output << read
+			  output.eql?(expected)
+		    end 	
+		rescue 
+		    puts 'diff exists between expected input and actual output:'
+		    puts Differ.diff_by_char expected, output
+		end
+		puts output
+		output
+	  end
+
+	  def end_talk
+		enter '[^'
+		@o.close
+		@i.close
+		begin
+		    Process.wait @pid
+		rescue PTY::ChildExited
+		rescue Errno::ECHILD 
+		    raise Exception.new 'console process not started, make sure domain is up and running'
+		end
+	  end
+
+	  def read 
+		buffer = ""
+		begin
+		    loop { buffer << @o.read_nonblock(10)}
+		rescue Errno::EAGAIN
+		    # nothing to read now
+		rescue EOFError
+		    # read was done
+		end
+		buffer
+	  end
+
+    end
+end

data/lib/console/expectancies.rb

@@ -0,0 +1,42 @@
+module Console
+    module Expectancies
+	  def connected 
+		res = <<-E
+Connected to domain puppet-client\r
+Escape character is ^]\r
+		E
+	  end
+
+	  def post_uname
+		res =  <<-E
+ronen
+Password: 
+E
+	  end
+
+	  def logged_in
+		res = <<-E
+Welcome to Ubuntu 11.04 (GNU/Linux 2.6.38-8-virtual x86_64)\r
+
+ * Documentation:  https://help.ubuntu.com/\r
+ronen@puppet-client:~$ 
+		E
+	  end
+
+	  def	logout
+		res = <<-E
+logout\r
+\r
+Ubuntu 11.04 puppet-client ttyS0\r
+\r
+puppet-client login: 
+		E
+	  end
+	  
+	  def waiting_shell
+            "\r\nronen@puppet-client:~$ "
+	  end
+
+    end
+end
+

data/lib/console/kvm_machine.rb

@@ -0,0 +1,43 @@
+require 'pty'
+
+module Console 
+    class KvmMachine
+	  def initialize(domain)
+           @domain = domain
+	  end
+
+	  def login
+		Console::Conversation.talk_with(@domain) do
+		    exepecting  connected
+		    enter 'ronen'
+		    exepecting post_uname.chomp!
+		    enter "xaL2ohM0"
+		    sleep 3
+		    puts read
+		    # exepecting logged_in -> we can't match since it contains date
+		end
+	  end 	
+  
+
+  	  def execute(cli)
+		Console::Conversation.talk_with(@domain) do
+		    exepecting connected
+		    enter ''
+		    exepecting waiting_shell
+		    enter cli
+		    exepecting waiting_shell
+		end
+  	    	
+  	  end
+
+  	  def logout
+		Console::Conversation.talk_with(@domain) do
+		    exepecting connected
+		    enter 'logout'
+		    return exepecting logout.chomp!
+		end
+  	  end
+
+      
+    end
+end

data/lib/console/libvirt.rb

@@ -0,0 +1,22 @@
+require 'etc'
+
+PTS_PATH = '/dev/pts' 
+LIBVIRT_USER= 'libvirt-qemu'
+LIBVIRT_GROUP= 'kvm'
+
+module Console 
+    class Libvirt
+
+    	  def initialize(path=PTS_PATH)
+    	     @pts_path=path	
+    	  end
+
+	  def libvirt_pts
+		Dir.entries(@pts_path).find_all do |pts|
+		    uid = File.stat("#{@pts_path}/#{pts}").uid
+		    Etc.getpwuid(uid).name.eql?(LIBVIRT_USER)
+		end
+	  end
+    end    	
+end
+

data/lib/puppet-testing.rb

@@ -0,0 +1,5 @@
+module Puppet
+  module Testing
+    # Your code goes here...
+  end
+end

data/lib/puppet-testing/version.rb

@@ -0,0 +1,5 @@
+module Puppet
+  module Testing
+    VERSION = "0.0.1"
+  end
+end

data/lib/ssh/port.rb

@@ -0,0 +1,39 @@
+require 'net/ssh'
+
+def ssh_to(host, options = {} , &block)
+    Net::SSH.start(host,options[:user], options) do|ssh|
+	  block.call(ssh)
+    end   
+end
+
+def ssh_after_boot(host, options = {} , &block)
+    execute_when_tcp_available(host,options) do |host|
+	  ssh_to(host,options) 	do |ssh|
+	    block.call(ssh)	
+	  end
+    end
+end
+
+# see http://tinyurl.com/3ht46m5 
+def execute_when_tcp_available(ip, options = { } , &block)
+    defaults={ :port => 22, :timeout => 2 , :pollrate => 5}
+    options=defaults.merge(options)
+    begin
+	  Timeout::timeout(options[:timeout]) do
+		connected=false
+		while !connected do
+		    begin
+			  s = TCPSocket.new(ip, options[:port])
+			  s.close
+			  block.call(ip)
+			  return true
+		    rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH, SocketError
+			  sleep options[:pollrate]
+		    end
+		end
+	  end
+    rescue Timeout::Error
+	  raise 'timeout connecting to port'
+    end
+    return false
+end

data/puppet-testing.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "puppet-testing/version"
+
+Gem::Specification.new do |s|
+  s.name        = "puppet-testing"
+  s.version     = Puppet::Testing::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["narkisr"]
+  s.email       = ["narkisr@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{A testing gem for puppet}
+  s.description = %q{Testing puppet using kvm and cucumber}
+
+  s.rubyforge_project = "puppet-testing"
+  s.add_dependency  'ruby-libvirt', '0.3'
+  s.add_dependency 'cucumber' , '0.3'
+  s.add_dependency 'rspec' , '1.3.1'
+  s.add_dependency 'net-ssh'
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+end

data/test/console/test_kvm_machine.rb

@@ -0,0 +1,24 @@
+require 'minitest/autorun'
+require 'test_helper'
+require 'console'
+
+class TestPtyShell < MiniTest::Unit::TestCase
+
+    def setup
+      @machine= Console::KvmMachine.new('puppet-client')
+    end
+
+    def test_domain_down
+       machine = Console::KvmMachine.new('random-name')
+       assert_raises Exception do
+          machine.login
+       end
+    end
+
+    def test_ls_la
+         @machine.login
+         assert_includes @machine.execute('ls -la'), 'total 36'
+         @machine.logout
+    end
+
+end

data/test/console/test_libvirt.rb

@@ -0,0 +1,23 @@
+require 'minitest/autorun'
+require 'test_helper'
+require 'console'
+require 'tmpdir'
+
+class TestConsole < MiniTest::Unit::TestCase
+    include FileUtils
+
+    def test_pts_filtering
+    	  mocked_pts do |dir|
+         pts = Console::Libvirt.new(dir).libvirt_pts
+         assert_includes_all ['1','0'], pts
+    	  end
+    end
+
+    def mocked_pts
+	  Dir.mktmpdir('pts') do |dir|
+		%w(1 0 2 3 4).each {|p| touch "#{dir}/#{p}" }
+		%w(1 0).each {|p| chown LIBVIRT_USER, LIBVIRT_GROUP, "#{dir}/#{p}"}
+		yield dir
+	  end      	
+    end
+end

data/test/test_console.rb

@@ -0,0 +1 @@
+

data/test/test_helper.rb

@@ -0,0 +1,9 @@
+$LOAD_PATH.unshift( 'lib') unless $LOAD_PATH.include?('lib')
+
+
+module MiniTest::Assertions
+    def assert_includes_all first , sec , msg =''
+      msg =  message(msg) {"Expected #{sec.inspect} to contain all the elements of #{first.inspect}" }
+      assert first.all?{|e| sec.include?(e)} , msg
+    end
+end

metadata

@@ -0,0 +1,149 @@
+--- !ruby/object:Gem::Specification 
+name: puppet-testing
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- narkisr
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-07-10 00:00:00 +03:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: ruby-libvirt
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 13
+        segments: 
+        - 0
+        - 3
+        version: "0.3"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: cucumber
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 13
+        segments: 
+        - 0
+        - 3
+        version: "0.3"
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 25
+        segments: 
+        - 1
+        - 3
+        - 1
+        version: 1.3.1
+  type: :runtime
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: net-ssh
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id004
+description: Testing puppet using kvm and cucumber
+email: 
+- narkisr@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- .rvmrc
+- Gemfile
+- Gemfile.lock
+- Guardfile
+- Rakefile
+- features/PuppetRun.feature
+- features/Sanity.feature
+- features/step_defenitions/puppet_run.rb
+- features/step_defenitions/sanity.rb
+- lib/console.rb
+- lib/console/conversation.rb
+- lib/console/expectancies.rb
+- lib/console/kvm_machine.rb
+- lib/console/libvirt.rb
+- lib/puppet-testing.rb
+- lib/puppet-testing/version.rb
+- lib/ssh/port.rb
+- puppet-testing.gemspec
+- test/console/test_kvm_machine.rb
+- test/console/test_libvirt.rb
+- test/test_console.rb
+- test/test_helper.rb
+has_rdoc: true
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: puppet-testing
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: A testing gem for puppet
+test_files: []
+