use redis::Commands;
use std::time::Duration;
use tokio::time::sleep;

#[tokio::test]
async fn test_setex_command() {
    let client = redis::Client::open("redis://127.0.0.1:6379/").unwrap();
    let mut con = client.get_connection().unwrap();

    // clean up any keys that may be present
    let _: () = redis::cmd("DEL").arg("test_setex_key").query(&mut con).unwrap_or(());

    // test SETEX - Set key values and specify expiration time (seconds)
    let result: String = redis::cmd("SETEX")
        .arg("test_setex_key")
        .arg(2)
        .arg("test_value")
        .query(&mut con)
        .unwrap();
    assert_eq!(result, "OK");

    // the verification value is present immediately
    let value: String = con.get("test_setex_key").unwrap();
    assert_eq!(value, "test_value");

    // wait for expiration
    sleep(Duration::from_secs(3)).await;

    // the verification key has expired
    let result: Option<String> = con.get("test_setex_key").ok();
    assert_eq!(result, None);
}

#[tokio::test]
async fn test_setex_overwrite_existing_key() {
    let client = redis::Client::open("redis://127.0.0.1:6379").unwrap();
    let mut con = client.get_connection().unwrap();

    // clean up any keys that may be present
    let _: () = redis::cmd("DEL").arg("test_setex_overwrite").query(&mut con).unwrap_or(());

    // set a key first
    let _: () = redis::cmd("SET").arg("test_setex_overwrite").arg("old_value").query(&mut con).unwrap();

    // Use SETEX to override and set a new expiration time
    let result: String = redis::cmd("SETEX")
        .arg("test_setex_overwrite")
        .arg(2)
        .arg("new_value")
        .query(&mut con)
        .unwrap();
    assert_eq!(result, "OK");

    // verify that the new value exists immediately
    let value: String = con.get("test_setex_overwrite").unwrap();
    assert_eq!(value, "new_value");

    // wait for expiration
    sleep(Duration::from_secs(3)).await;

    // the verification key has expired
    let result: Option<String> = con.get("test_setex_overwrite").ok();
    assert_eq!(result, None);
}