itamae-spec 0.0.2

data/lib/itamae-spec/task/local_itamae_task.rb

@@ -0,0 +1,84 @@
+require 'itamae-spec/task/base_task'
+
+module ItamaeSpec
+  module Task
+    class LocalItamaeTask < BaseTask
+      def create_itamae_command(node_name, hash)
+        command = 'bundle exec itamae-spec local'
+        command << " -j tmp-nodes/#{node_name}.json"
+
+        hash[:environments][:shell] = ENV['shell'] if ENV['shell']
+        command << if hash[:environments][:shell]
+                     " --shell=#{hash[:environments][:shell]}"
+                   else
+                     ' --shell=bash'
+                   end
+
+        command << ' --dry-run' if ENV['dry-run'] == 'true'
+        command << ' --log-level=debug' if ENV['debug'] == 'true'
+        command
+      end
+
+      Itamae.logger.formatter.colored = true
+      task = LocalItamaeTask.new
+
+      namespace :local_itamae do
+        all = []
+
+        Dir.glob('nodes/**/*.json').each do |node_file|
+          begin
+            node_name = File.basename(node_file, '.json')
+            node = task.load_node_attributes(node_file)
+            node_short = node[:environments][:hostname].split('.')[0]
+          rescue => e
+            Itamae.logger.error e.inspect
+            Itamae.logger.info "From node file: #{node_file}"
+            exit 2
+          end
+
+          all << node_short
+          desc 'Itamae local to all nodes'
+          task 'all' => all
+
+          desc "Itamae local to #{node_name}"
+          task node_short do
+            Itamae.logger.color(:cyan) do
+              Itamae.logger.info "Start local_itamae_task to #{node[:environments][:hostname]}"
+            end
+
+            begin
+              run_list = task.load_run_list(node_file)
+              environments = task.load_environments(node)
+              recipe_attributes_list = task.load_recipe_attributes(run_list)
+
+              merged_recipe = task.merge_attributes(recipe_attributes_list)
+              merged_environments = task.merge_attributes(merged_recipe, environments)
+              attributes = task.merge_attributes(merged_environments, node)
+              task.create_tmp_nodes(node_name, attributes)
+
+              command = task.create_itamae_command(node_name, attributes)
+              command_recipe = task.list_recipe_filepath(run_list)
+              command_recipe.sort_by! {|item| File.dirname(item) }
+              command << command_recipe.join
+
+              task.runner_display(attributes[:run_list], run_list, command)
+              st = system command
+              if st
+                Itamae.logger.color(:green) do
+                  Itamae.logger.info 'local_itamae_task is completed.'
+                end
+              else
+                Itamae.logger.error 'local_itamae_task is failed.'
+                exit 1
+              end
+            rescue => e
+              Itamae.logger.error e.inspect
+              Itamae.logger.info "From node file: #{node_file}"
+              exit 2
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/itamae-spec/task/local_serverspec_task.rb

@@ -0,0 +1,125 @@
+require 'itamae-spec/task/base_task'
+
+module ItamaeSpec
+  module Task
+    class LocalServerspecTask < BaseTask
+      ChangeTargetError = Class.new(StandardError)
+      LoadSpecError = Class.new(StandardError)
+
+      def create_spec_command(node_name, hash)
+        ENV['TARGET_HOST'] = if hash[:environments][:local_ipv4].nil?
+                               hash[:environments][:hostname]
+                             else
+                               hash[:environments][:local_ipv4]
+                             end
+
+        ENV['NODE_FILE'] = "tmp-nodes/#{node_name}.json"
+        ENV['SUDO_PASSWORD'] = hash[:environments][:sudo_password]
+        ENV['LOCAL_MODE'] = 'enable local mode'
+
+        command = 'bundle exec rspec'
+      end
+
+      def list_recipe_filepath(run_list)
+        recipes = []
+        run_list.each do |recipe|
+          target_list = Dir.glob("cookbooks/#{recipe.keys.join}/spec/#{recipe.values.join}_spec.rb")
+
+          raise LoadSpecError, "#{recipe.to_a.join('::')} cookbook or spec does not exist." if target_list.empty?
+
+          target_list.each do |target|
+            recipes << " #{target}"
+          end
+        end
+
+        recipes
+      end
+
+      Itamae.logger.formatter.colored = true
+      task = LocalServerspecTask.new
+
+      namespace :local_spec do
+        all = []
+
+        begin
+          project = { project: ARGV[1] }
+
+          if (ARGV[0] == '-T' || ARGV[0] == '--tasks') && !project[:project].nil?
+            unless Dir.exist?("nodes/#{project[:project]}")
+              raise ChangeTargetError, "'#{project[:project]}' project is not exist."
+            end
+
+            File.open 'Project.json', 'w' do |f|
+              f.flock File::LOCK_EX
+              f.puts project.to_json
+              f.flock File::LOCK_UN
+            end
+
+            Itamae.logger.color(:green) do
+              Itamae.logger.info "Changed target mode '#{project[:project]}'"
+            end
+          end
+
+          resp = JSON.parse(File.read('Project.json'))
+          target = resp['project'] << '/**'
+        rescue => e
+          Itamae.logger.error e.inspect
+          exit 2
+        end
+
+        Dir.glob("nodes/#{target}/*.json").each do |node_file|
+          begin
+            node_name = File.basename(node_file, '.json')
+            node = task.load_node_attributes(node_file)
+            node_short = node[:environments][:hostname].split('.')[0]
+          rescue => e
+            Itamae.logger.error e.inspect
+            Itamae.logger.info "From node file: #{node_file}"
+            exit 2
+          end
+
+          all << node_short
+          desc 'Serverspec to all nodes'
+          task 'all' => all
+
+          desc "Serverspec to #{node_name}"
+          task node_short do
+            Itamae.logger.color(:cyan) do
+              Itamae.logger.info "Start local_serverspec_task to #{node[:environments][:hostname]}"
+            end
+
+            begin
+              run_list = task.load_run_list(node_file)
+              environments = task.load_environments(node)
+              recipe_attributes_list = task.load_recipe_attributes(run_list)
+
+              merged_recipe = task.merge_attributes(recipe_attributes_list)
+              merged_environments = task.merge_attributes(merged_recipe, environments)
+              attributes = task.merge_attributes(merged_environments, node)
+              task.create_tmp_nodes(node_name, attributes)
+
+              command = task.create_spec_command(node_name, attributes)
+              command_recipe = task.list_recipe_filepath(run_list)
+              command << command_recipe.join
+
+              task.runner_display(attributes[:run_list], run_list, command)
+              st = system command
+              if st
+                Itamae.logger.color(:green) do
+                  Itamae.logger.info 'local_serverspec_task is completed.'
+                end
+              else
+                Itamae.logger.error 'local_serverspec_task is failed.'
+                exit 1
+              end
+            rescue => e
+              Itamae.logger.error e.inspect
+              Itamae.logger.info "From node file: #{node_file}"
+              exit 2
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/itamae-spec/task/serverspec_task.rb

@@ -0,0 +1,111 @@
+require 'itamae-spec/task/base_task'
+
+module ItamaeSpec
+  module Task
+    class ServerspecTask < BaseTask
+      ChangeTargetError = Class.new(StandardError)
+      LoadSpecError = Class.new(StandardError)
+
+      def list_recipe_filepath(run_list)
+        recipes = []
+        run_list.each do |recipe|
+          target_list = Dir.glob("cookbooks/#{recipe.keys.join}/spec/#{recipe.values.join}_spec.rb")
+
+          raise LoadSpecError, "#{recipe.to_a.join('::')} cookbook or spec does not exist." if target_list.empty?
+
+          target_list.each do |target|
+            recipes << " #{target}"
+          end
+        end
+
+        recipes
+      end
+
+      Itamae.logger.formatter.colored = true
+      task = ServerspecTask.new
+
+      namespace :spec do
+        all = []
+
+        begin
+          project = { project: ARGV[1] }
+
+          if (ARGV[0] == '-T' || ARGV[0] == '--tasks') && !project[:project].nil?
+            unless Dir.exist?("nodes/#{project[:project]}")
+              raise ChangeTargetError, "'#{project[:project]}' project is not exist."
+            end
+
+            File.open 'Project.json', 'w' do |f|
+              f.flock File::LOCK_EX
+              f.puts project.to_json
+              f.flock File::LOCK_UN
+            end
+
+            Itamae.logger.color(:green) do
+              Itamae.logger.info "Changed target mode '#{project[:project]}'"
+            end
+          end
+
+          resp = JSON.parse(File.read('Project.json'))
+          target = resp['project'] << '/**'
+        rescue => e
+          Itamae.logger.error e.inspect
+          exit 2
+        end
+
+        Dir.glob("nodes/#{target}/*.json").each do |node_file|
+          begin
+            node_name = File.basename(node_file, '.json')
+            node = task.load_node_attributes(node_file)
+            node_short = node[:environments][:hostname].split('.')[0]
+          rescue => e
+            Itamae.logger.error e.inspect
+            Itamae.logger.info "From node file: #{node_file}"
+            exit 2
+          end
+
+          all << node_short
+          desc 'Serverspec to all nodes'
+          task 'all' => all
+
+          desc "Serverspec to #{node_name}"
+          task node_short do
+            Itamae.logger.color(:cyan) do
+              Itamae.logger.info "Start serverspec_task to #{node[:environments][:hostname]}"
+            end
+
+            begin
+              run_list = task.load_run_list(node_file)
+              environments = task.load_environments(node)
+              recipe_attributes_list = task.load_recipe_attributes(run_list)
+
+              merged_recipe = task.merge_attributes(recipe_attributes_list)
+              merged_environments = task.merge_attributes(merged_recipe, environments)
+              attributes = task.merge_attributes(merged_environments, node)
+              task.create_tmp_nodes(node_name, attributes)
+
+              command = task.create_spec_command(node_name, attributes)
+              command_recipe = task.list_recipe_filepath(run_list)
+              command << command_recipe.join
+
+              task.runner_display(attributes[:run_list], run_list, command)
+              st = system command
+              if st
+                Itamae.logger.color(:green) do
+                  Itamae.logger.info 'serverspec_task is completed.'
+                end
+              else
+                Itamae.logger.error 'serverspec_task is failed.'
+                exit 1
+              end
+            rescue => e
+              Itamae.logger.error e.inspect
+              Itamae.logger.info "From node file: #{node_file}"
+              exit 2
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/itamae-spec/version.rb

@@ -0,0 +1,3 @@
+module ItamaeSpec
+  VERSION = File.read(File.expand_path("../version.txt", __FILE__)).strip
+end

data/lib/itamae-spec/version.txt

@@ -0,0 +1 @@
+0.0.2

data/spec/integration/Vagrantfile

@@ -0,0 +1,35 @@
+# -*- mode: ruby -*-
+# vi: set ft=ruby :
+require 'vagrant-digitalocean'
+
+# Vagrantfile API/syntax version. Don't touch unless you know what you're doing!
+VAGRANTFILE_API_VERSION = "2"
+
+Vagrant.configure(VAGRANTFILE_API_VERSION) do |config|
+  config.vm.define :trusty do |c|
+    c.vm.hostname  = 'itamae-trusty'
+    c.vm.hostname += "-#{ENV['WERCKER_BUILD_ID']}" if ENV['WERCKER_BUILD_ID']
+    c.vm.provider :virtualbox do |provider, override|
+      override.vm.box = "ubuntu/trusty64"
+      override.vm.provision :shell, inline: <<-EOC
+  cat /etc/apt/sources.list | sed -e 's|http://[^ ]*|mirror://mirrors.ubuntu.com/mirrors.txt|g' > /tmp/sources.list
+  if !(diff -q /etc/apt/sources.list /tmp/sources.list); then
+    mv /tmp/sources.list /etc/apt/sources.list
+    apt-get update
+  fi
+      EOC
+    end
+
+    c.vm.provider :digital_ocean do |provider, override|
+      override.ssh.private_key_path = '~/.ssh/id_rsa.vagrant'
+      override.vm.box = 'digital_ocean'
+      override.vm.box_url = "https://github.com/smdahlen/vagrant-digitalocean/raw/master/box/digital_ocean.box"
+
+      provider.ssh_key_name = ENV['WERCKER'] ? 'vagrant/wercker/itamae' : 'Vagrant'
+      provider.token = ENV['DIGITALOCEAN_TOKEN']
+      provider.image = 'ubuntu-14-04-x64' # ubuntu
+      provider.region = 'nyc3'
+      provider.size = '512mb'
+    end
+  end
+end

data/spec/integration/default_spec.rb

@@ -0,0 +1,226 @@
+require 'spec_helper'
+
+describe user("itamae") do
+  it { should exist }
+  it { should have_uid 1234 }
+  it { should have_home_directory '/home/itamae' }
+  it { should have_login_shell '/bin/dash' }
+end
+
+describe file('/tmp/included_recipe') do
+  it { should be_file }
+end
+
+describe package('dstat') do
+  it { should be_installed }
+end
+
+describe package('sl') do
+  it { should be_installed }
+end
+
+describe package('resolvconf') do
+  it { should_not be_installed }
+end
+
+%w!/tmp/remote_file /tmp/remote_file_auto!.each do |f|
+  describe file(f) do
+    it { should be_file }
+    its(:content) { should match(/Hello Itamae/) }
+  end
+end
+
+describe file('/tmp/directory') do
+  it { should be_directory }
+  it { should be_mode 700 }
+  it { should be_owned_by "itamae" }
+  it { should be_grouped_into "itamae" }
+end
+
+describe file('/tmp/directory_never_exist1') do
+  it { should_not be_directory }
+end
+
+%w!/tmp/template /tmp/template_auto!.each do |f|
+  describe file(f) do
+    it { should be_file }
+    its(:content) { should match(/Hello/) }
+    its(:content) { should match(/Good bye/) }
+    its(:content) { should match(/^total memory: \d+kB$/) }
+    its(:content) { should match(/^uninitialized node key: $/) }
+  end
+end
+
+describe file('/tmp/file') do
+  it { should be_file }
+  its(:content) { should match(/Hello World/) }
+  it { should be_mode 777 }
+end
+
+describe file('/tmp/execute') do
+  it { should be_file }
+  its(:content) { should match(/Hello Execute/) }
+end
+
+describe file('/tmp/never_exist1') do
+  it { should_not be_file }
+end
+
+describe file('/tmp/never_exist2') do
+  it { should_not be_file }
+end
+
+describe file('/tmp/http_request.html') do
+  it { should be_file }
+  its(:content) { should match(/"from": "itamae"/) }
+end
+
+describe file('/tmp/http_request_delete.html') do
+  it { should be_file }
+  its(:content) { should match(/"from": "itamae"/) }
+end
+
+describe file('/tmp/http_request_post.html') do
+  it { should be_file }
+  its(:content) do
+    should match(/"from": "itamae"/)
+    should match(/"love": "sushi"/)
+  end
+end
+
+describe file('/tmp/http_request_put.html') do
+  it { should be_file }
+  its(:content) do
+    should match(/"from": "itamae"/)
+    should match(/"love": "sushi"/)
+  end
+end
+
+describe file('/tmp/http_request_headers.html') do
+  it { should be_file }
+  its(:content) { should match(/"User-Agent": "Itamae"/) }
+end
+
+describe file('/tmp/http_request_redirect.html') do
+  it { should be_file }
+  its(:content) { should match(/"from": "itamae"/) }
+end
+
+describe file('/tmp/notifies') do
+  it { should be_file }
+  its(:content) { should eq("2431") }
+end
+
+describe file('/tmp/subscribes') do
+  it { should be_file }
+  its(:content) { should eq("2431") }
+end
+
+describe file('/tmp/cron_stopped') do
+  it { should be_file }
+  its(:content) do
+    expect(subject.content.lines.size).to eq 1
+  end
+end
+
+describe file('/tmp/cron_running') do
+  it { should be_file }
+  its(:content) do
+    expect(subject.content.lines.size).to eq 2
+  end
+end
+
+describe file('/tmp-link') do
+  it { should be_linked_to '/tmp' }
+  its(:content) do
+    expect(subject.content.lines.size).to eq 0
+  end
+end
+
+describe file('/tmp-link-force') do
+  it { should be_linked_to '/tmp' }
+end
+
+describe command('cd /tmp/git_repo && git rev-parse HEAD') do
+  its(:stdout) { should match(/3116e170b89dc0f7315b69c1c1e1fd7fab23ac0d/) }
+end
+
+describe command('cd /tmp/git_repo_submodule/empty_repo && cat README.md') do
+  its(:stdout) { should match(/Empty Repo/) }
+end
+
+describe file('/tmp/created_by_itamae_user') do
+  it { should be_file }
+  it { should be_owned_by 'itamae' }
+  its(:content) { should eq("/home/itamae\n/home/itamae") }
+end
+
+describe file('/tmp/created_in_default2') do
+  it { should be_file }
+end
+
+describe file('/tmp/never_exist3') do
+  it { should_not be_file }
+end
+
+describe file('/tmp/never_exist4') do
+  it { should_not be_file }
+end
+
+describe file('/tmp/created_in_redefine') do
+  it { should be_file }
+  its(:content) { should match(/first/) }
+end
+
+describe command('gem list') do
+  its(:stdout) { should include('tzinfo (1.2.2, 1.1.0)') }
+end
+
+describe command('ri Bundler') do
+  its(:stderr) { should eq("Nothing known about Bundler\n") }
+end
+
+describe file('/tmp/created_by_definition') do
+  it { should be_file }
+  its(:content) { should eq("name:name,key:value,message:Hello, Itamae\n") }
+end
+
+describe file('/tmp/remote_file_in_definition') do
+  it { should be_file }
+  its(:content) { should eq("definition_example\n") }
+end
+
+describe file('/tmp/multi_delayed_notifies') do
+  it { should be_file }
+  its(:content) { should eq("1\n2\n3\n4\n") }
+end
+
+describe file('/tmp/multi_immediately_notifies') do
+  it { should be_file }
+  its(:content) { should eq("1\n2\n3\n4\n") }
+end
+
+describe file('/tmp/file_edit_sample') do
+  it { should be_file }
+  its(:content) { should eq("Hello, Itamae") }
+  it { should be_mode 400 }
+  it { should be_owned_by "itamae2" }
+  it { should be_grouped_into "itamae2" }
+end
+
+describe file('/home/itamae2') do
+  it { should be_directory }
+  it { should be_owned_by "itamae2" }
+  it { should be_grouped_into "itamae2" }
+end
+
+describe file('/tmp/file_create_without_content') do
+  its(:content) { should eq("Hello, World") }
+  it { should be_mode 600 }
+  it { should be_owned_by "itamae" }
+  it { should be_grouped_into "itamae" }
+end
+
+describe file('/tmp/file_edit_notifies') do
+  its(:content) { should eq("1") }
+end