cloudstack_spec 0.0.4

data/lib/cloudstack_spec/resource/vpc.rb

@@ -0,0 +1,139 @@
+module CloudstackSpec::Resource
+  class Vpc < Base
+    # do nothing
+
+    def initialize(name='rspec-vpc1', zonename=nil)
+      @name   = name
+      @connection = CloudstackSpec::Helper::Api.new.connection
+      @zone = get_zone(zonename)
+      @runner = Specinfra::Runner
+      #vpc = @connection.list_vpcs(listall: true, name: @name)
+      if vpc.empty?
+        @router = nil
+      else
+        @router = @connection.list_routers(vpcid: vpc['id'])['router'].first
+      end
+      $vpc = vpc
+    end
+    
+    def exist?
+        if vpc.empty?
+          return false
+        else
+          return true
+        end
+    end
+
+    def ready?
+      begin
+        if @router['state'] == 'Running'
+          if @router['version'] == CloudstackSpec::Helper::Api.new.version
+            return true
+          else
+            return "VR version #{@router['version']}"
+          end
+        else
+          return false
+        end
+      rescue
+          return false
+      end
+    end
+
+    def reachable?(port, proto, timeout)
+      ip = @router['publicip']
+      @runner.check_host_is_reachable(ip, port, proto, timeout)
+    end
+
+    def created?(cidr='10.10.0.0/22')
+      if self.exist?
+        puts "  VPC already exist"
+        return true
+      else
+        job = @connection.create_vpc(
+                :name => @name,
+                :displaytext => @name,
+                :cidr => cidr,
+                :vpcofferingid => vpc_offering_id,
+                :zoneid => @zone['id']
+                )
+        job_status = job_status?(job['jobid'])
+        @router = @connection.list_routers(vpcid: vpc['id'])['router'].first
+        $vpc = vpc
+        return job_status
+
+      end
+    end
+
+    def destroy?
+      sleep(5)
+      if self.exist?
+        job = @connection.delete_vpc(id: vpc['id'])
+        return job_status?(job['jobid'])
+      else
+        puts "  Does not exist"
+        return false
+      end
+    end
+
+    def enable_remote_vpn
+      newvpn =  @connection.create_remote_access_vpn(publicipid: publicip_snat_id)
+      job_status?(newvpn['jobid'])
+      vpn = @connection.list_remote_access_vpns(publicipid: publicip_snat_id)
+      vpn = vpn['remoteaccessvpn'].first
+      if ! vpn.empty?
+        return true
+      end
+    end
+
+    def remote_vpn_enabled?
+      a = @connection.list_remote_access_vpns(listall: true, publicipid: publicip_snat_id)
+      if a.empty?
+        false
+      else
+        puts "    Pubic IP = #{a['remoteaccessvpn'].first['publicip']}"
+        puts "    PreShared Key = #{a['remoteaccessvpn'].first['presharedkey']}"
+        true
+      end
+    end
+
+
+    private
+
+      def vpc
+        vpc = @connection.list_vpcs(listall: true, name: @name)
+        if vpc.empty?
+          return {}
+        else
+          return vpc['vpc'].first
+        end
+      end
+
+      def first_vm_id
+        vm = @connection.list_virtual_machines(vpcid: vpc['id'])
+        if vm.empty?
+          return false
+        end
+        vm['virtualmachine'].first['id']
+      end
+
+      def first_tier_id
+        net = @connection.list_vpcs(id: vpc['id'])
+        net = net['vpc'].first['network'].first['id']
+        return net
+      end
+
+      def publicip_snat_id
+        # get the id of the sourceNAT public IP for the current VPC
+        public_ip = @connection.list_public_ip_addresses(vpcid: vpc['id'], issourcenat: true)
+        public_ip = public_ip['publicipaddress'].first
+        return public_ip['id']
+      end
+
+      def vpc_offering_id(offering_name="Default VPC offering")
+        offering = @connection.list_vpc_offerings(name: offering_name)['vpcoffering'].first
+        return offering['id']
+      end
+
+  end
+end

data/lib/cloudstack_spec/resource/vpc_tier.rb

@@ -0,0 +1,81 @@
+module CloudstackSpec::Resource
+  class VpcTier < Base
+    attr_reader :name
+
+    def initialize(name='tier11', zonename=nil)
+      @name   = name
+      @connection = CloudstackSpec::Helper::Api.new.connection
+      @zone = get_zone(zonename)
+      @runner = Specinfra::Runner
+    end
+
+    def exist?
+      begin  
+        if !get_tier.empty?
+          return true
+        else
+          return false
+        end
+      rescue Exception => e
+        return false
+      end
+    end
+
+    def created?
+      unless self.exist?
+        @connection.create_network(
+                    :name => @name, 
+                    :displaytext => @name, 
+                    :networkofferingid => get_offering_id, 
+                    :zoneid => @zone['id'],
+                    :vpcid => $vpc['id'],
+                    :gateway => '10.10.0.1',
+                    :netmask => '255.255.255.0',
+                    :aclid => get_acl_id
+                   )
+      else
+        print "  already exist"
+        return true
+        #return false
+      end
+    end
+
+    def destroy?
+      if self.exist?
+        sleep(5)
+        job = @connection.delete_network(id: get_tier['id'])
+        job_status?(job['jobid'])
+      else
+        puts "  Does not exist"
+        return false
+      end
+    end
+
+
+    private
+
+      def get_tier
+        # CloudStack API does not search by name for networks
+        networks = @connection.list_networks(:listall => true, vpcid: $vpc['id'])
+        networks = networks['network']
+        if networks.nil?
+          return {}
+        else
+          network = networks.select { |net| net['name'] == @name }
+          network = network.first
+        end
+      end
+
+      def get_offering_id
+        offering = @connection.list_network_offerings(:name => "DefaultIsolatedNetworkOfferingForVpcNetworks")
+        offering = offering["networkoffering"].first
+        return offering['id']
+      end
+
+      def get_acl_id(name='default_allow')
+        acl = @connection.list_network_a_c_l_lists(name: name)
+        acl = acl['networkacllist'].first
+        return acl['id']
+      end
+  end
+end

data/lib/cloudstack_spec/resource/zone.rb

@@ -0,0 +1,51 @@
+module CloudstackSpec::Resource
+  class Zone < Base
+    # All about zone...
+    #####
+    def exist?
+      #zone = @connection.list_zone(:name => name)
+      if @zonename.nil?
+        return false
+      else
+        return true
+      end
+    end
+
+    def allocated?
+      if @zonename['allocationstate'] == 'Enabled'
+        return true
+      else
+        return @zonename['allocationstate']
+      end
+    end
+
+    def local_storage
+      return @zonename['localstorageenabled']
+    end
+
+    def security_group
+      return @zonename['securitygroupsenabled']
+    end
+
+    def network_type
+      # return "Basic" or "Advanced"
+      return @zonename['networktype']
+    end
+
+    private
+
+      def this_zone(name)
+        if name.nil?
+          # if zone name not define, thake the first one
+          zone = @connection.list_zones['zone'].first
+          @name = zone['name']
+        else 
+          @name = name
+          zone = @connection.list_zones(:name => name)['zone'].first
+        end
+        return zone
+      end
+
+    # end private
+  end
+end

data/lib/cloudstack_spec/setup.rb

@@ -0,0 +1,191 @@
+require 'fileutils'
+require 'erb'
+
+module CloudstackSpec
+  class Setup
+    def self.run
+
+      create_initial
+    end
+
+    def self.create_initial
+      print "Create base for cloudstack_spec? y/n: "
+      auto_config = $stdin.gets.chomp
+      if auto_config =~ (/(true|t|yes|y|1)$/i)
+        [ 'spec', "spec/lib" ].each { |dir| safe_mkdir(dir) }
+        safe_create_config
+        safe_create_spec
+        safe_create_spec_helper
+#        safe_create_rakefile
+        puts "Make sure to Update file: spec/config.yml"
+      else
+        exit
+      end
+    end
+
+    def self.safe_create_config
+      # create config.yml file.
+      content = <<-EOF
+cloudstack:
+  url:        http://127.0.0.1:8080/client/api
+  api_key:    c0dD7KjmFF4ixekZyQZJa7xPrW6y5-egZU1FGqq2u0DITs1qSluyP3giLDJ2msg0y1gPeARq4ZyvK-j48QP8hQ
+  secret_key: GYfIZjsjSIHKO_227M6tvLns86QJEzqJVzc8xobqfgg03YeDUYk2QTGBgB6IeaHfSROW61uPm-koO2iDDvq5HQ
+  use_ssl:    false
+
+EOF
+
+      if File.exists? "spec/config.yml"
+        old_content = File.read("spec/config.yml")
+        if old_content != content
+          $stderr.puts "!! spec/config.yml already exists and differs from template"
+        end
+      else
+        File.open("spec/config.yml", 'w') do |f|
+          f.puts content
+        end
+        puts " + spec/config.yml"
+      end
+    end
+
+    def self.safe_create_spec
+      content = <<-EOF
+require 'spec_helper'
+
+describe zone do
+  it { should exist }
+  it { should be_allocated }
+#  its(:local_storage) { should_not be_set }
+#  its(:security_group) { should_not be_set }
+#  its(:network_type) { should match("Advanced") }
+end
+
+%w(consoleproxy secondarystoragevm).each do |svm|
+  describe system_vm(svm) do
+    it { should exist }
+    it { should be_running }
+    it { should be_reachable }
+  end
+end
+
+EOF
+
+      if File.exists? "spec/lib/001_zone_spec.rb"
+        old_content = File.read("spec/lib/001_zone_spec.rb")
+        if old_content != content
+          $stderr.puts "!! spec/lib/zone_spec.rb already exists and differs from template"
+        end
+      else
+        File.open("spec/lib/001_zone_spec.rb", 'w') do |f|
+          f.puts content
+        end
+        puts " + spec/lib/001_zone_spec.rb"
+      end
+    end
+
+    def self.safe_mkdir(dir)
+      if File.exists? dir
+        unless File.directory? dir
+          $stderr.puts "!! #{dir} already exists and is not a directory"
+        end
+      else
+        FileUtils.mkdir dir
+        puts " + #{dir}/"
+      end
+    end
+
+    def self.safe_create_spec_helper
+      requirements = []
+      content = ERB.new(spec_helper_template, nil, '-').result(binding)
+      if File.exists? 'spec/spec_helper.rb'
+        old_content = File.read('spec/spec_helper.rb')
+        if old_content != content
+          $stderr.puts "!! spec/spec_helper.rb already exists and differs from template"
+        end
+      else
+        File.open('spec/spec_helper.rb', 'w') do |f|
+          f.puts content
+        end
+        puts ' + spec/spec_helper.rb'
+      end
+    end
+
+    def self.safe_create_rakefile
+      content = <<-EOF
+require 'rake'
+require 'rspec/core/rake_task'
+
+task :spec    => 'spec:all'
+task :default => :spec
+
+namespace :spec do
+  targets = []
+  Dir.glob('./spec/*').each do |dir|
+    next unless File.directory?(dir)
+    targets << File.basename(dir)
+  end
+
+  task :all     => targets
+  task :default => :all
+
+  targets.each do |target|
+    desc "Run serverspec tests to #{target}"
+    RSpec::Core::RakeTask.new(target.to_sym) do |t|
+      ENV['TARGET_HOST'] = target
+      t.pattern = "spec/#{target}/*_spec.rb"
+    end
+  end
+end
+
+EOF
+      if File.exists? 'Rakefile'
+        old_content = File.read('Rakefile')
+        if old_content != content
+          $stderr.puts "!! Rakefile already exists and differs from template"
+        end
+      else
+        File.open('Rakefile', 'w') do |f|
+          f.puts content
+        end
+        puts ' + Rakefile'
+      end
+    end
+
+    def self.spec_helper_template
+      template = <<-EOF
+# This file was generated by the `rspec --init` command. Conventionally, all
+# specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
+# Require this file using `require "spec_helper"` to ensure that it is only
+# loaded once.
+#
+# See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
+require 'yaml'
+require 'cloudstack_ruby_client'
+require 'uuid'
+require 'cloudstack_spec'
+require 'rspec/expectations'
+
+require 'serverspec'
+#require 'specinfra'
+set :backend, :exec
+
+
+
+RSpec.configure do |config|
+  config.run_all_when_everything_filtered = true
+#  config.filter_run :focus
+
+  # Run specs in random order to surface order dependencies. If you find an
+  # order dependency and want to debug it, you can fix the order by providing
+  # the seed, which is printed after each run.
+  #     --seed 1234
+  config.order = 'defined'
+  config.formatter = 'documentation'
+  config.color = true
+end
+
+EOF
+      template
+    end
+
+  end
+end