From 281eea5df5c3512a870e7414affa2df7c59f7b2c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?K=C3=A9vin=20Commaille?= <zecakeh@tedomum.fr>
Date: Tue, 8 Aug 2023 11:19:32 +0200
Subject: [PATCH] Add tests for Matrix API scope tokens
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Kévin Commaille <zecakeh@tedomum.fr>
---
 crates/oidc-client/src/types/scope.rs | 39 +++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/crates/oidc-client/src/types/scope.rs b/crates/oidc-client/src/types/scope.rs
index 67c3ff88..0f13c07f 100644
--- a/crates/oidc-client/src/types/scope.rs
+++ b/crates/oidc-client/src/types/scope.rs
@@ -253,7 +253,46 @@ mod tests {
         assert_matches!(scope, ScopeToken::MatrixDevice(_));
         assert_eq!(scope.matrix_device_id(), Some("ABCDEFGHIJKL"));
 
+        let scope = ScopeToken::from_str("urn:matrix:org.matrix.msc2967.client:api:*").unwrap();
+        assert_eq!(scope, ScopeToken::MatrixApi(MatrixApiScopeToken::Full));
+
+        let scope = ScopeToken::from_str("urn:matrix:org.matrix.msc2967.client:api:guest").unwrap();
+        assert_eq!(scope, ScopeToken::MatrixApi(MatrixApiScopeToken::Guest));
+
+        let scope =
+            ScopeToken::from_str("urn:matrix:org.matrix.msc2967.client:api:my.custom.scope")
+                .unwrap();
+        let api_scope = assert_matches!(scope, ScopeToken::MatrixApi(s) => s);
+        assert_matches!(api_scope, MatrixApiScopeToken::Custom(_));
+        assert_eq!(api_scope.to_string(), "my.custom.scope");
+
         assert_eq!(ScopeToken::from_str("invalid\\scope"), Err(InvalidScope));
+        assert_eq!(
+            MatrixApiScopeToken::from_str("invalid\\scope"),
+            Err(InvalidScope)
+        );
+    }
+
+    #[test]
+    fn display_scope_token() {
+        let scope = ScopeToken::MatrixApi(MatrixApiScopeToken::Full);
+        assert_eq!(
+            scope.to_string(),
+            "urn:matrix:org.matrix.msc2967.client:api:*"
+        );
+
+        let scope = ScopeToken::MatrixApi(MatrixApiScopeToken::Guest);
+        assert_eq!(
+            scope.to_string(),
+            "urn:matrix:org.matrix.msc2967.client:api:guest"
+        );
+
+        let api_scope = MatrixApiScopeToken::from_str("my.custom.scope").unwrap();
+        let scope = ScopeToken::MatrixApi(api_scope);
+        assert_eq!(
+            scope.to_string(),
+            "urn:matrix:org.matrix.msc2967.client:api:my.custom.scope"
+        );
     }
 
     #[test]