Skip to content

Commit

Permalink
more and better tests
Browse files Browse the repository at this point in the history
  • Loading branch information
pront committed Aug 25, 2023
1 parent c124ff9 commit 627f561
Show file tree
Hide file tree
Showing 2 changed files with 48 additions and 4 deletions.
47 changes: 43 additions & 4 deletions src/convert_config.rs
Original file line number Diff line number Diff line change
Expand Up @@ -190,7 +190,7 @@ fn walk_dir_and_convert(
#[cfg(test)]
mod tests {
use crate::config::{format, ConfigBuilder, Format};
use crate::convert_config::walk_dir_and_convert;
use crate::convert_config::{check_paths, walk_dir_and_convert, Opts};
use std::path::{Path, PathBuf};
use std::str::FromStr;
use std::{env, fs};
Expand All @@ -200,16 +200,54 @@ mod tests {
PathBuf::from(env::var_os("CARGO_MANIFEST_DIR").unwrap()).join("tests/data/cmd/config")
}

// Read the contents of the specified `path` and deserialize them into a `ConfigBuilder`.
// Finally serialize them a string again. Configs do not implement equality,
// so for these tests we will rely on strings for comparisons.
fn convert_file_to_config_string(path: &Path) -> String {
let files_contents = fs::read_to_string(path).unwrap();
let extension = path.extension().unwrap().to_str().unwrap();
let file_format = Format::from_str(extension).unwrap();
let builder: ConfigBuilder = format::deserialize(&files_contents, file_format).unwrap();
let config = builder.build().unwrap();
// Configs do not implement equality, so will rely on strings for comparisons

format::serialize(&config, file_format).unwrap()
}

#[test]
fn invalid_path_opts() {
let check_error = |opts, pattern| {
let error = check_paths(&opts).unwrap_err();
assert!(error.contains(pattern));
};

check_error(
Opts {
input_path: [r"./"].iter().collect(),
output_path: [r"./"].iter().collect(),
output_format: Format::Yaml,
},
"already exists",
);

check_error(
Opts {
input_path: [r"./"].iter().collect(),
output_path: [r"./out.yaml"].iter().collect(),
output_format: Format::Yaml,
},
"points to a file.",
);

check_error(
Opts {
input_path: [r"./in.yaml"].iter().collect(),
output_path: [r"./"].iter().collect(),
output_format: Format::Yaml,
},
"points to a directory.",
);
}

#[test]
fn convert_all_from_dir() {
let input_path = test_data_dir();
Expand All @@ -218,21 +256,22 @@ mod tests {
.into_path();
walk_dir_and_convert(&input_path, &output_dir, Format::Yaml).unwrap();

let entries = fs::read_dir(&output_dir).unwrap();
let mut count: usize = 0;
let original_config = convert_file_to_config_string(&test_data_dir().join("config_1.yaml"));
for entry in entries {
for entry in fs::read_dir(&output_dir).unwrap() {
let entry = entry.unwrap();
let path = entry.path();
if path.is_file() {
let extension = path.extension().unwrap().to_str().unwrap();
if extension == Format::Yaml.to_string() {
// Note that here we read the converted string directly.
let converted_config = fs::read_to_string(&output_dir.join(&path)).unwrap();
assert_eq!(converted_config, original_config);
count += 1;
}
}
}
// There two non-yaml configs in the input directory.
assert_eq!(count, 2);
}
}
5 changes: 5 additions & 0 deletions tests/data/cmd/config/config_2.toml
Original file line number Diff line number Diff line change
@@ -1,3 +1,8 @@
# Commets are dropped.
#[api]
#enabled = true
#address = "127.0.0.1:8686"

data_dir = "/var/lib/vector/"

[sources.source0]
Expand Down

0 comments on commit 627f561

Please sign in to comment.