require File.dirname(__FILE__) + '/spec_helper.rb'

describe CsvMapper do
  describe "included" do
    before(:each) do    
      @mapped_klass = Class.new do
         include CsvMapper
         def upcase_name(row, index)
           row[index].upcase
         end
       end
      @mapped = @mapped_klass.new
    end

    it "should allow the creation of CSV mappings" do
      mapping = @mapped.map_csv do
        start_at_row 2      
      end
    
      mapping.should be_instance_of(CsvMapper::RowMap)
      mapping.start_at_row.should == 2
    end
  
    it "should import a CSV IO" do
      io = 'foo,bar,00,01'
      results = @mapped.import(io, :type => :io) do 
        first
        second
      end
    
      results.should be_kind_of(Enumerable)
      results.should have(1).things
      results[0].first.should == 'foo'
      results[0].second.should == 'bar'
    end
  
    it "should import a CSV File IO" do
      results = @mapped.import(File.dirname(__FILE__) + '/test.csv') do
        start_at_row 1
        [first_name, last_name, age]
      end
    
      results.size.should == 3
    end 
  
    it "should stop importing at a specified row" do
      results = @mapped.import(File.dirname(__FILE__) + '/test.csv') do
        start_at_row 1
        stop_at_row 2
        [first_name, last_name, age]
      end

      results.size.should == 2
    end

    it "should be able to read attributes from a csv file" do
      results = @mapped.import(File.dirname(__FILE__) + '/test.csv') do
        # we'll alias age here just as an example
        read_attributes_from_file('Age' => 'number_of_years_old')
      end
      results[1].first_name.should == 'Jane'
      results[1].last_name.should == 'Doe'
      results[1].number_of_years_old.should == '26'
    end

    it "should import non-comma delimited files" do
      piped_io = 'foo|bar|00|01'
    
      results = @mapped.import(piped_io, :type => :io) do
        delimited_by '|'
        [first, second]
      end
    
      results.should have(1).things
      results[0].first.should == 'foo'
      results[0].second.should == 'bar'
    end
    
    it "should allow named tranformation mappings" do
      def upcase_name(row)
        row[0].upcase
      end
      
      results = @mapped.import(File.dirname(__FILE__) + '/test.csv') do
        start_at_row 1
        
        first_name.map :upcase_name
      end
      
      results[0].first_name.should == 'JOHN'
    end
  end
  
  describe "extended" do
    it "should allow the creation of CSV mappings" do
      mapping = CsvMapper.map_csv do
        start_at_row 2      
      end
    
      mapping.should be_instance_of(CsvMapper::RowMap)
      mapping.start_at_row.should == 2
    end
  
    it "should import a CSV IO" do
      io = 'foo,bar,00,01'
      results = CsvMapper.import(io, :type => :io) do 
        first
        second
      end
    
      results.should be_kind_of(Enumerable)
      results.should have(1).things
      results[0].first.should == 'foo'
      results[0].second.should == 'bar'
    end
  
    it "should import a CSV File IO" do
      results = CsvMapper.import(File.dirname(__FILE__) + '/test.csv') do
        start_at_row 1
        [first_name, last_name, age]
      end
    
      results.size.should == 3
    end 
  
    it "should stop importing at a specified row" do
      results = CsvMapper.import(File.dirname(__FILE__) + '/test.csv') do
        start_at_row 1
        stop_at_row 2
        [first_name, last_name, age]
      end

      results.size.should == 2
    end

    it "should be able to read attributes from a csv file" do
      results = CsvMapper.import(File.dirname(__FILE__) + '/test.csv') do
        # we'll alias age here just as an example
        read_attributes_from_file('Age' => 'number_of_years_old')
      end
      results[1].first_name.should == 'Jane'
      results[1].last_name.should == 'Doe'
      results[1].number_of_years_old.should == '26'
    end

    it "should import non-comma delimited files" do
      piped_io = 'foo|bar|00|01'
    
      results = CsvMapper.import(piped_io, :type => :io) do
        delimited_by '|'
        [first, second]
      end
    
      results.should have(1).things
      results[0].first.should == 'foo'
      results[0].second.should == 'bar'
    end
    
    it "should not allow tranformation mappings" do
      def upcase_name(row)
        row[0].upcase
      end
      
      (lambda do 
        results = CsvMapper.import(File.dirname(__FILE__) + '/test.csv') do
          start_at_row 1
        
          first_name.map :upcase_name
        end
      end).should raise_error(Exception)
    end
  end
end