summaryrefslogtreecommitdiffstats
path: root/src/display/testutil.rs
blob: e4be6a90d1d03e10c2f6d70112e7546e775558fc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
use crate::config::Config;
use crate::display::{CrossTerm, Event, KeyCode, KeyEvent, KeyModifiers};
use crate::input::input_handler::InputHandler;
use crossterm::event::{MouseEvent, MouseEventKind};
use std::env::set_var;
use std::path::Path;

pub struct TestContext<'t> {
	pub config: &'t Config,
	pub crossterm: CrossTerm,
	pub input_handler: InputHandler<'t>,
}

pub fn display_module_test<F>(callback: F)
where F: FnOnce(TestContext<'_>) {
	set_var(
		"GIT_DIR",
		Path::new(env!("CARGO_MANIFEST_DIR"))
			.join("test")
			.join("fixtures")
			.join("simple")
			.to_str()
			.unwrap(),
	);
	let config = Config::new().unwrap();
	let input_handler = InputHandler::new(&config.key_bindings);
	let crossterm = CrossTerm::new();
	callback(TestContext {
		config: &config,
		crossterm,
		input_handler,
	});
}

pub fn _create_key_event(c: KeyCode, modifiers: &[String]) -> Event {
	let mut key_modifiers = KeyModifiers::NONE;
	for modifier in modifiers {
		match modifier.as_str() {
			"Control" => key_modifiers.insert(KeyModifiers::CONTROL),
			"Shift" => key_modifiers.insert(KeyModifiers::SHIFT),
			"Alt" => key_modifiers.insert(KeyModifiers::ALT),
			_ => panic!("Invalid modifier: {}", modifier),
		}
	}
	Event::Key(KeyEvent::new(c, key_modifiers))
}

#[macro_export]
macro_rules! create_key_event {
	($char:expr) => {
		crate::display::testutil::_create_key_event(crate::display::KeyCode::Char($char), &[])
	};
	($char:expr, $($modifiers:expr),*) => {
		{
			let mut modifiers = vec![];
			$( modifiers.push(String::from($modifiers)); )*
			crate::display::testutil::_create_key_event(crate::display::KeyCode::Char($char), &modifiers)
		}
	};
	(code $code:expr) => {
		crate::display::testutil::_create_key_event($code, &[])
	};
	(code $code:expr, $($modifiers:expr),*) => {
		let mut modifiers = vec![];
		$( modifiers.push(String::from($modifiers)); )*
		crate::display::testutil::_create_key_event($code, &modifiers)
	};
}

pub fn _create_mouse_event(kind: MouseEventKind, column: u16, row: u16, modifiers: &[String]) -> Event {
	let mut key_modifiers = KeyModifiers::NONE;
	for modifier in modifiers {
		match modifier.as_str() {
			"Control" => key_modifiers.insert(KeyModifiers::CONTROL),
			"Shift" => key_modifiers.insert(KeyModifiers::SHIFT),
			"Alt" => key_modifiers.insert(KeyModifiers::ALT),
			_ => panic!("Invalid modifier: {}", modifier),
		}
	}
	Event::Mouse(MouseEvent {
		kind,
		column,
		row,
		modifiers: key_modifiers,
	})
}

#[macro_export]
macro_rules! create_mouse_event {
	($kind:expr) => {
		Event::Mouse(MouseEvent {
			kind: $kind,
			column: 0,
			row: 0,
			modifiers: KeyModifiers::NONE,
		})
	};
}