chore(query): enable lib test (#15397)

* chore(query): enable lib test

* chore(query): enable lib test for all crate

* chore(query): enable lib test for all crate

Author: zhang2014

src/common/arrow/Cargo.toml

@@ -8,7 +8,7 @@ description = "Arrow implementation forked from arrow2 and native format impleme
 
 [lib]
 doctest = false
-test = false
+test = true
 
 [features]
 default = ["arrow-default", "parquet-default"]

src/common/auth/Cargo.toml

@@ -8,7 +8,7 @@ edition = { workspace = true }
 
 [lib]
 doctest = false
-test = false
+test = true
 
 [dependencies]
 base64 = "0.21.0"

src/common/base/Cargo.toml

@@ -8,7 +8,7 @@ edition = { workspace = true }
 
 [lib]
 doctest = false
-test = false
+test = true
 
 [features]
 tracing = ["tokio/tracing"]

src/common/base/src/runtime/memory/mem_stat.rs

@@ -98,7 +98,7 @@ impl MemStat {
         let mut used = self.used.fetch_add(batch_memory_used, Ordering::Relaxed);
 
         used += batch_memory_used;
-        self.peak_used.fetch_max(used, Ordering::Relaxed);
+        let old_peak_used = self.peak_used.fetch_max(used, Ordering::Relaxed);
 
         for (idx, parent_memory_stat) in self.parent_memory_stat.iter().enumerate() {
             if let Err(cause) = parent_memory_stat
@@ -108,6 +108,11 @@ impl MemStat {
                     // We only roll back the memory that alloc failed
                     self.used.fetch_sub(current_memory_alloc, Ordering::Relaxed);
 
+                    if used > old_peak_used {
+                        self.peak_used
+                            .fetch_sub(current_memory_alloc, Ordering::Relaxed);
+                    }
+
                     for index in 0..idx {
                         self.parent_memory_stat[index].rollback(current_memory_alloc);
                     }
@@ -119,6 +124,11 @@ impl MemStat {
 
         if let Err(cause) = self.check_limit(used) {
             if NEED_ROLLBACK {
+                if used > old_peak_used {
+                    self.peak_used
+                        .fetch_sub(current_memory_alloc, Ordering::Relaxed);
+                }
+
                 // NOTE: we cannot rollback peak_used of parent mem stat in this case
                 // self.peak_used.store(peak_used, Ordering::Relaxed);
                 self.rollback(current_memory_alloc);

src/common/building/Cargo.toml

@@ -8,7 +8,7 @@ edition = { workspace = true }
 
 [lib]
 doctest = false
-test = false
+test = true
 
 [dependencies]
 anyhow = { workspace = true }

src/common/cache/Cargo.toml

@@ -8,7 +8,7 @@ edition = { workspace = true }
 
 [lib]
 doctest = false
-test = false
+test = true
 
 [features]
 heapsize = ["heapsize_"]

src/common/cloud_control/Cargo.toml

@@ -8,7 +8,7 @@ edition = { workspace = true }
 
 [lib]
 doctest = false
-test = false
+test = true
 
 [dependencies]
 async-backtrace = { workspace = true }

src/common/exception/Cargo.toml

@@ -8,7 +8,7 @@ edition = { workspace = true }
 
 [lib]
 doctest = false
-test = false
+test = true
 
 [dependencies] # In alphabetical order
 databend-common-arrow = { path = "../arrow" }

src/common/grpc/Cargo.toml

@@ -8,7 +8,7 @@ edition = { workspace = true }
 
 [lib]
 doctest = false
-test = false
+test = true
 
 [dependencies] # In alphabetical order
 # Workspace dependencies

src/common/hashtable/Cargo.toml

@@ -8,7 +8,7 @@ edition = { workspace = true }
 
 [lib]
 doctest = false
-test = false
+test = true
 
 [dependencies] # In alphabetical order
 # Workspace dependencies