sbb-api 0.1.0

A library to interact with sbb.ch's unofficial API
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
137
138
139

use chrono::{Utc};
use simple_error::SimpleError;

use crate::make_request;
use crate::models::location::LocationType;
use crate::models::results::VerbindungenResults;

pub async fn get_connections(from: &str, from_type: LocationType, to: &str, to_type: LocationType,
                             on: &chrono::DateTime<Utc>) -> Result<VerbindungenResults, SimpleError> {
    let base_path = "/unauth/fahrplanservice/v1/verbindungen/";

    // s/Zurich/s/Bern/ab/2019-09-20/10-14/";
    let date = on.format("%Y-%m-%d");
    let time = on.format("%H-%M");

    let addition = format!("{}/{}/{}/{}/ab/{}/{}/", from_type,
                           from,
                           to_type,
                           to,
                           date,
                           time);
    let path = format!("{}{}", base_path, addition);

    let resp = make_request(&path).await.expect("Invalid request");

    if !resp.status().is_success() {
        bail!("status is not success")
    }

    let response = &resp.text().await.unwrap();

    Ok(serde_json::from_str(
        response
    ).unwrap())
}

mod tests {
    use chrono::{Datelike, TimeZone, Utc};
    use crate::connections::get_connections;
    use crate::models::location::LocationType;

    #[actix_rt::test]
    pub async fn test_get_connection() {
        let today = chrono::offset::Local::now();
        let date = chrono::Utc.ymd(today.year(), today.month(), today.day()).and_hms(12, 0, 0);
        let conn = get_connections("Zürich HB",
                                   LocationType::Station,
                                   "Basel",
                                   LocationType::Station,
                                   &date);
        let conn_res = conn.await;
        assert!(conn_res.is_ok());
        let verbindungen_res = conn_res.unwrap();
        assert!(verbindungen_res.verbindungen.len() > 0);
        println!("Connections = {:?}", verbindungen_res);
    }

    #[actix_rt::test]
    pub async fn test_get_connection2() {
        let today = chrono::offset::Local::now();
        let date = Utc.ymd(today.year(), today.month(), today.day()).and_hms(12, 0, 0);
        let conn = get_connections("Chiasso",
                                   LocationType::Station,
                                   "Zürich HB",
                                   LocationType::Station,
                                   &date);
        let conn_res = conn.await;
        assert!(conn_res.is_ok());
        let verbindungen_res = conn_res.unwrap();
        assert!(verbindungen_res.verbindungen.len() > 0);
        println!("Connections = {:?}", verbindungen_res);
    }

    #[actix_rt::test]
    pub async fn test_get_connections3() {
        let today = chrono::offset::Local::now();
        let date =
            Utc.ymd(today.year(), today.month(), today.day())
                .and_hms(12, 0, 0);
        let conn = get_connections(
            "Zürich HB",
            LocationType::Station,
            "Dübendorf, Bahnof",
            LocationType::Station,
            &date,
        );

        let conn_res = conn.await;
        assert!(conn_res.is_ok());
        let verbindungen_res = conn_res.unwrap();
        println!("Connections = {}", verbindungen_res)
    }

    #[actix_rt::test]
    pub async fn test_get_connections4() {
        let today = chrono::offset::Local::now();
        let date =
            Utc.ymd(today.year(), today.month(), today.day())
                .and_hms(12, 0, 0);
        let conn = get_connections("Chiasso",
                                   LocationType::Address,
                                   "Zürich",
                                   LocationType::Address,
                                   &date,
        );

        let conn_res = conn.await;
        assert!(conn_res.is_ok());
        let verbindungen_res = conn_res.unwrap();
        println!("Connections = {}", verbindungen_res);

        for c in verbindungen_res.verbindungen {
            println!("Conn: {}, Duration = {:?}", c, c.duration());
        }
    }

    #[actix_rt::test]
    pub async fn test_get_connections5() {
        let today = chrono::offset::Local::now();
        let date =
            Utc.ymd(today.year(), today.month(), today.day())
                .and_hms(12, 0, 0);
        let conn = get_connections("Im Tiergraten, 8055 Zürich, Zürich",
                                   LocationType::Address,
                                   "8005 Zürich, Hardturmstrasse 3",
                                   LocationType::Address,
                                   &date,
        );

        let conn_res = conn.await;
        assert!(conn_res.is_ok());
        let verbindungen_res = conn_res.unwrap();
        println!("Connections = {}", verbindungen_res);

        for c in verbindungen_res.verbindungen {
            println!("Conn: {}, Duration = {:?}", c, c.duration());
        }
    }
}