Replace version strings with Version objects

4 files changed, 41 insertions, 37 deletions

diff --git a/Library/Homebrew/test/test_bucket.rb b/Library/Homebrew/test/test_bucket.rb
index 0758d2623..2963a5532 100644
--- a/Library/Homebrew/test/test_bucket.rb
+++ b/Library/Homebrew/test/test_bucket.rb
@@ -22,6 +22,8 @@ end
 # separate TestCase classes.
 
 class BeerTasting < Test::Unit::TestCase
+  include VersionAssertions
+
   def test_supported_compressed_types
     assert_nothing_raised do
       MockFormula.new 'test-0.1.tar.gz'
@@ -129,12 +131,6 @@ class BeerTasting < Test::Unit::TestCase
     assert_equal 10.7, f+0.1
   end
 
-  def test_pathname_version
-    d=HOMEBREW_CELLAR+'foo-0.1.9'
-    d.mkpath
-    assert_equal '0.1.9', d.version
-  end
-
   def test_pathname_plus_yeast
     nostdout do
       assert_nothing_raised do
@@ -176,12 +172,12 @@ class BeerTasting < Test::Unit::TestCase
 
     assert_equal '.tar.gz', foo1.extname
     assert_equal 'foo-0.1', foo1.stem
-    assert_equal '0.1', foo1.version
+    assert_version_equal '0.1', foo1.version
 
     foo1 = HOMEBREW_CACHE/'foo-0.1.cpio.gz'
     assert_equal '.cpio.gz', foo1.extname
     assert_equal 'foo-0.1', foo1.stem
-    assert_equal '0.1', foo1.version
+    assert_version_equal '0.1', foo1.version
   end
 
   class MockMockFormula < Struct.new(:name); end
diff --git a/Library/Homebrew/test/test_formula.rb b/Library/Homebrew/test/test_formula.rb
index 686d8b0f0..e60ddd051 100644
--- a/Library/Homebrew/test/test_formula.rb
+++ b/Library/Homebrew/test/test_formula.rb
@@ -10,6 +10,7 @@ class MostlyAbstractFormula < Formula
 end
 
 class FormulaTests < Test::Unit::TestCase
+  include VersionAssertions
 
   def test_prefix
     nostdout do
@@ -65,7 +66,7 @@ class FormulaTests < Test::Unit::TestCase
     assert_equal 'http://example.com', f.homepage
     assert_equal 'file:///foo.com/testball-0.1.tbz', f.url
     assert_equal 1, f.mirrors.length
-    assert_equal '0.1', f.version
+    assert_version_equal '0.1', f.version
     assert_equal f.stable, f.active_spec
     assert_equal CurlDownloadStrategy, f.download_strategy
     assert_instance_of CurlDownloadStrategy, f.downloader
@@ -124,10 +125,10 @@ class FormulaTests < Test::Unit::TestCase
     assert !f.stable.explicit_version?
     assert !f.bottle.explicit_version?
     assert !f.devel.explicit_version?
-    assert_equal '0.1', f.stable.version
-    assert_equal '0.1', f.bottle.version
-    assert_equal '0.2', f.devel.version
-    assert_equal 'HEAD', f.head.version
+    assert_version_equal '0.1', f.stable.version
+    assert_version_equal '0.1', f.bottle.version
+    assert_version_equal '0.2', f.devel.version
+    assert_version_equal 'HEAD', f.head.version
     assert_equal 0, f.bottle.revision
   end
 
@@ -135,7 +136,7 @@ class FormulaTests < Test::Unit::TestCase
     ARGV.push '--devel'
     f = SpecTestBall.new
     assert_equal f.devel, f.active_spec
-    assert_equal '0.2', f.version
+    assert_version_equal '0.2', f.version
     assert_equal 'file:///foo.com/testball-0.2.tbz', f.url
     assert_equal CurlDownloadStrategy, f.download_strategy
     assert_instance_of CurlDownloadStrategy, f.downloader
@@ -146,7 +147,7 @@ class FormulaTests < Test::Unit::TestCase
     ARGV.push '--HEAD'
     f = SpecTestBall.new
     assert_equal f.head, f.active_spec
-    assert_equal 'HEAD', f.version
+    assert_version_equal 'HEAD', f.version
     assert_equal 'https://github.com/mxcl/homebrew.git', f.url
     assert_equal GitDownloadStrategy, f.download_strategy
     assert_instance_of GitDownloadStrategy, f.downloader
@@ -155,9 +156,9 @@ class FormulaTests < Test::Unit::TestCase
 
   def test_explicit_version_spec
     f = ExplicitVersionSpecTestBall.new
-    assert_equal '0.3', f.version
-    assert_equal '0.3', f.stable.version
-    assert_equal '0.4', f.devel.version
+    assert_version_equal '0.3', f.version
+    assert_version_equal '0.3', f.stable.version
+    assert_version_equal '0.4', f.devel.version
     assert f.stable.explicit_version?
     assert f.devel.explicit_version?
   end
@@ -183,7 +184,7 @@ class FormulaTests < Test::Unit::TestCase
 
       assert !f.bottle.explicit_version?
       assert_equal 0, f.bottle.revision
-      assert_equal '0.1', f.bottle.version
+      assert_version_equal '0.1', f.bottle.version
     else
       assert_nil f.bottle
     end
@@ -203,7 +204,7 @@ class FormulaTests < Test::Unit::TestCase
     assert_nil f.devel
 
     assert_equal f.head, f.active_spec
-    assert_equal 'HEAD', f.version
+    assert_version_equal 'HEAD', f.version
     assert_nil f.head.checksum
     assert_equal 'https://github.com/mxcl/homebrew.git', f.url
     assert_equal GitDownloadStrategy, f.download_strategy
@@ -220,7 +221,7 @@ class FormulaTests < Test::Unit::TestCase
     assert_nil f.devel
 
     assert_equal f.head, f.active_spec
-    assert_equal 'HEAD', f.version
+    assert_version_equal 'HEAD', f.version
     assert_nil f.head.checksum
     assert_equal 'https://github.com/mxcl/homebrew.git', f.url
     assert_equal GitDownloadStrategy, f.download_strategy
@@ -237,7 +238,7 @@ class FormulaTests < Test::Unit::TestCase
     assert_nil f.devel
 
     assert_equal f.head, f.active_spec
-    assert_equal 'HEAD', f.version
+    assert_version_equal 'HEAD', f.version
     assert_nil f.head.checksum
     assert_equal 'https://github.com/mxcl/homebrew.git', f.url
     assert_equal GitDownloadStrategy, f.download_strategy
diff --git a/Library/Homebrew/test/test_versions.rb b/Library/Homebrew/test/test_versions.rb
index a432b1824..ab6f83991 100644
--- a/Library/Homebrew/test/test_versions.rb
+++ b/Library/Homebrew/test/test_versions.rb
@@ -3,20 +3,6 @@ require 'formula'
 require 'test/testball'
 require 'version'
 
-module VersionAssertions
-  def assert_version_detected expected, url
-    assert_equal expected, Version.parse(url).to_s
-  end
-
-  def assert_version_nil url
-    assert_nil Version.parse(url)
-  end
-
-  def assert_version_comparison a, comparison, b
-    eval "assert Version.new(a) #{comparison} Version.new(b)"
-  end
-end
-
 class TestBadVersion < TestBall
   def initialize name=nil
     @stable = SoftwareSpec.new
@@ -34,6 +20,9 @@ class VersionComparisonTests < Test::Unit::TestCase
     assert_version_comparison '1.2.3', '>', '1.2.2'
     assert_version_comparison '1.2.3-p34', '>', '1.2.3-p33'
     assert_version_comparison '1.2.4', '<', '1.2.4.1'
+    assert_version_comparison 'HEAD', '==', 'HEAD'
+    assert_version_comparison 'HEAD', '>', '1.2.3'
+    assert_version_comparison '1.2.3', '<', 'HEAD'
   end
 end
 
@@ -43,7 +32,7 @@ class VersionParsingTests < Test::Unit::TestCase
   def test_pathname_version
     d = HOMEBREW_CELLAR/'foo-0.1.9'
     d.mkpath
-    assert_equal '0.1.9', d.version
+    assert_version_equal '0.1.9', d.version
   end
 
   def test_no_version
diff --git a/Library/Homebrew/test/testing_env.rb b/Library/Homebrew/test/testing_env.rb
index b01fe2295..d0dbd9401 100644
--- a/Library/Homebrew/test/testing_env.rb
+++ b/Library/Homebrew/test/testing_env.rb
@@ -65,3 +65,21 @@ ARGV.extend(HomebrewArgvExtension)
 
 require 'extend/ENV'
 ENV.extend(HomebrewEnvExtension)
+
+module VersionAssertions
+  def assert_version_equal expected, actual
+    assert_equal Version.new(expected), actual
+  end
+
+  def assert_version_detected expected, url
+    assert_equal expected, Version.parse(url).to_s
+  end
+
+  def assert_version_nil url
+    assert_nil Version.parse(url)
+  end
+
+  def assert_version_comparison a, comparison, b
+    eval "assert Version.new(a) #{comparison} Version.new(b)"
+  end
+end