feat(domain): init domain reader

unit-mesh · Aug 28, 2023 · 3e2be16 · 3e2be16
1 parent c84d2a8
commit 3e2be16
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 3 deletions.
diff --git a/_fixtures/domain-language/sample.csv → _fixtures/domain/sample.csv b/_fixtures/domain-language/sample.csv → _fixtures/domain/sample.csv
diff --git a/counit-server/src/domain/domain_loader.rs b/counit-server/src/domain/domain_loader.rs
@@ -1,3 +1,4 @@
+use std::path::Path;
 use crate::domain::domain_record::DomainRecord;
 
 pub struct DomainLoader {
@@ -11,7 +12,7 @@ impl DomainLoader {
         }
     }
 
-    pub fn load(&mut self, path: &str) {
+    pub fn load<P: AsRef<Path>>(&mut self, path: P) {
         use walkdir::WalkDir;
         for entry in WalkDir::new(path) {
             let entry = entry.unwrap();
@@ -50,4 +51,23 @@ impl DomainLoader {
             self.domain_records.push(record);
         });
     }
+}
+
+#[cfg(test)]
+mod tests {
+    use std::path::Path;
+    use super::*;
+
+    #[test]
+    fn load_csv() {
+        let model_dir = Path::new(env!("CARGO_MANIFEST_DIR")).parent()
+            .unwrap()
+            .join("_fixtures")
+            .join("domain");
+
+
+        let mut loader = DomainLoader::new();
+        loader.load(model_dir);
+        assert_eq!(loader.domain_records.len(), 29);
+    }
 }
diff --git a/counit-server/src/main.rs b/counit-server/src/main.rs
@@ -47,7 +47,7 @@ async fn main() -> anyhow::Result<()> {
         .nest("/agent", agent_api::router())
 
         // knowledge init
-        .nest("/translate/domain-language", translate_api::router())
+        .nest("/translate/domain", translate_api::router())
 
         //align to archguard api
         .nest("/scanner", archguard_api::router())

diff --git a/counit-server/src/server/agent_api.rs b/counit-server/src/server/agent_api.rs
@@ -1,4 +1,4 @@
-use axum::{body::HttpBody, extract::Query, Json, Router};
+use axum::{extract::Query, Json, Router};
 use axum::http::StatusCode;
 use serde::{Deserialize, Serialize};