1818// </copyright> 
1919
2020using  NUnit . Framework ; 
21+ using  System . Linq ; 
2122using  System . Threading . Tasks ; 
2223
2324namespace  OpenQA . Selenium . BiDi . Browser ; 
@@ -38,12 +39,15 @@ public async Task CanGetUserContexts()
3839        var  userContext1  =  await  bidi . Browser . CreateUserContextAsync ( ) ; 
3940        var  userContext2  =  await  bidi . Browser . CreateUserContextAsync ( ) ; 
4041
41-         var  userContexts  =  await  bidi . Browser . GetUserContextsAsync ( ) ; 
42+         var  userContextsResult  =  await  bidi . Browser . GetUserContextsAsync ( ) ; 
4243
43-         Assert . That ( userContexts ,  Is . Not . Null ) ; 
44-         Assert . That ( userContexts ,  Has . Count . GreaterThanOrEqualTo ( 2 ) ) ; 
45-         Assert . That ( userContexts ,  Does . Contain ( userContext1 ) ) ; 
46-         Assert . That ( userContexts ,  Does . Contain ( userContext2 ) ) ; 
44+         Assert . That ( userContextsResult ,  Is . Not . Null ) ; 
45+         Assert . That ( userContextsResult ,  Has . Count . GreaterThanOrEqualTo ( 2 ) ) ; 
46+ 
47+         var  userContexts  =  userContextsResult . Select ( uc =>  uc . UserContext ) ; 
48+ 
49+         Assert . That ( userContexts ,  Does . Contain ( userContext1 . UserContext ) ) ; 
50+         Assert . That ( userContexts ,  Does . Contain ( userContext2 . UserContext ) ) ; 
4751    } 
4852
4953    [ Test ] 
@@ -54,10 +58,10 @@ public async Task CanRemoveUserContext()
5458
5559        await  userContext2 . UserContext . RemoveAsync ( ) ; 
5660
57-         var  userContexts  =  await  bidi . Browser . GetUserContextsAsync ( ) ; 
61+         var  userContexts  =  ( await  bidi . Browser . GetUserContextsAsync ( ) ) . Select ( uc  =>   uc . UserContext ) ; 
5862
59-         Assert . That ( userContexts ,  Does . Contain ( userContext1 ) ) ; 
60-         Assert . That ( userContexts ,  Does . Not . Contain ( userContext2 ) ) ; 
63+         Assert . That ( userContexts ,  Does . Contain ( userContext1 . UserContext ) ) ; 
64+         Assert . That ( userContexts ,  Does . Not . Contain ( userContext2 . UserContext ) ) ; 
6165    } 
6266
6367    [ Test ] 
0 commit comments