| 5c6e92fd Andy McCall ↶ |
1 |
|
| 2 |
|
| 3 |
|
| 4 |
|
| 5 |
use oxy_core::repo::{IdentityScope, Repo}; |
| 6 |
use oxygenhub::config::Config; |
| 7 |
use oxygenhub::repos::RepoName; |
| 8 |
use oxygenhub::{db, Shared}; |
| 9 |
use std::path::{Path, PathBuf}; |
| 10 |
|
| 11 |
const ROOTFS: &str = "36fc4e3238044e424219a2b9002bf61bbe1805fec14d52cf37e17fae77c9ff16"; |
| 12 |
|
| 13 |
fn hermetic() -> PathBuf { |
| 14 |
static DIR: std::sync::OnceLock<tempfile::TempDir> = std::sync::OnceLock::new(); |
| 15 |
let d = DIR.get_or_init(|| tempfile::tempdir().unwrap()); |
| 16 |
std::env::set_var("OXY_HOME", d.path()); |
| 17 |
d.path().to_path_buf() |
| 18 |
} |
| 19 |
|
| 20 |
struct Hub { |
| 21 |
app: Shared, |
| 22 |
base: String, |
| 23 |
_dir: tempfile::TempDir, |
| 24 |
} |
| 25 |
|
| 26 |
async fn hub() -> Hub { |
| 27 |
hermetic(); |
| 28 |
let dir = tempfile::tempdir().unwrap(); |
| 29 |
let mut config = Config::load(dir.path()).unwrap(); |
| 30 |
config.auth.push_tokens = vec!["site".into()]; |
| 31 |
config.flow.repos = vec!["andy/*".into()]; |
| 32 |
let listener = tokio::net::TcpListener::bind("127.0.0.1:0").await.unwrap(); |
| 33 |
let port = listener.local_addr().unwrap().port(); |
| 34 |
config.server.base_url = format!("http://127.0.0.1:{port}"); |
| 35 |
let app = oxygenhub::build(config).await.unwrap(); |
| 36 |
let router = oxygenhub::router(app.clone()); |
| 37 |
tokio::spawn(async move { axum_serve(listener, router).await }); |
| 38 |
Hub { app, base: format!("http://127.0.0.1:{port}"), _dir: dir } |
| 39 |
} |
| 40 |
|
| 41 |
async fn axum_serve(listener: tokio::net::TcpListener, router: axum::Router) { |
| 42 |
axum::serve(listener, router).await.unwrap(); |
| 43 |
} |
| 44 |
|
| 45 |
|
| 46 |
async fn oh(dir: &Path, env: &[(&str, &str)], args: &[&str]) -> (i32, String, String) { |
| 47 |
let (dir, env, args): (PathBuf, Vec<(String, String)>, Vec<String>) = ( |
| 48 |
dir.to_path_buf(), |
| 49 |
env.iter().map(|(k, v)| (k.to_string(), v.to_string())).collect(), |
| 50 |
args.iter().map(|a| a.to_string()).collect(), |
| 51 |
); |
| 52 |
tokio::task::spawn_blocking(move || { |
| 53 |
let out = std::process::Command::new(env!("CARGO_BIN_EXE_oh")) |
| 54 |
.args(&args) |
| 55 |
.current_dir(&dir) |
| 56 |
.env_remove("OH_TOKEN") |
| 57 |
.env_remove("OXY_TOKEN") |
| 58 |
.env_remove("OH_HUB") |
| 59 |
.envs(env) |
| 60 |
.output() |
| 61 |
.unwrap(); |
| 62 |
( |
| 63 |
out.status.code().unwrap_or(-1), |
| 64 |
String::from_utf8_lossy(&out.stdout).into_owned(), |
| 65 |
String::from_utf8_lossy(&out.stderr).into_owned(), |
| 66 |
) |
| 67 |
}) |
| 68 |
.await |
| 69 |
.unwrap() |
| 70 |
} |
| 71 |
|
| 72 |
fn write(root: &Path, rel: &str, content: &str) { |
| 73 |
let p = root.join(rel); |
| 74 |
std::fs::create_dir_all(p.parent().unwrap()).unwrap(); |
| 75 |
std::fs::write(p, content).unwrap(); |
| 76 |
} |
| 77 |
|
| 78 |
#[tokio::test(flavor = "multi_thread")] |
| 79 |
async fn oh_drives_issues_runs_and_releases() { |
| 80 |
let h = hub().await; |
| 81 |
|
| 82 |
db::accounts::create_account(&h.app.db, "andy", "[email protected]", "x", "salt", true).await.unwrap(); |
| 83 |
h.app.repos.create(&RepoName::parse("andy/demo").unwrap(), None).unwrap(); |
| 84 |
db::repos::create(&h.app.db, "andy", "demo", "A demo", hub_api::Visibility::Public).await.unwrap(); |
| 85 |
|
| 86 |
|
| 87 |
let work = tempfile::tempdir().unwrap(); |
| 88 |
let mut c = Repo::init(work.path()).unwrap(); |
| 89 |
c.config.set("remote.origin.url", &format!("{}/andy/demo", h.base)).unwrap(); |
| 90 |
c.config.set("remote.origin.token", "site").unwrap(); |
| 91 |
c.save_config().unwrap(); |
| 92 |
c.create_identity("Andy", "[email protected]", IdentityScope::Repo).unwrap(); |
| 93 |
write( |
| 94 |
work.path(), |
| 95 |
".flow/pipeline.kdl", |
| 96 |
&format!("schema-version 1\nsandbox \"alpine\" {{\n rootfs \"cas://blob/{ROOTFS}\"\n}}\njob \"test\" {{\n sandbox \"alpine\"\n exec \"true\"\n}}\n"), |
| 97 |
); |
| 98 |
c.snapshot("test").unwrap(); |
| 99 |
c.commit("Add a pipeline", &[]).unwrap(); |
| 100 |
let heads = c.bookmark_heads().unwrap(); |
| 101 |
c.tag_create("v1", &heads, "first").unwrap(); |
| 102 |
assert!(c.push("origin", false).unwrap().iter().all(|r| r.ok)); |
| 103 |
let w = work.path(); |
| 104 |
|
| 105 |
let (code, out, err) = oh(w, &[], &["repo", "view", "--plain"]).await; |
| 106 |
assert_eq!(code, 0, "{err}"); |
| 107 |
assert!(out.contains("andy/demo") && out.contains("A demo") && out.contains("oxy clone"), "{out}"); |
| 108 |
|
| 109 |
|
| 110 |
let (code, out, err) = oh(w, &[], &["issue", "create", "--title", "Broken thing", "--body", "It breaks."]).await; |
| 111 |
assert_eq!(code, 0, "{err}"); |
| 112 |
assert!(out.contains("opened #1"), "{out}"); |
| 113 |
let (_, out, _) = oh(w, &[], &["issue", "list", "--json"]).await; |
| 114 |
let list: serde_json::Value = serde_json::from_str(&out).unwrap(); |
| 115 |
assert_eq!((list[0]["number"].as_i64(), list[0]["title"].as_str()), (Some(1), Some("Broken thing"))); |
| 116 |
let (_, out, _) = oh(w, &[], &["issue", "comment", "1", "--body", "Seen it too."]).await; |
| 117 |
assert!(out.contains("commented on #1")); |
| 118 |
let (_, out, _) = oh(w, &[], &["issue", "view", "1", "--plain"]).await; |
| 119 |
assert!(out.contains("It breaks.") && out.contains("Seen it too."), "{out}"); |
| 120 |
let (_, out, _) = oh(w, &[], &["issue", "close", "1"]).await; |
| 121 |
assert!(out.contains("closed #1")); |
| 122 |
let (_, out, _) = oh(w, &[], &["issue", "list", "--plain"]).await; |
| 123 |
assert!(out.contains("No open issues."), "{out}"); |
| 124 |
|
| 125 |
|
| 126 |
let (code, out, err) = oh(w, &[], &["run", "start"]).await; |
| 127 |
assert_eq!(code, 0, "{err}"); |
| 128 |
assert!(out.contains("started run #2"), "{out}"); |
| 129 |
let (_, out, _) = oh(w, &[], &["run", "list", "--json"]).await; |
| 130 |
let runs: serde_json::Value = serde_json::from_str(&out).unwrap(); |
| 131 |
assert_eq!(runs.as_array().unwrap().len(), 2); |
| 132 |
let (_, out, _) = oh(w, &[], &["run", "view", "2", "--plain"]).await; |
| 133 |
assert!(out.contains("test") && out.contains("#2"), "{out}"); |
| 134 |
|
| 135 |
|
| 136 |
let (code, out, err) = oh(w, &[], &["release", "create", "v1", "--title", "One", "--notes", "First."]).await; |
| 137 |
assert_eq!(code, 0, "{err}"); |
| 138 |
assert!(out.contains("released v1"), "{out}"); |
| 139 |
let asset = w.join("tool.bin"); |
| 140 |
std::fs::write(&asset, b"bytes").unwrap(); |
| 141 |
let (code, out, err) = oh(w, &[], &["release", "upload", "v1", asset.to_str().unwrap()]).await; |
| 142 |
assert_eq!(code, 0, "{err}"); |
| 143 |
assert!(out.contains("attached tool.bin to v1")); |
| 144 |
let (_, out, _) = oh(w, &[], &["release", "list", "--json"]).await; |
| 145 |
let rel: serde_json::Value = serde_json::from_str(&out).unwrap(); |
| 146 |
assert_eq!(rel[0]["assets"][0]["name"], "tool.bin"); |
| 147 |
let (code, out, _) = oh(w, &[], &["release", "trust", "v1", "--plain"]).await; |
| 148 |
assert_eq!(code, 0, "warnings only: {out}"); |
| 149 |
assert!(out.contains("Tag v1 is signed") && out.contains("tool.bin: no build found"), "{out}"); |
| 150 |
|
| 151 |
|
| 152 |
let elsewhere = tempfile::tempdir().unwrap(); |
| 153 |
let cfg = elsewhere.path().join("oh.toml"); |
| 154 |
let cfg_s = cfg.to_str().unwrap(); |
| 155 |
let (code, _, err) = oh(elsewhere.path(), &[("OH_CONFIG", cfg_s)], &["issue", "list"]).await; |
| 156 |
assert_ne!(code, 0); |
| 157 |
assert!(err.contains("--repo"), "{err}"); |
| 158 |
let (code, out, err) = |
| 159 |
oh(elsewhere.path(), &[("OH_CONFIG", cfg_s)], &["auth", "login", "--with-hub", &h.base, "--token", "site"]).await; |
| 160 |
assert_eq!(code, 0, "{err}"); |
| 161 |
assert!(out.contains("signed in"), "{out}"); |
| 162 |
#[cfg(unix)] |
| 163 |
{ |
| 164 |
use std::os::unix::fs::PermissionsExt; |
| 165 |
assert_eq!(std::fs::metadata(&cfg).unwrap().permissions().mode() & 0o777, 0o600, "tokens are owner-only"); |
| 166 |
} |
| 167 |
let (code, out, _) = |
| 168 |
oh(elsewhere.path(), &[("OH_CONFIG", cfg_s)], &["issue", "list", "--state", "all", "--repo", "andy/demo", "--plain"]).await; |
| 169 |
assert_eq!(code, 0); |
| 170 |
assert!(out.contains("Broken thing"), "{out}"); |
| 171 |
let (code, out, _) = oh(elsewhere.path(), &[("OH_CONFIG", cfg_s)], &["api", "/repos/andy/demo/labels"]).await; |
| 172 |
assert_eq!(code, 0); |
| 173 |
assert!(out.trim_start().starts_with('['), "{out}"); |
| 174 |
|
| 175 |
let (code, _, err) = oh(w, &[("OXY_TOKEN", "nope")], &["issue", "create", "--title", "x"]).await; |
| 176 |
assert_ne!(code, 0); |
| 177 |
assert!(err.contains("HTTP 401") && err.contains("oh auth login"), "{err}"); |
| 178 |
} |