CSV.swift/Tests/CSVTests/CSVWriterTests.swift

//
//  CSVWriterTests.swift
//  CSV
//
//  Created by Yasuhiro Hatta on 2017/05/28.
//  Copyright © 2017年 yaslab. All rights reserved.
//

import Foundation
import XCTest

import CSV

extension OutputStream {
    
    var data: Data? {
        guard let nsData = property(forKey: .dataWrittenToMemoryStreamKey) as? NSData else {
            return nil
        }
        return Data(referencing: nsData)
    }
    
}

class CSVWriterTests: XCTestCase {
    
    static let allTests = [
        ("testSingleFieldSingleRecord", testSingleFieldSingleRecord),
        ("testSingleFieldMultipleRecord", testSingleFieldMultipleRecord),
        ("testMultipleFieldSingleRecord", testMultipleFieldSingleRecord),
        ("testMultipleFieldMultipleRecord", testMultipleFieldMultipleRecord),
        ("testQuoted", testQuoted),
        ("testQuotedNewline", testQuotedNewline),
        ("testEscapeQuote", testEscapeQuote),
        ("testDelimiter", testDelimiter),
        ("testNewline", testNewline),
        ("testUTF16BE", testUTF16BE),
        ("testUTF16LE", testUTF16LE),
        ("testUTF32BE", testUTF32BE),
        ("testUTF32LE", testUTF32LE)
    ]
    
    let str = "TEST-test-1234-😄😆👨‍👩‍👧‍👦"
    
    /// xxxx
    func testSingleFieldSingleRecord() {
        let stream = OutputStream(toMemory: ())
        stream.open()
        
        let csv = try! CSVWriter(stream: stream)
        csv.beginNewRecord()
        try! csv.write(field: str)
        
        stream.close()
        let data = stream.data!
        let csvStr = String(data: data, encoding: .utf8)!

        XCTAssertEqual(csvStr, str)
    }

    /// xxxx
    /// xxxx
    func testSingleFieldMultipleRecord() {
        let stream = OutputStream(toMemory: ())
        stream.open()
        
        let csv = try! CSVWriter(stream: stream)
        csv.beginNewRecord()
        try! csv.write(field: str + "-1")
        csv.beginNewRecord()
        try! csv.write(field: str + "-2")
        
        stream.close()
        let data = stream.data!
        let csvStr = String(data: data, encoding: .utf8)!
        
        XCTAssertEqual(csvStr, "\(str)-1\n\(str)-2")
    }
    
    /// xxxx,xxxx
    func testMultipleFieldSingleRecord() {
        let stream = OutputStream(toMemory: ())
        stream.open()
        
        let csv = try! CSVWriter(stream: stream)
        csv.beginNewRecord()
        try! csv.write(field: str + "-1")
        try! csv.write(field: str + "-2")

        stream.close()
        let data = stream.data!
        let csvStr = String(data: data, encoding: .utf8)!
        
        XCTAssertEqual(csvStr, "\(str)-1,\(str)-2")
    }

    /// xxxx,xxxx
    /// xxxx,xxxx
    func testMultipleFieldMultipleRecord() {
        let stream = OutputStream(toMemory: ())
        stream.open()
        
        let csv = try! CSVWriter(stream: stream)
        csv.beginNewRecord()
        try! csv.write(field: str + "-1-1")
        try! csv.write(field: str + "-1-2")
        csv.beginNewRecord()
        try! csv.write(field: str + "-2-1")
        try! csv.write(field: str + "-2-2")
        
        stream.close()
        let data = stream.data!
        let csvStr = String(data: data, encoding: .utf8)!
        
        XCTAssertEqual(csvStr, "\(str)-1-1,\(str)-1-2\n\(str)-2-1,\(str)-2-2")
    }
    
    /// "xxxx",xxxx
    func testQuoted() {
        let stream = OutputStream(toMemory: ())
        stream.open()
        
        let csv = try! CSVWriter(stream: stream)
        csv.beginNewRecord()
        try! csv.write(field: str + "-1", quoted: true)
        try! csv.write(field: str + "-2") // quoted: false
        
        stream.close()
        let data = stream.data!
        let csvStr = String(data: data, encoding: .utf8)!
        
        XCTAssertEqual(csvStr, "\"\(str)-1\",\(str)-2")
    }
    
    /// xxxx,"xx\nxx"
    func testQuotedNewline() {
        let stream = OutputStream(toMemory: ())
        stream.open()
        
        let csv = try! CSVWriter(stream: stream)
        csv.beginNewRecord()
        try! csv.write(field: str + "-1") // quoted: false
        try! csv.write(field: str + "-\n-2", quoted: true)
        
        stream.close()
        let data = stream.data!
        let csvStr = String(data: data, encoding: .utf8)!
        
        XCTAssertEqual(csvStr, "\(str)-1,\"\(str)-\n-2\"")
    }
    
    /// xxxx,"xx""xx"
    func testEscapeQuote() {
        let stream = OutputStream(toMemory: ())
        stream.open()
        
        let csv = try! CSVWriter(stream: stream)
        csv.beginNewRecord()
        try! csv.write(field: str + "-1") // quoted: false
        try! csv.write(field: str + "-\"-2", quoted: true)
        
        stream.close()
        let data = stream.data!
        let csvStr = String(data: data, encoding: .utf8)!
        
        XCTAssertEqual(csvStr, "\(str)-1,\"\(str)-\"\"-2\"")
    }
    
    /// Test delimiter: "\t"
    func testDelimiter() {
        let stream = OutputStream(toMemory: ())
        stream.open()
        
        let config = CSVWriter.Configuration(delimiter: "\t")
        let csv = try! CSVWriter.init(stream: stream, configuration: config)
        csv.beginNewRecord()
        try! csv.write(field: str + "-1")
        try! csv.write(field: str + "-2")
        
        stream.close()
        let data = stream.data!
        let csvStr = String(data: data, encoding: .utf8)!
        
        XCTAssertEqual(csvStr, "\(str)-1\t\(str)-2")
    }

    /// Test newline: "\r\n"
    func testNewline() {
        let stream = OutputStream(toMemory: ())
        stream.open()
        
        let config = CSVWriter.Configuration(newline: "\r\n")
        let csv = try! CSVWriter.init(stream: stream, configuration: config)
        csv.beginNewRecord()
        try! csv.write(field: str + "-1")
        csv.beginNewRecord()
        try! csv.write(field: str + "-2")
        
        stream.close()
        let data = stream.data!
        let csvStr = String(data: data, encoding: .utf8)!
        
        XCTAssertEqual(csvStr, "\(str)-1\r\n\(str)-2")
    }
    
    /// UTF16 Big Endian
    func testUTF16BE() {
        let stream = OutputStream(toMemory: ())
        stream.open()
        
        let csv = try! CSVWriter(stream: stream, codecType: UTF16.self, endian: .big)
        csv.beginNewRecord()
        try! csv.write(field: str)
        
        stream.close()
        let data = stream.data!
        let csvStr = String(data: data, encoding: .utf16BigEndian)!
        
        XCTAssertEqual(csvStr, str)
    }

    /// UTF16 Little Endian
    func testUTF16LE() {
        let stream = OutputStream(toMemory: ())
        stream.open()
        
        let csv = try! CSVWriter(stream: stream, codecType: UTF16.self, endian: .little)
        csv.beginNewRecord()
        try! csv.write(field: str)
        
        stream.close()
        let data = stream.data!
        let csvStr = String(data: data, encoding: .utf16LittleEndian)!
        
        XCTAssertEqual(csvStr, str)
    }
    
    /// UTF32 Big Endian
    func testUTF32BE() {
        let stream = OutputStream(toMemory: ())
        stream.open()
        
        let csv = try! CSVWriter(stream: stream, codecType: UTF32.self, endian: .big)
        csv.beginNewRecord()
        try! csv.write(field: str)
        
        stream.close()
        let data = stream.data!
        let csvStr = String(data: data, encoding: .utf32BigEndian)!
        
        XCTAssertEqual(csvStr, str)
    }
    
    /// UTF32 Little Endian
    func testUTF32LE() {
        let stream = OutputStream(toMemory: ())
        stream.open()
        
        let csv = try! CSVWriter(stream: stream, codecType: UTF32.self, endian: .little)
        csv.beginNewRecord()
        try! csv.write(field: str)
        
        stream.close()
        let data = stream.data!
        let csvStr = String(data: data, encoding: .utf32LittleEndian)!
        
        XCTAssertEqual(csvStr, str)
    }
    
}