@@ -45,19 +45,19 @@ async fn test_backendai_env_auto_detection() {
4545 . await
4646 . expect ( "Config should load with Backend.AI env" ) ;
4747
48- // Check that backendai cluster was created
49- assert ! ( config. clusters. contains_key( "backendai " ) ) ;
48+ // Check that bai_auto cluster was created
49+ assert ! ( config. clusters. contains_key( "bai_auto " ) ) ;
5050
51- // Get the backendai cluster
52- let cluster = config. clusters . get ( "backendai " ) . unwrap ( ) ;
51+ // Get the bai_auto cluster
52+ let cluster = config. clusters . get ( "bai_auto " ) . unwrap ( ) ;
5353
5454 // Verify nodes were parsed correctly
5555 assert_eq ! ( cluster. nodes. len( ) , 3 ) ;
5656
57- // Resolve nodes for the backendai cluster
57+ // Resolve nodes for the bai_auto cluster
5858 let nodes = config
59- . resolve_nodes ( "backendai " )
60- . expect ( "Should resolve backendai nodes" ) ;
59+ . resolve_nodes ( "bai_auto " )
60+ . expect ( "Should resolve bai_auto nodes" ) ;
6161 assert_eq ! ( nodes. len( ) , 3 ) ;
6262
6363 // Check node details
@@ -114,11 +114,11 @@ async fn test_backendai_env_with_single_host() {
114114 . await
115115 . expect ( "Config should load" ) ;
116116
117- // Verify backendai cluster exists
118- assert ! ( config. clusters. contains_key( "backendai " ) ) ;
117+ // Verify bai_auto cluster exists
118+ assert ! ( config. clusters. contains_key( "bai_auto " ) ) ;
119119
120120 let nodes = config
121- . resolve_nodes ( "backendai " )
121+ . resolve_nodes ( "bai_auto " )
122122 . expect ( "Should resolve nodes" ) ;
123123 assert_eq ! ( nodes. len( ) , 1 ) ;
124124 assert_eq ! ( nodes[ 0 ] . host, "single-node.ai" ) ;
0 commit comments