use serde_json::json; use sqlx::{Pool, Postgres}; use windmill_test_utils::*; fn variable_url(port: u16, endpoint: &str, path: &str) -> String { format!("http://localhost:{port}/api/w/test-workspace/variables/{endpoint}/{path}") } fn client() -> reqwest::Client { reqwest::Client::new() } fn authed(builder: reqwest::RequestBuilder) -> reqwest::RequestBuilder { builder.header("Authorization", "Bearer SECRET_TOKEN") } async fn authed_get(port: u16, endpoint: &str, path: &str) -> reqwest::Response { authed(client().get(variable_url(port, endpoint, path))) .send() .await .unwrap() } #[sqlx::test(migrations = "../migrations", fixtures("base", "variables_test"))] async fn test_variable_endpoints(db: Pool) -> anyhow::Result<()> { initialize_tracing().await; let server = ApiServer::start(db.clone()).await?; let port = server.addr.port(); let base = format!("http://localhost:{port}/api/w/test-workspace/variables"); // --- exists --- let resp = authed_get(port, "exists", "u/test-user/plain_var").await; assert_eq!(resp.status(), 200); assert_eq!(resp.json::().await?, true); let resp = authed_get(port, "exists", "u/test-user/nonexistent").await; assert_eq!(resp.status(), 200); assert_eq!(resp.json::().await?, false); // --- get (plain) --- let resp = authed_get(port, "get", "u/test-user/plain_var").await; assert_eq!(resp.status(), 200); let body = resp.json::().await?; assert_eq!(body["path"], "u/test-user/plain_var"); assert_eq!(body["value"], "hello world"); assert_eq!(body["is_secret"], false); assert_eq!(body["description"], "A plain variable"); // --- get (secret, decrypt_secret=false) --- // fixture secrets are stored as plaintext, so skip decryption let resp = authed(client().get(format!( "{}/get/u/test-user/secret_var?decrypt_secret=false", base ))) .send() .await .unwrap(); assert_eq!(resp.status(), 200); let body = resp.json::().await?; assert_eq!(body["path"], "u/test-user/secret_var"); assert_eq!(body["is_secret"], true); assert_eq!(body["value"], serde_json::Value::Null); // get (secret, include_encrypted=true returns raw stored value) let resp = authed(client().get(format!( "{}/get/u/test-user/secret_var?decrypt_secret=false&include_encrypted=true", base ))) .send() .await .unwrap(); assert_eq!(resp.status(), 200); let body = resp.json::().await?; assert_eq!(body["is_secret"], true); assert!(body["value"].is_string(), "expected encrypted value string"); // --- get not found --- let resp = authed_get(port, "get", "u/test-user/nonexistent").await; assert_eq!(resp.status(), 404); // --- get_value (plain) --- let resp = authed_get(port, "get_value", "u/test-user/plain_var").await; assert_eq!(resp.status(), 200); assert_eq!(resp.json::().await?, "hello world"); let resp = authed_get(port, "get_value", "u/test-user/nonexistent").await; assert_eq!(resp.status(), 404); // --- list --- let resp = authed(client().get(format!("{base}/list"))) .send() .await .unwrap(); assert_eq!(resp.status(), 200); let list = resp.json::>().await?; assert!( list.len() >= 3, "expected at least 3 variables from fixture, got {}", list.len() ); assert!(list.iter().any(|v| v["path"] == "u/test-user/plain_var")); // secrets should have null value in list let secret = list .iter() .find(|v| v["path"] == "u/test-user/secret_var") .expect("secret_var missing from list"); assert_eq!(secret["value"], serde_json::Value::Null); // list with path_start filter let resp = authed(client().get(format!( "{base}/list?path_start=u/test-user/plain" ))) .send() .await .unwrap(); assert_eq!(resp.status(), 200); let list = resp.json::>().await?; assert_eq!(list.len(), 1); assert_eq!(list[0]["path"], "u/test-user/plain_var"); // --- list_contextual --- let resp = authed(client().get(format!("{base}/list_contextual"))) .send() .await .unwrap(); assert_eq!(resp.status(), 200); let list = resp.json::>().await?; assert!(!list.is_empty()); // should contain reserved variable names like WM_TOKEN, WM_WORKSPACE assert!( list.iter().any(|v| v["name"] == "WM_WORKSPACE"), "expected WM_WORKSPACE in contextual variables" ); // --- create (plain) --- let resp = authed(client().post(format!("{base}/create"))) .json(&json!({ "path": "u/test-user/new_var", "value": "new_value", "is_secret": false, "description": "Created in test" })) .send() .await .unwrap(); assert_eq!(resp.status(), 201); let resp = authed_get(port, "exists", "u/test-user/new_var").await; assert_eq!(resp.json::().await?, true); let resp = authed_get(port, "get_value", "u/test-user/new_var").await; assert_eq!(resp.json::().await?, "new_value"); // create duplicate -> 400 let resp = authed(client().post(format!("{base}/create"))) .json(&json!({ "path": "u/test-user/new_var", "value": "dup", "is_secret": false, "description": "Duplicate" })) .send() .await .unwrap(); assert_eq!(resp.status(), 400); // --- create (secret) --- let resp = authed(client().post(format!("{base}/create"))) .json(&json!({ "path": "u/test-user/new_secret", "value": "my_secret_val", "is_secret": true, "description": "A new secret" })) .send() .await .unwrap(); assert_eq!(resp.status(), 201); // secret is stored encrypted, get_value should decrypt it let resp = authed_get(port, "get_value", "u/test-user/new_secret").await; assert_eq!(resp.status(), 200); assert_eq!(resp.json::().await?, "my_secret_val"); // --- update (value) --- let resp = authed(client().post(variable_url(port, "update", "u/test-user/new_var"))) .json(&json!({"value": "updated_value"})) .send() .await .unwrap(); assert_eq!(resp.status(), 200); let resp = authed_get(port, "get_value", "u/test-user/new_var").await; assert_eq!(resp.json::().await?, "updated_value"); // --- update (description) --- let resp = authed(client().post(variable_url(port, "update", "u/test-user/new_var"))) .json(&json!({"description": "Updated desc"})) .send() .await .unwrap(); assert_eq!(resp.status(), 200); let resp = authed_get(port, "get", "u/test-user/new_var").await; let body = resp.json::().await?; assert_eq!(body["description"], "Updated desc"); // --- delete --- let resp = authed(client().delete(variable_url(port, "delete", "u/test-user/new_var"))) .send() .await .unwrap(); assert_eq!(resp.status(), 200); let resp = authed_get(port, "exists", "u/test-user/new_var").await; assert_eq!(resp.json::().await?, false); // delete nonexistent -> 200 (no-op, doesn't error) let resp = authed(client().delete(variable_url(port, "delete", "u/test-user/new_var"))) .send() .await .unwrap(); assert_eq!(resp.status(), 200); // --- delete_bulk --- let resp = authed(client().delete(format!("{base}/delete_bulk"))) .json(&json!({"paths": ["u/test-user/new_secret", "u/test-user/another_var"]})) .send() .await .unwrap(); assert_eq!(resp.status(), 200); let deleted = resp.json::>().await?; assert_eq!(deleted.len(), 2); let resp = authed_get(port, "exists", "u/test-user/new_secret").await; assert_eq!(resp.json::().await?, false); let resp = authed_get(port, "exists", "u/test-user/another_var").await; assert_eq!(resp.json::().await?, false); // --- encrypt --- let resp = authed(client().post(format!("{base}/encrypt"))) .json(&"test_plaintext".to_string()) .send() .await .unwrap(); assert_eq!(resp.status(), 200); let encrypted = resp.text().await?; assert!(!encrypted.is_empty()); assert_ne!(encrypted, "test_plaintext"); Ok(()) }