@@ -21,7 +21,8 @@ use openshell_core::proto::{
2121 GetSandboxProviderEnvironmentRequest , GetSandboxProviderEnvironmentResponse , GetSandboxRequest ,
2222 HealthRequest , HealthResponse , ListProvidersRequest , ListProvidersResponse ,
2323 ListSandboxProvidersRequest , ListSandboxProvidersResponse , ListSandboxesRequest ,
24- ListSandboxesResponse , Provider , ProviderCredentialRefreshStatus , ProviderProfile ,
24+ ListSandboxesResponse , Provider , ProviderCredentialRefresh , ProviderCredentialRefreshStatus ,
25+ ProviderCredentialRefreshStrategy , ProviderProfile , ProviderProfileCredential ,
2526 ProviderResponse , RevokeSshSessionRequest , RevokeSshSessionResponse ,
2627 RotateProviderCredentialRequest , RotateProviderCredentialResponse , Sandbox , SandboxResponse ,
2728 SandboxStreamEvent , ServiceStatus , SupervisorMessage , UpdateProviderRequest ,
@@ -1008,11 +1009,28 @@ async fn provider_refresh_cli_run_functions_wire_requests() {
10081009#[ tokio:: test]
10091010async fn provider_create_allows_empty_credentials_for_gateway_refresh_profiles ( ) {
10101011 let ts = run_server ( ) . await ;
1012+ ts. state . profiles . lock ( ) . await . insert (
1013+ "custom-refresh" . to_string ( ) ,
1014+ ProviderProfile {
1015+ id : "custom-refresh" . to_string ( ) ,
1016+ display_name : "Custom Refresh" . to_string ( ) ,
1017+ credentials : vec ! [ ProviderProfileCredential {
1018+ name: "ACCESS_TOKEN" . to_string( ) ,
1019+ required: true ,
1020+ refresh: Some ( ProviderCredentialRefresh {
1021+ strategy: ProviderCredentialRefreshStrategy :: Oauth2RefreshToken as i32 ,
1022+ ..Default :: default ( )
1023+ } ) ,
1024+ ..Default :: default ( )
1025+ } ] ,
1026+ ..Default :: default ( )
1027+ } ,
1028+ ) ;
10111029
10121030 run:: provider_create (
10131031 & ts. endpoint ,
1014- "my-graph " ,
1015- "outlook " ,
1032+ "custom-refresh-provider " ,
1033+ "custom-refresh " ,
10161034 false ,
10171035 & [ ] ,
10181036 & [ ] ,
@@ -1022,8 +1040,8 @@ async fn provider_create_allows_empty_credentials_for_gateway_refresh_profiles()
10221040 . expect ( "provider create" ) ;
10231041
10241042 let stored = ts. state . providers . lock ( ) . await ;
1025- let provider = stored. get ( "my-graph " ) . expect ( "provider" ) ;
1026- assert_eq ! ( provider. r#type, "outlook " ) ;
1043+ let provider = stored. get ( "custom-refresh-provider " ) . expect ( "provider" ) ;
1044+ assert_eq ! ( provider. r#type, "custom-refresh " ) ;
10271045 assert ! ( provider. credentials. is_empty( ) ) ;
10281046}
10291047
0 commit comments