Skip to content

Commit

Permalink
selector extension bug fix
Browse files Browse the repository at this point in the history
  • Loading branch information
kevin-valerio committed Sep 24, 2024
1 parent 2557f9d commit de1242a
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 6 deletions.
2 changes: 1 addition & 1 deletion src/contract/selectors/database.rs
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ impl SelectorDatabase {
self.messages.extend(messages);
}

pub fn messages_with_invariants(self) -> anyhow::Result<Vec<Selector>> {
pub fn get_unique_messages(self) -> anyhow::Result<Vec<Selector>> {
if !self.messages.is_empty() && !self.invariants.is_empty() {
return Ok(self
.messages
Expand Down
2 changes: 1 addition & 1 deletion src/fuzzer/environment.rs
Original file line number Diff line number Diff line change
Expand Up @@ -103,7 +103,7 @@ impl EnvironmentBuilder {

for (i, selector) in self
.database
.messages()
.get_unique_messages()
.with_context(|| "Couldn't load messages")?
.iter()
.enumerate()
Expand Down
8 changes: 4 additions & 4 deletions src/fuzzer/fuzz.rs
Original file line number Diff line number Diff line change
Expand Up @@ -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())?;
Expand Down

0 comments on commit de1242a

Please sign in to comment.