Add on start method

01-intro/anysystem-intro/ping-pong/impl_basic.py

@@ -5,6 +5,9 @@ class PingClient(Process):
     def __init__(self, proc_id: str, server_id: str):
         self._id = proc_id
         self._server_id = server_id
+
+    def on_start(self, ctx: Context):
+        pass
 
     def on_local_message(self, msg: Message, ctx: Context):
         # process messages from the local user
@@ -25,6 +28,9 @@ class PingServer(Process):
     def __init__(self, proc_id: str):
         self._id = proc_id
 
+    def on_start(self, ctx: Context):
+        pass
+
     def on_local_message(self, msg: Message, ctx: Context):
         # process messages from the local user (not used in this example)
         pass

01-intro/anysystem-intro/ping-pong/impl_retry.py

@@ -7,6 +7,9 @@ def __init__(self, proc_id: str, server_id: str):
         self._server_id = server_id
         self._ping = None
 
+    def on_start(self, ctx: Context):
+        pass
+
     def on_local_message(self, msg: Message, ctx: Context):
         # process messages from the local user
         if msg.type == 'PING':
@@ -32,6 +35,9 @@ class PingServer(Process):
     def __init__(self, proc_id: str):
         self._id = proc_id
 
+    def on_start(self, ctx: Context):
+        pass
+
     def on_local_message(self, msg: Message, ctx: Context):
         # process messages from the local user (not used in this example)
         pass

01-intro/anysystem-intro/ping-pong/tests/src/common.rs

@@ -21,6 +21,7 @@ pub fn build_system(config: &TestConfig) -> System {
     let client = client_f.build(("client", "server"), config.seed);
     sys.add_process("server", boxed!(server), "server-node");
     sys.add_process("client", boxed!(client), "client-node");
+
     sys
 }
 

anysystem/python/anysystem.py

@@ -102,6 +102,12 @@ def time(self) -> float:
 
 
 class Process:
+    @abc.abstractmethod
+    def on_start(self, ctx: Context):
+        """
+        This method is called after the process instance is created.
+        """
+
     @abc.abstractmethod
     def on_local_message(self, msg: Message, ctx: Context):
         """

homework/01-guarantees/solution/guarantees.py

@@ -8,6 +8,9 @@ def __init__(self, proc_id: str, receiver_id: str):
         self._id = proc_id
         self._receiver = receiver_id
 
+    def on_start(self, ctx: Context):
+        pass
+
     def on_local_message(self, msg: Message, ctx: Context):
         # receive message for delivery from local user
         pass
@@ -25,6 +28,9 @@ class AtMostOnceReceiver(Process):
     def __init__(self, proc_id: str):
         self._id = proc_id
 
+    def on_start(self, ctx: Context):
+        pass
+
     def on_local_message(self, msg: Message, ctx: Context):
         # not used in this task
         pass
@@ -46,6 +52,9 @@ def __init__(self, proc_id: str, receiver_id: str):
         self._id = proc_id
         self._receiver = receiver_id
 
+    def on_start(self, ctx: Context):
+        pass
+
     def on_local_message(self, msg: Message, ctx: Context):
         # receive message for delivery from local user
         pass
@@ -63,6 +72,9 @@ class AtLeastOnceReceiver(Process):
     def __init__(self, proc_id: str):
         self._id = proc_id
 
+    def on_start(self, ctx: Context):
+        pass
+
     def on_local_message(self, msg: Message, ctx: Context):
         # not used in this task
         pass
@@ -84,6 +96,9 @@ def __init__(self, proc_id: str, receiver_id: str):
         self._id = proc_id
         self._receiver = receiver_id
 
+    def on_start(self, ctx: Context):
+        pass
+
     def on_local_message(self, msg: Message, ctx: Context):
         # receive message for delivery from local user
         pass
@@ -101,6 +116,9 @@ class ExactlyOnceReceiver(Process):
     def __init__(self, proc_id: str):
         self._id = proc_id
 
+    def on_start(self, ctx: Context):
+        pass
+
     def on_local_message(self, msg: Message, ctx: Context):
         # not used in this task
         pass
@@ -122,6 +140,9 @@ def __init__(self, proc_id: str, receiver_id: str):
         self._id = proc_id
         self._receiver = receiver_id
 
+    def on_start(self, ctx: Context):
+        pass
+
     def on_local_message(self, msg: Message, ctx: Context):
         # receive message for delivery from local user
         pass
@@ -139,6 +160,9 @@ class ExactlyOnceOrderedReceiver(Process):
     def __init__(self, proc_id: str):
         self._id = proc_id
 
+    def on_start(self, ctx: Context):
+        pass
+
     def on_local_message(self, msg: Message, ctx: Context):
         # not used in this task
         pass

homework/01-guarantees/tests/Cargo.toml

@@ -8,6 +8,7 @@ anysystem = "0.1.2"
 assertables = "3.2.2"
 clap = { version = "3.1.17", features = ["cargo", "derive"] }
 env_logger = "0.9.0"
+indexmap = "2.12.1"
 log = "0.4.14"
 rand = "0.8.5"
 rand_pcg = "0.3.1"

homework/01-guarantees/tests/src/main.rs

@@ -2,12 +2,13 @@ mod common;
 mod tests;
 mod tests_mc;
 
-use std::collections::{BTreeMap, HashSet};
+use std::collections::HashSet;
 use std::env;
 use std::io::Write;
 
 use clap::Parser;
 use env_logger::Builder;
+use indexmap::IndexMap;
 use log::LevelFilter;
 
 use anysystem::test::{TestResult, TestSuite};
@@ -272,7 +273,7 @@ fn main() {
     }
 }
 
-fn score(results: BTreeMap<String, TestResult>) -> f32 {
+fn score(results: IndexMap<String, TestResult>) -> f32 {
     let guarantees = HashSet::from(["AT MOST ONCE", "AT LEAST ONCE", "EXACTLY ONCE", "EXACTLY ONCE ORDERED"]);
     let mut failed_guarantees: HashSet<&str> = HashSet::new();
     let mut failed_overheads: HashSet<&str> = HashSet::new();

homework/04-broadcast/solution/broadcast.py

@@ -7,6 +7,9 @@ def __init__(self, proc_id: str, processes: List[str]):
         self._id = proc_id
         self._processes = processes
 
+    def on_start(self, ctx: Context):
+        pass
+
     def on_local_message(self, msg: Message, ctx: Context):
         if msg.type == 'SEND':
             bcast_msg = Message('BCAST', {

homework/04-broadcast/tests/Cargo.toml

@@ -7,6 +7,7 @@ edition = "2021"
 anysystem = "0.1.2"
 clap = { version = "3.2.17", features = ["cargo", "derive"] }
 env_logger = "0.9.0"
+indexmap = "2.12.1"
 log = "0.4.17"
 rand = "0.8.5"
 rand_pcg = "0.3.1"

homework/04-broadcast/tests/src/main.rs

@@ -2,12 +2,13 @@ mod common;
 mod tests;
 mod tests_mc;
 
-use std::collections::{BTreeMap, HashSet};
+use std::collections::HashSet;
 use std::env;
 use std::io::Write;
 
 use clap::Parser;
 use env_logger::Builder;
+use indexmap::IndexMap;
 use log::LevelFilter;
 
 use anysystem::python::PyProcessFactory;
@@ -99,7 +100,7 @@ fn main() {
     }
 }
 
-fn score(results: BTreeMap<String, TestResult>) -> f32 {
+fn score(results: IndexMap<String, TestResult>) -> f32 {
     let mut violated = HashSet::new();
     for (_, result) in results {
         if let Err(e) = result {

homework/06-membership/solution/member.py

@@ -5,6 +5,9 @@ class GroupMember(Process):
     def __init__(self, proc_id: str):
         self._id = proc_id
 
+    def on_start(self, ctx: Context):
+        pass
+
     def on_local_message(self, msg: Message, ctx: Context):
         if msg.type == 'JOIN':
             # Add process to the group

homework/06-membership/tests/Cargo.toml

@@ -8,6 +8,7 @@ anysystem = "0.1.2"
 assertables = "3.2.2"
 clap = { version = "3.2.17", features = ["cargo", "derive"] }
 env_logger = "0.9.0"
+indexmap = "2.12.1"
 log = "0.4.17"
 rand = "0.8.5"
 rand_pcg = "0.3.1"

homework/06-membership/tests/src/main.rs

@@ -2,12 +2,13 @@ mod common;
 mod tests;
 mod tests_mc;
 
-use std::collections::{BTreeMap, HashSet};
+use std::collections::HashSet;
 use std::env;
 use std::io::Write;
 
 use clap::Parser;
 use env_logger::Builder;
+use indexmap::IndexMap;
 use log::LevelFilter;
 use rand::prelude::*;
 use rand_pcg::Pcg64;
@@ -152,7 +153,7 @@ macro_rules! string_set {
     );
 }
 
-fn score(results: BTreeMap<String, TestResult>, monkeys: u32, disable_mc_tests: bool) -> f32 {
+fn score(results: IndexMap<String, TestResult>, monkeys: u32, disable_mc_tests: bool) -> f32 {
     let basic_group = string_set! {
         "SIMPLE",
         "GET MEMBERS SEMANTICS",

homework/07-kv-sharding/solution/node.py

@@ -8,6 +8,9 @@ def __init__(self, node_id: str, nodes: List[str]):
         self._nodes = set(nodes)
         self._data = {}
 
+    def on_start(self, ctx: Context):
+        pass
+
     def on_local_message(self, msg: Message, ctx: Context):
         # Get value for the key.
         # Request:

homework/07-kv-sharding/tests/Cargo.toml

@@ -9,6 +9,7 @@ assertables = "3.2.2"
 clap = { version = "3.2.17", features = ["cargo", "derive"] }
 decorum = "0.3.1"
 env_logger = "0.9.0"
+indexmap = "2.12.1"
 log = "0.4.17"
 rand = "0.8.5"
 rand_pcg = "0.3.1"

homework/07-kv-sharding/tests/src/main.rs

@@ -2,12 +2,13 @@ mod common;
 mod tests;
 mod tests_mc;
 
-use std::collections::{BTreeMap, HashSet};
+use std::collections::HashSet;
 use std::env;
 use std::io::Write;
 
 use clap::Parser;
 use env_logger::Builder;
+use indexmap::IndexMap;
 use log::LevelFilter;
 
 use anysystem::python::PyProcessFactory;
@@ -112,7 +113,7 @@ macro_rules! string_set {
     );
 }
 
-fn score(results: BTreeMap<String, TestResult>) -> f32 {
+fn score(results: IndexMap<String, TestResult>) -> f32 {
     let basic_group = string_set! {
         "SINGLE NODE", "INSERTS", "DELETES", "MEMORY OVERHEAD", "MC NORMAL"
     };