Skip to content

Commit 0e3242f

Browse files
committed
Fix tests
1 parent e69518d commit 0e3242f

2 files changed

Lines changed: 11 additions & 10 deletions

File tree

guest-agent/src/rpc_service.rs

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -366,7 +366,9 @@ impl DstackGuestRpc for InternalRpcHandler {
366366
let verifying_key =
367367
k256::ecdsa::VerifyingKey::from_sec1_bytes(&request.public_key)?;
368368
let signature = k256::ecdsa::Signature::from_slice(&request.signature)?;
369-
verifying_key.verify_prehash(&request.data, &signature).is_ok()
369+
verifying_key
370+
.verify_prehash(&request.data, &signature)
371+
.is_ok()
370372
}
371373
_ => return Err(anyhow::anyhow!("Unsupported algorithm")),
372374
};
@@ -686,19 +688,12 @@ mod tests {
686688
event_log_file: "/tmp/sample_event_log.txt".to_string(),
687689
};
688690

689-
let dummy_docker_config = DockerConfig {
690-
registry: None,
691-
username: None,
692-
token_key: None,
693-
};
694-
695691
let dummy_appcompose = AppCompose {
696692
manifest_version: 0,
697693
name: String::new(),
698694
features: Vec::new(),
699695
runner: String::new(),
700696
docker_compose_file: None,
701-
docker_config: dummy_docker_config,
702697
public_logs: false,
703698
public_sysinfo: false,
704699
public_tcbinfo: false,
@@ -711,6 +706,7 @@ mod tests {
711706
no_instance_id: false,
712707
secure_time: false,
713708
storage_fs: None,
709+
swap_size: 0,
714710
};
715711

716712
let dummy_appcompose_wrapper = AppComposeWrapper {
@@ -805,7 +801,7 @@ pNs85uhOZE8z2jr8Pg==
805801
keys: dummy_keys,
806802
vm_config: String::new(),
807803
cert_client: dummy_cert_client,
808-
demo_cert: String::new(),
804+
demo_cert: RwLock::new(String::new()),
809805
};
810806

811807
AppState {

sdk/rust/tests/test_client.rs

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,12 @@ async fn test_async_client_sign_and_verify_ed25519() {
112112
let pub_key = sign_resp.decode_public_key().unwrap();
113113

114114
let verify_resp = client
115-
.verify(algorithm, data_to_sign.clone(), sig.clone(), pub_key.clone())
115+
.verify(
116+
algorithm,
117+
data_to_sign.clone(),
118+
sig.clone(),
119+
pub_key.clone(),
120+
)
116121
.await
117122
.unwrap();
118123
assert!(verify_resp.valid);

0 commit comments

Comments
 (0)