299 lines
11 KiB
Rust
299 lines
11 KiB
Rust
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
mod common;
|
|
|
|
use std::sync::Arc;
|
|
use std::time::Duration;
|
|
|
|
use async_trait::async_trait;
|
|
use chorus::errors::GatewayError;
|
|
use chorus::gateway::*;
|
|
use chorus::types::{
|
|
self, Channel, ChannelCreateSchema, ChannelModifySchema, GatewayReady, IntoShared,
|
|
RoleCreateModifySchema, RoleObject,
|
|
};
|
|
#[cfg(target_arch = "wasm32")]
|
|
use wasm_bindgen_test::*;
|
|
#[cfg(target_arch = "wasm32")]
|
|
wasm_bindgen_test_configure!(run_in_browser);
|
|
|
|
#[cfg(not(target_arch = "wasm32"))]
|
|
use tokio::time::sleep;
|
|
#[cfg(target_arch = "wasm32")]
|
|
use wasmtimer::tokio::sleep;
|
|
|
|
#[cfg_attr(target_arch = "wasm32", wasm_bindgen_test::wasm_bindgen_test)]
|
|
#[cfg_attr(not(target_arch = "wasm32"), tokio::test)]
|
|
/// Tests establishing a connection (hello and heartbeats) on the local gateway;
|
|
async fn test_gateway_establish() {
|
|
let bundle = common::setup().await;
|
|
|
|
let _: GatewayHandle = Gateway::spawn(bundle.urls.wss.clone(), GatewayOptions::default()).await.unwrap();
|
|
common::teardown(bundle).await
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
struct GatewayReadyObserver {
|
|
channel: tokio::sync::mpsc::Sender<()>,
|
|
}
|
|
|
|
#[async_trait]
|
|
impl Observer<GatewayReady> for GatewayReadyObserver {
|
|
async fn update(&self, _data: &GatewayReady) {
|
|
self.channel.send(()).await.unwrap();
|
|
}
|
|
}
|
|
|
|
#[cfg_attr(target_arch = "wasm32", wasm_bindgen_test::wasm_bindgen_test)]
|
|
#[cfg_attr(not(target_arch = "wasm32"), tokio::test)]
|
|
/// Tests establishing a connection and authenticating
|
|
async fn test_gateway_authenticate() {
|
|
let bundle = common::setup().await;
|
|
|
|
let gateway: GatewayHandle = Gateway::spawn(bundle.urls.wss.clone(), GatewayOptions::default()).await.unwrap();
|
|
|
|
let (ready_send, mut ready_receive) = tokio::sync::mpsc::channel(1);
|
|
|
|
let observer = Arc::new(GatewayReadyObserver {
|
|
channel: ready_send,
|
|
});
|
|
|
|
gateway
|
|
.events
|
|
.lock()
|
|
.await
|
|
.session
|
|
.ready
|
|
.subscribe(observer);
|
|
|
|
let mut identify = types::GatewayIdentifyPayload::common();
|
|
identify.token = bundle.user.token.clone();
|
|
|
|
gateway.send_identify(identify).await;
|
|
|
|
tokio::select! {
|
|
// Fail, we timed out waiting for it
|
|
() = sleep(Duration::from_secs(20)) => {
|
|
println!("Timed out waiting for event, failing..");
|
|
assert!(false);
|
|
}
|
|
// Success, we have received it
|
|
Some(_) = ready_receive.recv() => {}
|
|
};
|
|
|
|
common::teardown(bundle).await
|
|
}
|
|
|
|
#[cfg_attr(target_arch = "wasm32", wasm_bindgen_test::wasm_bindgen_test)]
|
|
#[cfg_attr(not(target_arch = "wasm32"), tokio::test)]
|
|
async fn test_self_updating_structs() {
|
|
// PRETTYFYME: This test is a bit of a mess, but it works. Ideally, each self-updating struct
|
|
// would have its own test.
|
|
let mut bundle = common::setup().await;
|
|
|
|
let received_channel = bundle
|
|
.user
|
|
.gateway
|
|
.observe_and_into_inner(bundle.channel.clone())
|
|
.await;
|
|
|
|
assert_eq!(received_channel, bundle.channel.read().unwrap().clone());
|
|
|
|
let modify_schema = ChannelModifySchema {
|
|
name: Some("selfupdating".to_string()),
|
|
..Default::default()
|
|
};
|
|
received_channel
|
|
.modify(modify_schema, None, &mut bundle.user)
|
|
.await
|
|
.unwrap();
|
|
assert_eq!(
|
|
bundle
|
|
.user
|
|
.gateway
|
|
.observe_and_into_inner(bundle.channel.clone())
|
|
.await
|
|
.name
|
|
.unwrap(),
|
|
"selfupdating".to_string()
|
|
);
|
|
|
|
let guild = bundle
|
|
.user
|
|
.gateway
|
|
.observe_and_into_inner(bundle.guild.clone())
|
|
.await;
|
|
assert!(guild.channels.is_none());
|
|
|
|
Channel::create(
|
|
&mut bundle.user,
|
|
guild.id,
|
|
None,
|
|
ChannelCreateSchema {
|
|
name: "selfupdating2".to_string(),
|
|
channel_type: Some(types::ChannelType::GuildText),
|
|
..Default::default()
|
|
},
|
|
)
|
|
.await
|
|
.unwrap();
|
|
|
|
let guild = bundle
|
|
.user
|
|
.gateway
|
|
.observe_and_into_inner(guild.into_shared())
|
|
.await;
|
|
assert!(guild.channels.is_some());
|
|
assert!(guild.channels.as_ref().unwrap().len() == 1);
|
|
|
|
common::teardown(bundle).await
|
|
}
|
|
|
|
#[cfg_attr(target_arch = "wasm32", wasm_bindgen_test::wasm_bindgen_test)]
|
|
#[cfg_attr(not(target_arch = "wasm32"), tokio::test)]
|
|
async fn test_recursive_self_updating_structs() {
|
|
// Setup
|
|
let mut bundle = common::setup().await;
|
|
let guild = bundle.guild.clone();
|
|
// Observe Guild, make sure it has no channels
|
|
let guild = bundle.user.gateway.observe(guild.clone()).await;
|
|
let inner_guild = guild.read().unwrap().clone();
|
|
assert!(inner_guild.roles.is_none());
|
|
// Create Role
|
|
let permissions = types::PermissionFlags::CONNECT | types::PermissionFlags::MANAGE_EVENTS;
|
|
let permissions = Some(permissions.to_string());
|
|
let mut role_create_schema: types::RoleCreateModifySchema = RoleCreateModifySchema {
|
|
name: Some("cool person".to_string()),
|
|
permissions,
|
|
hoist: Some(true),
|
|
icon: None,
|
|
unicode_emoji: Some("".to_string()),
|
|
mentionable: Some(true),
|
|
position: None,
|
|
color: None,
|
|
};
|
|
let guild_id = inner_guild.id;
|
|
let role = RoleObject::create(&mut bundle.user, guild_id, role_create_schema.clone())
|
|
.await
|
|
.unwrap();
|
|
// Watch role;
|
|
bundle
|
|
.user
|
|
.gateway
|
|
.observe(role.clone().into_shared())
|
|
.await;
|
|
// Update Guild and check for Guild
|
|
let inner_guild = guild.read().unwrap().clone();
|
|
assert!(inner_guild.roles.is_some());
|
|
// Update the Role
|
|
role_create_schema.name = Some("yippieee".to_string());
|
|
RoleObject::modify(&mut bundle.user, guild_id, role.id, role_create_schema)
|
|
.await
|
|
.unwrap();
|
|
let role_inner = bundle
|
|
.user
|
|
.gateway
|
|
.observe_and_into_inner(role.clone().into_shared())
|
|
.await;
|
|
assert_eq!(role_inner.name, "yippieee");
|
|
// Check if the change propagated
|
|
let guild = bundle.user.gateway.observe(bundle.guild.clone()).await;
|
|
let inner_guild = guild.read().unwrap().clone();
|
|
let guild_roles = inner_guild.roles;
|
|
let guild_role = guild_roles.unwrap();
|
|
let guild_role_inner = guild_role.get(0).unwrap().read().unwrap().clone();
|
|
assert_eq!(guild_role_inner.name, "yippieee".to_string());
|
|
common::teardown(bundle).await;
|
|
}
|
|
|
|
#[cfg_attr(target_arch = "wasm32", wasm_bindgen_test::wasm_bindgen_test)]
|
|
#[cfg_attr(not(target_arch = "wasm32"), test)]
|
|
fn test_error() {
|
|
let error = GatewayMessage("4000".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::Unknown);
|
|
let error = GatewayMessage("4001".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::UnknownOpcode);
|
|
let error = GatewayMessage("4002".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::Decode);
|
|
let error = GatewayMessage("4003".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::NotAuthenticated);
|
|
let error = GatewayMessage("4004".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::AuthenticationFailed);
|
|
let error = GatewayMessage("4005".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::AlreadyAuthenticated);
|
|
let error = GatewayMessage("4007".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::InvalidSequenceNumber);
|
|
let error = GatewayMessage("4008".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::RateLimited);
|
|
let error = GatewayMessage("4009".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::SessionTimedOut);
|
|
let error = GatewayMessage("4010".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::InvalidShard);
|
|
let error = GatewayMessage("4011".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::ShardingRequired);
|
|
let error = GatewayMessage("4012".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::InvalidAPIVersion);
|
|
let error = GatewayMessage("4013".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::InvalidIntents);
|
|
let error = GatewayMessage("4014".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::DisallowedIntents);
|
|
}
|
|
|
|
#[cfg_attr(target_arch = "wasm32", wasm_bindgen_test::wasm_bindgen_test)]
|
|
#[cfg_attr(not(target_arch = "wasm32"), test)]
|
|
fn test_error_message() {
|
|
let error = GatewayMessage("Unknown Error".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::Unknown);
|
|
let error = GatewayMessage("Unknown Opcode".to_string())
|
|
.error()
|
|
.unwrap();
|
|
assert_eq!(error, GatewayError::UnknownOpcode);
|
|
let error = GatewayMessage("Decode Error".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::Decode);
|
|
let error = GatewayMessage("Not Authenticated".to_string())
|
|
.error()
|
|
.unwrap();
|
|
assert_eq!(error, GatewayError::NotAuthenticated);
|
|
let error = GatewayMessage("Authentication Failed".to_string())
|
|
.error()
|
|
.unwrap();
|
|
assert_eq!(error, GatewayError::AuthenticationFailed);
|
|
let error = GatewayMessage("Already Authenticated".to_string())
|
|
.error()
|
|
.unwrap();
|
|
assert_eq!(error, GatewayError::AlreadyAuthenticated);
|
|
let error = GatewayMessage("Invalid Seq".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::InvalidSequenceNumber);
|
|
let error = GatewayMessage("Rate Limited".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::RateLimited);
|
|
let error = GatewayMessage("Session Timed Out".to_string())
|
|
.error()
|
|
.unwrap();
|
|
assert_eq!(error, GatewayError::SessionTimedOut);
|
|
let error = GatewayMessage("Invalid Shard".to_string()).error().unwrap();
|
|
assert_eq!(error, GatewayError::InvalidShard);
|
|
let error = GatewayMessage("Sharding Required".to_string())
|
|
.error()
|
|
.unwrap();
|
|
assert_eq!(error, GatewayError::ShardingRequired);
|
|
let error = GatewayMessage("Invalid API Version".to_string())
|
|
.error()
|
|
.unwrap();
|
|
assert_eq!(error, GatewayError::InvalidAPIVersion);
|
|
let error = GatewayMessage("Invalid Intent(s)".to_string())
|
|
.error()
|
|
.unwrap();
|
|
assert_eq!(error, GatewayError::InvalidIntents);
|
|
let error = GatewayMessage("Disallowed Intent(s)".to_string())
|
|
.error()
|
|
.unwrap();
|
|
assert_eq!(error, GatewayError::DisallowedIntents);
|
|
// Also test the dot thing
|
|
let error = GatewayMessage("Invalid Intent(s).".to_string())
|
|
.error()
|
|
.unwrap();
|
|
assert_eq!(error, GatewayError::InvalidIntents);
|
|
}
|