@@ -8,16 +8,6 @@ use crate::{
8
8
} ;
9
9
use std:: sync:: { Arc , Mutex } ;
10
10
11
- // simple_cb is a simple callback that increments the passed atomic integer and returns a static IdpServerResponse
12
- async fn simple_cb ( call_count : Arc < Mutex < u32 > > ) -> crate :: error:: Result < oidc:: IdpServerResponse > {
13
- * call_count. lock ( ) . unwrap ( ) += 1 ;
14
- Ok ( oidc:: IdpServerResponse {
15
- access_token : tokio:: fs:: read_to_string ( "/tmp/tokens/test_user1" ) . await ?,
16
- expires : None ,
17
- refresh_token : None ,
18
- } )
19
- }
20
-
21
11
// Machine Callback tests
22
12
// Prose test 1.1 Single Principal Implicit Username
23
13
#[ tokio:: test]
@@ -38,7 +28,16 @@ async fn machine_single_principal_implicit_username() -> anyhow::Result<()> {
38
28
opts. credential = Credential :: builder ( )
39
29
. mechanism ( AuthMechanism :: MongoDbOidc )
40
30
. oidc_callback ( oidc:: Callback :: machine ( move |_| {
41
- { simple_cb ( cb_call_count. clone ( ) ) } . boxed ( )
31
+ let call_count = cb_call_count. clone ( ) ;
32
+ async move {
33
+ * call_count. lock ( ) . unwrap ( ) += 1 ;
34
+ Ok ( oidc:: IdpServerResponse {
35
+ access_token : tokio:: fs:: read_to_string ( "/tmp/tokens/test_user1" ) . await ?,
36
+ expires : None ,
37
+ refresh_token : None ,
38
+ } )
39
+ }
40
+ . boxed ( )
42
41
} ) )
43
42
. build ( )
44
43
. into ( ) ;
@@ -80,7 +79,16 @@ async fn human_single_principal_implicit_username() -> anyhow::Result<()> {
80
79
opts. credential = Credential :: builder ( )
81
80
. mechanism ( AuthMechanism :: MongoDbOidc )
82
81
. oidc_callback ( oidc:: Callback :: human ( move |_| {
83
- { simple_cb ( cb_call_count. clone ( ) ) } . boxed ( )
82
+ let call_count = cb_call_count. clone ( ) ;
83
+ async move {
84
+ * call_count. lock ( ) . unwrap ( ) += 1 ;
85
+ Ok ( oidc:: IdpServerResponse {
86
+ access_token : tokio:: fs:: read_to_string ( "/tmp/tokens/test_user1" ) . await ?,
87
+ expires : None ,
88
+ refresh_token : None ,
89
+ } )
90
+ }
91
+ . boxed ( )
84
92
} ) )
85
93
. build ( )
86
94
. into ( ) ;
0 commit comments