metasploit-framework/lib/msf/core/exploit/tcp.rb.ut.rb

#!/usr/bin/ruby

$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..', '..'))

require 'test/unit'
require 'msf/core'

module Msf

class Exploit::Remote::Tcp::UnitTest < Test::Unit::TestCase

	class Stub < Msf::Exploit::Remote
		include Msf::Exploit::Remote::Tcp
	end

	def test_tcp
		e = Stub.new

		assert_equal(2, e.options.length, "invalid options length")
		assert_equal(true, e.options.get('RHOST').required?, "invalid RHOST requirement")
		assert_equal(true, e.options.get('RPORT').required?, "invalid RPORT requirement")
	end

end

end