From 248bc890f804cd4820b9b2a4d127eba344e9ae8c Mon Sep 17 00:00:00 2001
From: Joshua Peek <josh@joshpeek.com>
Date: Thu, 31 Dec 2009 09:44:01 -0600
Subject: [PATCH] Unit test for `brew cleanup`

---
 Library/Homebrew/test/unittest.rb | 29 ++++++++++++++++++++++++++++-
 1 file changed, 28 insertions(+), 1 deletion(-)

diff --git a/Library/Homebrew/test/unittest.rb b/Library/Homebrew/test/unittest.rb
index 8e4696fc04..c69dd5ad45 100755
--- a/Library/Homebrew/test/unittest.rb
+++ b/Library/Homebrew/test/unittest.rb
@@ -471,7 +471,34 @@ class BeerTasting <Test::Unit::TestCase
       end
     end
   end
-  
+
+  def test_brew_cleanup
+    f1=TestBall.new
+    f1.instance_eval { @version = "0.1" }
+    f2=TestBall.new
+    f2.instance_eval { @version = "0.2" }
+    f3=TestBall.new
+    f3.instance_eval { @version = "0.3" }
+
+    nostdout do
+      f1.brew { f1.install }
+      f2.brew { f2.install }
+      f3.brew { f3.install }
+    end
+
+    assert f1.installed?
+    assert f2.installed?
+    assert f3.installed?
+
+    nostdout do
+      cleanup f3
+    end
+
+    assert !f1.installed?
+    assert !f2.installed?
+    assert f3.installed?
+  end
+
   def test_my_float_assumptions
     # this may look ridiculous but honestly there's code in brewit that depends on 
     # this behaviour so I wanted to be certain Ruby floating points are behaving