Skip to content

Commit 143c2a3

Browse files
committed
fix /scan/deregister endpoint;
1 parent 30f0c1d commit 143c2a3

File tree

2 files changed

+28
-5
lines changed

2 files changed

+28
-5
lines changed

Cargo.toml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,3 +24,6 @@ base16 = "0.2.1"
2424
yaml-rust = "0.4.4"
2525
thiserror = "1.0.22"
2626
derive_more = "0.99"
27+
28+
[dev-dependencies]
29+
expect-test = "1.0.1"

src/scanning.rs

Lines changed: 25 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -27,11 +27,9 @@ impl NodeInterface {
2727
}
2828

2929
pub fn deregister_scan(&self, scan_id: ScanId) -> Result<ScanId> {
30-
let endpoint = "/scan/deregister/";
31-
let body = json! ({
32-
"scanId": scan_id,
33-
});
34-
let res = self.send_post_req(endpoint, to_string_pretty(&body).unwrap());
30+
let endpoint = "/scan/deregister";
31+
let body = generate_deregister_scan_json(scan_id);
32+
let res = self.send_post_req(endpoint, body);
3533
let res_json = self.parse_response_to_json(res)?;
3634

3735
if res_json["error"].is_null() {
@@ -87,3 +85,25 @@ impl NodeInterface {
8785
}
8886
}
8987
}
88+
89+
fn generate_deregister_scan_json(scan_id: ScanId) -> String {
90+
let body = json!({
91+
"scanId": scan_id,
92+
});
93+
to_string_pretty(&body).unwrap()
94+
}
95+
96+
#[cfg(test)]
97+
mod tests {
98+
use super::*;
99+
100+
#[test]
101+
fn test_generate_deregister_scan_json() {
102+
let scan_id = ScanId::from(100);
103+
expect_test::expect![[r#"
104+
{
105+
"scanId": 100
106+
}"#]]
107+
.assert_eq(&generate_deregister_scan_json(scan_id));
108+
}
109+
}

0 commit comments

Comments
 (0)