mirror of
https://github.com/MindWorkAI/AI-Studio.git
synced 2026-03-29 17:31:37 +00:00
Added another unit tests
This commit is contained in:
parent
a1aa16ba61
commit
422b31b3f4
@ -704,6 +704,37 @@ mod tests {
|
|||||||
assert_eq!(source.encryption_secret, "secret");
|
assert_eq!(source.encryption_secret, "secret");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn parse_enterprise_source_values_supports_gaps_between_indexed_slots() {
|
||||||
|
let mut values = HashMap::new();
|
||||||
|
values.insert(String::from("config_id0"), String::from(TEST_ID_A));
|
||||||
|
values.insert(
|
||||||
|
String::from("config_server_url0"),
|
||||||
|
String::from("https://slot0.example.org"),
|
||||||
|
);
|
||||||
|
values.insert(String::from("config_id4"), String::from(TEST_ID_B));
|
||||||
|
values.insert(
|
||||||
|
String::from("config_server_url4"),
|
||||||
|
String::from("https://slot4.example.org"),
|
||||||
|
);
|
||||||
|
|
||||||
|
let source = parse_enterprise_source_values("test", &values);
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
source.configs,
|
||||||
|
vec![
|
||||||
|
EnterpriseConfig {
|
||||||
|
id: String::from("9072b77d-ca81-40da-be6a-861da525ef7b"),
|
||||||
|
server_url: String::from("https://slot0.example.org"),
|
||||||
|
},
|
||||||
|
EnterpriseConfig {
|
||||||
|
id: String::from(TEST_ID_B),
|
||||||
|
server_url: String::from("https://slot4.example.org"),
|
||||||
|
},
|
||||||
|
]
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn select_effective_enterprise_source_uses_first_source_with_configs_only() {
|
fn select_effective_enterprise_source_uses_first_source_with_configs_only() {
|
||||||
let selected = select_effective_enterprise_source(vec![
|
let selected = select_effective_enterprise_source(vec![
|
||||||
@ -805,6 +836,39 @@ mod tests {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn load_policy_values_from_directories_supports_gaps_between_policy_slots() {
|
||||||
|
let directory = tempdir().unwrap();
|
||||||
|
|
||||||
|
fs::write(
|
||||||
|
directory.path().join("config0.yaml"),
|
||||||
|
"id: \"9072b77d-ca81-40da-be6a-861da525ef7b\"\nserver_url: \"https://slot0.example.org\"",
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
fs::write(
|
||||||
|
directory.path().join("config4.yaml"),
|
||||||
|
"id: \"a1b2c3d4-e5f6-7890-abcd-ef1234567890\"\nserver_url: \"https://slot4.example.org\"",
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let values = load_policy_values_from_directories(&[directory.path().to_path_buf()]);
|
||||||
|
let source = parse_enterprise_source_values("policy files", &values);
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
source.configs,
|
||||||
|
vec![
|
||||||
|
EnterpriseConfig {
|
||||||
|
id: String::from("9072b77d-ca81-40da-be6a-861da525ef7b"),
|
||||||
|
server_url: String::from("https://slot0.example.org"),
|
||||||
|
},
|
||||||
|
EnterpriseConfig {
|
||||||
|
id: String::from(TEST_ID_B),
|
||||||
|
server_url: String::from("https://slot4.example.org"),
|
||||||
|
},
|
||||||
|
]
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn load_policy_values_from_directories_ignores_invalid_and_incomplete_files() {
|
fn load_policy_values_from_directories_ignores_invalid_and_incomplete_files() {
|
||||||
let directory = tempdir().unwrap();
|
let directory = tempdir().unwrap();
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user