summaryrefslogtreecommitdiff
path: root/tests/follow_redirect.rs
blob: a421399bb31fe17f232be435ab008c1dbbce7efd (plain)
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
// Tests for calls to the /repos/{owner}/{repo}/stargazers API.
mod mock_error;

use mock_error::setup_error_handler;
use octocrab::{models::StarGazer, Octocrab, Page};
use serde::{Deserialize, Serialize};
use wiremock::{
    http::HeaderValue,
    matchers::{method, path},
    Mock, MockServer, ResponseTemplate,
};

#[derive(Serialize, Deserialize)]
struct FakePage<T> {
    items: Vec<T>,
}

async fn setup_api(template: ResponseTemplate) -> MockServer {
    let old_owner = "old-owner";
    let new_owner = "new-owner";
    let repo = "repo";
    let mock_server = MockServer::start().await;
    Mock::given(method("GET"))
        .and(path(format!("/repos/{old_owner}/{repo}/stargazers")))
        .respond_with(
            ResponseTemplate::new(301).append_header(
                "location",
                HeaderValue::from_bytes(format!("/repos/{new_owner}/{repo}/stargazers").as_bytes())
                    .unwrap(),
            ),
        )
        .mount(&mock_server)
        .await;
    Mock::given(method("GET"))
        .and(path(format!("/repos/{new_owner}/{repo}/stargazers")))
        .respond_with(template)
        .mount(&mock_server)
        .await;
    setup_error_handler(
        &mock_server,
        &format!("GET on /repos/{new_owner}/{repo}/stargazers was not received"),
    )
    .await;
    mock_server
}

fn setup_octocrab(uri: &str) -> Octocrab {
    Octocrab::builder().base_uri(uri).unwrap().build().unwrap()
}

const OWNER: &str = "old-owner";
const REPO: &str = "repo";

#[tokio::test]
#[cfg_attr(not(feature = "follow-redirect"), ignore)]
async fn should_return_page_with_users() {
    let star_gazers: Vec<StarGazer> =
        serde_json::from_str(include_str!("resources/stargazers.json")).unwrap();
    let login1: String = star_gazers[0].user.as_ref().unwrap().login.clone();
    let page_response = FakePage { items: star_gazers };
    let template = ResponseTemplate::new(200).set_body_json(&page_response);
    let mock_server = setup_api(template).await;
    let client = setup_octocrab(&mock_server.uri());
    let repos = client.repos(OWNER.to_owned(), REPO.to_owned());
    let result = repos.list_stargazers().send().await;
    assert!(
        result.is_ok(),
        "expected successful result, got error: {:#?}",
        result
    );
    let Page { items, .. } = result.unwrap();
    {
        assert_eq!(items.len(), 3);
        assert_eq!(items[0].user.as_ref().unwrap().login, login1);
    }
}