From 216becef1828794a1defe2ecefd570bceca53693 Mon Sep 17 00:00:00 2001
From: Matt Miller <mmiller@hick.org>
Date: Sat, 4 Jun 2005 18:32:51 +0000
Subject: [PATCH] unit tests for transformer

git-svn-id: file:///home/svn/incoming/trunk@2562 4d416f70-5f16-0410-b530-b9f4589650da
---
 lib/rex.rb.ts.rb             |  7 +++++++
 lib/rex/transformer.rb.ut.rb | 25 +++++++++++++++++++++++++
 2 files changed, 32 insertions(+)
 create mode 100644 lib/rex/transformer.rb.ut.rb

diff --git a/lib/rex.rb.ts.rb b/lib/rex.rb.ts.rb
index 89fc1ab036..02b691ffef 100644
--- a/lib/rex.rb.ts.rb
+++ b/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
diff --git a/lib/rex/transformer.rb.ut.rb b/lib/rex/transformer.rb.ut.rb
new file mode 100644
index 0000000000..60838c61df
--- /dev/null
+++ b/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