diff options
author | jeg2 <jeg2@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2007-12-25 02:46:26 +0000 |
---|---|---|
committer | jeg2 <jeg2@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2007-12-25 02:46:26 +0000 |
commit | 29ba327bd16fd7eb7114050724e4784bdcaac83a (patch) | |
tree | 2779c5d0c96275f397aaaa286686479ef4f83649 /test/csv/tc_interface.rb | |
parent | 0d570142ba2b7488804729b3cc1a9b4d87db1014 (diff) | |
download | ruby-29ba327bd16fd7eb7114050724e4784bdcaac83a.tar.gz |
* lib/csv.rb: Import the FasterCSV source as the new CSV class.
* test/csv/*: Added all applicable tests from FasterCSV.
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@14646 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'test/csv/tc_interface.rb')
-rw-r--r-- | test/csv/tc_interface.rb | 235 |
1 files changed, 235 insertions, 0 deletions
diff --git a/test/csv/tc_interface.rb b/test/csv/tc_interface.rb new file mode 100644 index 0000000000..e8cc920f9d --- /dev/null +++ b/test/csv/tc_interface.rb @@ -0,0 +1,235 @@ +#!/usr/local/bin/ruby -w + +# tc_interface.rb +# +# Created by James Edward Gray II on 2005-10-31. +# Copyright 2005 James Edward Gray II. You can redistribute or modify this code +# under the terms of Ruby's license. + +require "test/unit" + +require "csv" + +class TestCSVInterface < Test::Unit::TestCase + def setup + @path = File.join(File.dirname(__FILE__), "temp_test_data.csv") + + File.open(@path, "w") do |file| + file << "1\t2\t3\r\n" + file << "4\t5\r\n" + end + + @expected = [%w{1 2 3}, %w{4 5}] + end + + def teardown + File.unlink(@path) + end + + ### Test Read Interface ### + + def test_foreach + CSV.foreach(@path, :col_sep => "\t", :row_sep => "\r\n") do |row| + assert_equal(@expected.shift, row) + end + end + + def test_open_and_close + csv = CSV.open(@path, "r+", :col_sep => "\t", :row_sep => "\r\n") + assert_not_nil(csv) + assert_instance_of(CSV, csv) + assert_equal(false, csv.closed?) + csv.close + assert(csv.closed?) + + ret = CSV.open(@path) do |csv| + assert_instance_of(CSV, csv) + "Return value." + end + assert(csv.closed?) + assert_equal("Return value.", ret) + end + + def test_parse + data = File.read(@path) + assert_equal( @expected, + CSV.parse(data, :col_sep => "\t", :row_sep => "\r\n") ) + + CSV.parse(data, :col_sep => "\t", :row_sep => "\r\n") do |row| + assert_equal(@expected.shift, row) + end + end + + def test_parse_line + row = CSV.parse_line("1;2;3", :col_sep => ";") + assert_not_nil(row) + assert_instance_of(Array, row) + assert_equal(%w{1 2 3}, row) + + # shortcut interface + row = "1;2;3".parse_csv(:col_sep => ";") + assert_not_nil(row) + assert_instance_of(Array, row) + assert_equal(%w{1 2 3}, row) + end + + def test_read_and_readlines + assert_equal( @expected, + CSV.read(@path, :col_sep => "\t", :row_sep => "\r\n") ) + assert_equal( @expected, + CSV.readlines(@path, :col_sep => "\t", :row_sep => "\r\n") ) + + + data = CSV.open(@path, :col_sep => "\t", :row_sep => "\r\n") do |csv| + csv.read + end + assert_equal(@expected, data) + data = CSV.open(@path, :col_sep => "\t", :row_sep => "\r\n") do |csv| + csv.readlines + end + assert_equal(@expected, data) + end + + def test_table + table = CSV.table(@path, :col_sep => "\t", :row_sep => "\r\n") + assert_instance_of(CSV::Table, table) + assert_equal([[:"1", :"2", :"3"], [4, 5, nil]], table.to_a) + end + + def test_shift # aliased as gets() and readline() + CSV.open(@path, "r+", :col_sep => "\t", :row_sep => "\r\n") do |csv| + assert_equal(@expected.shift, csv.shift) + assert_equal(@expected.shift, csv.shift) + assert_equal(nil, csv.shift) + end + end + + ### Test Write Interface ### + + def test_generate + str = CSV.generate do |csv| # default empty String + assert_instance_of(CSV, csv) + assert_equal(csv, csv << [1, 2, 3]) + assert_equal(csv, csv << [4, nil, 5]) + end + assert_not_nil(str) + assert_instance_of(String, str) + assert_equal("1,2,3\n4,,5\n", str) + + CSV.generate(str) do |csv| # appending to a String + assert_equal(csv, csv << ["last", %Q{"row"}]) + end + assert_equal(%Q{1,2,3\n4,,5\nlast,"""row"""\n}, str) + end + + def test_generate_line + line = CSV.generate_line(%w{1 2 3}, :col_sep => ";") + assert_not_nil(line) + assert_instance_of(String, line) + assert_equal("1;2;3\n", line) + + # shortcut interface + line = %w{1 2 3}.to_csv(:col_sep => ";") + assert_not_nil(line) + assert_instance_of(String, line) + assert_equal("1;2;3\n", line) + end + + def test_write_header_detection + File.unlink(@path) + + headers = %w{a b c} + CSV.open(@path, "w", :headers => true) do |csv| + csv << headers + csv << %w{1 2 3} + assert_equal(headers, csv.instance_variable_get(:@headers)) + end + end + + def test_write_lineno + File.unlink(@path) + + CSV.open(@path, "w") do |csv| + lines = 20 + lines.times { csv << %w{a b c} } + assert_equal(lines, csv.lineno) + end + end + + def test_write_hash + File.unlink(@path) + + lines = [{:a => 1, :b => 2, :c => 3}, {:a => 4, :b => 5, :c => 6}] + CSV.open( @path, "w", :headers => true, + :converters => :all, + :header_converters => :symbol ) do |csv| + csv << lines.first.keys + lines.each { |line| csv << line } + end + CSV.open( @path, "w", :headers => true, + :converters => :all, + :header_converters => :symbol ) do |csv| + csv.each { |line| assert_equal(lines.shift, line.to_hash) } + end + end + + def test_append # aliased add_row() and puts() + File.unlink(@path) + + CSV.open(@path, "w", :col_sep => "\t", :row_sep => "\r\n") do |csv| + @expected.each { |row| csv << row } + end + + test_shift + + # same thing using CSV::Row objects + File.unlink(@path) + + CSV.open(@path, "w", :col_sep => "\t", :row_sep => "\r\n") do |csv| + @expected.each { |row| csv << CSV::Row.new(Array.new, row) } + end + + test_shift + end + + ### Test Read and Write Interface ### + + def test_filter + assert_respond_to(CSV, :filter) + + expected = [[1, 2, 3], [4, 5]] + CSV.filter( "1;2;3\n4;5\n", (result = String.new), + :in_col_sep => ";", :out_col_sep => ",", + :converters => :all ) do |row| + assert_equal(row, expected.shift) + row.map! { |n| n * 2 } + row << "Added\r" + end + assert_equal("2,4,6,\"Added\r\"\n8,10,\"Added\r\"\n", result) + end + + def test_instance + csv = String.new + + first = nil + assert_nothing_raised(Exception) do + first = CSV.instance(csv, :col_sep => ";") + first << %w{a b c} + end + + assert_equal("a;b;c\n", csv) + + second = nil + assert_nothing_raised(Exception) do + second = CSV.instance(csv, :col_sep => ";") + second << [1, 2, 3] + end + + assert_equal(first.object_id, second.object_id) + assert_equal("a;b;c\n1;2;3\n", csv) + + # shortcuts + assert_equal(STDOUT, CSV.instance.instance_eval { @io }) + assert_equal(STDOUT, CSV { |csv| csv.instance_eval { @io } }) + end +end |