centra
8f94aee1fa
- Remove gitlinks (160000 mode) for chainfire, flaredb, iam - Add workspace contents as regular tracked files - Update flake.nix to use simple paths instead of builtins.fetchGit This resolves the nix build failure where submodule directories appeared empty in the nix store. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
71 lines
2.2 KiB
Rust
71 lines
2.2 KiB
Rust
use flaredb_server::config::{encode_namespaced_key, ServerConfig};
|
|
use flaredb_server::store::Store;
|
|
use flaredb_storage::rocks_engine::RocksEngine;
|
|
use flaredb_types::RegionMeta;
|
|
use std::collections::HashMap;
|
|
use std::sync::Arc;
|
|
|
|
fn test_peer_addrs() -> Arc<HashMap<u64, String>> {
|
|
let mut addrs = HashMap::new();
|
|
addrs.insert(1, "127.0.0.1:50051".to_string());
|
|
Arc::new(addrs)
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn split_creates_two_regions_and_routes_keys() {
|
|
let dir = tempfile::TempDir::new().unwrap();
|
|
let engine = Arc::new(RocksEngine::new(dir.path().to_str().unwrap()).unwrap());
|
|
let cfg = Arc::new(ServerConfig::default_with_namespace());
|
|
let ns = cfg.get_namespace("default");
|
|
let store = Arc::new(Store::new(1, engine, cfg.clone(), test_peer_addrs()));
|
|
|
|
store
|
|
.bootstrap_regions(vec![(
|
|
RegionMeta {
|
|
id: 1,
|
|
start_key: encode_namespaced_key(ns.id, b""),
|
|
end_key: Vec::new(),
|
|
},
|
|
vec![1],
|
|
)])
|
|
.await
|
|
.unwrap();
|
|
|
|
store
|
|
.split_region(1, encode_namespaced_key(ns.id, b"m"), 2, vec![1])
|
|
.await
|
|
.unwrap();
|
|
|
|
let k1 = encode_namespaced_key(ns.id, b"a");
|
|
let k2 = encode_namespaced_key(ns.id, b"z");
|
|
assert_eq!(store.route_key(&k1).await, Some(1));
|
|
assert_eq!(store.route_key(&k2).await, Some(2));
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn split_rejects_invalid_keys() {
|
|
let dir = tempfile::TempDir::new().unwrap();
|
|
let engine = Arc::new(RocksEngine::new(dir.path().to_str().unwrap()).unwrap());
|
|
let cfg = Arc::new(ServerConfig::default_with_namespace());
|
|
let ns = cfg.get_namespace("default");
|
|
let store = Arc::new(Store::new(1, engine, cfg.clone(), test_peer_addrs()));
|
|
|
|
store
|
|
.bootstrap_regions(vec![(
|
|
RegionMeta {
|
|
id: 1,
|
|
start_key: encode_namespaced_key(ns.id, b""),
|
|
end_key: Vec::new(),
|
|
},
|
|
vec![1],
|
|
)])
|
|
.await
|
|
.unwrap();
|
|
|
|
// split key before start
|
|
let err = store
|
|
.split_region(1, encode_namespaced_key(ns.id, b""), 2, vec![1])
|
|
.await
|
|
.unwrap_err();
|
|
assert!(err.contains("invalid"));
|
|
}
|