test/errata_test.rb in errata-0.1.4 vs test/errata_test.rb in errata-0.1.5

@@ -1,34 +1,77 @@
 require 'test_helper'
 
-class Brewery
-  def self.is_microbrew?(row)
-    row[:volume] < 500
+class AutomobileVariant
+  class << self
+    def transmission_is_blank?(row)
+      row['transmission'].blank?
+    end
+
+    def is_a_2007_gmc_or_chevrolet?(row)
+      row['year'] == 2007 and %w(GMC CHEVROLET).include? row['MFR'].upcase
+    end
+
+    def is_a_porsche?(row)
+      row.slice('MFR', 'Manufacturer', 'carline_mfr_name').any? { |k, v| v.upcase == 'PORSCHE' }
+    end
+
+    def is_not_a_porsche?(row)
+      !is_a_porsche? row
+    end
+
+    def is_a_mercedes_benz?(row)
+      row.slice('MFR', 'Manufacturer', 'carline_mfr_name').any? { |k, v| v =~ /MERCEDES/i }
+    end
+
+    def is_a_lexus?(row)
+      row.slice('MFR', 'Manufacturer', 'carline_mfr_name').any? { |k, v| v.upcase == 'LEXUS' }
+    end
+
+    def is_a_bmw?(row)
+      row.slice('MFR', 'Manufacturer', 'carline_mfr_name').any? { |k, v| v.upcase == 'BMW' }
+    end
+
+    def is_a_ford?(row)
+      row.slice('MFR', 'Manufacturer', 'carline_mfr_name').any? { |k, v| v.upcase == 'FORD' }
+    end
+
+    def is_a_rolls_royce_and_model_contains_bentley?(row)
+      is_a_rolls_royce?(row) and model_contains_bentley?(row)
+    end
+
+    def is_a_bentley?(row)
+      row.slice('MFR', 'Manufacturer', 'carline_mfr_name').any? { |k, v| v.upcase == 'BENTLEY' }
+    end
+
+    def is_a_rolls_royce?(row)
+      row.slice('MFR', 'Manufacturer', 'carline_mfr_name').any? { |k, v| v =~ /ROLLS/i }
+    end
+
+    def is_a_turbo_brooklands?(row)
+      row.slice('CAR LINE', 'carline name', 'carline_name').any? { |k, v| v =~ /TURBO R\/RL BKLDS/i }
+    end
+
+    def model_contains_maybach?(row)
+      row.slice('CAR LINE', 'carline name', 'carline_name').any? { |k, v| v =~ /MAYBACH/i }
+    end
+    
+    def model_contains_bentley?(row)
+      row.slice('CAR LINE', 'carline name', 'carline_name').any? { |k, v| v =~ /BENTLEY/i }
+    end
   end
 end
 
 class ErrataTest < Test::Unit::TestCase
   def setup
-    @e = Errata.new(:url => 'http://static.brighterplanet.com/science/data/transport/automobiles/make_fleet_years/errata.csv')
+    @e = Errata.new(:url => 'http://spreadsheets.google.com/pub?key=t9WkYT39zjrStx7ruCFrZJg', :klass => AutomobileVariant)
   end
-  
-  should "return implied matching methods" do
-    flunk "create a fake errata for Brewery"
-  end
-  
-  should "use matching methods" do
-    flunk "create a fake errata for Brewery"
-  end
-  
+    
   should "correct rows" do
-    rover = { 'manufacturer_name' => 'foobar Austin Rover foobar' }
-    mercedes = { 'manufacturer_name' => 'MERCEDES' }
-    @e.correct!(mercedes)
-    @e.correct!(rover)
-    assert_equal 'Mercedes-Benz', mercedes['manufacturer_name']
-    assert_equal 'Rover',         rover['manufacturer_name']
+    alfa = { "carline_mfr_name"=>"ALFA ROMEO" }
+    @e.correct!(alfa)
+    assert_equal 'Alfa Romeo', alfa['carline_mfr_name']
   end
   
   should "reject rows" do
-    assert @e.rejects?('manufacturer_name' => 'AURORA CARS')
+    assert @e.rejects?('carline_mfr_name' => 'AURORA CARS')
   end
 end