diff --git a/src/contract/selectors/database.rs b/src/contract/selectors/database.rs index afb15f1..326ff88 100644 --- a/src/contract/selectors/database.rs +++ b/src/contract/selectors/database.rs @@ -41,7 +41,7 @@ impl SelectorDatabase { self.messages.extend(messages); } - pub fn messages_with_invariants(self) -> anyhow::Result> { + pub fn get_unique_messages(self) -> anyhow::Result> { if !self.messages.is_empty() && !self.invariants.is_empty() { return Ok(self .messages diff --git a/src/fuzzer/environment.rs b/src/fuzzer/environment.rs index 484498b..71481b5 100644 --- a/src/fuzzer/environment.rs +++ b/src/fuzzer/environment.rs @@ -103,7 +103,7 @@ impl EnvironmentBuilder { for (i, selector) in self .database - .messages() + .get_unique_messages() .with_context(|| "Couldn't load messages")? .iter() .enumerate() diff --git a/src/fuzzer/fuzz.rs b/src/fuzzer/fuzz.rs index 6198b3e..b90d58e 100644 --- a/src/fuzzer/fuzz.rs +++ b/src/fuzzer/fuzz.rs @@ -277,20 +277,20 @@ mod tests { env_builder.build_env(config.clone().fuzz_output())?; - let without_inv_counter = manager.database().messages_with_invariants()?.len(); + let get_unique_messages = manager.database().get_unique_messages()?.len(); assert_eq!( fs::read_dir(config.clone().fuzz_output().join("phink").join("corpus")) .expect("Failed to read directory") .count(), - without_inv_counter + get_unique_messages ); - assert_eq!(without_inv_counter, 5 + 1); // msg + constructor + assert_eq!(get_unique_messages, 5 + 1); // msg + constructor let inv_counter = manager.database().invariants()?.len(); assert_eq!(inv_counter, 1); - assert_eq!(manager.database().messages()?.len(), without_inv_counter); + assert_eq!(manager.database().messages()?.len(), get_unique_messages); let dict_path = config.fuzz_output().join("phink").join("selectors.dict"); let dict: String = fs::read_to_string(dict_path.clone())?;