Page MenuHomePhorge
Files F33202331

D9200.1768545061.diff
No OneTemporary
Actions

Size
2 KB
Referenced Files
None
Subscribers
None

D9200.1768545061.diff
View Options

diff --git a/services/commtest/tests/identity_tunnelbroker_tests.rs b/services/commtest/tests/identity_tunnelbroker_tests.rs
--- a/services/commtest/tests/identity_tunnelbroker_tests.rs
+++ b/services/commtest/tests/identity_tunnelbroker_tests.rs
@@ -8,8 +8,8 @@
use client::identity_client_service_client::IdentityClientServiceClient;
use client::UploadOneTimeKeysRequest;
use commtest::identity::device::create_device;
-use futures_util::SinkExt;
use futures_util::StreamExt;
+use futures_util::{SinkExt, TryStreamExt};
use tokio_tungstenite::{connect_async, tungstenite::Message};
use tonic::transport::Endpoint;
use tonic::Request;
@@ -17,6 +17,62 @@
ConnectionInitializationMessage, DeviceTypes, RefreshKeyRequest,
};
+#[tokio::test]
+#[should_panic]
+async fn test_tunnelbroker_invalid_auth() {
+ let (mut socket, _) = connect_async("ws://localhost:51001")
+ .await
+ .expect("Can't connect");
+
+ let session_request = ConnectionInitializationMessage {
+ device_id: "".to_string(),
+ access_token: "".to_string(),
+ user_id: "".to_string(),
+ notify_token: None,
+ device_type: DeviceTypes::Keyserver,
+ device_app_version: None,
+ device_os: None,
+ };
+
+ let serialized_request = serde_json::to_string(&session_request)
+ .expect("Failed to serialize connection request");
+
+ socket
+ .send(Message::Text(serialized_request))
+ .await
+ .expect("Failed to send message");
+
+ socket.next().await.unwrap().unwrap();
+}
+
+#[tokio::test]
+async fn test_tunnelbroker_valid_auth() {
+ let (mut socket, _) = connect_async("ws://localhost:51001")
+ .await
+ .expect("Can't connect");
+
+ let device_info = create_device().await;
+ let session_request = ConnectionInitializationMessage {
+ device_id: device_info.device_id.to_string(),
+ access_token: device_info.access_token.to_string(),
+ user_id: device_info.user_id.to_string(),
+ notify_token: None,
+ device_type: DeviceTypes::Keyserver,
+ device_app_version: None,
+ device_os: None,
+ };
+
+ let serialized_request = serde_json::to_string(&session_request)
+ .expect("Failed to serialize connection request");
+
+ socket
+ .send(Message::Text(serialized_request))
+ .await
+ .expect("Failed to send message");
+
+ socket.next().await.unwrap().unwrap();
+}
+
#[tokio::test]
async fn test_refresh_keys_request_upon_depletion() {
let device_info = create_device().await;

File Metadata

Mime Type
text/plain
Expires
Fri, Jan 16, 6:31 AM (13 h, 52 m)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
5942470
Default Alt Text
D9200.1768545061.diff (2 KB)

Event Timeline