#![allow(dead_code)]
use oxcache::backend::l2::L2Backend;
use oxcache::config::{ClusterConfig, L2Config, RedisMode, SentinelConfig};
use secrecy::SecretString;
use std::time::Duration;
pub(crate) async fn create_l2_backend_with_real_redis() -> Result<L2Backend, String> {
let config = create_standalone_config();
L2Backend::new(&config).await.map_err(|e| e.to_string())
}
pub(crate) async fn test_redis_connection() -> Result<(), String> {
let backend = match create_l2_backend_with_real_redis().await {
Ok(b) => b,
Err(e) => return Err(format!("无法创建Redis连接: {}", e)),
};
let test_key = "oxcache:test:connection";
if let Err(e) = backend
.set_bytes(test_key, b"test".to_vec(), Some(60))
.await
{
return Err(format!("SET操作失败: {}", e));
}
let value_opt = match backend.get_bytes(test_key).await {
Ok(v) => v,
Err(e) => return Err(format!("GET操作失败: {}", e)),
};
let value = match value_opt {
Some(v) => v,
None => return Err("Redis返回空值".to_string()),
};
if &value != b"test" {
return Err("Redis返回的值不正确".to_string());
}
if let Err(e) = backend.delete(test_key).await {
return Err(format!("DELETE操作失败: {}", e));
}
Ok(())
}
#[allow(dead_code)]
pub fn create_standalone_config() -> L2Config {
L2Config {
mode: RedisMode::Standalone,
connection_string: SecretString::new("redis://127.0.0.1:6379".into()),
connection_timeout_ms: 5000,
command_timeout_ms: 5000,
password: None,
enable_tls: false,
sentinel: None,
cluster: None,
default_ttl: Some(3600),
max_key_length: 256,
max_value_size: 1024 * 1024 * 10,
}
}
pub fn create_cluster_config() -> L2Config {
L2Config {
mode: RedisMode::Cluster,
connection_string: SecretString::new("redis://127.0.0.1:7000".into()),
connection_timeout_ms: 5000,
command_timeout_ms: 5000,
password: None,
enable_tls: false,
sentinel: None,
cluster: Some(ClusterConfig {
nodes: vec![
"127.0.0.1:7000".to_string(),
"127.0.0.1:7001".to_string(),
"127.0.0.1:7002".to_string(),
"127.0.0.1:7003".to_string(),
"127.0.0.1:7004".to_string(),
"127.0.0.1:7005".to_string(),
],
}),
default_ttl: Some(3600),
max_key_length: 256,
max_value_size: 1024 * 1024 * 10,
}
}
pub fn create_sentinel_config() -> L2Config {
L2Config {
mode: RedisMode::Sentinel,
connection_string: SecretString::new("redis://127.0.0.1:26379".into()),
connection_timeout_ms: 5000,
command_timeout_ms: 5000,
password: None,
enable_tls: false,
sentinel: Some(SentinelConfig {
master_name: "mymaster".to_string(),
nodes: vec![
"127.0.0.1:26379".to_string(),
"127.0.0.1:26380".to_string(),
"127.0.0.1:26381".to_string(),
],
}),
cluster: None,
default_ttl: Some(3600),
max_key_length: 256,
max_value_size: 1024 * 1024 * 10,
}
}
pub async fn cleanup_test_keys(pattern: &str) -> Result<(), String> {
let backend = create_l2_backend_with_real_redis().await?;
if pattern.contains("oxcache:test:*") {
let test_keys = ["oxcache:test:integration", "oxcache:test:ha:standalone"];
for key in test_keys {
match backend.delete(key).await {
Ok(()) => tracing::debug!("已清理测试键: {}", key),
Err(_) => tracing::debug!("键不存在或删除失败: {}", key),
}
}
}
Ok(())
}
#[allow(dead_code)]
pub fn is_redis_available() -> bool {
std::env::var("OXCACHE_SKIP_REDIS_TESTS").is_err()
}
pub async fn is_redis_available_url(url: &str) -> bool {
let client = match redis::Client::open(url) {
Ok(c) => c,
Err(_) => return false,
};
match tokio::time::timeout(
Duration::from_secs(1),
client.get_multiplexed_async_connection(),
)
.await
{
Ok(Ok(_)) => true,
Ok(Err(e)) => !e.is_connection_refusal(),
_ => false,
}
}
pub async fn wait_for_redis(url: &str) -> bool {
let start = std::time::Instant::now();
let timeout = Duration::from_secs(30);
while start.elapsed() < timeout {
if is_redis_available_url(url).await {
return true;
}
tokio::time::sleep(Duration::from_millis(500)).await;
}
false
}
pub async fn wait_for_redis_cluster(urls: &[&str]) -> bool {
let start = std::time::Instant::now();
let timeout = Duration::from_secs(60);
while start.elapsed() < timeout {
let mut all_ready = true;
for url in urls {
if !is_redis_available_url(url).await {
all_ready = false;
break;
}
}
if all_ready {
let nodes: Vec<String> = urls.iter().map(|s| s.to_string()).collect();
match redis::cluster::ClusterClient::new(nodes) {
Ok(client) => match client.get_async_connection().await {
Ok(mut conn) => {
match redis::cmd("CLUSTER")
.arg("INFO")
.query_async::<String>(&mut conn)
.await
{
Ok(info) => {
if info.contains("cluster_state:ok") {
println!("Redis Cluster is ready.");
return true;
}
}
Err(e) => {
println!("Failed to query cluster info: {}", e);
}
}
}
Err(e) => {
println!("Failed to get cluster connection: {}", e);
}
},
Err(e) => {
println!("Failed to create cluster client: {}", e);
}
}
}
tokio::time::sleep(Duration::from_secs(2)).await;
}
println!("Timeout waiting for Redis Cluster to be ready.");
false
}
pub async fn wait_for_sentinel() -> bool {
let sentinel_urls = vec![
"redis://127.0.0.1:26379",
"redis://127.0.0.1:26380",
"redis://127.0.0.1:26381",
];
let start = std::time::Instant::now();
let timeout = Duration::from_secs(60);
while start.elapsed() < timeout {
let mut all_ready = true;
for url in &sentinel_urls {
if !wait_for_redis(url).await {
all_ready = false;
break;
}
}
if all_ready {
let client = redis::Client::open(sentinel_urls[0]).unwrap();
if let Ok(mut conn) = client.get_multiplexed_async_connection().await {
let result: Result<Vec<String>, _> = redis::cmd("SENTINEL")
.arg("masters")
.query_async(&mut conn)
.await;
if let Ok(masters) = result {
if masters.iter().any(|m| m.contains("mymaster")) {
println!("Redis Sentinel is ready.");
return true;
}
}
}
}
tokio::time::sleep(Duration::from_secs(2)).await;
}
println!("Timeout waiting for Redis Sentinel to be ready.");
false
}
pub async fn is_redis_available_default() -> bool {
let redis_url =
std::env::var("REDIS_URL").unwrap_or_else(|_| "redis://127.0.0.1:6379".to_string());
is_redis_available_url(&redis_url).await
}
#[macro_export]
macro_rules! test_with_redis {
($name:ident, $body:block) => {
#[tokio::test]
async fn $name() {
if !redis_test_utils::is_redis_available() {
println!("跳过测试: Redis不可用 (设置 OXCACHE_SKIP_REDIS_TESTS=1 跳过)");
return;
}
match redis_test_utils::test_redis_connection().await {
Ok(()) => {
println!("Redis连接成功，开始执行测试...");
$body
}
Err(e) => {
println!("跳过测试: Redis连接失败 - {}", e);
println!("请确保Redis容器正在运行: docker start redis-test");
}
}
}
};
}