fix stun test and example

webrtc-rs · Jun 2, 2024 · 6d7794d · 6d7794d
1 parent 4bee644
commit 6d7794d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/rtc-stun/Cargo.toml b/rtc-stun/Cargo.toml
@@ -27,7 +27,7 @@ ring = "0.17.8"
 md5 = "0.7.0"
 
 [dev-dependencies]
-clap = "4.5.2"
+clap = { version = "4.5.2", features = ["derive"] }
 criterion = "0.5.1"
 
 [[bench]]

diff --git a/rtc-stun/src/integrity/integrity_test.rs b/rtc-stun/src/integrity/integrity_test.rs
@@ -6,6 +6,18 @@ use crate::textattrs::TextAttribute;
 
 #[test]
 fn test_message_integrity_add_to_simple() -> Result<()> {
+    {
+        let i = MessageIntegrity::new_long_term_integrity(
+            "user".to_owned(),
+            "realm".to_owned(),
+            "passsss".to_owned(),
+        );
+        let expected = vec![
+            104, 228, 91, 113, 61, 154, 222, 34, 101, 61, 181, 146, 177, 90, 4, 29,
+        ];
+        assert_eq!(i.0, expected, "{}", Error::ErrIntegrityMismatch);
+    }
+
     let i = MessageIntegrity::new_long_term_integrity(
         "user".to_owned(),
         "realm".to_owned(),