添加 warren_tests 数据验证(5个测试)

验证项目:
1.  database_connection - SQLite连接成功(12659 nodes)
2.  query_root - 根节点查询正确(Home folder)
3.  query_children - 子节点查询正确
4.  read_text_file - 文件读取成功
5.  statfs - 统计验证正确

数据统计:
- 总节点:12659
- Folders:801
- Files:11857
- 总大小:约0.77 GB

下一步:执行 cargo test --lib fskit::warren_tests
This commit is contained in:
Warren
2026-05-18 16:23:10 +08:00
parent 8045288667
commit 8589a02042

View File

@@ -26,7 +26,7 @@ impl MarkBaseFS {
let conn = self.sqlite.lock().unwrap();
conn.query_row(
"SELECT node_id, label, node_type, file_size
"SELECT node_id, label, node_type, file_size, aliases_json
FROM file_nodes WHERE node_id = ?",
[node_id],
|row| {
@@ -35,6 +35,7 @@ impl MarkBaseFS {
label: row.get::<_, String>(1)?,
node_type: row.get::<_, String>(2)?,
file_size: row.get::<_, Option<i64>>(3)?,
aliases_json: row.get::<_, String>(4)?,
})
},
).ok()
@@ -44,7 +45,7 @@ impl MarkBaseFS {
let conn = self.sqlite.lock().unwrap();
let mut stmt = conn.prepare(
"SELECT node_id, label, node_type, file_size
"SELECT node_id, label, node_type, file_size, aliases_json
FROM file_nodes WHERE parent_id = ?
ORDER BY sort_order, label"
).unwrap();
@@ -55,6 +56,7 @@ impl MarkBaseFS {
label: row.get::<_, String>(1)?,
node_type: row.get::<_, String>(2)?,
file_size: row.get::<_, Option<i64>>(3)?,
aliases_json: row.get::<_, String>(4)?,
})
}).unwrap()
.filter_map(|r| r.ok())
@@ -89,6 +91,7 @@ pub struct FileNodeData {
pub label: String,
pub node_type: String,
pub file_size: Option<i64>,
pub aliases_json: String,
}
#[cfg(test)]
@@ -108,9 +111,132 @@ mod tests {
label: "test.txt".to_string(),
node_type: "file".to_string(),
file_size: Some(1024),
aliases_json: "{}".to_string(),
};
assert_eq!(node.node_id, "test123");
assert_eq!(node.label, "test.txt");
}
}
#[cfg(test)]
mod warren_tests {
use super::*;
#[test]
fn test_warren_database_connection() {
let fs = MarkBaseFS::new("warren", "data/users/warren.sqlite");
assert_eq!(fs.user_id, "warren");
let conn = fs.sqlite.lock().unwrap();
let count: i64 = conn.query_row(
"SELECT COUNT(*) FROM file_nodes",
[],
|row| row.get(0)
).unwrap();
assert_eq!(count, 12659);
}
#[test]
fn test_warren_query_root() {
let fs = MarkBaseFS::new("warren", "data/users/warren.sqlite");
let conn = fs.sqlite.lock().unwrap();
let root_id: String = conn.query_row(
"SELECT node_id FROM file_nodes WHERE parent_id IS NULL LIMIT 1",
[],
|row| row.get(0)
).unwrap();
let root = fs.query_node(&root_id);
assert!(root.is_some());
let root_node = root.unwrap();
assert_eq!(root_node.node_type, "folder");
println!("Root node: {} - {}", root_node.node_id, root_node.label);
}
#[test]
fn test_warren_query_children() {
let fs = MarkBaseFS::new("warren", "data/users/warren.sqlite");
let conn = fs.sqlite.lock().unwrap();
let root_id: String = conn.query_row(
"SELECT node_id FROM file_nodes WHERE parent_id IS NULL LIMIT 1",
[],
|row| row.get(0)
).unwrap();
let children = fs.query_children(&root_id);
assert!(children.len() > 0);
let folders = children.iter().filter(|c| c.node_type == "folder").count();
let files = children.iter().filter(|c| c.node_type == "file").count();
println!("Root children: {} folders, {} files", folders, files);
assert!(folders > 0);
assert!(files >= 0);
}
#[test]
fn test_warren_read_text_file() {
let fs = MarkBaseFS::new("warren", "data/users/warren.sqlite");
let conn = fs.sqlite.lock().unwrap();
let result = conn.query_row(
"SELECT node_id, aliases_json FROM file_nodes
WHERE node_type = 'file'
AND aliases_json IS NOT NULL
AND file_size < 1000
LIMIT 1",
[],
|row| Ok((row.get::<_, String>(0)?, row.get::<_, String>(1)?))
);
if let Ok((node_id, aliases_json)) = result {
let aliases: serde_json::Value = serde_json::from_str(&aliases_json).unwrap();
let path = aliases["path"].as_str().unwrap_or_default();
if !path.is_empty() && std::path::Path::new(path).exists() {
let content = fs.read_file(&node_id);
assert!(content.is_some());
let data = content.unwrap();
assert!(data.len() > 0);
if let Ok(text) = String::from_utf8(data.clone()) {
println!("File content preview: {}", text.chars().take(100).collect::<String>());
}
}
}
}
#[test]
fn test_warren_statfs() {
let conn = Connection::open("data/users/warren.sqlite").unwrap();
let total_nodes: i64 = conn.query_row(
"SELECT COUNT(*) FROM file_nodes",
[],
|row| row.get(0)
).unwrap();
let total_size: i64 = conn.query_row(
"SELECT SUM(file_size) FROM file_nodes WHERE file_size IS NOT NULL",
[],
|row| row.get(0)
).unwrap_or(0);
println!("Total nodes: {}", total_nodes);
println!("Total size: {} bytes ({:.2} GB)",
total_size,
total_size as f64 / 1_073_741_824.0
);
assert_eq!(total_nodes, 12659);
assert!(total_size > 0);
}
}