toft-puppet 0.0.9

data/scripts/lxc-templates/lxc-natty

@@ -0,0 +1,237 @@
+#!/bin/bash
+
+echo "Creating natty node..."
+
+configure_ubuntu()
+{
+    rootfs=$1
+    hostname=$2
+
+	# disable selinux in ubuntu
+    mkdir -p $rootfs/selinux
+    echo 0 > $rootfs/selinux/enforce
+
+	# add host root ssh access
+    mkdir $rootfs/root/.ssh
+    chmod 0600 $rootfs/root/.ssh
+	cat <<-EOF > $rootfs/root/.ssh/authorized_keys
+ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQDCguB7XL3ARzLZYLsIMZe4UUO371m+H5C6V8MhtmSlgXtgHDo7eZhNSm5zCeoyGd32OKeLxuxCCEkXfDDF1aa2a6twcASE3pmWNdnBS7auiOH4P7g+eQ4Aw9v7DdESbIVgHF/NDiAEFFdmApYNM3oCX2FhEVNVKxkkIokUr4axYFJzmJ6Xoi5Sd8JtPC85FZVXqDucZDnHQlOcCkbSo0UOmsWQGwtu8eUHoDeUG0dB8ntb9xlBeLctdrAPhuFYCX8IfFkdcakkzv61ETPbKE6g9wdTDC/TEep7/AHGYmarziRnwKiVOL1jnE1coOJLqy8wOC3dKGmRZy9D4sTc+FRV root insecure public key
+EOF
+	
+	cp /usr/lib/lxc/templates/files/rc.local $rootfs/etc/rc.local
+
+    # disable selinux in ubuntu
+    mkdir -p $rootfs/selinux
+    echo 0 > $rootfs/selinux/enforce
+
+    # set the hostname
+    cat <<EOF > $rootfs/etc/hostname
+$hostname
+EOF
+    # set minimal hosts
+    cat <<EOF > $rootfs/etc/hosts
+127.0.0.1 $hostname localhost
+EOF
+
+    # suppress log level output for udev
+	sed -i "s/=\"err\"/=0/" $rootfs/etc/udev/udev.conf
+
+	# remove jobs for consoles 5 and 6 since we only create 4 consoles in
+	# this template
+	rm -f $rootfs/etc/init/tty{5,6}.conf
+
+    echo "Set root password to 'root'"
+    echo "root:root" | chroot $rootfs chpasswd
+
+    return 0
+}
+
+copy_ubuntu()
+{
+    cache=$1
+    arch=$2
+    rootfs=$3
+
+    # make a local copy of the miniubuntu
+    echo "Extracting rootfs image to $rootfs ..."
+    mkdir $rootfs
+    tar zxf $cache/natty-$arch.tar.gz -C $rootfs || return 1
+    return 0
+}
+
+install_ubuntu()
+{
+    cache="/var/cache/lxc"
+    rootfs=$1
+    mkdir -p /var/lock/subsys/
+    (
+		flock -n -x 200
+		if [ $? -ne 0 ]; then
+		    echo "Cache repository is busy."
+		    return 1
+		fi
+
+		arch=$(arch)
+		if [ "$arch" == "x86_64" ]; then
+		    arch=amd64
+		fi
+
+		if [ "$arch" == "i686" ]; then
+		    arch=i386
+		fi
+
+		echo "Checking image cache in $cache/rootfs-$arch ... "
+		if [ ! -e "$cache/rootfs-$arch" ]; then
+		    if [ $? -ne 0 ]; then
+			echo "Failed to download 'ubuntu base'"
+			return 1
+		    fi
+		fi
+
+		copy_ubuntu $cache $arch $rootfs
+		if [ $? -ne 0 ]; then
+		    echo "Failed to copy rootfs"
+		    return 1
+		fi
+
+		return 0
+
+	) 200>/var/lock/subsys/lxc
+
+    return $?
+}
+
+copy_configuration()
+{
+    path=$1
+    rootfs=$2
+    name=$3
+
+    cat <<EOF >> $path/config
+lxc.utsname = $name
+
+lxc.tty = 4
+lxc.pts = 1024
+lxc.rootfs = $rootfs
+lxc.mount  = $path/fstab
+
+lxc.cgroup.devices.deny = a
+# /dev/null and zero
+lxc.cgroup.devices.allow = c 1:3 rwm
+lxc.cgroup.devices.allow = c 1:5 rwm
+# consoles
+lxc.cgroup.devices.allow = c 5:1 rwm
+lxc.cgroup.devices.allow = c 5:0 rwm
+# lxc.cgroup.devices.allow = c 4:0 rwm
+# lxc.cgroup.devices.allow = c 4:1 rwm
+# /dev/{,u}random
+lxc.cgroup.devices.allow = c 1:9 rwm
+lxc.cgroup.devices.allow = c 1:8 rwm
+lxc.cgroup.devices.allow = c 136:* rwm
+lxc.cgroup.devices.allow = c 5:2 rwm
+# rtc
+lxc.cgroup.devices.allow = c 254:0 rwm
+EOF
+
+    cat <<EOF > $path/fstab
+proc            $rootfs/proc         proc    nodev,noexec,nosuid 0 0
+devpts          $rootfs/dev/pts      devpts defaults 0 0
+sysfs           $rootfs/sys          sysfs defaults  0 0
+EOF
+
+    if [ $? -ne 0 ]; then
+		echo "Failed to add configuration"
+		return 1
+    fi
+
+    return 0
+}
+
+clean()
+{
+    cache="/var/cache/lxc"
+
+    if [ ! -e $cache ]; then
+	exit 0
+    fi
+
+    # lock, so we won't purge while someone is creating a repository
+    (
+		flock -n -x 200
+		if [ $? != 0 ]; then
+		    echo "Cache repository is busy."
+		    exit 1
+		fi
+
+		echo -n "Purging the download cache..."
+		rm --preserve-root --one-file-system -rf $cache && echo "Done." || exit 1
+		exit 0
+    ) 200>/var/lock/subsys/lxc
+}
+
+usage()
+{
+    cat <<EOF
+$1 -h|--help -p|--path=<path> --clean
+EOF
+    return 0
+}
+
+options=$(getopt -o hp:n:c -l help,path:,name:,clean -- "$@")
+if [ $? -ne 0 ]; then
+    usage $(basename $0)
+    exit 1
+fi
+eval set -- "$options"
+
+while true
+do
+    case "$1" in
+		-h|--help)      usage $0 && exit 0;;
+		-p|--path)      path=$2; shift 2;;
+		-n|--name)      name=$2; shift 2;;
+		-c|--clean)     clean=$2; shift 2;;
+		--)             shift 1; break ;;
+        *)              break ;;
+    esac
+done
+
+if [ ! -z "$clean" -a -z "$path" ]; then
+    clean || exit 1
+    exit 0
+fi
+
+if [ -z "$path" ]; then
+    echo "'path' parameter is required"
+    exit 1
+fi
+
+if [ "$(id -u)" != "0" ]; then
+    echo "This script should be run as 'root'"
+    exit 1
+fi
+
+rootfs=$path/rootfs
+
+install_ubuntu $rootfs
+if [ $? -ne 0 ]; then
+    echo "failed to install ubuntu"
+    exit 1
+fi
+
+configure_ubuntu $rootfs $name
+if [ $? -ne 0 ]; then
+    echo "failed to configure ubuntu for a container"
+    exit 1
+fi
+
+copy_configuration $path $rootfs $name
+if [ $? -ne 0 ]; then
+    echo "failed write configuration file"
+    exit 1
+fi
+
+if [ ! -z $clean ]; then
+    clean || exit 1
+    exit 0
+fi

data/spec/fixtures/illegal_syntax.json

@@ -0,0 +1 @@
+illegal json file

data/spec/spec_helper.rb

@@ -0,0 +1,6 @@
+$:.unshift(File.dirname(__FILE__) + '/../lib')
+$:.unshift(File.dirname(__FILE__))
+
+PROJECT_ROOT = File.dirname(__FILE__) + '/..'
+
+require 'toft'

data/spec/toft/chef/chef_attributes_spec.rb

@@ -0,0 +1,39 @@
+require 'spec_helper'
+require 'cucumber/ast/table'
+
+describe "ChefAttributes" do
+  it "should parse chef dot connected attributes" do
+    ca = Toft::ChefAttributes.new(Cucumber::Ast::Table.new([[]]))
+    ca.add_attribute "one.two.three", "some"
+    ca.add_attribute "one.four", "another"
+    ca.attributes["one"]["two"]["three"].should == "some"
+    ca.attributes["one"]["four"].should == "another"
+  end
+  
+  it "should parse cucumber chef attributes ast table" do
+    table = Cucumber::Ast::Table.new([
+              ["key", "value"],
+              ["one.four", "one"],
+              ["one.two.three", "some"]
+            ])
+    ca = Toft::ChefAttributes.new(table)
+    ca.attributes["one"]["four"].should == "one"
+    ca.attributes["one"]["two"]["three"].should == "some"
+  end
+  
+  it "should return json" do
+    table = Cucumber::Ast::Table.new([
+              ["key", "value"],
+              ["one.four", "one"],
+              ["one.two.three", "some"]
+            ])
+    ca = Toft::ChefAttributes.new(table)
+    ca.to_json.should == "{\"one\":{\"two\":{\"three\":\"some\"},\"four\":\"one\"}}"
+  end
+  
+  it "should initialize with 0 parameter" do
+    ca =  Toft::ChefAttributes.new
+    ca.add_attribute "one.four", "one"
+    ca.attributes["one"]["four"].should == "one"
+  end
+end

data/spec/toft/chef/chef_runner_spec.rb

@@ -0,0 +1,34 @@
+require 'spec_helper'
+
+describe "ChefRunner" do
+  it "should throw exception if json file not exist" do
+    cf = Toft::Chef::ChefRunner.new "whatever"
+    lambda { cf.run "run list", :json => "non-exist-file" }.
+      should raise_error Errno::ENOENT, "No such file or directory - non-exist-file"
+  end
+
+  it "should throw exception if json file syntax is illegal" do
+    cf = Toft::Chef::ChefRunner.new "whatever"
+    lambda { cf.run "run list", :json => "#{PROJECT_ROOT}/spec/fixtures/illegal_syntax.json" }.
+      should raise_error JSON::ParserError
+  end
+  
+  it "should throw exception if cookbook path not set" do
+    cf = Toft::Chef::ChefRunner.new "whatever"
+    lambda { cf.run "run list" }.should raise_error ArgumentError, "Toft.cookbook_path can not be empty!"
+  end
+
+  it "should throw exception if cookbook not exist" do
+    Toft.cookbook_path = "non-exist-cookbook"
+    cf = Toft::Chef::ChefRunner.new "whatever"
+    lambda { cf.run "run list" }.should raise_error RuntimeError
+  end
+
+  it "should throw exception if roles not exist" do
+    Toft.cookbook_path = "#{PROJECT_ROOT}/fixtures/chef/cookbookse"
+    Toft.role_path = "non-exist-roles"
+    cf = Toft::Chef::ChefRunner.new "#{PROJECT_ROOT}/tmp"
+    lambda { cf.run "run list" }.should raise_error RuntimeError
+  end
+
+end

data/spec/toft/node_spec.rb

@@ -0,0 +1,18 @@
+require 'spec_helper'
+
+describe "Node" do
+
+  describe 'run_puppet' do
+
+    it 'should run block with puppet runner' do
+      puppet_runner = mock Toft::Puppet::PuppetRunner.class
+      puppet_runner.should_receive(:run).with('manifest', {})
+      
+      node = Toft::Node.new("my_host", {:runner => puppet_runner})
+      node.run_puppet('manifest')
+
+    end
+
+  end
+
+end

data/spec/toft/puppet/puppet_runner_spec.rb

@@ -0,0 +1,26 @@
+require 'spec_helper'
+
+describe "PuppetRunner" do
+
+  it "should run puppet apply to manifest" do
+    Toft.manifest_path = "#{PROJECT_ROOT}/fixtures/puppet/manifests"
+    runner = Toft::Puppet::PuppetRunner.new("whatever") do |command|
+      command.include?("puppet apply").should be_true
+    end    
+    lambda { runner.run "test" }.should_not raise_error
+  end
+
+  it "should throw exception if manifest path not set" do
+    Toft.manifest_path = ""
+    runner = Toft::Puppet::PuppetRunner.new "whatever"
+    lambda { runner.run "test" }.should raise_error ArgumentError, "Toft.manifest_path can not be empty!"
+  end
+
+
+  it "should throw exception if cookbook not exist" do
+    Toft.manifest_path = "non-exist-manifest"
+    cf = Toft::Puppet::PuppetRunner.new "whatever"
+    lambda { cf.run "test" }.should raise_error RuntimeError
+  end
+  
+end

metadata

@@ -0,0 +1,234 @@
+--- !ruby/object:Gem::Specification 
+name: toft-puppet
+version: !ruby/object:Gem::Version 
+  hash: 13
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 9
+  version: 0.0.9
+platform: ruby
+authors: 
+- Francisco Trindade
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-02-07 00:00:00 +11:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: fpm
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: rake
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: cucumber
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id004
+- !ruby/object:Gem::Dependency 
+  name: vagrant
+  prerelease: false
+  requirement: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 49
+        segments: 
+        - 0
+        - 8
+        - 7
+        version: 0.8.7
+  type: :development
+  version_requirements: *id005
+- !ruby/object:Gem::Dependency 
+  name: net-ssh
+  prerelease: false
+  requirement: &id006 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id006
+description: toft currently support testing chef, puppet, shell scripts using cucumber on lxc on ubuntu
+email: 
+- frank.trindade@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- Gemfile
+- Gemfile.lock
+- Rakefile
+- features/checker.feature
+- features/chef.feature
+- features/command.feature
+- features/node.feature
+- features/puppet.feature
+- features/step_definitions/centos/checks.rb
+- features/step_definitions/checker.rb
+- features/step_definitions/chef.rb
+- features/step_definitions/command.rb
+- features/step_definitions/node.rb
+- features/step_definitions/puppet.rb
+- features/support/env.rb
+- fixtures/chef/attributes.json
+- fixtures/chef/cookbooks/test/attributes/default.rb
+- fixtures/chef/cookbooks/test/recipes/attribute.rb
+- fixtures/chef/cookbooks/test/recipes/default.rb
+- fixtures/chef/cookbooks/test/recipes/role.rb
+- fixtures/chef/roles/test.rb
+- fixtures/puppet/conf/fileserver.conf
+- fixtures/puppet/conf/puppet.conf
+- fixtures/puppet/conf/puppet_exec.conf
+- fixtures/puppet/conf/puppet_fileserver.conf
+- fixtures/puppet/conf/puppet_modules.conf
+- fixtures/puppet/conf/puppet_template.conf
+- fixtures/puppet/manifests/fileserver/conf/test_fileserver
+- fixtures/puppet/manifests/nodes/test_node.pp
+- fixtures/puppet/manifests/site.pp
+- fixtures/puppet/manifests/templates/template_test
+- fixtures/puppet/manifests/test.pp
+- fixtures/puppet/manifests/test_fileserver.pp
+- fixtures/puppet/manifests/test_install.pp
+- fixtures/puppet/manifests/test_module.pp
+- fixtures/puppet/manifests/test_service.pp
+- fixtures/puppet/manifests/test_template.pp
+- fixtures/puppet/modules/test_module/manifests/init.pp
+- lib/toft.rb
+- lib/toft/chef/chef_attributes.rb
+- lib/toft/chef/chef_runner.rb
+- lib/toft/command_executor.rb
+- lib/toft/file_checker.rb
+- lib/toft/node.rb
+- lib/toft/node_controller.rb
+- lib/toft/puppet/puppet_runner.rb
+- lib/toft/version.rb
+- scripts/bin/centos/lxc-prepare-host
+- scripts/bin/centos/provision_vagrant
+- scripts/bin/share/install-chef-ubuntu.sh
+- scripts/bin/share/lxc-create-centos-image
+- scripts/bin/ubuntu/lxc-create-ubuntu-image
+- scripts/bin/ubuntu/lxc-prepare-host
+- scripts/bin/ubuntu/provision_vagrant
+- scripts/lxc-templates/files/rc.local
+- scripts/lxc-templates/lxc-centos-6
+- scripts/lxc-templates/lxc-lenny
+- scripts/lxc-templates/lxc-lucid
+- scripts/lxc-templates/lxc-natty
+- spec/fixtures/illegal_syntax.json
+- spec/spec_helper.rb
+- spec/toft/chef/chef_attributes_spec.rb
+- spec/toft/chef/chef_runner_spec.rb
+- spec/toft/node_spec.rb
+- spec/toft/puppet/puppet_runner_spec.rb
+has_rdoc: true
+homepage: https://github.com/frankmt/toft
+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: toft
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Fork from https://github.com/exceedhl/toft to add puppet support
+test_files: 
+- features/checker.feature
+- features/chef.feature
+- features/command.feature
+- features/node.feature
+- features/puppet.feature
+- features/step_definitions/centos/checks.rb
+- features/step_definitions/checker.rb
+- features/step_definitions/chef.rb
+- features/step_definitions/command.rb
+- features/step_definitions/node.rb
+- features/step_definitions/puppet.rb
+- features/support/env.rb
+- spec/fixtures/illegal_syntax.json
+- spec/spec_helper.rb
+- spec/toft/chef/chef_attributes_spec.rb
+- spec/toft/chef/chef_runner_spec.rb
+- spec/toft/node_spec.rb
+- spec/toft/puppet/puppet_runner_spec.rb