| 99 | |
| 100 | #[tokio::test] |
| 101 | async fn test_node_update() { |
| 102 | let storage = Arc::new(MemoryStorage::new()); |
| 103 | |
| 104 | let mut node = create_test_node("test-update-node"); |
| 105 | let key = build_key("nodes", None, "test-update-node"); |
| 106 | |
| 107 | // Create |
| 108 | storage.create(&key, &node).await.unwrap(); |
| 109 | |
| 110 | // Update to unschedulable |
| 111 | node.spec.as_mut().unwrap().unschedulable = Some(true); |
| 112 | let updated: Node = storage.update(&key, &node).await.unwrap(); |
| 113 | assert_eq!(updated.spec.as_ref().unwrap().unschedulable, Some(true)); |
| 114 | |
| 115 | // Verify update |
| 116 | let retrieved: Node = storage.get(&key).await.unwrap(); |
| 117 | assert_eq!(retrieved.spec.as_ref().unwrap().unschedulable, Some(true)); |
| 118 | |
| 119 | // Clean up |
| 120 | storage.delete(&key).await.unwrap(); |
| 121 | } |
| 122 | |
| 123 | #[tokio::test] |
| 124 | async fn test_node_delete() { |