CSV.swift/Tests/CSV/CSVTests.swift

//
//  CSVTests.swift
//  CSV
//
//  Created by Yasuhiro Hatta on 2016/06/11.
//
//

import XCTest
@testable import CSV

class CSVTests: XCTestCase {

    func test1Line() {
        let csv = "\"abc\",1,2"
        var i = 0
        for row in try! CSV(string: csv) {
            switch i {
            case 0: XCTAssertEqual(row, ["abc", "1", "2"])
            default: break
            }
            i += 1
        }
        XCTAssertEqual(i, 1)
    }
    
    func test2Lines() {
        let csv = "\"abc\",1,2\n\"cde\",3,4"
        var i = 0
        for row in try! CSV(string: csv) {
            switch i {
            case 0: XCTAssertEqual(row, ["abc", "1", "2"])
            case 1: XCTAssertEqual(row, ["cde", "3", "4"])
            default: break
            }
            i += 1
        }
        XCTAssertEqual(i, 2)
    }
    
    func testLastLineIsEmpty() {
        let csv = "\"abc\",1,2\n\"cde\",3,4\n"
        var i = 0
        for row in try! CSV(string: csv) {
            switch i {
            case 0: XCTAssertEqual(row, ["abc", "1", "2"])
            case 1: XCTAssertEqual(row, ["cde", "3", "4"])
            default: break
            }
            i += 1
        }
        XCTAssertEqual(i, 2)
    }

    func testLastLineIsWhiteSpace() {
        let csv = "\"abc\",1,2\n\"cde\",3,4\n "
        var i = 0
        for row in try! CSV(string: csv) {
            switch i {
            case 0: XCTAssertEqual(row, ["abc", "1", "2"])
            case 1: XCTAssertEqual(row, ["cde", "3", "4"])
            case 2: XCTAssertEqual(row, [" "])
            default: break
            }
            i += 1
        }
        XCTAssertEqual(i, 3)
    }

    func testMiddleLineEmpty() {
        let csv = "\"abc\",1,2\n\n\"cde\",3,4"
        var i = 0
        for row in try! CSV(string: csv) {
            switch i {
            case 0: XCTAssertEqual(row, ["abc", "1", "2"])
            case 1: XCTAssertEqual(row, [""])
            case 2: XCTAssertEqual(row, ["cde", "3", "4"])
            default: break
            }
            i += 1
        }
        XCTAssertEqual(i, 3)
    }
    
    func testDoubleQuoteBeforeLineBreak() {
        let csv = "\"abc\",1,\"2\"\n\n\"cde\",3,\"4\""
        var i = 0
        for row in try! CSV(string: csv) {
            switch i {
            case 0: XCTAssertEqual(row, ["abc", "1", "2"])
            case 1: XCTAssertEqual(row, [""])
            case 2: XCTAssertEqual(row, ["cde", "3", "4"])
            default: break
            }
            i += 1
        }
        XCTAssertEqual(i, 3)
    }
    
//    func testBufferSizeMod0() {
//        let csvString = "0,1,2,3,4,5,6,7,8,9\n"
//        let csv = try! CSV(string: csvString, bufferSize: 12)
//        XCTAssertEqual(csv.bufferSize, 12)
//    }
//    
//    func testBufferSizeMod1() {
//        let csvString = "0,1,2,3,4,5,6,7,8,9\n"
//        let csv = try! CSV(string: csvString, bufferSize: 13)
//        XCTAssertEqual(csv.bufferSize, 16)
//    }
//    
//    func testBufferSizeMod2() {
//        let csvString = "0,1,2,3,4,5,6,7,8,9\n"
//        let csv = try! CSV(string: csvString, bufferSize: 14)
//        XCTAssertEqual(csv.bufferSize, 16)
//    }
//    
//    func testBufferSizeMod3() {
//        let csvString = "0,1,2,3,4,5,6,7,8,9\n"
//        let csv = try! CSV(string: csvString, bufferSize: 15)
//        XCTAssertEqual(csv.bufferSize, 16)
//    }
//    
//    func testBufferSizeMod4() {
//        let csvString = "0,1,2,3,4,5,6,7,8,9\n"
//        let csv = try! CSV(string: csvString, bufferSize: 16)
//        XCTAssertEqual(csv.bufferSize, 16)
//    }
//    
//    func testBigDataAndSmallBufferSize() {
//        let line = "0,1,2,3,4,5,6,7,8,9\n"
//        var csv = ""
//        for _ in 0..<10000 {
//            csv += line
//        }
//        var i = 0
//        for row in try! CSV(string: csv, bufferSize: 10) {
//            XCTAssertEqual(row, ["0", "1", "2", "3", "4", "5", "6", "7", "8", "9"])
//            i += 1
//        }
//        XCTAssertEqual(i, 10000)
//    }

    func testSubscript() {
        let csvString = "id,name\n001,hoge\n002,fuga"
        var csv = try! CSV(string: csvString, hasHeaderRow: true)
        var i = 0
        while csv.next() != nil {
            switch i {
            case 0:
                XCTAssertEqual(csv["id"], "001")
                XCTAssertEqual(csv["name"], "hoge")
            case 1:
                XCTAssertEqual(csv["id"], "002")
                XCTAssertEqual(csv["name"], "fuga")
            default:
                break
            }
            i += 1
        }
        XCTAssertEqual(i, 2)
    }
    
    func testCSVState1() {
        let it = "あ,い1,\"う\",えお\n,,x,".unicodeScalars.makeIterator()
        var csv = try! CSV(iterator: it, hasHeaderRow: defaultHasHeaderRow, trimFields: defaultTrimFields, delimiter: defaultDelimiter)
        
        var rows = [[String]]()
        
        while let row = csv.next() {
            rows.append(row)
        }
        XCTAssertEqual(rows.count, 2)
        XCTAssertEqual(rows[0], ["あ", "い1", "う", "えお"])
        XCTAssertEqual(rows[1], ["", "", "x", ""])
    }
    
    func testTrimFields1() {
        let csvString = "abc,def,ghi"
        let csv = try! CSV(string: csvString, trimFields: true)
        for row in csv {
            XCTAssertEqual(row, ["abc", "def", "ghi"])
        }
    }
    
    func testTrimFields2() {
        let csvString = " abc,  def,   ghi"
        let csv = try! CSV(string: csvString, trimFields: true)
        for row in csv {
            XCTAssertEqual(row, ["abc", "def", "ghi"])
        }
    }
    
    func testTrimFields3() {
        let csvString = "abc ,def  ,ghi   "
        let csv = try! CSV(string: csvString, trimFields: true)
        for row in csv {
            XCTAssertEqual(row, ["abc", "def", "ghi"])
        }
    }
    
    func testTrimFields4() {
        let csvString = " abc ,  def  ,   ghi   "
        let csv = try! CSV(string: csvString, trimFields: true)
        for row in csv {
            XCTAssertEqual(row, ["abc", "def", "ghi"])
        }
    }
    
    func testTrimFields5() {
        let csvString = "\"abc\",\"def\",\"ghi\""
        let csv = try! CSV(string: csvString, trimFields: true)
        for row in csv {
            XCTAssertEqual(row, ["abc", "def", "ghi"])
        }
    }

    func testTrimFields6() {
        let csvString = " \"abc\",  \"def\",   \"ghi\""
        let csv = try! CSV(string: csvString, trimFields: true)
        for row in csv {
            XCTAssertEqual(row, ["abc", "def", "ghi"])
        }
    }
    
    func testTrimFields7() {
        let csvString = "\"abc\" ,\"def\"  ,\"ghi\"   "
        let csv = try! CSV(string: csvString, trimFields: true)
        for row in csv {
            XCTAssertEqual(row, ["abc", "def", "ghi"])
        }
    }
    
    func testTrimFields8() {
        let csvString = " \"abc\" ,  \"def\"  ,   \"ghi\"   "
        let csv = try! CSV(string: csvString, trimFields: true)
        for row in csv {
            XCTAssertEqual(row, ["abc", "def", "ghi"])
        }
    }
    
    func testTrimFields9() {
        let csvString = "\" abc \",\" def \",\" ghi \""
        let csv = try! CSV(string: csvString, trimFields: true)
        for row in csv {
            XCTAssertEqual(row, [" abc ", " def ", " ghi "])
        }
    }

    func testTrimFields10() {
        let csvString = "\tabc,\t\tdef\t,ghi\t"
        let csv = try! CSV(string: csvString, trimFields: true)
        for row in csv {
            XCTAssertEqual(row, ["abc", "def", "ghi"])
        }
    }
    
    func testTrimFields11() {
        let csvString = " abc \n def "
        var csv = try! CSV(string: csvString, trimFields: true)
        
        let row1 = csv.next()!
        XCTAssertEqual(row1, ["abc"])
        let row2 = csv.next()!
        XCTAssertEqual(row2, ["def"])
    }
    
    func testTrimFields12() {
        let csvString = " \"abc \" \n \" def\" "
        var csv = try! CSV(string: csvString, trimFields: true)
        
        let row1 = csv.next()!
        XCTAssertEqual(row1, ["abc "])
        let row2 = csv.next()!
        XCTAssertEqual(row2, [" def"])
    }
    
    func testTrimFields13() {
        let csvString = " abc \t\tdef\t ghi "
        let csv = try! CSV(string: csvString, trimFields: true, delimiter: UnicodeScalar("\t")!)
        for row in csv {
            XCTAssertEqual(row, ["abc", "", "def", "ghi"])
        }
    }
    
}