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
|
use wiremock::{
matchers::{method, path},
Mock, MockServer, ResponseTemplate,
};
use mock_error::setup_error_handler;
use octocrab::models::code_scannings::CodeScanningAlert;
use octocrab::Octocrab;
mod mock_error;
const OWNER: &str = "org";
const REPO: &str = "some-repo";
async fn setup_codescanning_list_api(template: ResponseTemplate, is_org: bool) -> MockServer {
let mock_server = MockServer::start().await;
if is_org {
Mock::given(method("GET"))
.and(path(format!(
"/orgs/{owner}/code-scanning/alerts",
owner = OWNER
)))
.respond_with(template.clone())
.mount(&mock_server)
.await;
setup_error_handler(
&mock_server,
&format!(
"GET on /org/{owner}/code-scanning/alerts was not received",
owner = OWNER
),
)
.await;
} else {
Mock::given(method("GET"))
.and(path(format!(
"/repos/{owner}/{repo}/code-scanning/alerts",
owner = OWNER,
repo = REPO
)))
.respond_with(template.clone())
.mount(&mock_server)
.await;
setup_error_handler(
&mock_server,
&format!(
"GET on /repos/{owner}/{repo}/code-scanning/alerts was not received",
owner = OWNER,
repo = REPO
),
)
.await;
}
mock_server
}
fn setup_octocrab(uri: &str) -> Octocrab {
Octocrab::builder().base_uri(uri).unwrap().build().unwrap()
}
#[tokio::test]
async fn check_list_200() {
let s = include_str!("resources/codescanning_alerts_multiple.json");
let alert: Vec<CodeScanningAlert> = serde_json::from_str(s).unwrap();
let template = ResponseTemplate::new(200).set_body_json(&alert);
let mock_server = setup_codescanning_list_api(template, false).await;
let client = setup_octocrab(&mock_server.uri());
let result = client
.code_scannings(OWNER.to_owned(), REPO.to_owned())
.list()
.send()
.await;
assert!(
result.is_ok(),
"expected successful result, got error: {:?}",
result
);
}
#[tokio::test]
async fn check_list_organisation_200() {
let s = include_str!("resources/codescanning_alerts_multiple.json");
let alert: Vec<CodeScanningAlert> = serde_json::from_str(s).unwrap();
let template = ResponseTemplate::new(200).set_body_json(&alert);
let mock_server = setup_codescanning_list_api(template, true).await;
let client = setup_octocrab(&mock_server.uri());
let result = client
.code_scannings_organisation(OWNER.to_owned())
.list()
.send()
.await;
assert!(
result.is_ok(),
"expected successful result, got error: {:?}",
result
);
}
|