-
-
Notifications
You must be signed in to change notification settings - Fork 1.6k
/
expansion.rs
67 lines (52 loc) · 1.52 KB
/
expansion.rs
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
#[macro_use] extern crate rocket;
use rocket::local::blocking::Client;
#[get("/easy/<id>")]
fn easy(id: i32) -> String {
format!("easy id: {}", id)
}
macro_rules! make_handler {
() => {
#[get("/hard/<id>")]
fn hard(id: i32) -> String {
format!("hard id: {}", id)
}
}
}
make_handler!();
macro_rules! foo {
($addr:expr, $name:ident) => {
#[get($addr)]
fn hi($name: String) -> String {
$name
}
};
}
// regression test for `#[get] panicking if used inside a macro
foo!("/hello/<name>", name);
#[test]
fn test_reexpansion() {
let rocket = rocket::build().mount("/", routes![easy, hard, hi]);
let client = Client::debug(rocket).unwrap();
let response = client.get("/easy/327").dispatch();
assert_eq!(response.into_string().unwrap(), "easy id: 327");
let response = client.get("/hard/72").dispatch();
assert_eq!(response.into_string().unwrap(), "hard id: 72");
let response = client.get("/hello/fish").dispatch();
assert_eq!(response.into_string().unwrap(), "fish");
}
macro_rules! index {
($type:ty) => {
#[get("/")]
fn index(thing: &rocket::State<$type>) -> String {
format!("Thing: {}", thing)
}
}
}
index!(i32);
#[test]
fn test_index() {
let rocket = rocket::build().mount("/", routes![index]).manage(100i32);
let client = Client::debug(rocket).unwrap();
let response = client.get("/").dispatch();
assert_eq!(response.into_string().unwrap(), "Thing: 100");
}