moq-transport: Update SETUP message tests

englishm · Nov 3, 2024 · 54ae8d9 · 54ae8d9
1 parent 2151545
commit 54ae8d9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/moq-transport/src/setup/client.rs b/moq-transport/src/setup/client.rs
@@ -79,15 +79,15 @@ mod tests {
 	fn encode_decode() {
 		let mut buf = BytesMut::new();
 		let client = Client {
-			versions: [Version::DRAFT_03].into(),
+			versions: [Version::DRAFT_06].into(),
 			role: Role::Both,
 			params: Params::default(),
 		};
 
 		client.encode(&mut buf).unwrap();
 		assert_eq!(
 			buf.to_vec(),
-			vec![0x40, 0x40, 0x01, 0xC0, 0x00, 0x00, 0x00, 0xFF, 0x00, 0x00, 0x03, 0x01, 0x00, 0x01, 0x03]
+			vec![0x40, 0x40, 0x0D, 0x01, 0xC0, 0x00, 0x00, 0x00, 0xFF, 0x00, 0x00, 0x06, 0x01, 0x00, 0x01, 0x03]
 		);
 
 		let decoded = Client::decode(&mut buf).unwrap();

diff --git a/moq-transport/src/setup/server.rs b/moq-transport/src/setup/server.rs
@@ -74,20 +74,19 @@ mod tests {
 	use crate::setup::Role;
 	use bytes::BytesMut;
 
-	// TODO: Update this test to use the draft 06 version
 	#[test]
 	fn encode_decode() {
 		let mut buf = BytesMut::new();
 		let client = Server {
-			version: Version::DRAFT_03,
+			version: Version::DRAFT_06,
 			role: Role::Both,
 			params: Params::default(),
 		};
 
 		client.encode(&mut buf).unwrap();
 		assert_eq!(
 			buf.to_vec(),
-			vec![0x40, 0x41, 0xC0, 0x00, 0x00, 0x00, 0xFF, 0x00, 0x00, 0x03, 0x01, 0x00, 0x01, 0x03]
+			vec![0x40, 0x41, 0x0C, 0xC0, 0x00, 0x00, 0x00, 0xFF, 0x00, 0x00, 0x06, 0x01, 0x00, 0x01, 0x03]
 		);
 
 		let decoded = Server::decode(&mut buf).unwrap();