tengine_resource 0.5.13

data/spec/tengine/resource/credential_spec.rb

@@ -0,0 +1,205 @@
+# -*- coding: utf-8 -*-
+require 'spec_helper'
+require 'net/ssh'
+
+describe Tengine::Resource::Credential do
+
+  valid_attributes1 = {
+    :name => "ssh-private_key",
+    :auth_type_key => :ssh_password,
+    :auth_values => {
+      :username => 'user1',
+      :password => "password1",
+    }
+  }.freeze
+
+  context "name、auth_type_cd、auth_values は必須" do
+    it "正常系" do
+      Tengine::Resource::Credential.delete_all
+      credential1 = Tengine::Resource::Credential.new(valid_attributes1)
+      credential1.valid?.should == true
+    end
+
+    [:name, :auth_type_key].each do |key|
+      it "#{key}なし" do
+        attrs = valid_attributes1.dup
+        attrs.delete(key)
+        credential1 = Tengine::Resource::Credential.new(attrs)
+        credential1.valid?.should == false
+      end
+    end
+
+  end
+
+  context "nameはユニーク" do
+    before do
+      Tengine::Resource::Credential.delete_all
+      @credential1 = Tengine::Resource::Credential.create!(valid_attributes1)
+    end
+
+    it "同じ名前で登録されているものが存在する場合エラー" do
+      expect{
+        @credential1 = Tengine::Resource::Credential.create!(valid_attributes1)
+      }.to raise_error(Mongoid::Errors::Validations, "Validation failed - Name is already taken.")
+    end
+  end
+
+  context "nameはベース名として定義される文字列です" do
+    it "スラッシュ'/’はリソース識別子で使われるのでnameには使用できません" do
+      server1 = Tengine::Resource::Credential.new(:name => "foo/bar")
+      server1.valid?.should == false
+      server1.errors[:name].should == [Tengine::Core::Validation::BASE_NAME.message]
+    end
+
+    it "コロン':'はリソース識別子で使われるのでnameには使用できません" do
+      server1 = Tengine::Resource::Credential.new(:name => "foo:bar")
+      server1.valid?.should == false
+      server1.errors[:name].should == [Tengine::Core::Validation::BASE_NAME.message]
+    end
+  end
+
+
+  context "nameで検索" do
+    before do
+      Tengine::Resource::Credential.delete_all
+      @credential = Tengine::Resource::Credential.create!(valid_attributes1)
+    end
+
+    context "見つかる場合" do
+      it "name で検索できるか" do
+        found_credential = nil
+        lambda{
+          found_credential = Tengine::Resource::Credential.first(:conditions => {:name => "ssh-private_key"})
+        }.should_not raise_error
+        found_credential.should_not be_nil
+        found_credential.id.should == @credential.id
+      end
+
+      it "find_by_name" do
+        found_credential = nil
+        lambda{
+          found_credential = Tengine::Resource::Credential.find_by_name("ssh-private_key")
+        }.should_not raise_error
+        found_credential.should_not be_nil
+        found_credential.id.should == @credential.id
+      end
+
+      it "find_by_name!" do
+        found_credential = nil
+        lambda{
+          found_credential = Tengine::Resource::Credential.find_by_name!("ssh-private_key")
+        }.should_not raise_error
+        found_credential.should_not be_nil
+        found_credential.id.should == @credential.id
+      end
+    end
+
+    context "見つからない場合" do
+      it "find_by_name" do
+        found_credential = Tengine::Resource::Credential.find_by_name("unexist_name").should == nil
+      end
+
+      it "find_by_name!" do
+        lambda{
+          found_credential = Tengine::Resource::Credential.find_by_name!("unexist_name")
+        }.should raise_error(Tengine::Core::FindByName::Error)
+      end
+    end
+
+  end
+
+
+  describe :secure_auth_values, "外部に見せても良いように見せちゃ拙い情報は取り除く" do
+    it :ssh_password do
+      credential = Tengine::Resource::Credential.new(:name => "ssh-pass1",
+        :auth_type_key => :ssh_password,
+        :auth_values => {:username => 'goku', :password => "password1"})
+      credential.secure_auth_values.should == {'username' => 'goku'}
+    end
+
+    it :ssh_public_key do
+      credential = Tengine::Resource::Credential.new(:name => "ssh-pk1",
+        :auth_type_key => :ssh_public_key,
+        :auth_values => {:username => 'goku', :private_keys => "pkxxxxx", :passphrase => "abc"})
+      credential.secure_auth_values.should == {'username' => 'goku'}
+    end
+
+  end
+
+  describe :auth_values_validation do
+    context "valid" do
+
+      describe :ssh_password do
+        it "キーがSymbol" do
+          Tengine::Resource::Credential.new(:name => "ssh-pass1",
+            :auth_type_key => :ssh_password,
+            :auth_values => {:username => 'goku', :password => "password1"}).valid?.should == true
+        end
+        it "キーが文字列" do
+          Tengine::Resource::Credential.new(:name => "ssh-pass1",
+            :auth_type_key => :ssh_password,
+            :auth_values => {'username' => 'goku', 'password' => "password1"}).valid?.should == true
+        end
+      end
+
+      describe :ssh_public_key do
+        it "キーがSymbol" do
+          Tengine::Resource::Credential.new(:name => "ssh-pk1",
+            :auth_type_key => :ssh_public_key,
+            :auth_values => {:username => 'goku', :private_keys => "pkxxxxx"}).valid?.should == true
+        end
+        it "キーが文字列" do
+          Tengine::Resource::Credential.new(:name => "ssh-pk1",
+            :auth_type_key => :ssh_public_key,
+            :auth_values => {'username' => 'goku', 'private_keys' => "pkxxxxx"}).valid?.should == true
+        end
+      end
+
+    end
+
+    context "invalid" do
+      describe :ssh_password do
+        it ":usernameが空文字列" do
+          Tengine::Resource::Credential.new(:name => "ssh-pass1",
+            :auth_type_key => :ssh_password,
+            :auth_values => {:username => '', :password => "password1"}).valid?.should == false
+        end
+        it "passwordなし" do
+          Tengine::Resource::Credential.new(:name => "ssh-pass1",
+            :auth_type_key => :ssh_password,
+            :auth_values => {'username' => 'goku'}).valid?.should == false
+        end
+      end
+
+      describe :ssh_public_key do
+        it ":usernameなし" do
+          credential1 = Tengine::Resource::Credential.new(:name => "ssh-pk1",
+            :auth_type_key => :ssh_public_key,
+            :auth_values => {:private_keys => "pkxxxxx"})
+          credential1.valid?.should == false
+        end
+        it "private_keyが空文字列" do
+          credential1 = Tengine::Resource::Credential.new(:name => "ssh-pk1",
+            :auth_type_key => :ssh_public_key,
+            :auth_values => {'username' => 'goku', 'private_keys' => ""})
+          credential1.valid?.should == false
+        end
+      end
+
+      describe :ec2_access_key do
+        it "access_keyが空文字列" do
+          credential1 = Tengine::Resource::Credential.new(:name => "ec2-access-key1",
+            :auth_type_key => :ec2_access_key,
+            :auth_values => {:access_key => '', :secret_access_key => "abcdef"})
+          credential1.valid?.should == false
+        end
+        it "secret_access_keyがnil" do
+          credential1 = Tengine::Resource::Credential.new(:name => "ec2-access-key1",
+            :auth_type_key => :ec2_access_key,
+            :auth_values => {'access_key' => '12345', 'secret_access_key' => nil})
+          credential1.valid?.should == false
+        end
+      end
+    end
+  end
+end

data/spec/tengine/resource/drivers/resource_control_driver_spec.rb

@@ -0,0 +1,84 @@
+# -*- coding: utf-8 -*-
+__END__
+require 'spec_helper'
+require 'tengine/rspec'
+
+describe 'resource_control_driver' do
+  include Tengine::RSpec::Extension
+
+  target_dsl File.expand_path("../../../../lib/tengine/resource/drivers/resource_control_driver.rb", File.dirname(__FILE__))
+  driver :resource_control_driver
+
+  before do
+    Tengine::Resource::Provider::Wakame.delete_all(:name => 'tama0001')
+    @provider = Tengine::Resource::Provider::Wakame.create!(
+      :name => "tama0001",
+      :description => "provided by wakame / tama",
+      :connection_settings => {
+        :account => "a-shpoolxx",
+        :ec2_host => "192.168.2.22",
+        :ec2_port => 9005,
+        :ec2_protocol => "https",
+        :wakame_host => "192.168.2.22",
+        :wakame_port => 9001,
+        :wakame_protocol => "https",},
+      :properties => { :key_name => "ssh-xxxxx" })
+  end
+
+  context '仮想サーバ起動リクエストイベント' do
+    it "反応する" do
+      vi = @provider.virtual_server_images.create!(:provided_id => "wmi-lucid5", :name => "wmi-lucid5")
+      vt = @provider.virtual_server_types.create!(:provided_id => "is-small", :name => "is-small")
+      ps = @provider.physical_servers.create!(:provided_id => "foo-dc", :name => "foo-dc")
+      @provider.
+        should_receive(:create_virtual_servers).
+        with(:virtual_server_image => vi,
+             :virtual_server_type => vt,
+             :physical_server => ps,
+             :count => 1)
+      Tengine::Resource::Provider::Wakame.stub(:instantiate).with(anything).and_return(@provider)
+
+      tengine.receive('仮想サーバ起動リクエストイベント', :properties => {
+        :provider_id => @provider._id,
+        :virtual_server_image_id => vi._id,
+        :virtual_server_type_id => vt._id,
+        :physical_server_id => ps._id,
+        :count => 1,})
+    end
+  end
+
+  context '仮想サーバ停止リクエストイベント' do
+    before do
+      Tengine::Resource::VirtualServer.delete_all
+    end
+
+    it "反応する" do
+      v1 = @provider.virtual_servers.create!(:provided_id => 'i-f222222d', :name => 'i-f222222d')
+      v2 = @provider.virtual_servers.create!(:provided_id => 'i-f222222e', :name => 'i-f222222e')
+      @provider.
+        should_receive(:terminate_virtual_servers).
+        with([v1, v2])
+      Tengine::Resource::Provider::Wakame.stub(:instantiate).with(anything).and_return(@provider)
+
+      tengine.receive('仮想サーバ停止リクエストイベント', :properties => {
+        :provider_id => @provider._id,
+        :virtual_servers => [v1._id, v2._id] })
+    end
+
+    it "存在しない仮想サーバを停止しようとすると例外がraiseされる" do
+      v1 = @provider.virtual_servers.create!(:provided_id => 'i-f222222d', :name => 'i-f222222d')
+      v2 = @provider.virtual_servers.create!(:provided_id => 'i-f222222e', :name => 'i-f222222e')
+
+      # 存在するはずの仮想サーバのデータを削除して例外を起こします
+      Tengine::Resource::VirtualServer.delete_all
+
+      begin
+        tengine.receive('仮想サーバ停止リクエストイベント', :properties => {
+            :provider_id => @provider._id,
+            :virtual_servers => [v1._id, v2._id] })
+        fail
+      rescue Mongoid::Errors::DocumentNotFound => e
+      end
+    end
+  end
+end

data/spec/tengine/resource/net_ssh_spec.rb

@@ -0,0 +1,148 @@
+# -*- coding: utf-8 -*-
+require 'spec_helper'
+require 'tengine/resource/net_ssh'
+
+describe Net::SSH do
+  describe :start do
+    before do
+      @e = Exception.new("ok")
+      Net::SSH::Transport::Session.stub(:new).with(anything, anything).and_raise(@e)
+    end
+
+    after do
+      Tengine::Resource::Credential.delete_all(:name => "ssh1")
+    end
+
+    describe "``Net::SSH.start(hostname, credential) {|ctx| ... }''" do
+      describe "auth_type_key: :ssh_password" do
+        it "starts" do
+          c = Tengine::Resource::Credential.new(
+            :name => "ssh1",
+            :auth_type_key => :ssh_password,
+            :auth_values => {:username => 'goku', :password => "password1"})
+
+          expect { Net::SSH.start "localhost", c }.to raise_error(@e)
+        end
+      end
+
+      describe "auth_type_key: :ssh_public_key" do
+        it "starts" do
+          c = Tengine::Resource::Credential.new(
+            :name => "ssh1",
+            :auth_type_key => :ssh_public_key,
+            :auth_values => {:username => 'goku', :private_keys => <<END, :passphrase => ""})
+-----BEGIN RSA PRIVATE KEY-----
+MIIEoAIBAAKCAQEAvdBvdpWxMumGZsyWpnsPetEW0da5/5sIfSgP8pAVgibp492O
+l++LQlITsSf3Lt/WOQNEfOrfU7HFZuRk3b7dPTJFneG+AIXS8PdrxDO8e9DSG1D4
+xDAgSrG4inuSQju1EPaPhIz7r0eg1tupP6UM8GVUu90zO7OCnvjYOPjNRvibW5ao
+dzo4T5cOLBZuUm3RqclQLehLAQpG6p+d6DbADTVVNpvgk+VfPpKI9E+q944ivJIM
+0KLNWyfEbxjoXuYnZYn1em6MHcaXsR1I5rlzuhfOaukerqFulG3MkU6A2t/CfbXM
+PiOtwPSse7ZuoBtdFhnxUpGRmlPoSvezfSz23wIBIwKCAQA7p+iEWu6MWAWrR50P
+wEaxkiu+Ssy+DCc9Rxr0dm0o56iYEm6kx6DLs2VGTmOhBIUn3HScSdEo7rpiKoYZ
+zkWIQv/vyprM9tvlVRM9qdwm6duiL2QgZuWT0XSDS2h7N1Yp5xcpqKbfUQap7UPS
+LI8JuXJm7b+lDIgjVYXIwzkzj2qEkoF+p8hHWD7X3v6db6Gz7+HMQx/rc6yMrZ0H
+RF/CKHOWvriY/41CxsYvpAK0ksXTXupxzEJdItvHYIJsQW6gxBGtxWo7Y55si8w/
+VBgmkiQ/wPVZsQ5omo3lwrLohW0k7xO8ZX2sAweb/NUKhZi0KcrwTHFT5MbwxrHi
+GGcbAoGBAPvjaPtg9171h+xu/MlpOBivhfApdhrEoRfn5Np8OmSRtOC5hK/UyQtk
+Ot4G10KQziK98xkSQ3OphR/kQA8oiq+TfZvPKJ69GYqLpWkBdM2rNXph5BCahri+
+qO04wqgpZkG7obCMrKXPNQj1MWhCAy5mRNYcuWSEeIXZZ66TT1SZAoGBAMDpoVbg
+W4qlmUKCme8i0ofsSB00s+HyJ6zmBriDaJ8o/77jQ/ypwKmox+6Ebo5WvnNGzgnI
+puOgkNH5MWjGoowvnH6CXWzSQYlvAOGnZ58IhbIm5OolRHC3fASUw0JZRVQGvlFC
+v73ltf+td9TfhmAVfqzE5PMK7C9E6UNJBFo3AoGAVlyZBbrYeFQumjSucPDgCHax
+lCt48zwZ+ZFVx0CJDIm6W61SEGY2TQxr9Fod6u/RpYL0Qxw0Yit+GZAV9pGOopj3
+3aYcjjI0pIembSUSGqEZpk1yw901gSttHiIW2pHZ6qa7F/W3iU5bU4dEI77chO/d
+FjW75/Lnflkq3MTKvesCgYALBgk4KhPcCXZ41ENPflUsVqUI+7KB8JSNiXy6FiM8
+S3xUDPySGFQm87OnOsR9KYc5yYgd0POX+ovuvchPIUsd9BeSM1XLtD5CXh1OuRvd
+M6/eS12JuPyYugcWNGLt2TcpX3iW7d8SKmIr3gbY9tR6hOKqyWwrJTIRVGUZyah6
+LwKBgA+Fdse5WibNPG1smoS/JzmPdrdD46x2D7HoLtbcNF4rPlduVI8yS3dRSqi4
+Gh5L65ukAbWSC8yuQpKtq3EfpNcf83a+Xc5XXANbTgN/+sJjgd+ycmPzLv5Zcsg9
+8xlb43zLCiEf5TyrlbmqfM97hxuPbrgifyU7jMaTdmHHGsDx
+-----END RSA PRIVATE KEY-----
+END
+
+          expect { Net::SSH.start "localhost", c }.to raise_error(@e)
+        end
+      end
+    end
+
+    describe "``Net::SSH.start(hostname, user, credential) {|ctx| ... }''" do
+      describe "normal path" do
+        it "starts" do
+          c = Tengine::Resource::Credential.new(
+            :name => "ssh1",
+            :auth_type_key => :ssh_password,
+            :auth_values => {:password => "password1"})
+
+          expect { Net::SSH.start "localhost", "goku", c }.to raise_error(@e)
+        end
+      end
+
+      describe "conflicted username" do
+        subject do
+          c = Tengine::Resource::Credential.new(
+            :name => "ssh1",
+            :auth_type_key => :ssh_password,
+            :auth_values => {:username => "goku", :password => "password1"})
+
+          expect { Net::SSH.start "localhost", "goku", c }
+        end
+        it { should raise_error(ArgumentError) }
+      end
+    end
+
+    describe "``Net::SSH.start(hostname, credential, other_opts) {|ctx| ... }''" do
+      describe "normal path" do
+        it "starts" do
+          c = Tengine::Resource::Credential.new(
+            :name => "ssh1",
+            :auth_type_key => :ssh_password,
+            :auth_values => {:username => "goku"})
+
+          expect { Net::SSH.start "localhost", c, :password => "passowrd1" }.to raise_error(@e)
+        end
+      end
+
+      describe "conflicted option" do
+        subject do
+          c = Tengine::Resource::Credential.new(
+            :name => "ssh1",
+            :auth_type_key => :ssh_password,
+            :auth_values => {:username => "goku", :password => "password1"})
+
+          expect { Net::SSH.start "localhost", c, :password => "password1" }
+        end
+        it { should raise_error(ArgumentError) }
+      end
+    end
+
+    describe "``Net::SSH.start(hostname, user, other_opts) {|ctx| ... }''" do
+      describe "normal path" do
+        it "starts" do
+          expect { Net::SSH.start "localhost", "goku", :password => "passowrd1" }.to raise_error(@e)
+        end
+      end
+
+      describe "unknown key" do
+        subject do
+          expect { Net::SSH.start "localhost", "goku", :foo => "bar" }
+        end
+        it { should raise_error(ArgumentError) }
+      end
+    end
+
+    describe "``Net::SSH.start(hostname, other_opts) {|ctx| ... }''" do
+      describe "normal path" do
+        it "starts" do
+          expect { Net::SSH.start "localhost", :username => "goku", :password => "passowrd1" }.to raise_error(@e)
+        end
+      end
+
+      describe "missing username" do
+        subject do
+          expect { Net::SSH.start "localhost", :password => "password1" }
+        end
+        it { should raise_error(ArgumentError) }
+      end
+    end
+  end
+end

data/spec/tengine/resource/physical_server_spec.rb

@@ -0,0 +1,47 @@
+# -*- coding: utf-8 -*-
+require 'spec_helper'
+
+describe Tengine::Resource::PhysicalServer do
+  context "nameで検索" do
+    before do
+      Tengine::Resource::Server.delete_all
+      @fixture = GokuAtEc2ApNortheast.new
+      @physical1 = @fixture.availability_zone(1)
+      @virtual1 = @fixture.hadoop_master_node
+    end
+
+    context "見つかる場合" do
+      it "find_by_name" do
+        found_credential = nil
+        lambda{
+          found_credential = Tengine::Resource::PhysicalServer.find_by_name(@physical1.name)
+        }.should_not raise_error
+        found_credential.should_not be_nil
+        found_credential.id.should == @physical1.id
+      end
+
+      it "find_by_name!" do
+        found_credential = nil
+        lambda{
+          found_credential = Tengine::Resource::PhysicalServer.find_by_name!(@physical1.name)
+        }.should_not raise_error
+        found_credential.should_not be_nil
+        found_credential.id.should == @physical1.id
+      end
+    end
+
+    context "見つからない場合" do
+      it "find_by_name" do
+        found_credential = Tengine::Resource::PhysicalServer.find_by_name(@virtual1.name).should == nil
+      end
+
+      it "find_by_name!" do
+        lambda{
+          found_credential = Tengine::Resource::PhysicalServer.find_by_name!(@virtual1.name)
+        }.should raise_error(Tengine::Core::FindByName::Error)
+      end
+    end
+
+  end
+
+end