test/unit/test_struct.rb

   1 require "#{File.expand_path(File.dirname(__FILE__))}/../test_helper"
   2
   3 module Dummy
   4 end
   5 class TestStruct < Test::Unit::TestCase
   6
   7    EXPECT_DB = {
   8     :SenderPort => LWES::U_INT_16,
   9     :st => LWES::STRING,
  10     :enc => LWES::INT_16,
  11     :SiteID => LWES::U_INT_16,
  12     :ReceiptTime => LWES::INT_64,
  13     :SenderIP => LWES::IP_ADDR,
  14
  15     :t_ip_addr => LWES::IP_ADDR,
  16     :t_bool => LWES::BOOLEAN,
  17     :t_uint64 => LWES::U_INT_64,
  18     :t_uint32 => LWES::U_INT_32,
  19     :t_int64 => LWES::INT_64,
  20     :t_string => LWES::STRING,
  21     :t_int32 => LWES::INT_32,
  22     :t_uint16 => LWES::U_INT_16,
  23     :t_int16 => LWES::INT_16
  24   }
  25   EXPECT_LIST = [
  26     # class-specific fields first
  27     [ :t_bool, "t_bool", LWES::BOOLEAN],
  28     [ :t_int16, "t_int16", LWES::INT_16 ],
  29     [ :t_int32, "t_int32", LWES::INT_32],
  30     [ :t_int64, "t_int64", LWES::INT_64],
  31     [ :t_ip_addr, "t_ip_addr", LWES::IP_ADDR],
  32     [ :t_string, "t_string", LWES::STRING],
  33     [ :t_uint16, "t_uint16", LWES::U_INT_16],
  34     [ :t_uint32, "t_uint32", LWES::U_INT_32],
  35     [ :t_uint64, "t_uint64", LWES::U_INT_64],
  36
  37     # MetaEventInfo
  38     [ :ReceiptTime, "ReceiptTime", LWES::INT_64],
  39     [ :SenderIP, "SenderIP", LWES::IP_ADDR],
  40     [ :SenderPort, "SenderPort", LWES::U_INT_16],
  41     [ :SiteID, "SiteID", LWES::U_INT_16],
  42     [ :enc, "enc", LWES::INT_16],
  43     [ :st, "st", LWES::STRING],
  44   ]
  45
  46   ESF_FILE = "#{File.dirname(__FILE__)}/test1.esf"
  47   MULTI_EVENT_ESF = "#{File.dirname(__FILE__)}/test2.esf"
  48
  49   def test_constants
  50     assert_instance_of Module, LWES, "LWES is not a module"
  51   end
  52
  53   def test_multiple_events_in_esf
  54     a = LWES::Struct.new(:file=>MULTI_EVENT_ESF, :class => :EventBool)
  55     assert_equal "EventBool", a.name
  56     b = LWES::Struct.new(:file=>MULTI_EVENT_ESF,
  57                          :class => :EVENT_BOOL,
  58                          :name => :EventBool)
  59     assert_equal "EVENT_BOOL", b.name
  60
  61     e = assert_raises(RuntimeError) {
  62       LWES::Struct.new(:file=>MULTI_EVENT_ESF)
  63     }
  64   end
  65
  66   def test_new_with_defaults
  67     type_db = LWES::TypeDB.new(ESF_FILE)
  68     assert type_db.instance_of?(LWES::TypeDB)
  69     a = LWES::Struct.new(:db=>type_db, :class => :TypeDB_Event)
  70     assert a.instance_of?(Class)
  71     assert_equal "TypeDB_Event", a.name
  72     assert a.const_get(:TYPE_DB).instance_of?(LWES::TypeDB)
  73     assert a.const_get(:EVENT_DEF).instance_of?(Hash)
  74     assert_equal EXPECT_DB, a.const_get(:EVENT_DEF)
  75     assert_equal EXPECT_LIST, a.const_get(:TYPE_LIST)
  76     y = a.new
  77     assert_equal "TypeDB_Event", y.class.name
  78     assert_kind_of(::Struct, y)
  79   end
  80
  81   def test_new_with_defaults
  82     a = LWES::Struct.new(:file=>ESF_FILE)
  83     assert a.instance_of?(Class)
  84     assert_equal "Event1", a.name
  85     assert a.const_get(:TYPE_DB).instance_of?(LWES::TypeDB)
  86     assert a.const_get(:EVENT_DEF).instance_of?(Hash)
  87     assert_equal EXPECT_DB, a.const_get(:EVENT_DEF)
  88     assert_equal EXPECT_LIST, a.const_get(:TYPE_LIST)
  89     y = a.new
  90     assert_equal "Event1", y.class.name
  91     assert_kind_of(::Struct, y)
  92   end
  93
  94   def test_new_with_alt_name
  95     a = LWES::Struct.new(:file=>ESF_FILE,:class=>:Event,:name=>:Event1)
  96     assert a.instance_of?(Class)
  97     assert_equal "Event", a.name
  98     assert a.const_get(:EVENT_DEF).instance_of?(Hash)
  99     assert_equal EXPECT_DB, a.const_get(:EVENT_DEF)
 100     y = a.new
 101     assert_equal "Event", y.class.name
 102     assert_kind_of(::Struct, y)
 103   end
 104
 105   def test_new_in_module
 106     a = LWES::Struct.new(:file=>ESF_FILE,:class=>:Event1,:parent=>LWES)
 107     assert a.instance_of?(Class)
 108     assert_equal "LWES::Event1", a.name
 109     assert a.const_get(:EVENT_DEF).instance_of?(Hash)
 110     assert_equal EXPECT_DB, a.const_get(:EVENT_DEF)
 111     y = a.new
 112     assert_equal "LWES::Event1", y.class.name
 113     assert_kind_of(::Struct, y)
 114   end
 115
 116   def test_eval
 117     a = LWES::Struct.new(:file=>ESF_FILE,:class=>:Eval,:name=>:Event1) do
 118       def aaaaaaaa
 119         true
 120       end
 121
 122       class << self
 123         def get_event_def
 124           const_get(:EVENT_DEF)
 125         end
 126       end
 127     end
 128     y = a.new
 129     assert y.respond_to?(:aaaaaaaa)
 130     assert_instance_of TrueClass, y.aaaaaaaa
 131     assert_equal a.const_get(:EVENT_DEF).object_id, a.get_event_def.object_id
 132     assert_equal EXPECT_DB, a.get_event_def
 133   end
 134
 135   def test_new_initialize
 136     a = LWES::Struct.new(:file=>ESF_FILE,
 137                          :class=>:Init,
 138                          :name=>:Event1,
 139                          :parent=>Dummy)
 140     y = Dummy::Init.new(:t_ip_addr => "192.168.0.1", :t_bool => true)
 141     assert_equal "192.168.0.1", y.t_ip_addr
 142     assert_equal true, y.t_bool
 143   end
 144
 145   def test_new_initialize_defaults
 146     a = LWES::Struct.new(:file=>ESF_FILE,
 147                          :class=> :LWES_Blah,
 148                          :name=>:Event1,
 149                          :defaults => { :enc => 1 })
 150     y = LWES_Blah.new
 151     assert_equal 1, y.enc
 152   end
 153
 154   def test_new_initialize_defaults_and_hash
 155     a = LWES::Struct.new(:file=>ESF_FILE,
 156                          :class=> :LWES_Foo,
 157                          :name=>:Event1,
 158                          :defaults => { :enc => 1 })
 159     y = LWES_Foo.new(:enc => 2)
 160     assert_equal 2, y.enc
 161   end
 162
 163   def test_new_initialize_struct_style
 164     a = LWES::Struct.new(:file=>ESF_FILE,
 165                          :class=> :LWES_New,
 166                          :name=>:Event1)
 167     y = LWES_Foo.new(true, -16, -32, -64, "127.0.0.1", "HI", 16, 32, 64)
 168     assert_equal true, y.t_bool
 169     assert_equal -16, y.t_int16
 170     assert_equal -32, y.t_int32
 171     assert_equal -64, y.t_int64
 172     assert_equal "127.0.0.1", y.t_ip_addr
 173     assert_equal "HI", y.t_string
 174     assert_equal 16, y.t_uint16
 175     assert_equal 32, y.t_uint32
 176     assert_equal 64, y.t_uint64
 177   end
 178
 179   def test_skip_attr
 180     a = LWES::Struct.new(:file=>ESF_FILE,
 181                          :class=> :LWES_Skip,
 182                          :name=>:Event1,
 183                          :skip => %w(SenderIP SenderPort SiteID ReceiptTime))
 184     y = LWES_Skip.new
 185     assert ! y.respond_to?(:SenderIP)
 186     assert ! y.respond_to?(:SenderPort)
 187     assert ! y.respond_to?(:SiteID)
 188     assert ! y.respond_to?(:ReceiptTime)
 189     assert y.respond_to?(:enc)
 190     assert y.respond_to?(:st)
 191   end
 192
 193   def test_skip_regexp
 194     a = LWES::Struct.new(:file=>ESF_FILE,
 195                          :class=> :LWES_SkipRe,
 196                          :name=>:Event1,
 197                          :skip => %r(\AS))
 198     y = LWES_SkipRe.new
 199     assert ! y.respond_to?(:SenderIP)
 200     assert ! y.respond_to?(:SenderPort)
 201     assert ! y.respond_to?(:SiteID)
 202     assert y.respond_to?(:ReceiptTime)
 203     assert y.respond_to?(:enc)
 204     assert y.respond_to?(:st)
 205   end
 206
 207 end