require 'runit/testcase'
require 'runit/cui/testrunner'

require 'image_size'

class TestImageSize < RUNIT::TestCase

	def setup
		@files = ['4_1_2.gif', '2-4-7.png', 'tokyo_tower.jpg', 'bmp.bmp', 
		          'ppm.ppm', 'pgm.pgm', 'pbm.pbm', 
		          'cursor.xbm', 'tiff.tiff', 'test.xpm', 
		          'tower_e.gif.psd', 'pcx.pcx', 'detect.swf']
		@results = [
		  ['GIF',668,481],
		  ['PNG',640,532],
		  ['JPEG',320,240],
		  ['BMP',50,50],
		  ['PPM',80,50],
		  ['PGM',90,55],
		  ['PBM',85,55],
		  ['XBM',16,16],
		  ['TIFF',64,64],
		  ['XPM',32,32],
		  ['PSD',20,20],
		  ['PCX',70,60],
		  ['SWF',450,450],
		]
	end

	def teardown
	end

	def test_0_string
		print "\n" if $VERBOSE

		@files.each_index do |i|
			file = @files[i]
			result = @results[i]

			open(file, "rb") do |fh|
				img_data = fh.read
				print "file  =#{file}\n" if $VERBOSE

				img = ImageSize.new(img_data, result[0])

				print "type  =#{img.get_type}\n" if $VERBOSE
				print "width =#{img.get_width}\n" if $VERBOSE
				print "height=#{img.get_height}\n" if $VERBOSE

				assert_equal(result[1], img.get_width)
				assert_equal(result[2], img.get_height)

				img = ImageSize.new(img_data)
				assert_equal(result[0], img.get_type)
				assert_equal(result[1], img.get_width)
				assert_equal(result[2], img.get_height)
			end
		end
	end

	def test_1_io
		print "\n" if $VERBOSE

		@files.each_index do |i|
			file = @files[i]
			result = @results[i]

			open(file, "rb") do |fh|
				print "file  =#{file}\n" if $VERBOSE

				img = ImageSize.new(fh)
				assert_equal(result[0], img.get_type)
				assert_equal(result[1], img.get_width)
				assert_equal(result[2], img.get_height)
			end
		end
	end
end


suite = TestImageSize::suite
RUNIT::CUI::TestRunner.run(suite)