rustlings/solutions/08_enums/enums3.rs

93 lines
2.1 KiB
Rust
Raw Normal View History

2024-07-05 15:11:03 +03:00
struct Point {
x: u64,
y: u64,
}
2024-06-22 00:18:25 +03:00
enum Message {
2024-07-05 15:11:03 +03:00
Resize { width: u64, height: u64 },
2024-06-22 00:18:25 +03:00
Move(Point),
2024-07-05 15:11:03 +03:00
Echo(String),
ChangeColor(u8, u8, u8),
2024-06-22 00:18:25 +03:00
Quit,
}
struct State {
2024-07-05 15:11:03 +03:00
width: u64,
height: u64,
2024-06-22 00:18:25 +03:00
position: Point,
message: String,
2024-07-05 15:11:03 +03:00
color: (u8, u8, u8),
quit: bool,
2024-06-22 00:18:25 +03:00
}
impl State {
2024-07-05 15:11:03 +03:00
fn resize(&mut self, width: u64, height: u64) {
self.width = width;
self.height = height;
2024-06-22 00:18:25 +03:00
}
2024-07-05 15:11:03 +03:00
fn move_position(&mut self, point: Point) {
self.position = point;
2024-06-22 00:18:25 +03:00
}
fn echo(&mut self, s: String) {
self.message = s;
}
fn change_color(&mut self, red: u8, green: u8, blue: u8) {
self.color = (red, green, blue);
2024-07-05 15:11:03 +03:00
}
fn quit(&mut self) {
self.quit = true;
2024-06-22 00:18:25 +03:00
}
fn process(&mut self, message: Message) {
match message {
2024-07-05 15:11:03 +03:00
Message::Resize { width, height } => self.resize(width, height),
2024-06-22 00:18:25 +03:00
Message::Move(point) => self.move_position(point),
2024-10-09 16:27:36 +03:00
Message::Echo(string) => self.echo(string),
Message::ChangeColor(red, green, blue) => self.change_color(red, green, blue),
2024-06-22 00:18:25 +03:00
Message::Quit => self.quit(),
}
}
}
fn main() {
// You can optionally experiment here.
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_match_message_call() {
let mut state = State {
2024-07-05 15:11:03 +03:00
width: 0,
height: 0,
2024-06-22 00:18:25 +03:00
position: Point { x: 0, y: 0 },
message: String::from("hello world"),
2024-07-05 15:11:03 +03:00
color: (0, 0, 0),
quit: false,
2024-06-22 00:18:25 +03:00
};
2024-07-05 15:11:03 +03:00
state.process(Message::Resize {
width: 10,
height: 30,
});
2024-06-22 00:18:25 +03:00
state.process(Message::Move(Point { x: 10, y: 15 }));
2024-07-05 15:11:03 +03:00
state.process(Message::Echo(String::from("Hello world!")));
state.process(Message::ChangeColor(255, 0, 255));
2024-06-22 00:18:25 +03:00
state.process(Message::Quit);
2024-07-05 15:11:03 +03:00
assert_eq!(state.width, 10);
assert_eq!(state.height, 30);
2024-06-22 00:18:25 +03:00
assert_eq!(state.position.x, 10);
assert_eq!(state.position.y, 15);
assert_eq!(state.message, "Hello world!");
2024-07-05 15:11:03 +03:00
assert_eq!(state.color, (255, 0, 255));
assert!(state.quit);
2024-06-22 00:18:25 +03:00
}
}