# $Id$
#
#


$:.unshift "lib"

require 'net/snmp'
require 'stringio'


class TestSnmp < Test::Unit::TestCase

  SnmpGetRequest = "0'\002\001\000\004\006public\240\032\002\002?*\002\001\000\002\001\0000\0160\f\006\b+\006\001\002\001\001\001\000\005\000"
  SnmpGetResponse = "0+\002\001\000\004\006public\242\036\002\002'\017\002\001\000\002\001\0000\0220\020\006\b+\006\001\002\001\001\001\000\004\004test"


  def setup
  end

  def teardown
  end

  def test_invalid_packet
      data = "xxxx"
      assert_raise( Net::BER::BerError ) {
	ary = data.read_ber(Net::SNMP::AsnSyntax)
      }

  end

  # The method String#read_ber! added by Net::BER consumes a well-formed BER object
  # from the head of a string. If it doesn't find a complete, well-formed BER object,
  # it returns nil and leaves the string unchanged. If it finds an object, it returns
  # the object and removes it from the head of the string. This is good for handling
  # partially-received data streams, such as from network connections.
  def test_consume_string
      data = "xxx"
      assert_equal( nil, data.read_ber! )
      assert_equal( "xxx", data )

      data = SnmpGetRequest + "!!!"
      ary = data.read_ber!( Net::SNMP::AsnSyntax )
      assert_equal( "!!!", data )
      assert ary.is_a?(Array)
      assert ary.is_a?(Net::BER::BerIdentifiedArray)
  end

  def test_weird_packet
      assert_raise( Net::SnmpPdu::Error ) {
	Net::SnmpPdu.parse("aaaaaaaaaaaaaa")
      }
  end

  def test_get_request
      data = SnmpGetRequest.dup
      pkt = data.read_ber(Net::SNMP::AsnSyntax)
      assert pkt.is_a?(Net::BER::BerIdentifiedArray)
      assert_equal( 48, pkt.ber_identifier) # Constructed [0], signifies GetRequest

      pdu = Net::SnmpPdu.parse(pkt)
      assert_equal(:get_request, pdu.pdu_type )
      assert_equal(16170, pdu.request_id ) # whatever was in the test data. 16170 is not magic.
      assert_equal( [[[1,3,6,1,2,1,1,1,0],nil]], pdu.variables )

      assert_equal( pdu.to_ber_string, SnmpGetRequest )
  end

  def test_empty_pdu
      pdu = Net::SnmpPdu.new
      assert_raise( Net::SnmpPdu::Error ) {
	pdu.to_ber_string
      }
  end

  def test_malformations
      pdu = Net::SnmpPdu.new
	pdu.version = 0
	pdu.version = 2
      assert_raise( Net::SnmpPdu::Error ) {
	pdu.version = 100
      }

      pdu.pdu_type = :get_request
      pdu.pdu_type = :get_next_request
      pdu.pdu_type = :get_response
      pdu.pdu_type = :set_request
      pdu.pdu_type = :trap
      assert_raise( Net::SnmpPdu::Error ) {
	pdu.pdu_type = :something_else
      }
  end

  def test_make_response
      pdu = Net::SnmpPdu.new
      pdu.version = 0
      pdu.community = "public"
      pdu.pdu_type = :get_response
      pdu.request_id = 9999
      pdu.error_status = 0
      pdu.error_index = 0
      pdu.add_variable_binding [1,3,6,1,2,1,1,1,0], "test"

      assert_equal( SnmpGetResponse,  pdu.to_ber_string )
  end

  def test_make_bad_response
      pdu = Net::SnmpPdu.new
      assert_raise(Net::SnmpPdu::Error) {pdu.to_ber_string}
      pdu.pdu_type = :get_response
      pdu.request_id = 999
      pdu.to_ber_string
      # Not specifying variables doesn't create an error. (Maybe it should?)
  end

end