mirror of
https://github.com/storytold/artcraft.git
synced 2026-11-04 09:28:11 +00:00
messages were floats
This commit is contained in:
1 parent
c8152ce364
commit
b4dafa5926
1 file changed
+5
-5
+5
-5
@@ -32,7 +32,7 @@ pub struct ImageDataMessage {
|
||||
pub request_id: Option<String>,
|
||||
|
||||
/// Eg. "0", "50", "100"
|
||||
pub percentage_complete: Option<i32>,
|
||||
pub percentage_complete: Option<f32>,
|
||||
|
||||
pub prompt: Option<String>,
|
||||
pub full_prompt: Option<String>,
|
||||
@@ -60,7 +60,7 @@ pub struct JsonDataMessage {
|
||||
pub request_id: Option<String>,
|
||||
|
||||
/// Eg. "0", "50", "100"
|
||||
pub percentage_complete: Option<i32>,
|
||||
pub percentage_complete: Option<f32>,
|
||||
|
||||
pub prompt: Option<String>,
|
||||
pub full_prompt: Option<String>,
|
||||
@@ -97,7 +97,7 @@ mod tests {
|
||||
|
||||
match message {
|
||||
WebsocketServerMessage::Image(image_data) => {
|
||||
assert_eq!(image_data.percentage_complete, Some(50));
|
||||
assert_eq!(image_data.percentage_complete, Some(50.0));
|
||||
assert!(image_data.url.is_some());
|
||||
},
|
||||
_ => panic!("Expected ImageData message"),
|
||||
@@ -115,7 +115,7 @@ mod tests {
|
||||
|
||||
match message {
|
||||
WebsocketServerMessage::Json(image_data) => {
|
||||
assert_eq!(image_data.percentage_complete, Some(100));
|
||||
assert_eq!(image_data.percentage_complete, Some(100.0));
|
||||
},
|
||||
_ => panic!("Expected JsonData message"),
|
||||
}
|
||||
@@ -132,7 +132,7 @@ mod tests {
|
||||
|
||||
match message {
|
||||
WebsocketServerMessage::Json(image_data) => {
|
||||
assert_eq!(image_data.percentage_complete, Some(0));
|
||||
assert_eq!(image_data.percentage_complete, Some(0.0));
|
||||
},
|
||||
_ => panic!("Expected JsonData message"),
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user