rocket_lamb 0.6.0

A crate to allow running a Rocket webserver as an AWS Lambda Function with API Gateway
Documentation 
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123

#![feature(proc_macro_hygiene, decl_macro)]

#[macro_use]
extern crate rocket;

use lambda_http::{Body, Handler, Request};
use lambda_runtime::Context;
use rocket::http::uri::Origin;
use rocket_lamb::{BasePathBehaviour, RocketExt};
use std::error::Error;
use std::fs::File;

#[catch(404)]
fn not_found(req: &rocket::Request) -> String {
    req.uri().to_string()
}

#[get("/path")]
fn get_path<'r>(origin: &'r Origin<'r>) -> &'r str {
    origin.path()
}

fn make_rocket() -> rocket::Rocket {
    rocket::ignite()
        .mount("/", routes![get_path])
        .register(catchers![not_found])
}

fn get_request(json_file: &str) -> Result<Request, Box<dyn Error>> {
    let file = File::open(format!("tests/requests/{}.json", json_file))?;
    Ok(lambda_http::request::from_reader(file)?)
}

macro_rules! test_case {
    ($name:ident, $file:expr, $status:expr, $path:expr) => {
        test_case!($name, RemountAndInclude, $file, $status, $path);
    };
    ($name:ident, $path_behaviour:ident, $file:expr, $status:expr, $path:expr) => {
        #[test]
        fn $name() -> Result<(), Box<dyn Error>> {
            let mut handler = make_rocket()
                .lambda()
                .base_path_behaviour(BasePathBehaviour::$path_behaviour)
                .into_handler();

            let req = get_request($file)?;
            let res = handler.run(req, Context::default())?;

            assert_eq!(res.status(), $status);
            assert_eq!(*res.body(), Body::Text($path.to_string()));
            Ok(())
        }
    };
}

test_case!(api_gateway, "path_api_gateway", 200, "/Prod/path/");
test_case!(
    api_gateway_include_base,
    Include,
    "path_api_gateway",
    404,
    "/Prod/path/"
);
test_case!(
    api_gateway_exclude_base,
    Exclude,
    "path_api_gateway",
    200,
    "/path/"
);

test_case!(custom_domain, "path_custom_domain", 200, "/path/");
test_case!(
    custom_domain_include_empty_base,
    Include,
    "path_custom_domain",
    200,
    "/path/"
);
test_case!(
    custom_domain_exclude_empty_base,
    Exclude,
    "path_custom_domain",
    200,
    "/path/"
);

test_case!(
    custom_domain_with_base_path,
    "path_custom_domain_with_base",
    200,
    "/base-path/path/"
);
test_case!(
    custom_domain_with_base_path_include,
    Include,
    "path_custom_domain_with_base",
    404,
    "/base-path/path/"
);
test_case!(
    custom_domain_with_base_path_exclude,
    Exclude,
    "path_custom_domain_with_base",
    200,
    "/path/"
);

test_case!(application_load_balancer, "path_alb", 200, "/path/");
test_case!(
    application_load_balancer_include_empty_base,
    Include,
    "path_alb",
    200,
    "/path/"
);
test_case!(
    application_load_balancer_exclude_empty_base,
    Exclude,
    "path_alb",
    200,
    "/path/"
);