bendns' repos / rust-fsm
Finite state machines in rust; bendns fork to add types.
Diffstat (limited to 'rust_fsm/examples/circuit_breaker.rs')
-rw-r--r--rust_fsm/examples/circuit_breaker.rs10
1 files changed, 5 insertions, 5 deletions
diff --git a/rust_fsm/examples/circuit_breaker.rs b/rust_fsm/examples/circuit_breaker.rs
index b2814d3..e318b30 100644
--- a/rust_fsm/examples/circuit_breaker.rs
+++ b/rust_fsm/examples/circuit_breaker.rs
@@ -69,7 +69,7 @@ fn main() {
let machine = Arc::new(Mutex::new(machine));
{
let mut lock = machine.lock().unwrap();
- let res = lock.consume_anyway(&CircuitBreakerInput::Unsuccessful);
+ let res = lock.consume(&CircuitBreakerInput::Unsuccessful).unwrap();
assert_eq!(res, Some(CircuitBreakerOutputSetTimer));
assert_eq!(lock.state(), &CircuitBreakerState::Open);
}
@@ -79,7 +79,7 @@ fn main() {
std::thread::spawn(move || {
std::thread::sleep(Duration::new(5, 0));
let mut lock = machine_wait.lock().unwrap();
- let res = lock.consume_anyway(&CircuitBreakerInput::TimerTriggered);
+ let res = lock.consume(&CircuitBreakerInput::TimerTriggered).unwrap();
assert_eq!(res, None);
assert_eq!(lock.state(), &CircuitBreakerState::HalfOpen);
});
@@ -89,8 +89,8 @@ fn main() {
std::thread::spawn(move || {
std::thread::sleep(Duration::new(1, 0));
let mut lock = machine_try.lock().unwrap();
- let res = lock.consume_anyway(&CircuitBreakerInput::Successful);
- assert_eq!(res, None);
+ let res = lock.consume(&CircuitBreakerInput::Successful);
+ assert_eq!(res, Err(()));
assert_eq!(lock.state(), &CircuitBreakerState::Open);
});
@@ -98,7 +98,7 @@ fn main() {
std::thread::sleep(Duration::new(7, 0));
{
let mut lock = machine.lock().unwrap();
- let res = lock.consume_anyway(&CircuitBreakerInput::Successful);
+ let res = lock.consume(&CircuitBreakerInput::Successful).unwrap();
assert_eq!(res, None);
assert_eq!(lock.state(), &CircuitBreakerState::Closed);
}