hosttag 0.12

data/lib/hosttag/server.rb

@@ -0,0 +1,35 @@
+# Hosttag::Server class, extending Redis
+
+# Blech - shouldn't need to require rubygems here, but fails on 1.8.x otherwise
+require 'rubygems'
+require 'redis'
+require 'resolv'
+
+module Hosttag
+  class Server < Redis
+    def initialize(options)
+      @defaults = { 
+        :server     => ENV['HOSTTAG_SERVER']    || 'hosttag', 
+        :port       => ENV['HOSTTAG_PORT']      || 6379, 
+        :namespace  => ENV['HOSTTAG_NAMESPACE'] || 'hosttag',
+      }
+      @defaults.merge!(options)
+       
+      # Check :server name resolves
+      begin
+        Resolv.getaddress( @defaults[:server] )
+      rescue Resolv::ResolvError
+        raise Resolv::ResolvError, 
+          "Host '#{@defaults[:server]}' does not resolve (try --server <hostname>?)", caller
+      end
+
+      # Connect to redis
+      super( :host => @defaults[:server], :port => @defaults[:port] )
+    end
+
+    def get_key(*elt)
+      "#{@defaults[:namespace]}::#{elt.join(':')}"
+    end
+  end
+end
+

data/test/test_hosttag_bin.rb

@@ -0,0 +1,70 @@
+#!/usr/bin/ruby
+
+require 'test/unit'
+require 'fileutils'
+
+require 'hosttag'
+include Hosttag
+
+class TestHosttagBin < Test::Unit::TestCase
+  def setup
+    @test_args = '--server localhost --ns hosttag_testing'
+    @test_opts = { :server => 'localhost', :namespace => 'hosttag_testing',
+      :delete => true, :autoconfirm => true, :debug => false }
+    @bindir = File.join(File.dirname(__FILE__), '..', 'bin')
+    datadir = "#{File.dirname(__FILE__)}/data_hosttag"
+    File.directory?(datadir) or throw "missing datadir #{datadir}"
+    hosttag_import_from_directory(datadir, @test_opts)
+  end
+
+  # format: args (string) => expected (either string or regex)
+  TESTS = [
+    [ 'centos',                 'a m n' ],
+    [ 'laptop',                 'n' ],
+    [ 'vps',                    'm' ],
+    [ 'laptop vps',             '' ],
+    [ '-a laptop vps',          '' ],
+    [ '-o laptop vps',          'm n' ],
+    [ '-t n',                   'centos centos5 centos5-i386 laptop' ],
+    [ '-t a',                   'centos centos5 centos5-x86_64 public' ],
+    [ '-t m',                   'centos centos4 centos4-x86_64 public vps' ],
+    [ '-t a n',                 'centos centos5 centos5-i386 centos5-x86_64 laptop public' ],
+    [ '-to a n',                'centos centos5 centos5-i386 centos5-x86_64 laptop public' ],
+    [ '-t -o a n',              'centos centos5 centos5-i386 centos5-x86_64 laptop public' ],
+    [ '-ta a n',                'centos centos5' ],
+    [ '-t -a a n',              'centos centos5' ],
+    [ '-t -a a n m',            'centos' ],
+    [ '-1 centos',              "a\nm\nn" ],
+    [ '-o -1 laptop vps',       "m\nn" ],
+    # List mode
+    [ '-l centos5',             'centos5: a n' ],
+    # All cases
+    [ '-A',                     'a m n' ],
+    [ '-A -A',                  'a g h m n' ],
+    [ '-T',                     'centos centos4 centos4-x86_64 centos5 centos5-i386 centos5-x86_64 laptop public vps' ],
+    [ '-A -1',                  "a\nm\nn" ],
+    [ '-A -1 -A',               "a\ng\nh\nm\nn" ],
+    [ '-1 -T',                  "centos\ncentos4\ncentos4-x86_64\ncentos5\ncentos5-i386\ncentos5-x86_64\nlaptop\npublic\nvps" ],
+    # Try unrecognised tags as hosts
+    [ 'n',                      'centos centos5 centos5-i386 laptop' ],
+    [ 'a n',                    'centos centos5 centos5-i386 centos5-x86_64 laptop public' ],
+    # Invalid data
+    [ 'foo',                    %r{ \btag\b  .* not\sfound }x ],
+    [ 'centos foo',             %r{ \btag\b  .* not\sfound }x ],
+    [ '-t foo',                 %r{ \bhost\b .* not\sfound }x ],
+    [ '-t n foo',               %r{ \bhost\b .* not\sfound }x ],
+  ]
+
+  def test_hosttag
+    TESTS.each do |args, expected|
+      got = `#{@bindir}/hosttag #{@test_args} #{args} 2>&1`.chomp
+      if expected.is_a?(String)
+        assert_equal(expected, got, "args: #{args}")
+      elsif expected.is_a?(Regexp)
+        assert_match(expected, got, "args: #{args}")
+      end
+    end
+  end
+
+end
+

data/test/test_hosttag_lib.rb

@@ -0,0 +1,119 @@
+#!/usr/bin/ruby
+
+require 'test/unit'
+require 'fileutils'
+
+require 'hosttag'
+include Hosttag
+
+class TestHosttagLib < Test::Unit::TestCase
+  def setup
+    @test_args = '--server localhost --ns hosttag_testing'
+    @test_opts = { :server => 'localhost', :namespace => 'hosttag_testing',
+      :delete => true, :autoconfirm => true, :debug => false }
+    datadir = "#{File.dirname(__FILE__)}/data_hosttag"
+    bindir = File.join(File.dirname(__FILE__), '..', 'bin')
+    File.directory?(datadir) or throw "missing datadir #{datadir}"
+    hosttag_import_from_directory(datadir, @test_opts)
+  end
+
+  # format: args (array), options (hash) => expected (array)
+  TAGSET = [
+    [ %w{centos},           {},                 %w{a m n} ],
+    [ %w{laptop},           {},                 %w{n} ],
+    [ %w{vps},              {},                 %w{m} ],
+    [ %w{laptop vps},       {},                 %w{} ],
+    [ %w{laptop vps},       { :rel => :and },   %w{} ],
+    [ %w{laptop vps},       { :rel => :or },    %w{m n} ],
+  ]
+  HOSTSET = [
+    [ %w{n},                {},                 %w{centos centos5 centos5-i386 laptop} ],
+    [ %w{a},                {},                 %w{centos centos5 centos5-x86_64 public} ],
+    [ %w{m},                {},                 %w{centos centos4 centos4-x86_64 public vps} ],
+    [ %w{a n},              {},                 %w{centos centos5 centos5-i386 centos5-x86_64 laptop public} ],
+    [ %w{a n},              { :rel => :or },    %w{centos centos5 centos5-i386 centos5-x86_64 laptop public} ],
+    [ %w{a n},              { :rel => :and },   %w{centos centos5} ],
+    [ %w{a n m},            { :rel => :and },   %w{centos} ],
+  ]
+
+  def test_hosttag_lookup
+    TAGSET.each do |args, opts, expected|
+      opts.merge!(@test_opts)
+      got = hosttag_lookup(args, opts)
+      assert_equal(expected, got, "args: #{args}")
+    end
+    HOSTSET.each do |args, opts, expected|
+      opts.merge!(@test_opts)
+      got = hosttag_lookup(args, opts)
+      assert_equal(expected, got, "args: #{args}")
+    end
+  end
+
+  def test_hosttag_lookup_tags
+    TAGSET.each do |args, opts, expected|
+      opts.merge!(@test_opts)
+      got = hosttag_lookup_tags(args, opts)
+      assert_equal(expected, got, "args: #{args}")
+    end
+  end
+
+  def test_hosttag_lookup_hosts
+    HOSTSET.each do |args, opts, expected|
+      opts.merge!(@test_opts)
+      got = hosttag_lookup_hosts(args, opts)
+      assert_equal(expected, got, "args: #{args}")
+    end
+  end
+
+  # format: options (hash) => expected (array)
+  TAGSET_ALL = [
+    [ {},                           %w{centos centos4 centos4-x86_64 centos5 centos5-i386 centos5-x86_64 laptop public vps} ],
+    [ { :include_skip? => true },   %w{SKIP centos centos4 centos4-i386 centos4-x86_64 centos5 centos5-i386 centos5-x86_64 laptop public vps} ],
+  ]
+  def test_hosttag_all_tags
+    TAGSET_ALL.each do |opts, expected|
+      mopts = opts.merge(@test_opts)
+      got = hosttag_all_tags(mopts)
+      assert_equal(expected, got, "opts: #{opts}")
+    end
+  end
+
+  # format: options (hash) => expected (array)
+  HOSTSET_ALL = [
+    [ {},                           %w{a m n} ],
+    [ { :include_skip? => true },   %w{a g h m n} ],
+  ]
+  def test_hosttag_all_hosts
+    HOSTSET_ALL.each do |opts, expected|
+      mopts = opts.merge(@test_opts)
+      got = hosttag_all_hosts(mopts)
+      assert_equal(expected, got, "opts: #{opts}")
+    end
+  end
+
+    # Invalid data
+  INVALID_TAGS = [
+    [ %w{foo},                  %r{ \btag\b  .* not\sfound }x ],
+    [ %w{centos foo},           %r{ \btag\b  .* not\sfound }x ],
+  ]
+  INVALID_HOSTS = [
+    [ %w{foo},                  %r{ \bhost\b .* not\sfound }x ],
+    [ %w{n foo},                %r{ \bhost\b .* not\sfound }x ],
+  ]
+
+  def test_invalid_data
+    INVALID_TAGS.each do |args, expected|
+      exception = assert_raise RuntimeError do
+        hosttag_lookup_tags(args)
+      end
+      assert_match expected, exception.message, "args: #{args}"
+    end
+    INVALID_HOSTS.each do |args, expected|
+      exception = assert_raise RuntimeError do
+        hosttag_lookup_hosts(args)
+      end
+      assert_match expected, exception.message, "args: #{args}"
+    end
+  end
+end
+

data/test/test_htset_bin.rb

@@ -0,0 +1,174 @@
+#!/usr/bin/ruby
+
+require 'test/unit'
+require 'fileutils'
+
+require 'hosttag'
+include Hosttag
+
+class TestHtsetBin < Test::Unit::TestCase
+
+  def setup
+    @test_args = '--server localhost --ns hosttag_testing'
+    @test_opts = { :server => 'localhost', :namespace => 'hosttag_testing',
+      :delete => true, :autoconfirm => true, :debug => false }
+    datadir = "#{File.dirname(__FILE__)}/data_reset"
+    @bindir = File.join(File.dirname(__FILE__), '..', 'bin')
+    File.directory?(datadir) or throw "missing datadir #{datadir}"
+    hosttag_import_from_directory(datadir, @test_opts)
+  end
+
+  TESTS = [
+    [
+      %w{htset foo centos},         # operation
+      %w{foo},                      # ht -A output
+      %w{foo},                      # ht -A -A output
+      %w{centos},                   # ht -T output
+      %w{centos},                   # ht -T -T output
+    ],
+    [
+      %w{htset foo centos5},
+      %w{foo},
+      %w{foo},
+      %w{centos centos5},
+      %w{centos centos5},
+    ],
+    [
+      %w{htset bar centos},
+      %w{bar foo},
+      %w{bar foo},
+      %w{centos centos5},
+      %w{centos centos5},
+    ],
+    [
+      %w{htset bar centos4},
+      %w{bar foo},
+      %w{bar foo},
+      %w{centos centos4 centos5},
+      %w{centos centos4 centos5},
+    ],
+    [
+      %w{htset bar SKIP},
+      %w{foo},
+      %w{bar foo},
+      %w{centos centos5},
+      %w{SKIP centos centos4 centos5},
+    ],
+    [
+      %w{htset bar centos4-i386},
+      %w{foo},
+      %w{bar foo},
+      %w{centos centos5},
+      %w{SKIP centos centos4 centos4-i386 centos5},
+    ],
+    [
+      %w{htdel bar centos4},
+      %w{foo},
+      %w{bar foo},
+      %w{centos centos5},
+      %w{SKIP centos centos4-i386 centos5},
+    ],
+    [
+      %w{htdel bar SKIP},
+      %w{bar foo},
+      %w{bar foo},
+      %w{centos centos4-i386 centos5},
+      %w{centos centos4-i386 centos5},
+    ],
+    [
+      %w{htdel foo centos5},
+      %w{bar foo},
+      %w{bar foo},
+      %w{centos centos4-i386},
+      %w{centos centos4-i386},
+    ],
+    [
+      %w{htdel foo centos},
+      %w{bar},
+      %w{bar},
+      %w{centos centos4-i386},
+      %w{centos centos4-i386},
+    ],
+    [
+      %w{htdel bar centos},
+      %w{bar},
+      %w{bar},
+      %w{centos4-i386},
+      %w{centos4-i386},
+    ],
+    [
+      %w{htdel bar centos4-i386},
+      [],
+      [],
+      [],
+      [],
+    ],
+  ]
+
+  def test_htset
+    TESTS.each do |op, all_hosts, all_hosts_full, all_tags, all_tags_full|
+      cmd = op.join(' ')
+      bin = op.shift()
+      error = %x{#{@bindir}/#{bin} #{@test_args} #{op.join(' ')}}
+
+      # Check results
+      assert_equal("", error, "Error output found: #{error}")
+      assert_equal(all_hosts, hosttag_all_hosts(@test_opts), "all_hosts, #{cmd}")
+      assert_equal(all_hosts_full, hosttag_all_hosts(@test_opts.merge({ :include_skip? => true })),
+        "all_hosts include_skip, #{cmd}")
+      assert_equal(all_tags, hosttag_all_tags(@test_opts),  "all_tags, #{cmd}")
+      assert_equal(all_tags_full, hosttag_all_tags(@test_opts.merge({ :include_skip? => true })),
+        "all_tags_full, #{cmd}")
+    end
+  end
+
+  # Test argument classification
+  # Format: operation, error, all_hosts, all_tags
+  CLASSIFY_TESTS = [
+    [
+      %w{htset foo centos},
+      '',
+      %w{foo},
+      %w{centos},
+    ],
+    [
+      %w{htset cat foo dog centos5},            # unclassifiable host (dog) should give gives error
+      "Error: can't auto-classify 'dog' - aborting\n",
+      %w{foo},
+      %w{centos},
+    ],
+    [
+      %w{htset cat dell centos centos5-x86_64}, # unclassifiable tag (dell) should give error
+      "Error: can't auto-classify 'dell' - aborting\n",
+      %w{foo},
+      %w{centos},
+    ],
+    [
+      %w{htset cat dog foo centos5},            # known host allows classification of former args as hosts
+      '',
+      %w{cat dog foo},
+      %w{centos centos5},
+    ],
+    [
+      %w{htset cat centos dell centos5-x86_64}, # known tag allows classification of later args as tags
+      '',
+      %w{cat dog foo},
+      %w{centos centos5 centos5-x86_64 dell},
+    ],
+  ]
+
+  def test_htset_argument_classification
+    CLASSIFY_TESTS.each do |op, error, all_hosts, all_tags|
+      cmd = op.join(' ')
+      bin = op.shift()
+      output = %x{#{@bindir}/#{bin} #{@test_args} #{op.join(' ')}}
+
+      # Check results
+      assert_equal(error, output, "Error output mismatch")
+      assert_equal(all_hosts, hosttag_all_hosts(@test_opts), "all_hosts, #{cmd}")
+      assert_equal(all_tags,  hosttag_all_tags(@test_opts),  "all_tags, #{cmd}")
+    end
+  end
+
+end
+

data/test/test_htset_lib.rb

@@ -0,0 +1,183 @@
+#!/usr/bin/ruby
+
+require 'test/unit'
+require 'fileutils'
+
+require 'hosttag'
+include Hosttag
+
+class TestHtsetLib < Test::Unit::TestCase
+
+  def setup
+    @test_args = '--server localhost --ns hosttag_testing'
+    @test_opts = { :server => 'localhost', :namespace => 'hosttag_testing',
+      :delete => true, :autoconfirm => true, :debug => false }
+    datadir = "#{File.dirname(__FILE__)}/data_reset"
+    @bindir = File.join(File.dirname(__FILE__), '..', 'bin')
+    File.directory?(datadir) or throw "missing datadir #{datadir}"
+    hosttag_import_from_directory(datadir, @test_opts)
+  end
+
+  # -----------------------------------------------------------------------
+  # Testing hosttag_add_tags and hosttag_delete_tags
+  # -----------------------------------------------------------------------
+
+  TESTS = [
+    [
+      %w{htset foo centos},         # operation
+      %w{foo},                      # ht -A output
+      %w{foo},                      # ht -A -A output
+      %w{centos},                   # ht -T output
+      %w{centos},                   # ht -T -T output
+    ],
+    [
+      %w{htset foo centos5},
+      %w{foo},
+      %w{foo},
+      %w{centos centos5},
+      %w{centos centos5},
+    ],
+    [
+      %w{htset bar centos},
+      %w{bar foo},
+      %w{bar foo},
+      %w{centos centos5},
+      %w{centos centos5},
+    ],
+    [
+      %w{htset bar centos4},
+      %w{bar foo},
+      %w{bar foo},
+      %w{centos centos4 centos5},
+      %w{centos centos4 centos5},
+    ],
+    [
+      %w{htset bar SKIP},
+      %w{foo},
+      %w{bar foo},
+      %w{centos centos5},
+      %w{SKIP centos centos4 centos5},
+    ],
+    [
+      %w{htset bar centos4-i386},
+      %w{foo},
+      %w{bar foo},
+      %w{centos centos5},
+      %w{SKIP centos centos4 centos4-i386 centos5},
+    ],
+    [
+      %w{htdel bar centos4},
+      %w{foo},
+      %w{bar foo},
+      %w{centos centos5},
+      %w{SKIP centos centos4-i386 centos5},
+    ],
+    [
+      %w{htdel bar SKIP},
+      %w{bar foo},
+      %w{bar foo},
+      %w{centos centos4-i386 centos5},
+      %w{centos centos4-i386 centos5},
+    ],
+    [
+      %w{htdel foo centos5},
+      %w{bar foo},
+      %w{bar foo},
+      %w{centos centos4-i386},
+      %w{centos centos4-i386},
+    ],
+    [
+      %w{htdel foo centos},
+      %w{bar},
+      %w{bar},
+      %w{centos centos4-i386},
+      %w{centos centos4-i386},
+    ],
+    [
+      %w{htdel bar centos},
+      %w{bar},
+      %w{bar},
+      %w{centos4-i386},
+      %w{centos4-i386},
+    ],
+    [
+      %w{htdel bar centos4-i386},
+      [],
+      [],
+      [],
+      [],
+    ],
+  ]
+
+  def test_hosttag_add_delete_tags
+    TESTS.each do |op, all_hosts, all_hosts_full, all_tags, all_tags_full|
+      cmd = op.join(' ')
+      bin = op.shift()
+      if bin == 'htset'
+        hosttag_add_tags([ op[0] ], [ op[1] ], @test_opts)
+      else
+        hosttag_delete_tags([ op[0] ], [ op[1] ], @test_opts)
+      end
+
+      # Check results
+      assert_equal(all_hosts, hosttag_all_hosts(@test_opts), "all_hosts, #{cmd}")
+      assert_equal(all_hosts_full, hosttag_all_hosts(@test_opts.merge({ :include_skip? => true })),
+        "all_hosts_full, #{cmd}")
+      assert_equal(all_tags, hosttag_all_tags(@test_opts),  "all_tags, #{cmd}")
+      assert_equal(all_tags_full, hosttag_all_tags(@test_opts.merge({ :include_skip? => true })),
+        "all_tags_full, #{cmd}")
+    end
+  end
+
+  # -----------------------------------------------------------------------
+  # Testing hosttag_delete_all_tags
+  # -----------------------------------------------------------------------
+
+  # Format: host => tags
+  SETUP2 = [
+    %w{foo centos centos5 centos5-x86_64 dell},
+    %w{bar ubuntu ubuntu-lucid hp},
+    %w{cat debian debian4 dell},
+    %w{dog fedora fedora14 supermicro},
+  ]
+  TESTS2 = [
+    # Format: hosts to delete, ht -A output, ht -T output
+    [
+      %w{},
+      %w{bar cat dog foo},
+      %w{centos centos5 centos5-x86_64 debian debian4 dell fedora fedora14 hp supermicro ubuntu ubuntu-lucid},
+    ],
+    [
+      %w{foo},
+      %w{bar cat dog},
+      %w{debian debian4 dell fedora fedora14 hp supermicro ubuntu ubuntu-lucid},
+    ],
+    [
+      %w{cat dog},
+      %w{bar},
+      %w{hp ubuntu ubuntu-lucid},
+    ],
+    [
+      %w{bar},
+      %w{},
+      %w{},
+    ],
+  ]
+
+  def test_hosttag_delete_all_tags
+    SETUP2.each do |host, *tags|
+      hosttag_add_tags([ host ], tags, @test_opts)
+    end
+    TESTS2.each do |hosts, all_hosts, all_tags|
+      if hosts.length > 0
+        hosttag_delete_all_tags(hosts, @test_opts)
+      end
+
+      # Check results
+      assert_equal(all_hosts, hosttag_all_hosts(@test_opts), "all_hosts after deleting #{hosts}")
+      assert_equal(all_tags,  hosttag_all_tags(@test_opts),  "all_tags after deleting #{hosts}")
+    end
+  end
+
+end
+