@@ -3,19 +3,19 @@ use image_service::{IMAGE_DIR, IMAGE_SERVICE_SOCKET};
33use log:: { error, info, warn} ;
44use nix:: sys:: signal:: { kill, Signal } ;
55use nix:: unistd:: { self , Pid } ;
6+ use prost:: Message ;
67use proto_definitions:: {
78 host_service:: { host_service_client:: HostServiceClient , Empty } ,
89 image_service:: {
910 image_service_client:: ImageServiceClient , DeleteImageRequest , ImageState ,
1011 ListImagesRequest , PullImageRequest , WatchImageStatusRequest ,
1112 } ,
1213 vm_service:: {
13- vm_service_client:: VmServiceClient , CreateVmRequest , DeleteVmRequest , GetVmRequest ,
14- PingVmRequest , StartVmRequest , StreamVmEventsRequest , CpuConfig , MemoryConfig , VmConfig ,
14+ vm_service_client:: VmServiceClient , CpuConfig , CreateVmRequest , DeleteVmRequest ,
15+ GetVmRequest , MemoryConfig , PingVmRequest , StartVmRequest , StreamVmEventsRequest , VmConfig ,
1516 VmEvent , VmState , VmStateChangedEvent ,
1617 } ,
1718} ;
18- use prost:: Message ;
1919use std:: path:: Path ;
2020use std:: process:: Command ;
2121use std:: sync:: Arc ;
@@ -24,8 +24,8 @@ use tokio::net::UnixStream;
2424use tokio:: sync:: OnceCell ;
2525use tokio:: time:: timeout;
2626use tokio_stream:: StreamExt ;
27- use tower:: service_fn;
2827use tonic:: transport:: { Channel , Endpoint , Uri } ;
28+ use tower:: service_fn;
2929use vm_service:: { VM_API_SOCKET_DIR , VM_CH_BIN } ;
3030
3131const PUBLIC_SERVER_ADDRESS : & str = "http://[::1]:1337" ;
@@ -73,7 +73,9 @@ async fn get_public_clients() -> Result<(VmServiceClient<Channel>, HostServiceCl
7373async fn get_image_service_client ( ) -> Result < ImageServiceClient < Channel > > {
7474 let endpoint = Endpoint :: from_static ( "http://[::1]:50051" ) ;
7575 let channel = endpoint
76- . connect_with_connector ( service_fn ( |_: Uri | UnixStream :: connect ( IMAGE_SERVICE_SOCKET ) ) )
76+ . connect_with_connector ( service_fn ( |_: Uri | {
77+ UnixStream :: connect ( IMAGE_SERVICE_SOCKET )
78+ } ) )
7779 . await ?;
7880 Ok ( ImageServiceClient :: new ( channel) )
7981}
@@ -146,10 +148,7 @@ async fn wait_for_vm_state(
146148 }
147149
148150 if new_state == VmState :: Crashed {
149- let err_msg = format ! (
150- "VM entered Crashed state. Reason: {}" ,
151- state_change. reason
152- ) ;
151+ let err_msg = format ! ( "VM entered Crashed state. Reason: {}" , state_change. reason) ;
153152 error ! ( "{}" , & err_msg) ;
154153 return Err ( anyhow:: anyhow!( err_msg) ) ;
155154 }
@@ -197,10 +196,7 @@ async fn test_create_and_start_vm() -> Result<()> {
197196 cleanup_disabled : false ,
198197 } ;
199198
200- info ! (
201- "Connecting to StreamVmEvents stream for vm_id: {}" ,
202- & vm_id
203- ) ;
199+ info ! ( "Connecting to StreamVmEvents stream for vm_id: {}" , & vm_id) ;
204200 let events_req = StreamVmEventsRequest {
205201 vm_id : vm_id. clone ( ) ,
206202 ..Default :: default ( )
@@ -386,4 +382,4 @@ where
386382 }
387383 }
388384 anyhow:: bail!( "Stream ended before image became ready" ) ;
389- }
385+ }
0 commit comments