2020-09-10 08:39:43 +00:00
|
|
|
// use async_graphql::*;
|
|
|
|
|
// use futures::{SinkExt, Stream, StreamExt};
|
|
|
|
|
//
|
|
|
|
|
// #[async_std::test]
|
|
|
|
|
// pub async fn test_subscription_ws_transport() {
|
|
|
|
|
// struct QueryRoot;
|
|
|
|
|
//
|
|
|
|
|
// #[Object]
|
|
|
|
|
// impl QueryRoot {}
|
|
|
|
|
//
|
|
|
|
|
// struct SubscriptionRoot;
|
|
|
|
|
//
|
|
|
|
|
// #[Subscription]
|
|
|
|
|
// impl SubscriptionRoot {
|
|
|
|
|
// async fn values(&self) -> impl Stream<Item = i32> {
|
|
|
|
|
// futures::stream::iter(0..10)
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
//
|
|
|
|
|
// let schema = Schema::new(QueryRoot, EmptyMutation, SubscriptionRoot);
|
|
|
|
|
// let (mut sink, mut stream) = schema.subscription_connection(WebSocketTransport::default());
|
|
|
|
|
//
|
|
|
|
|
// sink.send(
|
|
|
|
|
// serde_json::to_vec(&serde_json::json!({
|
|
|
|
|
// "type": "connection_init",
|
|
|
|
|
// "payload": { "token": "123456" }
|
|
|
|
|
// }))
|
|
|
|
|
// .unwrap(),
|
|
|
|
|
// )
|
|
|
|
|
// .await
|
|
|
|
|
// .unwrap();
|
|
|
|
|
//
|
|
|
|
|
// assert_eq!(
|
|
|
|
|
// Some(serde_json::json!({
|
|
|
|
|
// "type": "connection_ack",
|
|
|
|
|
// })),
|
|
|
|
|
// serde_json::from_slice(&stream.next().await.unwrap()).unwrap()
|
|
|
|
|
// );
|
|
|
|
|
//
|
|
|
|
|
// sink.send(
|
|
|
|
|
// serde_json::to_vec(&serde_json::json!({
|
|
|
|
|
// "type": "start",
|
|
|
|
|
// "id": "1",
|
|
|
|
|
// "payload": {
|
|
|
|
|
// "query": "subscription { values }"
|
|
|
|
|
// },
|
|
|
|
|
// }))
|
|
|
|
|
// .unwrap(),
|
|
|
|
|
// )
|
|
|
|
|
// .await
|
|
|
|
|
// .unwrap();
|
|
|
|
|
//
|
|
|
|
|
// for i in 0..10 {
|
|
|
|
|
// assert_eq!(
|
|
|
|
|
// Some(serde_json::json!({
|
|
|
|
|
// "type": "data",
|
|
|
|
|
// "id": "1",
|
|
|
|
|
// "payload": { "data": { "values": i } },
|
|
|
|
|
// })),
|
|
|
|
|
// serde_json::from_slice(&stream.next().await.unwrap()).unwrap()
|
|
|
|
|
// );
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
//
|
|
|
|
|
// #[async_std::test]
|
|
|
|
|
// pub async fn test_subscription_ws_transport_with_token() {
|
|
|
|
|
// struct Token(String);
|
|
|
|
|
//
|
|
|
|
|
// struct QueryRoot;
|
|
|
|
|
//
|
|
|
|
|
// #[Object]
|
|
|
|
|
// impl QueryRoot {}
|
|
|
|
|
//
|
|
|
|
|
// struct SubscriptionRoot;
|
|
|
|
|
//
|
|
|
|
|
// #[Subscription]
|
|
|
|
|
// impl SubscriptionRoot {
|
|
|
|
|
// async fn values(&self, ctx: &Context<'_>) -> FieldResult<impl Stream<Item = i32>> {
|
|
|
|
|
// if ctx.data_unchecked::<Token>().0 != "123456" {
|
|
|
|
|
// return Err("forbidden".into());
|
|
|
|
|
// }
|
|
|
|
|
// Ok(futures::stream::iter(0..10))
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
//
|
|
|
|
|
// let schema = Schema::new(QueryRoot, EmptyMutation, SubscriptionRoot);
|
|
|
|
|
//
|
|
|
|
|
// let (mut sink, mut stream) = schema.subscription_connection(WebSocketTransport::new(|value| {
|
|
|
|
|
// #[derive(serde::Deserialize)]
|
|
|
|
|
// struct Payload {
|
|
|
|
|
// token: String,
|
|
|
|
|
// }
|
|
|
|
|
//
|
|
|
|
|
// let payload: Payload = serde_json::from_value(value).unwrap();
|
|
|
|
|
// let mut data = Data::default();
|
|
|
|
|
// data.insert(Token(payload.token));
|
|
|
|
|
// Ok(data)
|
|
|
|
|
// }));
|
|
|
|
|
//
|
|
|
|
|
// sink.send(
|
|
|
|
|
// serde_json::to_vec(&serde_json::json!({
|
|
|
|
|
// "type": "connection_init",
|
|
|
|
|
// "payload": { "token": "123456" }
|
|
|
|
|
// }))
|
|
|
|
|
// .unwrap(),
|
|
|
|
|
// )
|
|
|
|
|
// .await
|
|
|
|
|
// .unwrap();
|
|
|
|
|
//
|
|
|
|
|
// assert_eq!(
|
|
|
|
|
// Some(serde_json::json!({
|
|
|
|
|
// "type": "connection_ack",
|
|
|
|
|
// })),
|
|
|
|
|
// serde_json::from_slice(&stream.next().await.unwrap()).unwrap()
|
|
|
|
|
// );
|
|
|
|
|
//
|
|
|
|
|
// sink.send(
|
|
|
|
|
// serde_json::to_vec(&serde_json::json!({
|
|
|
|
|
// "type": "start",
|
|
|
|
|
// "id": "1",
|
|
|
|
|
// "payload": {
|
|
|
|
|
// "query": "subscription { values }"
|
|
|
|
|
// },
|
|
|
|
|
// }))
|
|
|
|
|
// .unwrap(),
|
|
|
|
|
// )
|
|
|
|
|
// .await
|
|
|
|
|
// .unwrap();
|
|
|
|
|
//
|
|
|
|
|
// for i in 0..10 {
|
|
|
|
|
// assert_eq!(
|
|
|
|
|
// Some(serde_json::json!({
|
|
|
|
|
// "type": "data",
|
|
|
|
|
// "id": "1",
|
|
|
|
|
// "payload": { "data": { "values": i } },
|
|
|
|
|
// })),
|
|
|
|
|
// serde_json::from_slice(&stream.next().await.unwrap()).unwrap()
|
|
|
|
|
// );
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
//
|
|
|
|
|
// #[async_std::test]
|
|
|
|
|
// pub async fn test_subscription_ws_transport_error() {
|
|
|
|
|
// struct QueryRoot;
|
|
|
|
|
//
|
|
|
|
|
// struct Event {
|
|
|
|
|
// value: i32,
|
|
|
|
|
// }
|
|
|
|
|
//
|
|
|
|
|
// #[Object]
|
|
|
|
|
// impl Event {
|
|
|
|
|
// async fn value(&self) -> FieldResult<i32> {
|
|
|
|
|
// if self.value < 5 {
|
|
|
|
|
// Ok(self.value)
|
|
|
|
|
// } else {
|
|
|
|
|
// Err("TestError".into())
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
//
|
|
|
|
|
// #[Object]
|
|
|
|
|
// impl QueryRoot {}
|
|
|
|
|
//
|
|
|
|
|
// struct SubscriptionRoot;
|
|
|
|
|
//
|
|
|
|
|
// #[Subscription]
|
|
|
|
|
// impl SubscriptionRoot {
|
|
|
|
|
// async fn events(&self) -> impl Stream<Item = Event> {
|
|
|
|
|
// futures::stream::iter((0..10).map(|n| Event { value: n }))
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
//
|
|
|
|
|
// let schema = Schema::new(QueryRoot, EmptyMutation, SubscriptionRoot);
|
|
|
|
|
//
|
|
|
|
|
// let (mut sink, mut stream) =
|
|
|
|
|
// schema.subscription_connection(WebSocketTransport::new(|_| Ok(Data::default())));
|
|
|
|
|
//
|
|
|
|
|
// sink.send(
|
|
|
|
|
// serde_json::to_vec(&serde_json::json!({
|
|
|
|
|
// "type": "connection_init"
|
|
|
|
|
// }))
|
|
|
|
|
// .unwrap(),
|
|
|
|
|
// )
|
|
|
|
|
// .await
|
|
|
|
|
// .unwrap();
|
|
|
|
|
//
|
|
|
|
|
// assert_eq!(
|
|
|
|
|
// Some(serde_json::json!({
|
|
|
|
|
// "type": "connection_ack",
|
|
|
|
|
// })),
|
|
|
|
|
// serde_json::from_slice(&stream.next().await.unwrap()).unwrap()
|
|
|
|
|
// );
|
|
|
|
|
//
|
|
|
|
|
// sink.send(
|
|
|
|
|
// serde_json::to_vec(&serde_json::json!({
|
|
|
|
|
// "type": "start",
|
|
|
|
|
// "id": "1",
|
|
|
|
|
// "payload": {
|
|
|
|
|
// "query": "subscription { events { value } }"
|
|
|
|
|
// },
|
|
|
|
|
// }))
|
|
|
|
|
// .unwrap(),
|
|
|
|
|
// )
|
|
|
|
|
// .await
|
|
|
|
|
// .unwrap();
|
|
|
|
|
//
|
|
|
|
|
// for i in 0i32..5 {
|
|
|
|
|
// assert_eq!(
|
|
|
|
|
// Some(serde_json::json!({
|
|
|
|
|
// "type": "data",
|
|
|
|
|
// "id": "1",
|
|
|
|
|
// "payload": { "data": { "events": { "value": i } } },
|
|
|
|
|
// })),
|
|
|
|
|
// serde_json::from_slice(&stream.next().await.unwrap()).unwrap()
|
|
|
|
|
// );
|
|
|
|
|
// }
|
|
|
|
|
//
|
|
|
|
|
// assert_eq!(
|
|
|
|
|
// Some(serde_json::json!({
|
|
|
|
|
// "type": "error",
|
|
|
|
|
// "id": "1",
|
|
|
|
|
// "payload": [{
|
|
|
|
|
// "message": "TestError",
|
|
|
|
|
// "locations": [{"line": 1, "column": 25}],
|
|
|
|
|
// "path": ["events", "value"],
|
|
|
|
|
// }],
|
|
|
|
|
// })),
|
|
|
|
|
// serde_json::from_slice(&stream.next().await.unwrap()).unwrap()
|
|
|
|
|
// );
|
|
|
|
|
// }
|
|
|
|
|
//
|
|
|
|
|
// #[async_std::test]
|
|
|
|
|
// pub async fn test_query_over_websocket() {
|
|
|
|
|
// struct QueryRoot;
|
|
|
|
|
//
|
|
|
|
|
// #[Object]
|
|
|
|
|
// impl QueryRoot {
|
|
|
|
|
// async fn value(&self) -> i32 {
|
|
|
|
|
// 999
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
//
|
|
|
|
|
// let schema = Schema::new(QueryRoot, EmptyMutation, EmptySubscription);
|
|
|
|
|
// let (mut sink, mut stream) = schema.subscription_connection(WebSocketTransport::default());
|
|
|
|
|
//
|
|
|
|
|
// sink.send(
|
|
|
|
|
// serde_json::to_vec(&serde_json::json!({
|
|
|
|
|
// "type": "connection_init",
|
|
|
|
|
// }))
|
|
|
|
|
// .unwrap(),
|
|
|
|
|
// )
|
|
|
|
|
// .await
|
|
|
|
|
// .unwrap();
|
|
|
|
|
//
|
|
|
|
|
// assert_eq!(
|
|
|
|
|
// Some(serde_json::json!({
|
|
|
|
|
// "type": "connection_ack",
|
|
|
|
|
// })),
|
|
|
|
|
// serde_json::from_slice(&stream.next().await.unwrap()).unwrap()
|
|
|
|
|
// );
|
|
|
|
|
//
|
|
|
|
|
// sink.send(
|
|
|
|
|
// serde_json::to_vec(&serde_json::json!({
|
|
|
|
|
// "type": "start",
|
|
|
|
|
// "id": "1",
|
|
|
|
|
// "payload": {
|
|
|
|
|
// "query": "query { value }"
|
|
|
|
|
// },
|
|
|
|
|
// }))
|
|
|
|
|
// .unwrap(),
|
|
|
|
|
// )
|
|
|
|
|
// .await
|
|
|
|
|
// .unwrap();
|
|
|
|
|
//
|
|
|
|
|
// assert_eq!(
|
|
|
|
|
// Some(serde_json::json!({
|
|
|
|
|
// "type": "data",
|
|
|
|
|
// "id": "1",
|
|
|
|
|
// "payload": { "data": { "value": 999 } },
|
|
|
|
|
// })),
|
|
|
|
|
// serde_json::from_slice(&stream.next().await.unwrap()).unwrap()
|
|
|
|
|
// );
|
|
|
|
|
//
|
|
|
|
|
// assert_eq!(
|
|
|
|
|
// Some(serde_json::json!({
|
|
|
|
|
// "type": "complete",
|
|
|
|
|
// "id": "1",
|
|
|
|
|
// })),
|
|
|
|
|
// serde_json::from_slice(&stream.next().await.unwrap()).unwrap()
|
|
|
|
|
// );
|
|
|
|
|
// }
|
