aurelius 0.7.5

A complete solution for previewing markdown.
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
124
125
126
127
128
129
130
131
132
133
134
135
136

use std::error::Error;
use std::fs;
use std::io::prelude::*;
use std::net::TcpStream;

use reqwest::StatusCode;
use socket2::Socket;

use aurelius::Server;

#[test]
fn not_found() -> Result<(), Box<dyn Error>> {
    let server = Server::bind("localhost:0")?;
    let addr = server.addr();

    let res = reqwest::blocking::get(&format!("http://{}/non-existent", addr))?;

    assert_eq!(res.status(), StatusCode::NOT_FOUND);

    Ok(())
}

#[test]
fn static_file_unknown_mime_type() -> Result<(), Box<dyn Error>> {
    let tmp_dir = tempfile::tempdir()?;
    let file_path = tmp_dir.path().join("file-of-unknown-type");
    fs::write(file_path, ":)")?;

    let mut server = Server::bind("localhost:0")?;
    server.set_static_root(tmp_dir.path());
    let addr = server.addr();

    let res = reqwest::blocking::get(&format!("http://{}/file-of-unknown-type", addr))?;

    assert_eq!(res.status(), StatusCode::OK);

    Ok(())
}

#[test]
fn change_static_root() -> Result<(), Box<dyn Error>> {
    let tmp_dir = tempfile::tempdir()?;

    fs::write(tmp_dir.path().join("file.txt"), "Lorem ipsum")?;

    let mut server = Server::bind("localhost:0")?;

    let file_url = format!("http://{}/file.txt", server.addr());

    let res = reqwest::blocking::get(&file_url)?;
    assert_eq!(res.status(), StatusCode::NOT_FOUND);

    server.set_static_root(tmp_dir.path());

    let res = reqwest::blocking::get(&file_url)?;
    assert_eq!(res.status(), StatusCode::OK);
    assert_eq!(res.headers()["Content-Type"], "text/plain");
    assert_eq!(res.text()?, "Lorem ipsum");

    Ok(())
}

#[test]
fn change_static_root_to_file() -> Result<(), Box<dyn Error>> {
    let tmp_dir = tempfile::tempdir()?;
    let file_path = tmp_dir.path().join("file.txt");

    fs::write(&file_path, "Lorem ipsum")?;

    let mut server = Server::bind("localhost:0")?;

    server.set_static_root(file_path);

    let res = reqwest::blocking::get(&format!("http://{}/", server.addr()))?;
    assert_eq!(res.status(), StatusCode::OK);
    assert!(!res.text()?.contains("Lorem ipsum"));

    let res = reqwest::blocking::get(&format!("http://{}/non-existent", server.addr()))?;
    assert_eq!(res.status(), StatusCode::NOT_FOUND);

    Ok(())
}

#[test]
fn static_files() -> Result<(), Box<dyn Error>> {
    let server = Server::bind("localhost:0")?;
    let addr = server.addr();

    let res = reqwest::blocking::get(&format!("http://{}/__/css/styles.css", addr))?;
    assert!(res.status().is_success());
    assert_eq!(res.headers()["Content-Type"], "text/css");
    res.text()?;

    // File in a submodule, make sure that it's included too.
    let res = reqwest::blocking::get(&format!(
        "http://{}/__/vendor/highlight.js/build/highlight.min.js",
        addr
    ))?;
    assert!(res.status().is_success());
    assert_eq!(res.headers()["Content-Type"], "application/javascript");
    res.text()?;

    Ok(())
}

/// Tests that the server gracefully handles clients that disconnect in the middle of reading a
/// response. It's a bit hacky (and thus flaky), but the test triggers the desired conditions
/// enough to be valuable.
#[test]
fn partial_read() -> Result<(), Box<dyn Error>> {
    let server = Server::bind("localhost:0")?;

    let mut conn: Socket = TcpStream::connect(server.addr())?.into();

    // Use a very small buffer to make it likely that the socket closes before the write completes.
    conn.set_recv_buffer_size(1)?;

    write!(conn, "GET /__/css/styles.css HTTP/1.1\r\n\r\n")?;
    conn.flush()?;

    let _ = conn.read(&mut [])?;
    drop(conn);

    Ok(())
}

#[test]
fn static_file_not_found() -> Result<(), Box<dyn Error>> {
    let server = Server::bind("localhost:0")?;
    let addr = server.addr();

    let res = reqwest::blocking::get(&format!("http://{}/__/does-not-exist.js", addr))?;
    assert_eq!(res.status(), StatusCode::NOT_FOUND);

    Ok(())
}