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
|
/*
Social program for Ramaskrik.
Copyright (C) 2019 Harald Eilertsen <haraldei@anduin.net>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
use ramaskrik;
use ramaskrik::models::{
film::{ Film, NewFilm },
room::Room
};
use rocket::http::ContentType;
use rocket_contrib::databases::diesel::prelude::*;
use serde_json;
pub fn server_with_db<TestFn>(f: TestFn)
where
TestFn: Fn(rocket::Rocket, ramaskrik::db::Connection)
{
let db_url = dotenv::var("TEST_DATABASE_URL")
.map_err(|_| "No database! Set TEST_DATABASE_URL env var and try again.").unwrap();
let server = ramaskrik::build_rocket(&db_url).unwrap();
let db = ramaskrik::db::Connection::get_one(&server).expect("Could not get db connection");
diesel::dsl::sql_query("TRUNCATE TABLE rooms, films").execute(&*db).unwrap();
f(server, db);
}
#[test]
fn getting_rooms_from_api() {
server_with_db(|server, db| {
use ramaskrik::schema::rooms::dsl::*;
let new_rooms = vec![
name.eq("Main room"),
name.eq("Small room"),
name.eq("Neverland")];
diesel::insert_into(rooms)
.values(&new_rooms)
.execute(&*db)
.unwrap();
let client = rocket::local::Client::new(server).unwrap();
let mut response = client.get("/rooms").dispatch();
assert_eq!(response.content_type(), Some(ContentType::JSON));
let fetched_rooms: Vec<Room> =
serde_json::from_str(&response.body_string().unwrap()).unwrap();
assert_eq!(fetched_rooms.len(), 3);
let room_names: Vec<&str> = fetched_rooms.iter().map(|room| room.name.as_str()).collect();
assert!(room_names.contains(&"Main room"));
assert!(room_names.contains(&"Small room"));
assert!(room_names.contains(&"Neverland"));
})
}
#[test]
fn getting_films_from_api() {
server_with_db(|server, db| {
use ramaskrik::schema::films::dsl::*;
let new_films = vec![
NewFilm { title: "Hellraiser", url: Some("https://www.imdb.com/title/tt0093177") },
NewFilm { title: "Huset", url: Some("https://www.imdb.com/title/tt3425402") },
NewFilm { title: "Skuld", url: Some("https://www.imdb.com/title/tt4344456") }];
diesel::insert_into(films)
.values(&new_films)
.execute(&*db)
.unwrap();
let client = rocket::local::Client::new(server).unwrap();
let mut response = client.get("/films").dispatch();
assert_eq!(response.content_type(), Some(ContentType::JSON));
let fetched_films: Vec<Film> =
serde_json::from_str(&response.body_string().unwrap()).unwrap();
assert_eq!(fetched_films.len(), 3);
let film_names: Vec<&str> = fetched_films.iter().map(|film| film.title.as_str()).collect();
assert!(film_names.contains(&"Hellraiser"));
assert!(film_names.contains(&"Huset"));
assert!(film_names.contains(&"Skuld"));
});
}
|
