extern crate csv;
extern crate qsv_sniffer;

use std::path::Path;

use qsv_sniffer::metadata::*;
use qsv_sniffer::{DatePreference, SampleSize, Sniffer, Type};

#[test]
fn test_utf8() {
    let data_filepath = Path::new(file!())
        .parent()
        .unwrap()
        .join("data/test-utf8.csv");
    let metadata = Sniffer::new()
        .sample_size(SampleSize::All)
        .sniff_path(data_filepath)
        .unwrap();
    assert_eq!(
        metadata,
        Metadata {
            dialect: Dialect {
                delimiter: b';',
                header: Header {
                    has_header_row: true,
                    num_preamble_rows: 0
                },
                quote: Quote::None,
                flexible: false,
                is_utf8: false
            },
            avg_record_len: 137,
            num_fields: 11,
            fields: vec![
                "DIA.DESEMB".to_string(),
                "COD.SUBITEM.NCM".to_string(),
                "VMLE.DOLAR.BAL.EXP".to_string(),
                "PESO.LIQ.MERC.BAL.EXP".to_string(),
                "COD.IMPDR.EXPDR".to_string(),
                "NOME.IMPDR.EXPDR".to_string(),
                "PAIS.ORIGEM.DESTINO".to_string(),
                "UA.LOCAL.DESBQ.EMBQ".to_string(),
                "NOME.IMPORTADOR.ESTRANGEIRO".to_string(),
                "NUM.DDE".to_string(),
                "NUM.RE".to_string()
            ],
            types: vec![
                Type::Text,
                Type::Unsigned,
                Type::Float,
                Type::Float,
                Type::Text,
                Type::Text,
                Type::Text,
                Type::Text,
                Type::Text,
                Type::Text,
                Type::Text
            ]
        }
    );
}

#[test]
fn test_flexible_again() {
    let data_filepath = Path::new(file!())
        .parent()
        .unwrap()
        .join("data/gotriangle-routes-cary-ch-duke-durham-raleigh-wofline.csv");
    let metadata = Sniffer::new()
        .sample_size(SampleSize::All)
        .sniff_path(data_filepath)
        .unwrap();
    assert_eq!(
        metadata,
        Metadata {
            dialect: Dialect {
                delimiter: b',',
                header: Header {
                    has_header_row: true,
                    num_preamble_rows: 0,
                },
                quote: Quote::None,
                flexible: true,
                is_utf8: true,
            },
            avg_record_len: 112,
            num_fields: 7,
            fields: vec![
                "municipality".to_string(),
                "agency_id".to_string(),
                "route_id".to_string(),
                "route_short_name".to_string(),
                "route_long_name".to_string(),
                "min_headway_minutes".to_string(),
                "route_url".to_string()
            ],
            types: vec![
                Type::Text,
                Type::Unsigned,
                Type::Unsigned,
                Type::Text,
                Type::Text,
                Type::Unsigned,
                Type::Text
            ]
        }
    );
}

#[test]
fn test_date_sniffing_dmy() {
    let data_filepath = Path::new(file!())
        .parent()
        .unwrap()
        .join("data/dmy-test.csv");
    let metadata = Sniffer::new()
        .sample_size(SampleSize::All)
        .date_preference(DatePreference::DmyFormat)
        .sniff_path(data_filepath)
        .unwrap();
    assert_eq!(
        metadata,
        Metadata {
            dialect: Dialect {
                delimiter: b',',
                header: Header {
                    has_header_row: true,
                    num_preamble_rows: 0
                },
                quote: Quote::None,
                flexible: false,
                is_utf8: true
            },
            avg_record_len: 11,
            num_fields: 3,
            fields: vec![
                "starttime".to_string(),
                "letter".to_string(),
                "number".to_string(),
            ],
            types: vec![Type::Date, Type::Text, Type::Unsigned,]
        }
    );
}