integration-tests: add test-utils feature for key-value-storage

The test cases in integration-test shares a same process, while the
storage registry is a static variable thus the lifetime is the same as
the whole process.

We need to do clear operation everytime we do a new test item.

Signed-off-by: Xynnn007 <xynnn@linux.alibaba.com>

Author: Xynnn007

deps/key-value-storage/Cargo.toml

@@ -7,6 +7,7 @@ edition = "2021"
 default = ["postgres", "redis"]
 postgres = ["sqlx/postgres", "pg-connection-string"]
 redis = ["dep:redis"]
+test-utils = []
 
 [dependencies]
 anyhow.workspace = true

deps/key-value-storage/src/global.rs

@@ -49,3 +49,13 @@ pub async fn get_namespace(namespace: &str) -> Result<KeyValueStorageInstance> {
     })?;
     Ok(storage)
 }
+
+#[cfg(feature = "test-utils")]
+pub async fn clear_all_namespaces() -> Result<()> {
+    let registry = REGISTRY.get_or_init(|| RegistryInner {
+        storages: RwLock::new(HashMap::new()),
+    });
+    let mut storages = registry.storages.write().await;
+    storages.clear();
+    Ok(())
+}

integration-tests/Cargo.toml

@@ -8,7 +8,9 @@ edition.workspace = true
 
 [dependencies]
 kbs = { path = "../kbs" }
-key-value-storage = { path = "../deps/key-value-storage" }
+key-value-storage = { path = "../deps/key-value-storage", features = [
+    "test-utils",
+] }
 policy-engine = { path = "../deps/policy-engine" }
 reference-value-provider-service = { path = "../rvps" }
 

integration-tests/src/common.rs

@@ -202,6 +202,9 @@ impl TestHarness {
     }
 
     pub async fn new(test_parameters: TestParameters) -> Result<TestHarness> {
+        // integration-tests is a shared test environment, so we need to clear all the namespaces to avoid conflicts between tests.
+        key_value_storage::global::clear_all_namespaces().await?;
+
         let auth_keypair = PKey::generate_ed25519()?;
         let auth_pubkey = String::from_utf8(auth_keypair.public_key_to_pem()?)?;
         let auth_privkey = String::from_utf8(auth_keypair.private_key_to_pem_pkcs8()?)?;