beet_net 0.0.8

Cross-platform networking utilities
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

use beet_core::prelude::*;
use eventsource_stream::EventStream;
use eventsource_stream::Eventsource;

#[extend::ext(name=ResponseClientExt)]
pub impl Response {
	/// Server Sent Events, checks for Ok status before parsing
	#[allow(async_fn_in_trait)]
	async fn event_source(self) -> Result<EventStream<Body>> {
		self.into_result().await?.body.eventsource().xok()
	}
}

#[cfg(test)]
mod test {
	use crate::prelude::*;
	use beet_core::prelude::*;
	use bevy::tasks::futures_lite::StreamExt;

	#[beet_core::test]
	// TODO spin up our own server for tests
	#[ignore = "hits network"]
	async fn works() {
		let mut ev = Request::get("https://sse.dev/test")
			.send()
			.await
			.unwrap()
			.event_source()
			.await
			.unwrap();

		let mut count = 0;
		while let Some(Ok(event)) = ev.next().await {
			event.data.xref().xpect_contains("It works!");
			if count == 2 {
				break;
			} else {
				count += 1;
			}
		}
	}
}