rea-netscaler-cli 0.5.9

data/lib/netscaler/vserver/request.rb

@@ -0,0 +1,67 @@
+require 'netscaler/logging'
+require 'netscaler/base_request'
+require 'netscaler/vserver/response'
+
+module Netscaler::VServer
+  class Request < Netscaler::BaseRequest
+    include Netscaler::Logging
+
+    def enable(vserver, options)
+      params = { :name => vserver }
+      send_request('enablelbvserver', params)
+    end
+
+    def disable(vserver, options)
+      params = { :name => vserver }
+      send_request('disablelbvserver', params)
+    end
+
+    def list(vserver, options)
+      vservers = []
+      send_request('getlbvserver', {:empty => :ok}) do |response|
+        response[:return][:list][:item].each do |vserver|
+          vservers << Response.new(vserver).to_hash
+        end
+      end
+      yield vservers if block_given?
+    end
+
+    def status(vserver, options)
+      params = { :name => vserver }
+      send_request('getlbvserver', params) do |response|
+        yield Response.new(response).to_hash if block_given?
+      end
+    end
+
+    def bind(vserver, options)
+      params = { 
+        :name => vserver,
+        :policyname => options[:policy],
+        :priority => options[:Priority],
+        :gotopriorityexpression => 'END' 
+      }
+
+      method = if options[:netscaler].version == "9.2"
+                 'bindlbvserver'
+               else
+                 'bindlbvserver_policy'
+               end
+      send_request(method, params)
+    end
+
+    def unbind(vserver, options)
+      params = {
+        :name => vserver,
+        :policyname => options[:policy], 
+        :type => 'REQUEST'
+      }
+
+      method = if options[:netscaler].version == "9.2"
+                 'unbindlbvserver'
+               else
+                 'unbindlbvserver_policy'
+               end
+      send_request(method, params)
+    end
+  end
+end

data/lib/netscaler/vserver/response.rb

@@ -0,0 +1,115 @@
+module Netscaler::VServer
+  class Response
+    def initialize(raw_response)
+      @info = if raw_response[:return]
+                raw_response[:return][:list][:item]
+              else
+                raw_response
+              end
+    end
+
+    def name
+      @info[:name]
+    end
+
+    def ip_address
+      if @info[:ipaddress] =~ /0\.0\.0\.0/
+        @info[:ipaddress2]
+      else
+        @info[:ipaddress]
+      end
+    end
+
+    def type
+      @info[:servicetype]
+    end
+
+    def port
+      @info[:port].to_i
+    end
+
+    def state
+      @info[:state]
+    end
+
+    def services
+      @parsed_services ||= []
+      return @parsed_services if !@parsed_services.empty? || @info[:servicename].nil?
+
+      if @info[:servicename][:item].is_a?(String)
+        @parsed_services << ServiceInfo.new(@info, nil)
+      else
+        @info[:servicename][:item].each_with_index do |name, i|
+          @parsed_services << ServiceInfo.new(@info, i)
+        end
+      end
+
+      @parsed_services
+    end
+
+    def service_groups
+      @service_groups ||= if @info[:servicegroupname] && @info[:servicegroupname][:item]
+                            groups = @info[:servicegroupname][:item]
+                            if groups.is_a?(Array)
+                              groups
+                            else
+                              [groups]
+                            end
+                          else
+                            []
+                          end
+    end
+
+    def to_hash
+      hash = { :name => name,
+        :ip_address => ip_address,
+        :state => state,
+        :port => port,
+        :type => type,
+      }
+
+      if !services.empty?
+        hash[:services] = services.map {|s| s.to_hash}
+      end
+
+      if !service_groups.empty?
+        hash[:service_groups] = service_groups
+      end
+      
+      hash
+    end
+  end
+
+  class ServiceInfo
+    attr_accessor :name, :ip_address, :state, :type
+
+    def initialize(raw_response, index)
+      @name = raw_response[:servicename][:item]
+      @ip_address = raw_response[:svcipaddress][:item]
+      @state = raw_response[:svcstate][:item]
+      @port = raw_response[:svcport][:item]
+      @type = raw_response[:svctype][:item]
+
+      if !index.nil?
+        @name = @name[index]
+        @ip_address = @ip_address[index]
+        @state = @state[index]
+        @port = @port[index]
+        @type = @type[index]
+      end
+    end
+
+    def port
+      @port.to_i
+    end
+
+    def to_hash
+      { :name => name,
+        :ip_address => ip_address,
+        :state => state,
+        :port => port,
+        :type => type
+      }
+    end
+  end
+end

data/spec/netscaler/cli_spec.rb

@@ -0,0 +1,204 @@
+require 'spec_helpers'
+require 'netscaler/cli'
+
+module Netscaler
+  
+  module CLIHelper
+    def parse(*args)
+      args << '-c'
+      args << File.join(File.dirname(__FILE__), 'configs', 'simple-config.yml')
+      cli = CLI.new(args)
+      cli.parse!(true)
+    end
+  end
+
+  describe CLI do
+    include CLIHelper
+
+    it "should fail when the netscaler alias doesn't exist" do
+      attempting {
+        parse('-n', 'asdf', 'server', '--action', 'list')
+      }.should raise_error(Netscaler::ConfigurationError, /host was not found/)
+    end
+
+    it "should be able to set the netscaler configuration correctly" do
+      res = parse('-n', 'else', 'server', '--action', 'list')
+      res.options[:netscaler].password.should eql('blah')
+    end
+
+    it "should set the json flag correctly" do
+      res = parse('-n', 'else', 'server', '--action', 'list', '--json')
+      res.options[:json].should be(true)
+    end
+
+    it "should set the debug flag correctly" do
+      res = parse('-n', 'else', 'server', '-a', 'list', '--debug')
+      res.options[:debug].should be(true)
+    end
+
+    describe "for servers" do
+      it "should fail when the server is not given to a non-list action" do
+        attempting {
+          parse('-n', 'else', 'server', '-a', 'status')
+        }.should raise_error(Choosy::ValidationError, /no server given/)
+      end
+
+      it "should succeed when given the list action with no arguments" do
+        res = parse('-n', 'else', 'server', '-a', 'list')
+        res.subresults[0][:action].should eql(:list)
+      end
+
+      it "should have the server argument on a non-list action" do
+        res = parse('-n', 'else', 'server', '-a', 'status', 'blarg')
+        res.subresults[0].args[0].should eql('blarg')
+        res.subresults[0][:action].should eql(:status)
+      end
+    end
+
+    describe "for vserver" do
+      it "should fail when the vserver is given to a non-list option" do
+        attempting {
+          parse('-n', 'else', 'vserver', '-a', 'disable')
+        }.should raise_error(Choosy::ValidationError, /no virtual server/)
+      end
+
+      it "should fail when the --policy flag is unset on a bind/unbind" do
+        attempting {
+          parse('-n', 'else', 'vserver', '-a', 'bind', 'vserv')
+        }.should raise_error(Choosy::ValidationError, /required by the 'bind\/unbind' actions/)
+      end
+
+      it "should fail when the --policy flag is used on anything other than bind/unbind" do
+        attempting {
+          parse('-n', 'else', 'vserver', '-a', 'disable', '-p', 'pol', 'vserv')
+        }.should raise_error(Choosy::ValidationError, /only used with bind/)
+      end
+
+      it "should fail when priority flag is used on anything other than a bind" do
+        attempting {
+          parse('-n', 'else', 'vserver', '-a', 'disable', '-P', '20', 'vserv')
+        }.should raise_error(Choosy::ValidationError, /only used with the bind action/)
+      end
+
+      it "should set the default value for the priority to 100" do
+        res = parse('-n', 'else', 'vserver', '-a', 'bind', '-p', 'pol', 'vserv')
+        res.subresults[0][:Priority].should eql(100)
+      end
+
+      it "should set the policy name" do
+        res = parse('-n', 'else', 'vserver', '-a', 'bind', '-p', 'pol', 'vserv')
+        res.subresults[0][:policy].should eql('pol')
+      end
+
+      it "should fail on a bad action" do
+        attempting {
+          parse('-n', 'else', 'vserver', '-a', 'asdf', 'vserv')
+        }.should raise_error(Choosy::ValidationError, /unrecognized value/)
+      end
+    end
+
+    describe "for services" do
+      it "should fail when the service is given to a non-list action" do
+        attempting {
+          parse('-n', 'else', 'service', '-a', 'disable')
+        }.should raise_error(Choosy::ValidationError, /no services given to act/)
+      end
+      
+      it "should fail when --vserver flag is used with a non-bind/unbind action" do
+        attempting {
+          parse('-n', 'else', 'service', '-a', 'disable', '-v', 'blarg', 'a-service')
+        }.should raise_error(Choosy::ValidationError, /only used with bind/)
+      end
+
+      it "should fail when the --vserver flag is not present with the bind/unbind action" do
+        attempting {
+          parse('-n', 'else', 'service', '-a', 'bind', 'a-service')
+        }.should raise_error(Choosy::ValidationError, /requires the -v\/--vserver flag/)
+      end
+
+      it "should succeed in setting the service name as an argument" do
+        res = parse('-n', 'else', 'service', '-a', 'bind', '-v', 'blah', 'a-service')
+        res.subresults[0].args[0].should eql('a-service')
+      end
+
+      it "should succed in setting the vserver name" do
+        res = parse('-n', 'else', 'service', '-a', 'bind', '-v', 'blah', 'a-service')
+        res.subresults[0][:vserver].should eql('blah')
+      end
+
+      it "should succeed in setting the action" do
+        res = parse('-n', 'else', 'service', '-a', 'bind', '-v', 'blah', 'a-service')
+        res.subresults[0][:action].should eql(:bind)
+      end
+
+      it "should set the default action to status" do
+        res = parse('-n', 'else', 'service', 'a-service')
+        res.subresults[0][:action].should eql(:status)
+      end
+    end
+
+    describe "for servicegroups" do
+      it "should fail when the service is given to a non-list action" do
+        attempting {
+          parse('-n', 'else', 'servicegroup', '-a', 'disable')
+        }.should raise_error(Choosy::ValidationError, /no service group given to act/)
+      end
+      
+      it "should fail when --vserver flag is used with a non-bind/unbind action" do
+        attempting {
+          parse('-n', 'else', 'servicegroup', '-a', 'disable', '-v', 'blarg', 'a-service-group')
+        }.should raise_error(Choosy::ValidationError, /only used with bind/)
+      end
+
+      it "should fail when the --vserver flag is not present with the bind/unbind action" do
+        attempting {
+          parse('-n', 'else', 'servicegroup', '-a', 'bind', 'a-service-group')
+        }.should raise_error(Choosy::ValidationError, /requires the -v\/--vserver flag/)
+      end
+
+      it "should succeed in setting the service name as an argument" do
+        res = parse('-n', 'else', 'servicegroup', '-a', 'bind', '-v', 'blah', 'a-service-group')
+        res.subresults[0].args[0].should eql('a-service-group')
+      end
+
+      it "should succeed in setting the vserver name" do
+        res = parse('-n', 'else', 'servicegroup', '-a', 'bind', '-v', 'blah', 'a-service-group')
+        res.subresults[0][:vserver].should eql('blah')
+      end
+
+      it "should succeed in setting the action" do
+        res = parse('-n', 'else', 'servicegroup', '-a', 'bind', '-v', 'blah', 'a-service-group')
+        res.subresults[0][:action].should eql(:bind)
+      end
+
+      it "should set the default action to status" do
+        res = parse('-n', 'else', 'servicegroup', 'a-service-group')
+        res.subresults[0][:action].should eql(:status)
+      end
+
+      describe "scoping to an individual service" do
+        before :each do 
+          @res = parse('-n', 'else', 'servicegroup', '-a', 'enable', 'a-service-group', '-s', 'a-server-name', '-p', '8080', '-d', '180')
+        end
+        
+        it "should set the servername to a-server-name" do
+          @res.subresults[0][:servername].should eql('a-server-name')
+        end
+        
+        it "should set the port to 8080" do
+          @res.subresults[0][:port].should eql('8080')
+        end
+        
+        it "should set the delay to 180" do
+          @res.subresults[0][:delay].should eql('180')
+        end
+        
+        it "should set the delay to 0 if not specified" do
+          @res = parse('-n', 'else', 'servicegroup', '-a', 'enable', 'a-service-group', '-s', 'a-server-name', '-p', '8080')
+          @res.subresults[0][:delay].should eql('0')
+        end
+      end
+    end
+
+  end
+end

data/spec/netscaler/config_spec.rb

@@ -0,0 +1,63 @@
+require 'spec_helpers'
+require 'netscaler/config'
+require 'netscaler/errors'
+
+module Netscaler
+
+  module ConfigurationHelper
+    def reading(file)
+      actual_file = File.expand_path("./configs/#{file}", File.dirname(__FILE__))
+      Netscaler::ConfigurationReader.read_config_file(actual_file)
+    end
+  end
+
+  describe "Configuration Reader" do
+    include ConfigurationHelper
+
+    before :each do
+      @config = reading('simple-config.yml')
+    end
+
+    describe "when reading an existing file" do
+      it "should be able to load the basic config file" do
+        @config.load_balancers.length.should eql(2)
+      end
+
+      it "should set the username and password correctly when set in the file." do
+        ns = @config['something.goes.here']
+        ns.username.should eql('some_user')
+        ns.password.should eql('somepass')
+      end
+
+      it "should load via an alias" do
+        ns = @config['else']
+        ns.alias.should eql('else')
+        ns.username.should eql('here')
+      end
+
+      it "should set the default version to 9.2" do
+        ns = @config['something.goes.here']
+        ns.version.should eql("9.2")
+      end
+
+      it "should read the version if present" do
+        ns = @config['else']
+        ns.version.should eql("9.1")
+      end
+    end
+
+    describe "when reading a non-existent or bad file" do
+      it "should fail when the config doesn't exist" do
+        attempting { reading('non-config.yml') }.should raise_error(Netscaler::ConfigurationError)
+      end
+
+      it "should fail when the config is poorly formed" do
+        attempting { reading('bad-yaml.yml') }.should raise_error(Netscaler::ConfigurationError)
+      end
+
+      it "should fail when the username is missing" do
+        attempting { reading('missing-username.yml')['load-balancer'] }.should raise_error(Netscaler::ConfigurationError)
+      end
+    end
+  end
+end