unit tests for transformer

git-svn-id: file:///home/svn/incoming/trunk@2562 4d416f70-5f16-0410-b530-b9f4589650da

lib/rex.rb.ts.rb

@@ -2,11 +2,15 @@
 
 require 'test/unit'
 require 'Rex'
+
+require 'Rex/Transformer.rb.ut'
+
 require 'Rex/Encoding/Xor/Generic.rb.ut'
 require 'Rex/Encoding/Xor/Byte.rb.ut'
 require 'Rex/Encoding/Xor/Word.rb.ut'
 require 'Rex/Encoding/Xor/DWord.rb.ut'
 require 'Rex/Encoding/Xor/DWordAdditive.rb.ut'
+
 require 'Rex/Socket.rb.ut'
 require 'Rex/Socket/Tcp.rb.ut'
 require 'Rex/Socket/SslTcp.rb.ut'
@@ -19,6 +23,9 @@ class Rex::TestSuite
 	def self.suite
 		suite = Test::Unit::TestSuite.new("Rex")
 
+		# General
+		suite << Rex::Transformer::UnitTest.suite
+
 		# Encoding
 		suite << Rex::Encoding::Xor::Generic::UnitTest.suite
 		suite << Rex::Encoding::Xor::Byte::UnitTest.suite

lib/rex/transformer.rb.ut.rb

@@ -0,0 +1,25 @@
+#!/usr/bin/ruby
+
+$:.unshift(File.join('..', File.dirname(__FILE__)))
+
+require 'test/unit'
+require 'Rex/Transformer'
+
+class Rex::Transformer::UnitTest < Test::Unit::TestCase
+	class Pizza
+		def Pizza.from_s(str)
+		end
+	end
+
+	def test_transformer
+		a = Rex::Transformer.transform([ 'yo', 'ho' ], Array, [ String ], 'Jones')
+
+		assert_equal(2, a.length, "valid array length")
+		assert_equal('yo', a[0], "valid first element")
+		assert_equal('ho', a[1], "valid second element")
+
+		assert_raises(ArgumentError, "invalid transform") {
+			Rex::Transformer.transform('dog', Array, [ Pizza ], 'bob')
+		}
+	end
+end