Handle numeric version numbers #13

Author: JasonBarnabe

Rakefile

@@ -1 +1,6 @@
 require "bundler/gem_tasks"
+
+require 'rake/testtask'
+Rake::TestTask.new do |t|
+  t.pattern = 'test/**/*_test.rb'
+end

lib/qbxml/qbxml.rb

@@ -86,6 +86,8 @@ def parse_schema(key, version)
   end
 
   def select_schema(schema_key, version)
+    # Try to handle it if a user gave us a numeric version. Assume 1 decimal.
+    version = '%.1f' % version if version.is_a?(Numeric)
     raise "invalid schema '#{schema_key}', must be one of #{SCHEMAS.keys.inspect}" if !SCHEMAS.has_key?(schema_key)
     raise "invalid version '#{version}' for schema #{schema_key}, must be one of #{SCHEMAS[schema_key].keys.inspect}" if !SCHEMAS[schema_key].has_key?(version)
     return SCHEMAS[schema_key][version]

test/unit/version_test.rb

@@ -0,0 +1,24 @@
+require 'minitest/autorun'
+require 'qbxml'
+
+class VersionTest < Minitest::Test
+
+  def test_string_version
+    Qbxml.new(:qb, '7.0')
+  end
+
+  def test_bad_version
+    assert_raises RuntimeError do
+      Qbxml.new(:qb, '3.14')
+    end
+  end
+
+  def test_float_version
+    Qbxml.new(:qb, 7.0)
+  end
+
+  def test_int_version
+    Qbxml.new(:qb, 7)
+  end
+
+end