Skip to content

Commit fb47363

Browse files
committed
fix: update tests for renamed bai_auto cluster
1 parent ec359a9 commit fb47363

1 file changed

Lines changed: 10 additions & 10 deletions

File tree

tests/backendai_env_test.rs

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)