From 31ccb052d77bb9aa3f762896f2e40b285ac16b34 Mon Sep 17 00:00:00 2001
From: Jack Nagel
Date: Mon, 16 Apr 2012 16:43:42 -0500
Subject: tests: setup and teardown cleanups

Signed-off-by: Jack Nagel <jacknagel@gmail.com>
---
 Library/Homebrew/test/test_compilers.rb | 35 +++++++--------------------------
 1 file changed, 7 insertions(+), 28 deletions(-)

(limited to 'Library/Homebrew/test/test_compilers.rb')

diff --git a/Library/Homebrew/test/test_compilers.rb b/Library/Homebrew/test/test_compilers.rb
index 7ad6f2df7..dcc2332d4 100644
--- a/Library/Homebrew/test/test_compilers.rb
+++ b/Library/Homebrew/test/test_compilers.rb
@@ -8,17 +8,16 @@ ENV.extend(HomebrewEnvExtension)
 
 require 'test/testball'
 
-module CompilerTestsEnvExtension
-  def unset_use_cc
-    vars = %w{HOMEBREW_USE_CLANG HOMEBREW_USE_LLVM HOMEBREW_USE_GCC}
-    vars.each { |v| ENV.delete(v) }
+class CompilerTests < Test::Unit::TestCase
+  def setup
+    %w{HOMEBREW_USE_CLANG HOMEBREW_USE_LLVM HOMEBREW_USE_GCC}.each { |v| ENV.delete(v) }
+  end
+
+  def teardown
+    ENV.send MacOS.default_compiler
   end
-end
-ENV.extend(CompilerTestsEnvExtension)
 
-class CompilerTests < Test::Unit::TestCase
   def test_llvm_failure
-    ENV.unset_use_cc
     f = TestLLVMFailure.new
     cs = CompilerSelector.new(f)
 
@@ -32,12 +31,9 @@ class CompilerTests < Test::Unit::TestCase
       when 0..210 then :gcc
       else :clang
       end, ENV.compiler
-
-    ENV.send MacOS.default_compiler
   end
 
   def test_all_compiler_failures
-    ENV.unset_use_cc
     f = TestAllCompilerFailures.new
     cs = CompilerSelector.new(f)
 
@@ -48,12 +44,9 @@ class CompilerTests < Test::Unit::TestCase
     cs.select_compiler
 
     assert_equal MacOS.default_compiler, ENV.compiler
-
-    ENV.send MacOS.default_compiler
   end
 
   def test_no_compiler_failures
-    ENV.unset_use_cc
     f = TestNoCompilerFailures.new
     cs = CompilerSelector.new(f)
 
@@ -67,12 +60,9 @@ class CompilerTests < Test::Unit::TestCase
     cs.select_compiler
 
     assert_equal MacOS.default_compiler, ENV.compiler
-
-    ENV.send MacOS.default_compiler
   end
 
   def test_mixed_compiler_failures
-    ENV.unset_use_cc
     f = TestMixedCompilerFailures.new
     cs = CompilerSelector.new(f)
 
@@ -83,12 +73,9 @@ class CompilerTests < Test::Unit::TestCase
     cs.select_compiler
 
     assert_equal :llvm, ENV.compiler
-
-    ENV.send MacOS.default_compiler
   end
 
   def test_more_mixed_compiler_failures
-    ENV.unset_use_cc
     f = TestMoreMixedCompilerFailures.new
     cs = CompilerSelector.new(f)
 
@@ -99,12 +86,9 @@ class CompilerTests < Test::Unit::TestCase
     cs.select_compiler
 
     assert_equal :clang, ENV.compiler
-
-    ENV.send MacOS.default_compiler
   end
 
   def test_even_more_mixed_compiler_failures
-    ENV.unset_use_cc
     f = TestEvenMoreMixedCompilerFailures.new
     cs = CompilerSelector.new(f)
 
@@ -121,12 +105,9 @@ class CompilerTests < Test::Unit::TestCase
       when 0..210 then :gcc
       else :clang
       end, ENV.compiler
-
-    ENV.send MacOS.default_compiler
   end
 
   def test_block_with_no_build_compiler_failures
-    ENV.unset_use_cc
     f = TestBlockWithoutBuildCompilerFailure.new
     cs = CompilerSelector.new(f)
 
@@ -137,7 +118,5 @@ class CompilerTests < Test::Unit::TestCase
     cs.select_compiler
 
     assert_equal MacOS.default_compiler, ENV.compiler
-
-    ENV.send MacOS.default_compiler
   end
 end
-- 
cgit v1.2.3