more-config 3.0.0

use config::{prelude::*, Error, FileSource, Reloadable, ReloadableConfiguration};
use serde_json::json;
use std::convert::TryInto;
use std::env::temp_dir;
use std::fs::{remove_file, File};
use std::io::Write;
use std::path::PathBuf;
use std::sync::{Arc, Condvar, Mutex};
use std::time::Duration;
use tokens::ChangeToken;

#[test]
fn add_json_file_should_load_settings_from_file() {
    // arrange
    let json = json!({"service": {
       "enabled": false},
     "feature": {
         "nativeCopy": {
             "disabled": true}}
    });
    let path = temp_dir().join("test_settings_1.json");
    let mut file = File::create(&path).unwrap();

    file.write_all(json.to_string().as_bytes()).unwrap();

    let config = config::builder().add_json_file(&path).build().unwrap();
    let section = config.section("Feature").section("NativeCopy");

    // act
    let actual = section.get("Disabled");

    // assert
    if path.exists() {
        remove_file(&path).ok();
    }

    assert_eq!(actual, Some("true"));
}

#[test]
fn add_json_file_should_fail_if_file_does_not_exist() {
    // arrange
    let path = PathBuf::from(r"C:\fake\settings.json");

    // act
    let result = config::builder().add_json_file(&path).build();

    // assert
    if let Err(error) = result {
        if matches!(error, Error::MissingFile(_)) {
            assert_eq!(
                &error.to_string(),
                r"The configuration file 'C:\fake\settings.json' was not found, but is required."
            )
        } else {
            panic!("{:?}", error)
        }
    } else {
        panic!("No error occurred.")
    }
}

#[test]
fn add_optional_json_file_should_load_settings_from_file() {
    let json = json!({"service": {
       "enabled": false},
     "feature": {
       "nativeCopy": {
           "disabled": true}}
    });
    let path = temp_dir().join("test_settings_2.json");
    let mut file = File::create(&path).unwrap();

    file.write_all(json.to_string().as_bytes()).unwrap();

    let config = config::builder()
        .add_json_file(FileSource::optional(&path))
        .build()
        .unwrap();
    let section = config.section("Feature").section("NativeCopy");

    // act
    let actual = section.get("Disabled");

    // assert
    if path.exists() {
        remove_file(&path).ok();
    }

    assert_eq!(actual, Some("true"));
}

#[test]
fn add_json_file_should_succeed_if_optional_file_does_not_exist() {
    // arrange
    let path = PathBuf::from(r"C:\fake\settings.json");

    // act
    let config = config::builder()
        .add_json_file(FileSource::optional(&path))
        .build()
        .unwrap();

    // assert
    assert_eq!(config.sections().len(), 0);
}

#[test]
fn simple_json_array_should_be_converted_to_key_value_pairs() {
    // arrange
    let json = json!({"ip": ["1.2.3.4", "7.8.9.10", "11.12.13.14"]});
    let path = temp_dir().join("array_settings_1.json");
    let mut file = File::create(&path).unwrap();

    file.write_all(json.to_string().as_bytes()).unwrap();

    // act
    let config = config::builder().add_json_file(&path).build().unwrap();

    // assert
    if path.exists() {
        remove_file(&path).ok();
    }

    assert_eq!(config.get("ip:0"), Some("1.2.3.4"));
    assert_eq!(config.get("ip:1"), Some("7.8.9.10"));
    assert_eq!(config.get("ip:2"), Some("11.12.13.14"));
}

#[test]
fn complex_json_array_should_be_converted_to_key_value_pairs() {
    // arrange
    let json = json!({"ip": [
        {"address": "1.2.3.4", "hidden": false},
        {"address": "5.6.7.8", "hidden": true}
    ]});
    let path = temp_dir().join("array_settings_2.json");
    let mut file = File::create(&path).unwrap();

    file.write_all(json.to_string().as_bytes()).unwrap();

    // act
    let config = config::builder().add_json_file(&path).build().unwrap();

    // assert
    if path.exists() {
        remove_file(&path).ok();
    }

    assert_eq!(config.get("ip:0:address"), Some("1.2.3.4"));
    assert_eq!(config.get("ip:0:hidden"), Some("false"));
    assert_eq!(config.get("ip:1:address"), Some("5.6.7.8"));
    assert_eq!(config.get("ip:1:hidden"), Some("true"));
}

#[test]
fn nested_json_array_should_be_converted_to_key_value_pairs() {
    // arrange
    let json = json!({"ip": [
        ["1.2.3.4", "5.6.7.8"],
        ["9.10.11.12", "13.14.15.16"]
    ]});
    let path = temp_dir().join("array_settings_3.json");
    let mut file = File::create(&path).unwrap();

    file.write_all(json.to_string().as_bytes()).unwrap();

    // act
    let config = config::builder().add_json_file(&path).build().unwrap();

    // assert
    if path.exists() {
        remove_file(&path).ok();
    }

    assert_eq!(config.get("ip:0:0"), Some("1.2.3.4"));
    assert_eq!(config.get("ip:0:1"), Some("5.6.7.8"));
    assert_eq!(config.get("ip:1:0"), Some("9.10.11.12"));
    assert_eq!(config.get("ip:1:1"), Some("13.14.15.16"));
}

#[test]
fn json_array_item_should_be_implicitly_replaced() {
    // arrange
    let json1 = json!({"ip": ["1.2.3.4", "7.8.9.10", "11.12.13.14"]});
    let json2 = json!({"ip": ["15.16.17.18"]});
    let path1 = temp_dir().join("array_settings_4.json");
    let path2 = temp_dir().join("array_settings_5.json");
    let mut file = File::create(&path1).unwrap();

    file.write_all(json1.to_string().as_bytes()).unwrap();
    file = File::create(&path2).unwrap();
    file.write_all(json2.to_string().as_bytes()).unwrap();

    // act
    let config = config::builder()
        .add_json_file(&path1)
        .add_json_file(&path2)
        .build()
        .unwrap();

    // assert
    if path1.exists() {
        remove_file(&path1).ok();
    }

    if path2.exists() {
        remove_file(&path2).ok();
    }

    assert_eq!(config.section("ip").sections().len(), 3);
    assert_eq!(config.get("ip:0"), Some("15.16.17.18"));
    assert_eq!(config.get("ip:1"), Some("7.8.9.10"));
    assert_eq!(config.get("ip:2"), Some("11.12.13.14"));
}

#[test]
fn json_array_item_should_be_explicitly_replaced() {
    // arrange
    let json1 = json!({"ip": ["1.2.3.4", "7.8.9.10", "11.12.13.14"]});
    let json2 = json!({"ip": {"1": "15.16.17.18"}});
    let path1 = temp_dir().join("array_settings_6.json");
    let path2 = temp_dir().join("array_settings_7.json");
    let mut file = File::create(&path1).unwrap();

    file.write_all(json1.to_string().as_bytes()).unwrap();
    file = File::create(&path2).unwrap();
    file.write_all(json2.to_string().as_bytes()).unwrap();

    // act
    let config = config::builder()
        .add_json_file(&path1)
        .add_json_file(&path2)
        .build()
        .unwrap();

    // assert
    if path1.exists() {
        remove_file(&path1).ok();
    }

    if path2.exists() {
        remove_file(&path2).ok();
    }

    assert_eq!(config.section("ip").sections().len(), 3);
    assert_eq!(config.get("ip:0"), Some("1.2.3.4"));
    assert_eq!(config.get("ip:1"), Some("15.16.17.18"));
    assert_eq!(config.get("ip:2"), Some("11.12.13.14"));
}

#[test]
fn json_arrays_should_be_merged() {
    // arrange
    let json1 = json!({"ip": ["1.2.3.4", "7.8.9.10", "11.12.13.14"]});
    let json2 = json!({"ip": {"3": "15.16.17.18"}});
    let path1 = temp_dir().join("array_settings_8.json");
    let path2 = temp_dir().join("array_settings_9.json");
    let mut file = File::create(&path1).unwrap();

    file.write_all(json1.to_string().as_bytes()).unwrap();
    file = File::create(&path2).unwrap();
    file.write_all(json2.to_string().as_bytes()).unwrap();

    // act
    let config = config::builder()
        .add_json_file(&path1)
        .add_json_file(&path2)
        .build()
        .unwrap();

    // assert
    if path1.exists() {
        remove_file(&path1).ok();
    }

    if path2.exists() {
        remove_file(&path2).ok();
    }

    assert_eq!(config.section("ip").sections().len(), 4);
    assert_eq!(config.get("ip:0"), Some("1.2.3.4"));
    assert_eq!(config.get("ip:1"), Some("7.8.9.10"));
    assert_eq!(config.get("ip:2"), Some("11.12.13.14"));
    assert_eq!(config.get("ip:3"), Some("15.16.17.18"));
}

#[test]
fn json_file_should_reload_when_changed() {
    // arrange
    let path = temp_dir().join("reload_settings_1.json");
    let mut json = json!(
    {
        "service": {
            "enabled": false
        },
        "feature": {
            "nativeCopy": {
                "disabled": true
            }
        }
    });

    let mut file = File::create(&path).unwrap();
    file.write_all(json.to_string().as_bytes()).unwrap();
    drop(file);

    let builder = config::builder().add_json_file(&path.is().reloadable());
    let config: ReloadableConfiguration = builder.try_into().unwrap();
    let initial = config
        .current()
        .section("Feature")
        .section("NativeCopy")
        .get("Disabled")
        .unwrap_or_default()
        .to_owned();
    let token = config.reload_token();
    let state = Arc::new((Mutex::new(false), Condvar::new()));
    let _registration = token.register(
        Box::new(|s| {
            let data = s.unwrap();
            let (reloaded, event) = &*(data.downcast_ref::<(Mutex<bool>, Condvar)>().unwrap());
            *reloaded.lock().unwrap() = true;
            event.notify_one();
        }),
        Some(state.clone()),
    );

    json = json!(
    {
        "service": {
            "enabled": false
        },
        "feature": {
            "nativeCopy": {
                "disabled": false
            }
        }
    });

    file = File::create(&path).unwrap();
    file.write_all(json.to_string().as_bytes()).unwrap();
    drop(file);

    let (mutex, event) = &*state;
    let mut reloaded = mutex.lock().unwrap();

    while !*reloaded {
        reloaded = event.wait_timeout(reloaded, Duration::from_secs(1)).unwrap().0;
    }

    // act
    let current = config
        .current()
        .section("Feature")
        .section("NativeCopy")
        .get("Disabled")
        .unwrap_or_default()
        .to_owned();

    // assert
    assert_eq!(initial, "true");
    assert_eq!(current, "false");
}