From 0108133876702e21203d9e02eb282b0ba93c43b4 Mon Sep 17 00:00:00 2001
From: KunoiSayami <46131041+KunoiSayami@users.noreply.github.com>
Date: Tue, 13 Jul 2021 15:42:39 +0800
Subject: feat(test): Add query_is_all_protected() method for test *
 feat(build): Add optimize option for build release

---
 src/datastructures.rs | 10 ++++++++++
 src/test.rs           |  6 ++----
 2 files changed, 12 insertions(+), 4 deletions(-)

(limited to 'src')

diff --git a/src/datastructures.rs b/src/datastructures.rs
index ba73d96..e004514 100644
--- a/src/datastructures.rs
+++ b/src/datastructures.rs
@@ -230,6 +230,11 @@ impl Config {
     pub(crate) fn get_white_list_mode_status(&self) -> bool {
         self.protect_config.get_white_list_mode_status()
     }
+
+    #[cfg(test)]
+    pub(crate) fn query_is_all_protected(&self) -> bool {
+        self.protect_config.query_is_all_protected()
+    }
 }
 
 impl TestSuite for Config {
@@ -379,6 +384,11 @@ impl ProtectSettings {
     pub(crate) fn get_white_list_mode_status(&self) -> bool {
         self.protect_white_list_mode
     }
+
+    #[cfg(test)]
+    pub(crate) fn query_is_all_protected(&self) -> bool {
+        self.protect_enabled && self.protect_white_list_mode && self.repos.is_empty()
+    }
 }
 
 impl TestSuite for ProtectSettings {
diff --git a/src/test.rs b/src/test.rs
index 68ba29d..8ae631a 100644
--- a/src/test.rs
+++ b/src/test.rs
@@ -323,8 +323,7 @@ mod core {
 
         assert!(cfg.check_repo_protect("test"), "struct: {:#?}", cfg);
         assert!(!cfg.get_white_list_mode_status());
-        // TODO:
-        //assert!(!cfg.query_is_all_protected());
+        assert!(!cfg.query_is_all_protected());
 
         write_to_specify_file(
             &tmpdir.path().join("REPO_SETTING"),
@@ -336,8 +335,7 @@ mod core {
 
         assert!(!cfg.check_repo_protect("test"));
         assert!(!cfg.get_white_list_mode_status());
-        // TODO:
-        //assert!(!cfg.query_is_all_protected());
+        assert!(!cfg.query_is_all_protected());
 
         tmpdir.close().unwrap();
     }
-- 
cgit v1.3.1