42fa9554 Andy McCall 1 month ago ↶
1
2
3
4
5
use axum:: body:: Body;
6
use axum:: http:: { header, Request, StatusCode} ;
7
use http_body_util:: BodyExt;
8
use oxy_core:: protocol:: { open_remote, RemoteOptions} ;
9
use oxy_core:: repo:: Repo;
10
use oxygenhub:: config:: Config;
11
use oxygenhub:: repos:: RepoName;
12
use oxygenhub:: { db, Shared} ;
13
use std:: path:: Path;
14
use tower:: ServiceExt;
15
16
const ROOTFS : & str = " 36fc4e3238044e424219a2b9002bf61bbe1805fec14d52cf37e17fae77c9ff16" ;
17
18
fn hermetic ( ) {
19
static ONCE : std:: sync:: Once = std:: sync:: Once:: new( ) ;
20
static DIR : std:: sync:: OnceLock< tempfile:: TempDir> = std:: sync:: OnceLock:: new( ) ;
21
ONCE .call_once ( | | {
22
let d = tempfile:: tempdir( ) .unwrap ( ) ;
23
std:: env:: set_var( " OXY_HOME" , d.path ( ) ) ;
24
DIR .set ( d) .ok ( ) ;
25
} ) ;
26
}
27
28
fn write ( root : & Path, rel : & str , content : & str ) {
29
let p = root.join ( rel) ;
30
std:: fs:: create_dir_all( p.parent ( ) .unwrap ( ) ) .unwrap ( ) ;
31
std:: fs:: write( p, content) .unwrap ( ) ;
32
}
33
34
struct Hub {
35
app : Shared,
36
base : String,
37
_dir : tempfile:: TempDir,
38
}
39
40
async fn hub ( ) -> Hub {
41
hermetic ( ) ;
42
let dir = tempfile:: tempdir( ) .unwrap ( ) ;
43
let mut config = Config:: load( dir.path ( ) ) .unwrap ( ) ;
c8684d44 Andy McCall 20 days ago
44
45
config.registration.setup_code = " test-setup" .into ( ) ;
42fa9554 Andy McCall 1 month ago ↶
46
config.auth.push_tokens = vec! [ " site" .into ( ) ] ;
4d2e44d7 Andy McCall 1 month ago ↶
47
config.flow.runner_tokens = vec! [ " runner-secret" .into ( ) , " runner-two" .into ( ) , " runner-three" .into ( ) ] ;
42fa9554 Andy McCall 1 month ago ↶
48
config.flow.repos = vec! [ " andy/*" .into ( ) ] ;
10708ab8 Andy McCall 1 month ago ↶
49
config.registration.mode = " open" .into ( ) ;
42fa9554 Andy McCall 1 month ago ↶
50
let listener = tokio:: net:: TcpListener:: bind( " 127.0.0.1:0" ) .await.unwrap ( ) ;
51
let port = listener.local_addr ( ) .unwrap ( ) .port ( ) ;
52
config.server.base_url = format! ( " http://127.0.0.1:{port} " ) ;
53
let app = oxygenhub:: build( config) .await.unwrap ( ) ;
54
let router = oxygenhub:: router( app.clone ( ) ) ;
55
tokio:: spawn( async move { axum:: serve( listener, router) .await.unwrap ( ) } ) ;
56
Hub { app, base: format! ( " http://127.0.0.1:{port} " ) , _dir: dir }
57
}
58
59
async fn call ( app : & Shared, method : & str , path : & str , bearer : Option< & str > , json : Option< String> ) -> ( StatusCode, String) {
60
let mut req = Request:: builder( ) .method ( method) .uri ( path) ;
61
if let Some ( b) = bearer {
62
req = req.header ( header:: AUTHORIZATION , format! ( " Bearer {b} " ) ) ;
63
}
64
let body = match json {
65
Some ( j) => {
66
req = req.header ( header:: CONTENT_TYPE , " application/json" ) ;
67
Body:: from( j)
68
}
69
None => Body:: empty( ) ,
70
} ;
71
let res = oxygenhub:: router( app.clone ( ) ) .oneshot ( req.body ( body) .unwrap ( ) ) .await.unwrap ( ) ;
72
let status = res.status ( ) ;
73
let bytes = res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ;
74
( status, String :: from_utf8_lossy( & bytes) .into_owned ( ) )
75
}
76
736f6813 Andy McCall 1 month ago ↶
77
78
async fn head ( app : & Shared, path : & str ) -> ( StatusCode, String, String) {
79
let res = oxygenhub:: router( app.clone ( ) ) .oneshot ( Request:: get( path) .body ( Body:: empty( ) ) .unwrap ( ) ) .await.unwrap ( ) ;
80
let h = | k : header:: HeaderName| res.headers ( ) .get ( k) .and_then ( | v | v.to_str ( ) .ok ( ) ) .unwrap_or ( " " ) .to_string ( ) ;
81
( res.status ( ) , h ( header:: LOCATION ) , h ( header:: CONTENT_TYPE ) )
82
}
83
42fa9554 Andy McCall 1 month ago ↶
84
async fn claim ( app : & Shared) -> Option< serde_json:: Value> {
85
let ( st, body) =
86
call ( app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( r # "{"runner":"test-1","runtimes":["microvm"]}"# .into ( ) ) ) .await;
87
match st {
88
StatusCode:: NO_CONTENT => None ,
89
StatusCode:: OK => Some ( serde_json:: from_str( & body) .unwrap ( ) ) ,
90
other => panic! ( " claim: {other} {body}" ) ,
91
}
92
}
93
94
fn receipt ( job : & str , pipeline_hash : & str , exit : u32 ) -> serde_json:: Value {
95
serde_json:: json! ( {
96
" job_name" : job, " pipeline_hash" : pipeline_hash, " scope_tree_hash" : " 11" .repeat ( 32 ) ,
97
" inputs" : [ ] , " environment" : " 22" .repeat ( 32 ) , " outputs" : [ ] , " exit_code" : exit, " duration_ms" : 640
98
} )
99
}
100
101
async fn finish ( app : & Shared, id : i64 , body : serde_json:: Value) -> StatusCode {
102
call ( app, " POST" , & format! ( " /api/v1/flow/jobs/{id} /finish" ) , Some ( " runner-secret" ) , Some ( body.to_string ( ) ) ) .await.0
103
}
104
105
async fn create ( app : & Shared, owner : & str , name : & str ) {
106
app.repos.create ( & RepoName:: parse( & format! ( " {owner} /{name} " ) ) .unwrap ( ) , None ) .unwrap ( ) ;
107
db:: repos:: create( & app.db, owner, name, " " , hub_api:: Visibility:: Public) .await.unwrap ( ) ;
108
}
109
110
fn client ( url : & str , work : & Path) -> Repo {
111
let mut c = Repo:: init( work) .unwrap ( ) ;
112
c.config.set ( " remote.origin.url" , url) .unwrap ( ) ;
113
c.config.set ( " remote.origin.token" , " site" ) .unwrap ( ) ;
114
c.save_config ( ) .unwrap ( ) ;
115
c
116
}
117
118
fn pipeline ( extra : & str ) -> String {
119
format! (
120
" schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" build\" {{ \n sandbox \" alpine\" \n exec \" make\" \n }} \n job \" test\" {{ \n sandbox \" alpine\" \n depends-on \" build\" \n exec \" sh\" \" -eu\" \" test.sh\" \n }} \n job \" lint\" {{ \n sandbox \" alpine\" \n exec \" true\" \n }} \n {extra} "
121
)
122
}
123
124
async fn runs_of ( app : & Shared, owner : & str , name : & str ) -> Vec< db:: flow:: Run> {
125
let row = db:: repos:: get( & app.db, owner, name) .await.unwrap ( ) .unwrap ( ) ;
126
for _ in 0 .. 100 {
127
let r = db:: flow:: runs( & app.db, row.id, 10 , None ) .await.unwrap ( ) ;
128
if ! r.is_empty ( ) {
129
return r;
130
}
131
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
132
}
133
vec! [ ]
134
}
135
90326c96 Andy McCall 1 month ago ↶
136
137
async fn checks_when ( app : & Shared, repo_id : i64 , state : & str , n : usize ) -> Vec< db:: hooks:: Check> {
138
for _ in 0 .. 100 {
139
let c = db:: hooks:: checks_for_state( & app.db, repo_id, state) .await.unwrap ( ) ;
140
if c.len ( ) > = n {
141
return c;
142
}
143
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
144
}
145
db:: hooks:: checks_for_state( & app.db, repo_id, state) .await.unwrap ( )
146
}
147
42fa9554 Andy McCall 1 month ago ↶
148
# [ tokio ::test ( flavor = " multi_thread" ) ]
149
async fn a_push_runs_the_pipeline_through_a_runner ( ) {
150
let h = hub ( ) .await;
151
create ( & h.app, " andy" , " demo" ) .await;
152
create ( & h.app, " andy" , " secret" ) .await;
153
create ( & h.app, " other" , " plain" ) .await;
154
let url = format! ( " {} /andy/demo" , h.base) ;
155
let work = tempfile:: tempdir( ) .unwrap ( ) ;
156
let mut c = client ( & url, work.path ( ) ) ;
157
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
158
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
159
c.snapshot ( " test" ) .unwrap ( ) ;
160
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
161
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
162
163
164
let runs = runs_of ( & h.app, " andy" , " demo" ) .await;
165
assert_eq! ( runs.len ( ) , 1 ) ;
166
let run = & runs[ 0 ] ;
167
assert_eq! ( ( run.bookmark.as_str ( ) , run.status.as_str ( ) , run.actor.as_str ( ) ) , ( " main" , " queued" , " site token" ) ) ;
168
assert_eq! ( run.pipeline_hash.len ( ) , 64 ) ;
169
let jobs = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) ;
170
assert_eq! ( jobs.iter ( ) .map ( | j | ( j.name.as_str ( ) , j.depth) ) .collect:: < Vec< _ > > ( ) , [ ( " build" , 0 ) , ( " lint" , 0 ) , ( " test" , 1 ) ] ) ;
90326c96 Andy McCall 1 month ago ↶
171
172
let checks = checks_when ( & h.app, run.repo_id, & run.state, 3 ) .await;
42fa9554 Andy McCall 1 month ago ↶
173
assert_eq! (
174
checks.iter ( ) .map ( | c | ( c.name.as_str ( ) , c.status.as_str ( ) ) ) .collect:: < Vec< _ > > ( ) ,
175
[ ( " flow/build" , " queued" ) , ( " flow/lint" , " queued" ) , ( " flow/test" , " queued" ) ]
176
) ;
177
178
179
assert_eq! (
180
call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " site" ) , Some ( r # "{"runner":"x","runtimes":["microvm"]}"# .into ( ) ) ) .await.0 ,
181
StatusCode:: UNAUTHORIZED
182
) ;
183
assert_eq! (
184
call ( & h.app, " POST" , " /api/v1/flow/claim" , None , Some ( r # "{"runner":"x","runtimes":["microvm"]}"# .into ( ) ) ) .await.0 ,
185
StatusCode:: UNAUTHORIZED
186
) ;
187
188
let ( st, _ ) =
189
call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( r # "{"runner":"x","runtimes":["wasm"]}"# .into ( ) ) ) .await;
190
assert_eq! ( st, StatusCode:: NO_CONTENT ) ;
191
192
193
let a = claim ( & h.app) .await.unwrap ( ) ;
194
let b = claim ( & h.app) .await.unwrap ( ) ;
195
assert_eq! ( ( a[ " job" ] [ " name" ] .as_str ( ) , b[ " job" ] [ " name" ] .as_str ( ) ) , ( Some ( " build" ) , Some ( " lint" ) ) ) ;
196
assert! ( claim ( & h.app) .await.is_none ( ) , " test waits for build" ) ;
197
assert_eq! ( a[ " repo" ] , " andy/demo" ) ;
198
assert_eq! ( a[ " clone_url" ] , url.as_str ( ) ) ;
199
assert_eq! ( a[ " sandbox" ] [ " rootfs" ] , ROOTFS ) ;
200
assert_eq! ( a[ " job" ] [ " exec" ] , serde_json:: json! ( [ " make" ] ) ) ;
201
assert_eq! ( a[ " timeout_s" ] , 3600 ) ;
9a065bc0 Andy McCall 1 month ago ↶
202
203
assert_eq! (
204
( a[ " state" ] .as_str ( ) , a[ " bookmark" ] .as_str ( ) , a[ " run_number" ] .as_i64 ( ) ) ,
205
( Some ( run.state.as_str ( ) ) , Some ( " main" ) , Some ( run.number) )
206
) ;
42fa9554 Andy McCall 1 month ago ↶
207
let heads: Vec< String> = a[ " heads" ] .as_array ( ) .unwrap ( ) .iter ( ) .map ( | v | v.as_str ( ) .unwrap ( ) .to_string ( ) ) .collect ( ) ;
208
assert_eq! ( heads.join ( " +" ) , run.heads) ;
209
let checks = db:: hooks:: checks_for_state( & h.app.db, run.repo_id, & run.state) .await.unwrap ( ) ;
210
assert_eq! ( checks.iter ( ) .find ( | c | c.name = = " flow/build" ) .unwrap ( ) .status, " running" ) ;
211
212
213
let token = a[ " token" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
214
assert! ( token.starts_with ( " flowjob_" ) ) ;
e8fa5369 Andy McCall 22 days ago ↶
215
let opts = RemoteOptions { token: Some ( token.clone ( ) ) , .. Default :: default( ) } ;
42fa9554 Andy McCall 1 month ago ↶
216
open_remote ( & url, & opts) .unwrap ( ) .refs ( ) .unwrap ( ) ;
217
assert! ( open_remote ( & format! ( " {} /andy/secret" , h.base) , & opts) .unwrap ( ) .refs ( ) .is_err ( ) , " another repository" ) ;
e8fa5369 Andy McCall 22 days ago ↶
218
let bad = RemoteOptions { token: Some ( " flowjob_00" .into ( ) ) , .. Default :: default( ) } ;
42fa9554 Andy McCall 1 month ago ↶
219
assert! ( open_remote ( & url, & bad) .unwrap ( ) .refs ( ) .is_err ( ) , " an unknown job token" ) ;
220
221
c.config.set ( " remote.origin.token" , & token) .unwrap ( ) ;
222
c.save_config ( ) .unwrap ( ) ;
223
write ( work.path ( ) , " more.txt" , " x\n " ) ;
224
c.snapshot ( " test" ) .unwrap ( ) ;
225
c.commit ( " More" , & [ ] ) .unwrap ( ) ;
226
let pushed = c.push ( " origin" , false ) ;
227
assert! ( pushed.is_err ( ) | | pushed.unwrap ( ) .iter ( ) .all ( | r | ! r.ok) , " a job token cannot push" ) ;
228
c.config.set ( " remote.origin.token" , " site" ) .unwrap ( ) ;
229
c.save_config ( ) .unwrap ( ) ;
230
231
232
let a_id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
233
let b_id = b[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
234
let log = serde_json:: json! ( { " records" : [
235
{ " t" : 5 , " stream" : " flow" , " text" : " booting" } ,
236
{ " t" : 120 , " stream" : " out" , " text" : " cc -o demo demo.c" } ,
237
{ " t" : 130 , " stream" : " weird" , " text" : " x" .repeat ( 20_000 ) } ,
238
] } ) ;
239
assert_eq! (
240
call ( & h.app, " POST" , & format! ( " /api/v1/flow/jobs/{a_id} /log" ) , Some ( " runner-secret" ) , Some ( log.to_string ( ) ) ) .await.0 ,
241
StatusCode:: NO_CONTENT
242
) ;
243
let lines = db:: flow:: log( & h.app.db, a_id, 0 , 100 ) .await.unwrap ( ) ;
244
assert_eq! ( lines.len ( ) , 3 ) ;
245
assert_eq! ( ( lines[ 2 ] .stream.as_str ( ) , lines[ 2 ] .text.len ( ) ) , ( " console" , 16 * 1024 ) , " streams are known ones, text is capped" ) ;
246
let ph = a[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) ;
247
assert_eq! (
248
finish ( & h.app, a_id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 , " receipt" : receipt ( " lint" , ph, 0 ) } ) ) .await,
249
StatusCode:: BAD_REQUEST ,
250
" a receipt for another job"
251
) ;
252
assert_eq! (
253
finish ( & h.app, a_id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 , " receipt" : receipt ( " build" , ph, 0 ) } ) ) .await,
254
StatusCode:: NO_CONTENT
255
) ;
256
assert_eq! ( finish ( & h.app, a_id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 } ) ) .await, StatusCode:: CONFLICT , " only once" ) ;
257
assert! ( open_remote ( & url, & opts) .unwrap ( ) .refs ( ) .is_err ( ) , " the token ends with its job" ) ;
258
assert_eq! (
259
finish ( & h.app, b_id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 300 , " receipt" : receipt ( " lint" , ph, 0 ) } ) ) .await,
260
StatusCode:: NO_CONTENT
261
) ;
262
let t = claim ( & h.app) .await.unwrap ( ) ;
263
assert_eq! ( t[ " job" ] [ " name" ] , " test" ) ;
264
let t_id = t[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
265
assert_eq! (
266
finish ( & h.app, t_id, serde_json:: json! ( { " exit_code" : 3 , " duration_ms" : 500 , " receipt" : receipt ( " test" , ph, 3 ) } ) ) .await,
267
StatusCode:: NO_CONTENT
268
) ;
269
270
let run = db:: flow:: run_by_id( & h.app.db, run.id) .await.unwrap ( ) .unwrap ( ) ;
271
assert_eq! ( run.status, " failed" ) ;
272
let jobs = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) ;
273
let build = jobs.iter ( ) .find ( | j | j.name = = " build" ) .unwrap ( ) ;
274
assert_eq! ( ( build.status.as_str ( ) , build.exit_code, build.receipt_hash.len ( ) ) , ( " passed" , Some ( 0 ) , 64 ) ) ;
275
assert_eq! ( jobs.iter ( ) .find ( | j | j.name = = " test" ) .unwrap ( ) .status, " failed" ) ;
276
let checks = db:: hooks:: checks_for_state( & h.app.db, run.repo_id, & run.state) .await.unwrap ( ) ;
277
assert_eq! (
278
checks.iter ( ) .map ( | c | ( c.name.as_str ( ) , c.status.as_str ( ) ) ) .collect:: < Vec< _ > > ( ) ,
279
[ ( " flow/build" , " passed" ) , ( " flow/lint" , " passed" ) , ( " flow/test" , " failed" ) ]
280
) ;
a0d4bbfb Andy McCall 1 month ago ↶
281
assert! ( checks[ 2 ] .summary.contains ( " exit 3" ) & & checks[ 2 ] .url.ends_with ( & format! ( " /andy/demo/flow/{} " , run.number) ) ) ;
42fa9554 Andy McCall 1 month ago ↶
282
283
284
let ( st, list) = call ( & h.app, " GET" , " /andy/demo/flow" , None , None ) .await;
285
assert_eq! ( st, StatusCode:: OK ) ;
a0d4bbfb Andy McCall 1 month ago ↶
286
assert! ( list.contains ( & format! ( " /andy/demo/flow/{} " , run.number) ) & & list.contains ( " st-failed" ) , " {list}" ) ;
287
let ( st, page) = call ( & h.app, " GET" , & format! ( " /andy/demo/flow/{} " , run.number) , None , None ) .await;
42fa9554 Andy McCall 1 month ago ↶
288
assert_eq! ( st, StatusCode:: OK ) ;
289
assert! ( page.contains ( " <svg class=\" flow-graph\" " ) , " graph" ) ;
290
assert_eq! ( page.matches ( " data-job-node=" ) .count ( ) , 3 ) ;
291
assert_eq! ( page.matches ( " <path class=\" edge" ) .count ( ) , 1 , " one dependency edge" ) ;
292
assert! ( page.contains ( " cc -o demo demo.c" ) , " log shown" ) ;
293
assert! ( ! page.contains ( " data-flow-live" ) , " a finished run does not poll" ) ;
a0d4bbfb Andy McCall 1 month ago ↶
294
let ( st, live) = call ( & h.app, " GET" , & format! ( " /andy/demo/flow/{} /live?after={a_id} :1" , run.number) , None , None ) .await;
42fa9554 Andy McCall 1 month ago ↶
295
assert_eq! ( st, StatusCode:: OK ) ;
296
let live: serde_json:: Value = serde_json:: from_str( & live) .unwrap ( ) ;
297
assert_eq! ( live[ " status" ] , " failed" ) ;
298
let build_live = live[ " jobs" ] .as_array ( ) .unwrap ( ) .iter ( ) .find ( | j | j[ " name" ] = = " build" ) .unwrap ( ) .clone ( ) ;
299
assert_eq! ( build_live[ " lines" ] .as_array ( ) .unwrap ( ) .len ( ) , 2 , " lines after seq 1" ) ;
300
assert_eq! ( call ( & h.app, " GET" , " /andy/demo/flow/999" , None , None ) .await.0 , StatusCode:: NOT_FOUND ) ;
301
302
303
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
304
write ( work.path ( ) , " test.sh" , " echo changed\n " ) ;
305
c.snapshot ( " test" ) .unwrap ( ) ;
306
c.commit ( " Change the test" , & [ ] ) .unwrap ( ) ;
307
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
308
let mut second = None ;
309
for _ in 0 .. 100 {
310
let r = runs_of ( & h.app, " andy" , " demo" ) .await;
311
if r.len ( ) = = 2 {
312
second = Some ( r[ 0 ] .clone ( ) ) ;
313
break ;
314
}
315
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
316
}
317
let second = second.expect ( " a second run" ) ;
318
let x = claim ( & h.app) .await.unwrap ( ) ;
319
let y = claim ( & h.app) .await.unwrap ( ) ;
320
let ( build_id, lint_id) = if x[ " job" ] [ " name" ] = = " build" {
321
( x[ " job_id" ] .as_i64 ( ) .unwrap ( ) , y[ " job_id" ] .as_i64 ( ) .unwrap ( ) )
322
} else {
323
( y[ " job_id" ] .as_i64 ( ) .unwrap ( ) , x[ " job_id" ] .as_i64 ( ) .unwrap ( ) )
324
} ;
325
assert_eq! (
326
finish ( & h.app, build_id, serde_json:: json! ( { " exit_code" : null, " duration_ms" : 10 , " error" : " rootfs is not on this runner" } ) )
327
.await,
328
StatusCode:: NO_CONTENT
329
) ;
330
assert_eq! ( finish ( & h.app, lint_id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 10 } ) ) .await, StatusCode:: NO_CONTENT ) ;
331
assert! ( claim ( & h.app) .await.is_none ( ) , " test was skipped" ) ;
332
let jobs = db:: flow:: jobs( & h.app.db, second.id) .await.unwrap ( ) ;
333
assert_eq! (
334
jobs.iter ( ) .map ( | j | ( j.name.as_str ( ) , j.status.as_str ( ) ) ) .collect:: < Vec< _ > > ( ) ,
335
[ ( " build" , " error" ) , ( " lint" , " passed" ) , ( " test" , " skipped" ) ]
336
) ;
337
assert_eq! ( db:: flow:: run_by_id( & h.app.db, second.id) .await.unwrap ( ) .unwrap ( ) .status, " failed" ) ;
338
339
340
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " job \" broken\" {\n sandbox \" missing\" \n exec \" x\" \n }\n " ) ) ;
341
c.snapshot ( " test" ) .unwrap ( ) ;
342
c.commit ( " Break the pipeline" , & [ ] ) .unwrap ( ) ;
343
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
344
let mut third = None ;
345
for _ in 0 .. 100 {
346
let r = runs_of ( & h.app, " andy" , " demo" ) .await;
347
if r.len ( ) = = 3 {
348
third = Some ( r[ 0 ] .clone ( ) ) ;
349
break ;
350
}
351
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
352
}
353
let third = third.expect ( " a third run" ) ;
354
assert_eq! ( third.status, " error" ) ;
355
assert! ( third.error.contains ( " line 18" ) & & third.error.contains ( " \" missing\" " ) , " {}" , third.error) ;
a0d4bbfb Andy McCall 1 month ago ↶
356
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/demo/flow/{} " , third.number) , None , None ) .await;
42fa9554 Andy McCall 1 month ago ↶
357
assert! ( page.contains ( " The pipeline is not valid" ) ) ;
358
736f6813 Andy McCall 1 month ago ↶
359
360
write ( work.path ( ) , " README.md" , " # Demo\n \n [](.flow/run)\n \n \n " ) ;
361
std:: fs:: create_dir_all( work.path ( ) .join ( " img" ) ) .unwrap ( ) ;
362
std:: fs:: write( work.path ( ) .join ( " img/logo.png" ) , b " \x89 PNG\r \n \x1a \n \0 \0 \0 \r IHDR\0 \0 \0 \x01 \0 \0 \0 \x01 \x08 \x06 \0 \0 \0 \x1f \x15 \xc4 \x89 " )
363
.unwrap ( ) ;
364
c.snapshot ( " test" ) .unwrap ( ) ;
365
c.commit ( " A README with a stamp" , & [ ] ) .unwrap ( ) ;
366
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
367
let mut fourth = None ;
368
for _ in 0 .. 100 {
369
let r = runs_of ( & h.app, " andy" , " demo" ) .await;
370
if r.len ( ) = = 4 {
371
fourth = Some ( r[ 0 ] .clone ( ) ) ;
372
break ;
373
}
374
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
375
}
376
let fourth = fourth.expect ( " a fourth run" ) ;
377
let ( _ , overview) = call ( & h.app, " GET" , " /andy/demo" , None , None ) .await;
378
assert! ( overview.contains ( " src=\" /andy/demo/raw/main/.flow/stamp\" " ) , " {overview}" ) ;
379
assert! ( overview.contains ( " href=\" /andy/demo/file/main/.flow/run\" " ) ) ;
380
assert! ( overview.contains ( " src=\" /andy/demo/raw/main/img/logo.png\" " ) , " images use the raw route" ) ;
381
let ( st, _ , ct) = head ( & h.app, " /andy/demo/raw/main/img/logo.png" ) .await;
382
assert_eq! ( ( st, ct.as_str ( ) ) , ( StatusCode:: OK , " image/png" ) ) ;
383
let ( st, loc, _ ) = head ( & h.app, " /andy/demo/raw/main/.flow/stamp" ) .await;
384
assert! ( st.is_redirection ( ) & & loc = = " /andy/demo/flow/stamp/main" , " {st} {loc}" ) ;
385
let ( st, loc, _ ) = head ( & h.app, " /andy/demo/flow/stamp/main" ) .await;
386
assert! ( st.is_redirection ( ) & & loc = = format! ( " /andy/demo/flow/stamp/{} " , fourth.heads) , " a bookmark's stamp names its state: {loc}" ) ;
387
let ( st, _ , ct) = head ( & h.app, & loc) .await;
388
assert_eq! ( ( st, ct.as_str ( ) ) , ( StatusCode:: OK , " image/svg+xml; charset=utf-8" ) ) ;
389
let ( _ , svg) = call ( & h.app, " GET" , & loc, None , None ) .await;
390
assert! ( svg.starts_with ( " <svg" ) & & svg.contains ( " error ·" ) & & svg.contains ( & fourth.heads[ .. 12 ] ) , " {svg}" ) ;
391
let ( _ , svg1) = call ( & h.app, " GET" , & format! ( " /andy/demo/flow/stamp/{} " , run.heads) , None , None ) .await;
392
assert! ( svg1.contains ( " failed ·" ) & & svg1.contains ( & format! ( " run #{} failed" , run.id) ) , " an older state keeps its own result: {svg1}" ) ;
393
assert! ( svg1.contains ( & build.receipt_hash) , " receipt hashes in the metadata" ) ;
394
assert_eq! ( svg1.matches ( " text-anchor=\" middle\" " ) .count ( ) , 3 , " one glyph per job" ) ;
395
let ( st, loc, _ ) = head ( & h.app, " /andy/demo/file/main/.flow/run" ) .await;
396
assert! ( st.is_redirection ( ) & & loc = = " /andy/demo/flow/at/main" , " {loc}" ) ;
397
let ( st, loc, _ ) = head ( & h.app, " /andy/demo/flow/at/main" ) .await;
398
assert! ( st.is_redirection ( ) & & loc = = format! ( " /andy/demo/flow/{} " , fourth.id) , " {loc}" ) ;
399
42fa9554 Andy McCall 1 month ago ↶
400
401
let work2 = tempfile:: tempdir( ) .unwrap ( ) ;
402
let c2 = client ( & format! ( " {} /other/plain" , h.base) , work2.path ( ) ) ;
403
write ( work2.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
404
c2.snapshot ( " test" ) .unwrap ( ) ;
405
c2.commit ( " Pipeline" , & [ ] ) .unwrap ( ) ;
406
assert! ( c2.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
407
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 500 ) ) .await;
408
let row = db:: repos:: get( & h.app.db, " other" , " plain" ) .await.unwrap ( ) .unwrap ( ) ;
409
assert! ( db:: flow:: runs( & h.app.db, row.id, 10 , None ) .await.unwrap ( ) .is_empty ( ) ) ;
410
let ( _ , page) = call ( & h.app, " GET" , " /other/plain/flow" , None , None ) .await;
411
assert! ( page.contains ( " not enabled for this repository" ) ) ;
736f6813 Andy McCall 1 month ago ↶
412
let ( _ , loc, _ ) = head ( & h.app, " /other/plain/flow/stamp/main" ) .await;
413
let ( _ , svg) = call ( & h.app, " GET" , & loc, None , None ) .await;
414
assert! ( svg.contains ( " no run ·" ) , " {svg}" ) ;
415
let ( _ , loc, _ ) = head ( & h.app, " /other/plain/flow/at/main" ) .await;
416
assert_eq! ( loc, " /other/plain/flow" ) ;
42fa9554 Andy McCall 1 month ago ↶
417
}
fd3db644 Andy McCall 1 month ago ↶
418
419
# [ tokio ::test ( flavor = " multi_thread" ) ]
420
async fn claims_carry_pinned_subrepos_and_the_job_token_reads_only_those ( ) {
421
let h = hub ( ) .await;
25091e5c Andy McCall 1 month ago ↶
422
for n in [ " base" , " lib" , " vault" , " other" , " app" ] {
fd3db644 Andy McCall 1 month ago ↶
423
create ( & h.app, " andy" , n) .await;
424
}
425
let push = | name : & str , file : & str | {
426
let work = tempfile:: tempdir( ) .unwrap ( ) ;
427
let c = client ( & format! ( " {} /andy/{name} " , h.base) , work.path ( ) ) ;
428
write ( work.path ( ) , file, " x\n " ) ;
429
c.snapshot ( " t" ) .unwrap ( ) ;
430
c.commit ( " first" , & [ ] ) .unwrap ( ) ;
431
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
432
let heads = c.bookmark_heads ( ) .unwrap ( ) ;
433
( work, heads)
434
} ;
25091e5c Andy McCall 1 month ago ↶
435
let ( _b, base_heads) = push ( " base" , " base.rs" ) ;
436
437
let lib_work = tempfile:: tempdir( ) .unwrap ( ) ;
438
let lib = client ( & format! ( " {} /andy/lib" , h.base) , lib_work.path ( ) ) ;
439
lib.submodule_add ( & format! ( " {} /andy/base" , h.base) , " deps/base" , Some ( " main" ) ) .unwrap ( ) ;
440
write ( lib_work.path ( ) , " lib.rs" , " x\n " ) ;
441
lib.snapshot ( " t" ) .unwrap ( ) ;
442
lib.commit ( " lib on base" , & [ ] ) .unwrap ( ) ;
443
assert! ( lib.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
444
let lib_heads = lib.bookmark_heads ( ) .unwrap ( ) ;
fd3db644 Andy McCall 1 month ago ↶
445
let ( _v, _ ) = push ( " vault" , " secret.rs" ) ;
446
let ( _o, _ ) = push ( " other" , " o.rs" ) ;
447
448
449
let work = tempfile:: tempdir( ) .unwrap ( ) ;
450
let app = client ( & format! ( " {} /andy/app" , h.base) , work.path ( ) ) ;
451
app.submodule_add ( & format! ( " {} /andy/lib" , h.base) , " deps/lib" , Some ( " main" ) ) .unwrap ( ) ;
a4f01d51 Andy McCall 1 month ago ↶
452
2bbe88a8 Andy McCall 1 month ago ↶
453
454
fd3db644 Andy McCall 1 month ago ↶
455
sqlx:: query( " UPDATE repositories SET visibility = 'private' WHERE name = 'vault'" ) .execute ( & h.app.db) .await.unwrap ( ) ;
a4f01d51 Andy McCall 1 month ago ↶
456
let bare_work = tempfile:: tempdir( ) .unwrap ( ) ;
457
let mut bare = Repo:: init( bare_work.path ( ) ) .unwrap ( ) ;
458
bare.config.set ( " remote.origin.url" , & format! ( " {} /andy/bare" , h.base) ) .unwrap ( ) ;
459
bare.save_config ( ) .unwrap ( ) ;
460
assert! ( bare.submodule_add ( & format! ( " {} /andy/vault" , h.base) , " deps/vault" , Some ( " main" ) ) .is_err ( ) , " not without a token" ) ;
370035a4 Andy McCall 1 month ago ↶
461
app.submodule_add ( & format! ( " {} /andy/vault" , h.base) , " deps/vault" , Some ( " main" ) ) .unwrap ( ) ;
a4f01d51 Andy McCall 1 month ago ↶
462
let nested = Repo:: open( & work.path ( ) .join ( " deps/vault" ) ) .unwrap ( ) ;
2bbe88a8 Andy McCall 1 month ago ↶
463
assert! ( nested.config.remote_tokens.is_empty ( ) , " no token is copied into a subrepo" ) ;
464
nested.fetch ( " origin" ) .unwrap ( ) ;
370035a4 Andy McCall 1 month ago ↶
465
fd3db644 Andy McCall 1 month ago ↶
466
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
467
write ( work.path ( ) , " Makefile" , " all:\n " ) ;
468
app.snapshot ( " t" ) .unwrap ( ) ;
469
app.commit ( " app with deps" , & [ ] ) .unwrap ( ) ;
470
assert! ( app.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
471
assert! ( ! runs_of ( & h.app, " andy" , " app" ) .await.is_empty ( ) ) ;
472
473
let job = claim ( & h.app) .await.unwrap ( ) ;
474
let subs = job[ " subrepos" ] .as_array ( ) .cloned ( ) .unwrap_or_default ( ) ;
25091e5c Andy McCall 1 month ago ↶
475
assert_eq! ( subs.len ( ) , 2 , " lib and what lib pins; vault is private and the pusher has no account: {subs:?}" ) ;
476
assert_eq! ( subs[ 1 ] [ " path" ] , " deps/lib/deps/base" , " nested under its parent, after it" ) ;
477
assert_eq! ( subs[ 1 ] [ " heads" ] , serde_json:: json! ( base_heads.iter ( ) .map ( | h | h.to_hex ( ) ) .collect:: < Vec< _ > > ( ) ) ) ;
370035a4 Andy McCall 1 month ago ↶
478
let _ = std:: fs:: remove_dir_all( work.path ( ) .join ( " deps/vault2" ) ) ;
fd3db644 Andy McCall 1 month ago ↶
479
assert_eq! ( subs[ 0 ] [ " path" ] , " deps/lib" ) ;
480
assert_eq! ( subs[ 0 ] [ " repo" ] , " andy/lib" ) ;
481
assert_eq! ( subs[ 0 ] [ " heads" ] , serde_json:: json! ( lib_heads.iter ( ) .map ( | h | h.to_hex ( ) ) .collect:: < Vec< _ > > ( ) ) ) ;
482
assert! ( subs[ 0 ] [ " clone_url" ] .as_str ( ) .unwrap ( ) .ends_with ( " /andy/lib" ) ) ;
afd0be38 Andy McCall 1 month ago ↶
483
484
let job_id = job[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
485
let lines = db:: flow:: log( & h.app.db, job_id, 0 , 100 ) .await.unwrap ( ) ;
486
assert! (
487
lines.iter ( ) .any ( | l | l.stream = = " flow" & & l.text.contains ( " subrepo deps/vault (andy/vault) is left out" ) ) ,
488
" {:?}" ,
489
lines.iter ( ) .map ( | l | & l.text) .collect:: < Vec< _ > > ( )
490
) ;
fd3db644 Andy McCall 1 month ago ↶
491
492
493
let token = job[ " token" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
494
let read = | name : & str | {
495
let opts = RemoteOptions { token: Some ( token.clone ( ) ) , .. Default :: default( ) } ;
496
open_remote ( & format! ( " {} /andy/{name} " , h.base) , & opts) .and_then ( | r | r.refs ( ) ) .is_ok ( )
497
} ;
25091e5c Andy McCall 1 month ago ↶
498
assert! ( read ( " app" ) & & read ( " lib" ) & & read ( " base" ) ) ;
fd3db644 Andy McCall 1 month ago ↶
499
assert! ( ! read ( " vault" ) , " not pinned for this pusher" ) ;
500
assert! ( ! read ( " other" ) , " public, but not the job's to read" ) ;
501
}
941abdd6 Andy McCall 1 month ago ↶
502
503
# [ tokio ::test ( flavor = " multi_thread" , worker_threads = 8 ) ]
504
async fn concurrent_log_appends_never_fail_on_a_locked_database ( ) {
505
let h = hub ( ) .await;
506
create ( & h.app, " andy" , " demo" ) .await;
507
let work = tempfile:: tempdir( ) .unwrap ( ) ;
508
let c = client ( & format! ( " {} /andy/demo" , h.base) , work.path ( ) ) ;
509
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
510
write ( work.path ( ) , " Makefile" , " all:\n " ) ;
511
c.snapshot ( " t" ) .unwrap ( ) ;
512
c.commit ( " pipeline" , & [ ] ) .unwrap ( ) ;
513
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
514
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
515
let job = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) .remove ( 0 ) ;
516
517
518
let mut tasks = Vec :: new( ) ;
519
for w in 0 .. 16 {
520
let pool = h.app.db.clone ( ) ;
521
tasks.push ( tokio:: spawn( async move {
522
let records: Vec< flow_core:: api:: Record> =
523
( 0 .. 20 ) .map ( | i | flow_core:: api:: Record { t: i, stream: " out" .into ( ) , text: format! ( " writer {w} line {i} " ) } ) .collect ( ) ;
524
db:: flow:: append_log( & pool, job.id, & records, 100_000 ) .await
525
} ) ) ;
526
}
527
for t in tasks {
528
assert_eq! ( t.await.unwrap ( ) .unwrap ( ) , 20 , " every append succeeds" ) ;
529
}
530
let lines = db:: flow:: log( & h.app.db, job.id, 0 , 1_000 ) .await.unwrap ( ) ;
531
assert_eq! ( lines.len ( ) , 320 ) ;
532
let seqs: std:: collections:: BTreeSet< i64 > = lines.iter ( ) .map ( | l | l.seq ) .collect ( ) ;
533
assert_eq! ( seqs.len ( ) , 320 , " sequence numbers are unique" ) ;
534
}
9008bf6c Andy McCall 1 month ago ↶
535
536
537
# [ tokio ::test ( flavor = " multi_thread" ) ]
538
async fn a_repository_setting_turns_flow_on_or_off ( ) {
539
let h = hub ( ) .await;
540
create ( & h.app, " other" , " plain" ) .await;
541
create ( & h.app, " andy" , " quiet" ) .await;
a0d4bbfb Andy McCall 1 month ago ↶
542
543
create ( & h.app, " andy" , " first" ) .await;
544
{
545
let work = tempfile:: tempdir( ) .unwrap ( ) ;
546
let c = client ( & format! ( " {} /andy/first" , h.base) , work.path ( ) ) ;
547
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
548
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
549
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
550
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
551
assert_eq! ( runs_of ( & h.app, " andy" , " first" ) .await.len ( ) , 1 ) ;
552
}
9008bf6c Andy McCall 1 month ago ↶
553
let row = db:: repos:: get( & h.app.db, " other" , " plain" ) .await.unwrap ( ) .unwrap ( ) ;
554
db:: repos:: set_flow( & h.app.db, row.id, " on" ) .await.unwrap ( ) ;
555
let quiet = db:: repos:: get( & h.app.db, " andy" , " quiet" ) .await.unwrap ( ) .unwrap ( ) ;
556
db:: repos:: set_flow( & h.app.db, quiet.id, " off" ) .await.unwrap ( ) ;
557
for ( owner, name) in [ ( " other" , " plain" ) , ( " andy" , " quiet" ) ] {
558
let work = tempfile:: tempdir( ) .unwrap ( ) ;
559
let c = client ( & format! ( " {} /{owner} /{name} " , h.base) , work.path ( ) ) ;
560
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
561
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
562
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
563
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
564
}
a0d4bbfb Andy McCall 1 month ago ↶
565
let plain = runs_of ( & h.app, " other" , " plain" ) .await;
566
assert_eq! ( plain.len ( ) , 1 , " turned on" ) ;
55673c65 Andy McCall 19 days ago ↶
567
568
bf3d0754 Andy McCall 1 month ago ↶
569
let a = claim ( & h.app) .await.unwrap ( ) ;
570
assert_eq! ( a[ " repo" ] , " andy/first" ) ;
571
let ( _ , page) = call ( & h.app, " GET" , " /other/plain/flow/1" , None , None ) .await;
55673c65 Andy McCall 19 days ago ↶
572
assert! ( page.contains ( " Waiting for a runner: 1st in line" ) & & page.contains ( " Running now: andy/first #1" ) , " {page}" ) ;
bf3d0754 Andy McCall 1 month ago ↶
573
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/repos/other/plain/flow/runs/1" , None , None ) .await;
55673c65 Andy McCall 19 days ago ↶
574
assert_eq! ( serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " queue_position" ] , 1 ) ;
bf3d0754 Andy McCall 1 month ago ↶
575
let ( _ , list) = call ( & h.app, " GET" , " /other/plain/flow" , None , None ) .await;
55673c65 Andy McCall 19 days ago ↶
576
assert! ( list.contains ( " queued, 1st in line" ) , " {list}" ) ;
a0d4bbfb Andy McCall 1 month ago ↶
577
578
579
assert_eq! ( plain[ 0 ] .number, 1 ) ;
90326c96 Andy McCall 1 month ago ↶
580
let checks = checks_when ( & h.app, row.id, & plain[ 0 ] .state, 3 ) .await;
a0d4bbfb Andy McCall 1 month ago ↶
581
assert! ( checks.iter ( ) .all ( | c | c.url.ends_with ( " /other/plain/flow/1" ) ) , " {checks:?}" ) ;
582
let ( st, page) = call ( & h.app, " GET" , " /other/plain/flow/1" , None , None ) .await;
583
assert! ( st = = StatusCode:: OK & & page.contains ( " Run #1" ) , " {st}" ) ;
9008bf6c Andy McCall 1 month ago ↶
584
assert! ( runs_of ( & h.app, " andy" , " quiet" ) .await.is_empty ( ) , " turned off despite andy/*" ) ;
585
}
1edac4b6 Andy McCall 1 month ago ↶
586
587
588
589
# [ tokio ::test ( flavor = " multi_thread" ) ]
590
async fn runs_start_without_a_push_and_finished_runs_re_run ( ) {
591
let h = hub ( ) .await;
592
593
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
594
create ( & h.app, " andy" , " demo" ) .await;
595
let url = format! ( " {} /andy/demo" , h.base) ;
596
let work = tempfile:: tempdir( ) .unwrap ( ) ;
597
let c = client ( & url, work.path ( ) ) ;
598
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
599
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
600
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
601
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
602
let first = runs_of ( & h.app, " andy" , " demo" ) .await;
603
assert_eq! ( ( first.len ( ) , first[ 0 ] .trigger.as_str ( ) ) , ( 1 , " push" ) ) ;
604
605
606
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( " {}" .into ( ) ) ) .await;
607
assert_eq! ( st, StatusCode:: CREATED , " {body}" ) ;
608
let v: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
609
assert_eq! ( v[ " number" ] , 2 ) ;
610
assert! ( v[ " url" ] .as_str ( ) .unwrap ( ) .ends_with ( " /andy/demo/flow/2" ) ) ;
611
let second = db:: flow:: run( & h.app.db, first[ 0 ] .repo_id, 2 ) .await.unwrap ( ) .unwrap ( ) ;
612
assert_eq! ( ( second.trigger.as_str ( ) , second.actor.as_str ( ) , second.state.as_str ( ) ) , ( " manual" , " andy" , first[ 0 ] .state.as_str ( ) ) ) ;
3d086bc3 Andy McCall 8 days ago ↶
613
614
let ( st, body) = call ( & h.app, " GET" , " /api/v1/user/flow/runs" , Some ( " site" ) , None ) .await;
615
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
616
let v: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
617
assert_eq! ( v.as_array ( ) .unwrap ( ) .len ( ) , 2 , " {v}" ) ;
618
assert_eq! (
619
( v[ 0 ] [ " repository" ] .as_str ( ) , v[ 0 ] [ " number" ] .as_i64 ( ) , v[ 1 ] [ " number" ] .as_i64 ( ) ) ,
620
( Some ( " andy/demo" ) , Some ( 2 ) , Some ( 1 ) ) ,
621
" {v}"
622
) ;
623
assert! ( ! v[ 0 ] [ " jobs" ] .as_array ( ) .unwrap ( ) .is_empty ( ) , " with its jobs: {v}" ) ;
624
let ( _ , body) = call ( & h.app, " GET" , & format! ( " /api/v1/user/flow/runs?before={} " , v[ 0 ] [ " created_ms" ] ) , Some ( " site" ) , None ) .await;
625
let older: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
626
assert! ( older.as_array ( ) .unwrap ( ) .iter ( ) .all ( | r | r[ " number" ] ! = 2 ) , " before pages back: {older}" ) ;
627
assert_eq! ( call ( & h.app, " GET" , " /api/v1/user/flow/runs" , None , None ) .await.0 , StatusCode:: UNAUTHORIZED ) ;
1edac4b6 Andy McCall 1 month ago ↶
628
assert_eq! ( db:: flow:: jobs( & h.app.db, second.id) .await.unwrap ( ) .len ( ) , 3 ) ;
629
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/flow/2" , None , None ) .await;
630
assert! ( page.contains ( " started by andy" ) , " the run page says what started it" ) ;
631
4f692533 Andy McCall 1 month ago ↶
632
633
let head = first[ 0 ] .head_list ( ) [ 0 ] .to_string ( ) ;
634
let ( st, body) =
635
call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( format! ( r # "{{ "change":"{head} "}} "# ) ) ) .await;
636
assert_eq! ( st, StatusCode:: CREATED , " {body}" ) ;
637
let on_change = db:: flow:: run( & h.app.db, first[ 0 ] .repo_id, 3 ) .await.unwrap ( ) .unwrap ( ) ;
638
assert_eq! (
639
( on_change.bookmark.as_str ( ) , on_change.heads.as_str ( ) , on_change.state.as_str ( ) ) ,
640
( " " , head.as_str ( ) , first[ 0 ] .state.as_str ( ) )
641
) ;
642
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/flow/3" , None , None ) .await;
643
assert! ( page.contains ( " title=\" Run on one change, not a bookmark\" >change</span>" ) , " run page" ) ;
644
let ( st, body) =
645
call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( format! ( r # "{{ "change":"{} "}} "# , " ab" .repeat ( 32 ) ) ) )
646
.await;
647
assert! ( st = = StatusCode:: CONFLICT & & body.contains ( " no change" ) , " {st} {body}" ) ;
648
649
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/flow" , None , None ) .await;
650
assert! ( page.contains ( " Sign in with write access to run the pipeline." ) & & ! page.contains ( " primary-btn" ) ) ;
651
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/bookmarks" , None , None ) .await;
652
assert! ( ! page.contains ( " /flow/start" ) , " no run buttons for anonymous viewers" ) ;
653
1edac4b6 Andy McCall 1 month ago ↶
654
655
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs/1/rerun" , Some ( " site" ) , None ) .await;
656
assert! ( st = = StatusCode:: CONFLICT & & body.contains ( " still queued" ) , " {st} {body}" ) ;
657
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( r # "{"bookmark":"nope"}"# .into ( ) ) ) .await;
658
assert_eq! ( st, StatusCode:: CONFLICT ) ;
659
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs/99/rerun" , Some ( " site" ) , None ) .await;
660
assert_eq! ( st, StatusCode:: NOT_FOUND ) ;
661
662
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , None , Some ( " {}" .into ( ) ) ) .await;
663
assert_ne! ( st, StatusCode:: CREATED ) ;
664
665
666
let row = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
667
for r in [ & first[ 0 ] , & second] {
668
for j in db:: flow:: jobs( & h.app.db, r.id) .await.unwrap ( ) {
3b1bc27f Andy McCall 20 days ago
669
sqlx:: query( " UPDATE flow_jobs SET status = 'passed' WHERE id = $1" ) .bind ( j.id) .execute ( & h.app.db) .await.unwrap ( ) ;
1edac4b6 Andy McCall 1 month ago ↶
670
}
3b1bc27f Andy McCall 20 days ago
671
sqlx:: query( " UPDATE flow_runs SET status = 'passed', finished_ms = 1 WHERE id = $1" ) .bind ( r.id) .execute ( & h.app.db) .await.unwrap ( ) ;
1edac4b6 Andy McCall 1 month ago ↶
672
}
673
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs/1/rerun" , Some ( " site" ) , None ) .await;
674
assert_eq! ( st, StatusCode:: CREATED , " {body}" ) ;
4f692533 Andy McCall 1 month ago ↶
675
let third = db:: flow:: run( & h.app.db, row.id, 4 ) .await.unwrap ( ) .unwrap ( ) ;
348f8c4b Andy McCall 1 month ago ↶
676
assert_eq! (
677
( third.trigger_label ( ) .as_str ( ) , third.status.as_str ( ) , third.state.as_str ( ) ) ,
678
( " rerun:1" , " queued" , first[ 0 ] .state.as_str ( ) )
679
) ;
c889911b Andy McCall 1 month ago ↶
680
assert_eq! ( third.trigger, " rerun:1:push" , " a rerun keeps the event it repeats" ) ;
1edac4b6 Andy McCall 1 month ago ↶
681
let checks = db:: hooks:: checks_for_state( & h.app.db, row.id, & third.state) .await.unwrap ( ) ;
682
assert! (
4f692533 Andy McCall 1 month ago ↶
683
checks.iter ( ) .all ( | c | c.status = = " queued" & & c.url.ends_with ( " /flow/4" ) ) ,
1edac4b6 Andy McCall 1 month ago ↶
684
" the re-run's checks replace the old ones: {checks:?}"
685
) ;
4f692533 Andy McCall 1 month ago ↶
686
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/flow/4" , None , None ) .await;
1edac4b6 Andy McCall 1 month ago ↶
687
assert! ( page.contains ( " re-run of #1 by andy" ) ) ;
688
assert! ( ! page.contains ( " RE-RUN" ) , " anonymous viewers get no button" ) ;
689
0650d94e Andy McCall 25 days ago ↶
690
691
692
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( r # "{"jobs":["test"]}"# .into ( ) ) ) .await;
693
assert_eq! ( st, StatusCode:: CREATED , " {body}" ) ;
694
let n = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " number" ] .as_i64 ( ) .unwrap ( ) ;
695
let only = db:: flow:: run( & h.app.db, first[ 0 ] .repo_id, n) .await.unwrap ( ) .unwrap ( ) ;
696
let mut names: Vec< String> = db:: flow:: jobs( & h.app.db, only.id) .await.unwrap ( ) .into_iter ( ) .map ( | j | j.name ) .collect ( ) ;
697
names.sort ( ) ;
698
assert_eq! ( names, [ " build" , " test" ] , " test and the job it depends on, not lint" ) ;
699
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( r # "{"jobs":["deploy"]}"# .into ( ) ) ) .await;
700
assert! ( st = = StatusCode:: CONFLICT & & body.contains ( " not in the pipeline" ) & & body.contains ( " build, lint, test" ) , " {st} {body}" ) ;
701
1edac4b6 Andy McCall 1 month ago ↶
702
703
db:: repos:: set_flow( & h.app.db, row.id, " off" ) .await.unwrap ( ) ;
704
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( " {}" .into ( ) ) ) .await;
705
assert! ( st = = StatusCode:: CONFLICT & & body.contains ( " not enabled" ) , " {body}" ) ;
4f692533 Andy McCall 1 month ago ↶
706
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/flow" , None , None ) .await;
707
assert! ( page.contains ( " Flow is off for this repository" ) , " and the Flow tab says so" ) ;
90326c96 Andy McCall 1 month ago ↶
708
}
709
710
711
712
# [ tokio ::test ( flavor = " multi_thread" ) ]
713
async fn runs_jobs_and_logs_are_in_the_api ( ) {
714
let h = hub ( ) .await;
715
create ( & h.app, " andy" , " demo" ) .await;
716
let url = format! ( " {} /andy/demo" , h.base) ;
717
let work = tempfile:: tempdir( ) .unwrap ( ) ;
718
let c = client ( & url, work.path ( ) ) ;
719
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
720
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
721
let head = c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
722
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
723
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
724
725
726
let a = claim ( & h.app) .await.unwrap ( ) ;
727
let id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
728
let recs =
729
serde_json:: json! ( { " records" : [ { " t" : 1 , " stream" : " stdout" , " text" : " one" } , { " t" : 3 , " stream" : " stderr" , " text" : " three" } ] } ) ;
730
let ( st, _ ) = call ( & h.app, " POST" , & format! ( " /api/v1/flow/jobs/{id} /log" ) , Some ( " runner-secret" ) , Some ( recs.to_string ( ) ) ) .await;
731
assert! ( st.is_success ( ) , " {st}" ) ;
732
733
let ( st, body) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/flow/runs" , None , None ) .await;
734
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
735
let runs: Vec< hub_api:: FlowRun> = serde_json:: from_str( & body) .unwrap ( ) ;
736
assert_eq! ( runs.len ( ) , 1 ) ;
737
assert_eq! ( ( runs[ 0 ] .number, runs[ 0 ] .state.as_str ( ) , runs[ 0 ] .trigger.as_str ( ) ) , ( 1 , run.state.as_str ( ) , " push" ) ) ;
738
assert! ( runs[ 0 ] .url.ends_with ( " /andy/demo/flow/1" ) & & runs[ 0 ] .jobs.is_empty ( ) ) ;
739
740
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/flow/runs/1" , None , None ) .await;
741
let one: hub_api:: FlowRun = serde_json:: from_str( & body) .unwrap ( ) ;
742
let names: Vec< ( & str , & str ) > = one.jobs.iter ( ) .map ( | j | ( j.name.as_str ( ) , j.status.as_str ( ) ) ) .collect ( ) ;
743
assert_eq! ( names.len ( ) , 3 ) ;
744
let running = one.jobs.iter ( ) .find ( | j | j.status = = " running" ) .unwrap ( ) ;
745
assert_eq! ( running.name, a[ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) ) ;
746
assert! ( one.jobs.iter ( ) .any ( | j | j.name = = " test" & & j.depends_on = = [ " build" ] ) ) ;
747
748
let job = running.name.clone ( ) ;
749
let ( st, body) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/demo/flow/runs/1/jobs/{job} /log" ) , None , None ) .await;
750
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
751
let lines: Vec< hub_api:: FlowLogLine> = serde_json:: from_str( & body) .unwrap ( ) ;
752
assert! ( lines.iter ( ) .any ( | l | l.text = = " three" ) , " {body}" ) ;
753
let last = lines.last ( ) .unwrap ( ) .seq;
754
let ( _ , body) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/demo/flow/runs/1/jobs/{job} /log?after={last} " ) , None , None ) .await;
755
assert_eq! ( body, " []" , " nothing after the last line" ) ;
756
assert_eq! ( call ( & h.app, " GET" , " /api/v1/repos/andy/demo/flow/runs/9" , None , None ) .await.0 , StatusCode:: NOT_FOUND ) ;
757
assert_eq! ( call ( & h.app, " GET" , " /api/v1/repos/andy/demo/flow/runs/1/jobs/nope/log" , None , None ) .await.0 , StatusCode:: NOT_FOUND ) ;
758
759
760
let by = | key : String| {
761
let app = h.app.clone ( ) ;
762
async move {
763
let ( st, body) = call ( & app, " GET" , & format! ( " /api/v1/repos/andy/demo/checks/{key} " ) , None , None ) .await;
764
assert_eq! ( st, StatusCode:: OK , " {key}: {body}" ) ;
765
serde_json:: from_str:: < Vec< hub_api:: Check> > ( & body) .unwrap ( ) .len ( )
766
}
767
} ;
768
checks_when ( & h.app, run.repo_id, & run.state, 3 ) .await;
769
assert_eq! ( by ( run.state.clone ( ) ) .await, 3 ) ;
770
assert_eq! ( by ( head.to_hex ( ) ) .await, 3 , " a single head names its state" ) ;
771
assert_eq! ( by ( " main" .into ( ) ) .await, 3 ) ;
772
let ( st, _ ) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/checks/no-such-bookmark" , None , None ) .await;
773
assert_eq! ( st, StatusCode:: NOT_FOUND ) ;
774
775
776
let ( st, body) = call ( & h.app, " GET" , " /api/v1/users/andy/repos" , None , None ) .await;
777
assert_eq! ( st, StatusCode:: NOT_FOUND ) ;
778
let v: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
779
assert_eq! ( v[ " status" ] , 404 , " {body}" ) ;
1edac4b6 Andy McCall 1 month ago ↶
780
}
cc19697f Andy McCall 1 month ago ↶
781
782
async fn get_bytes ( app : & Shared, path : & str ) -> ( StatusCode, Vec< u8 > ) {
783
let res = oxygenhub:: router( app.clone ( ) ) .oneshot ( Request:: get( path) .body ( Body:: empty( ) ) .unwrap ( ) ) .await.unwrap ( ) ;
784
let status = res.status ( ) ;
785
( status, res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) .to_vec ( ) )
786
}
787
788
# [ tokio ::test ( flavor = " multi_thread" ) ]
789
async fn signed_receipts_are_checked_kept_and_shown ( ) {
790
use oxy_core:: hash:: { hash_object, ChangeHash, ObjectRef, ObjectType} ;
791
use oxy_core:: identity:: SecretKey;
792
use oxy_core:: objects:: StateObject;
793
use oxy_core:: receipt:: SignedReceipt;
794
let h = hub ( ) .await;
795
create ( & h.app, " andy" , " demo" ) .await;
796
let url = format! ( " {} /andy/demo" , h.base) ;
797
let work = tempfile:: tempdir( ) .unwrap ( ) ;
798
let c = client ( & url, work.path ( ) ) ;
799
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
800
c.snapshot ( " test" ) .unwrap ( ) ;
801
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
802
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
803
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
804
let heads: Vec< ChangeHash> = run.head_list ( ) .into_iter ( ) .filter_map ( ChangeHash:: from_hex) .collect ( ) ;
805
let state = hash_object ( ObjectType:: State, & StateObject:: new( heads.clone ( ) ) .encode ( ) ) ;
806
let repo_id = h.app.repos.open ( & RepoName:: parse( " andy/demo" ) .unwrap ( ) ) .unwrap ( ) .id ( ) .unwrap ( ) ;
807
let sk = SecretKey:: generate( ) ;
808
let runner = sk.identity ( " test-1 (OxygenFlow runner)" , " " , 1 ) ;
809
let sign = | r : & serde_json:: Value, state : ObjectRef, sk : & SecretKey, id : & oxy_core:: identity:: Identity| {
810
let r: flow_core:: Receipt = serde_json:: from_value( r.clone ( ) ) .unwrap ( ) ;
811
let run_url = format! ( " {url} /flow/{} " , run.number) ;
812
hex:: encode( SignedReceipt:: sign( r.to_oxy ( ) , repo_id, state, & run_url, id, sk) .encode ( ) )
813
} ;
814
815
let a = claim ( & h.app) .await.unwrap ( ) ;
816
let b = claim ( & h.app) .await.unwrap ( ) ;
817
let ph = a[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
818
let ( a_id, b_id) = ( a[ " job_id" ] .as_i64 ( ) .unwrap ( ) , b[ " job_id" ] .as_i64 ( ) .unwrap ( ) ) ;
819
820
let rb = receipt ( " build" , & ph, 0 ) ;
821
let body = serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 , " receipt" : rb, " signed_receipt" : sign ( & rb, state, & sk, & runner) } ) ;
822
assert_eq! ( finish ( & h.app, a_id, body) .await, StatusCode:: NO_CONTENT ) ;
823
824
let other = SecretKey:: generate( ) ;
825
let rl = receipt ( " lint" , & ph, 0 ) ;
826
let body = serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 300 , " receipt" : rl,
827
" signed_receipt" : sign ( & rl, state, & other, & other.identity ( " mallory" , " " , 1 ) ) } ) ;
828
assert_eq! ( finish ( & h.app, b_id, body) .await, StatusCode:: NO_CONTENT ) ;
829
830
let t = claim ( & h.app) .await.unwrap ( ) ;
831
let rt = receipt ( " test" , & ph, 0 ) ;
832
let body = serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 500 , " receipt" : rt,
833
" signed_receipt" : sign ( & rt, ObjectRef( [ 9 ; 32 ] ) , & sk, & runner) } ) ;
834
assert_eq! ( finish ( & h.app, t[ " job_id" ] .as_i64 ( ) .unwrap ( ) , body) .await, StatusCode:: NO_CONTENT ) ;
835
836
let jobs = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) ;
837
let job = | n : & str | jobs.iter ( ) .find ( | j | j.name = = n ) .unwrap ( ) .clone ( ) ;
838
assert_eq! ( job ( " build" ) .status, " passed" ) ;
839
assert_eq! ( ( job ( " build" ) .receipt_attest.len ( ) , job ( " build" ) .receipt_signer.as_str ( ) ) , ( 64 , " test-1 (OxygenFlow runner)" ) ) ;
840
assert_eq! ( job ( " lint" ) .status, " passed" , " the job's result stands" ) ;
841
assert! ( job ( " lint" ) .receipt_attest.is_empty ( ) ) ;
842
assert! ( job ( " lint" ) .receipt_note.contains ( " not the key this runner signs with" ) , " {}" , job ( " lint" ) .receipt_note) ;
843
assert! ( job ( " test" ) .receipt_note.contains ( " another state" ) , " {}" , job ( " test" ) .receipt_note) ;
844
845
846
847
let kept = h.app.repos.open ( & RepoName:: parse( " andy/demo" ) .unwrap ( ) ) .unwrap ( ) .receipts ( & heads) .unwrap ( ) ;
848
assert_eq! ( kept.len ( ) , 1 ) ;
849
let attest = job ( " build" ) .receipt_attest;
850
let ( st, bytes) = get_bytes ( & h.app, & format! ( " /andy/demo/receipts/{attest} " ) ) .await;
851
assert_eq! ( st, StatusCode:: OK ) ;
852
let sr = SignedReceipt:: decode( & bytes) .unwrap ( ) ;
853
sr.verify ( ) .unwrap ( ) ;
854
assert_eq! ( ( sr.state ( ) , sr.receipt.job_name.as_str ( ) ) , ( state, " build" ) ) ;
855
assert_eq! ( get_bytes ( & h.app, & format! ( " /andy/demo/receipts/{} " , " 0" .repeat ( 64 ) ) ) .await.0 , StatusCode:: NOT_FOUND ) ;
856
857
858
859
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/demo/flow/{} " , run.number) , None , None ) .await;
860
assert! ( page.contains ( " signed receipt</a> by test-1" ) , " run page" ) ;
861
assert! ( page.contains ( " receipt not kept" ) , " run page" ) ;
862
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/demo/change/{} " , heads[ 0 ] .to_hex ( ) ) , None , None ) .await;
863
assert! ( page.contains ( " <dt>Receipts</dt>" ) & & page.contains ( & format! ( " /andy/demo/receipts/{attest} " ) ) , " change page" ) ;
864
}
a3c8288b Andy McCall 1 month ago ↶
865
866
async fn upload ( app : & Shared, job : i64 , name : & str , bytes : & [u8 ]) -> ( StatusCode, String) {
867
let req = Request:: post( format! ( " /api/v1/flow/jobs/{job} /artifacts/{name} " ) )
868
.header ( header:: AUTHORIZATION , " Bearer runner-secret" )
869
.header ( header:: CONTENT_TYPE , " application/octet-stream" )
870
.body ( Body:: from( bytes.to_vec ( ) ) )
871
.unwrap ( ) ;
872
let res = oxygenhub:: router( app.clone ( ) ) .oneshot ( req) .await.unwrap ( ) ;
873
let status = res.status ( ) ;
874
( status, String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) )
875
}
876
877
# [ tokio ::test ( flavor = " multi_thread" ) ]
878
async fn artifacts_are_kept_named_by_the_receipt_and_served ( ) {
879
let h = hub ( ) .await;
880
create ( & h.app, " andy" , " demo" ) .await;
881
let url = format! ( " {} /andy/demo" , h.base) ;
882
let work = tempfile:: tempdir( ) .unwrap ( ) ;
883
let c = client ( & url, work.path ( ) ) ;
884
write (
885
work.path ( ) ,
886
" .flow/pipeline.kdl" ,
887
& format! (
888
" schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" build\" {{ \n sandbox \" alpine\" \n exec \" make\" \n output {{ \n artifact \" tool\" {{ \n path \" /src/out/tool\" \n }} \n }} \n }} \n "
889
) ,
890
) ;
891
c.snapshot ( " test" ) .unwrap ( ) ;
892
c.commit ( " Build a tool" , & [ ] ) .unwrap ( ) ;
893
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
894
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
895
let a = claim ( & h.app) .await.unwrap ( ) ;
896
assert_eq! ( a[ " job" ] [ " outputs" ] , serde_json:: json! ( [ { " name" : " tool" , " path" : " /src/out/tool" } ] ) , " the runner is told" ) ;
897
let id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
898
899
let body: Vec< u8 > = ( 0 .. 300_000 u32 ) .map ( | i | ( i % 253 ) as u8 ) .collect ( ) ;
900
let ( st, text) = upload ( & h.app, id, " other" , & body) .await;
901
assert_eq! ( st, StatusCode:: BAD_REQUEST , " an undeclared artifact: {text}" ) ;
902
let ( st, text) = upload ( & h.app, id, " tool" , & body) .await;
903
assert_eq! ( st, StatusCode:: OK , " {text}" ) ;
904
let blob = serde_json:: from_str:: < serde_json:: Value> ( & text) .unwrap ( ) [ " blob" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
905
assert_eq! ( blob, oxy_core:: repo:: Repo:: blob_ref( & body) .to_hex ( ) , " the same blob the runner computes" ) ;
906
907
908
let mut r = receipt ( " build" , a[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) , 0 ) ;
909
assert_eq! (
910
finish ( & h.app, id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 , " receipt" : r.clone ( ) } ) ) .await,
911
StatusCode:: BAD_REQUEST
912
) ;
913
r[ " outputs" ] = serde_json:: json! ( [ blob] ) ;
914
assert_eq! ( finish ( & h.app, id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 , " receipt" : r } ) ) .await, StatusCode:: NO_CONTENT ) ;
915
916
917
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/demo/flow/{} " , run.number) , None , None ) .await;
918
let link = format! ( " /andy/demo/flow/{} /artifacts/build/tool" , run.number) ;
919
assert! ( page.contains ( & link) , " run page lists the artifact" ) ;
16ee5d19 Andy McCall 18 days ago ↶
920
921
922
let ( _ , list) = call ( & h.app, " GET" , " /andy/demo/artifacts" , None , None ) .await;
923
assert! ( list.contains ( & link) & & list.contains ( & format! ( " #{} </a>" , run.number) ) , " {list}" ) ;
924
assert! ( list.contains ( " <summary class=\" on\" >Artifacts</summary>" ) & & list.contains ( " href=\" /andy/demo/images\" " ) , " {list}" ) ;
a3c8288b Andy McCall 1 month ago ↶
925
let ( st, got) = get_bytes ( & h.app, & link) .await;
926
assert_eq! ( ( st, got = = body) , ( StatusCode:: OK , true ) ) ;
927
assert_eq! ( get_bytes ( & h.app, & format! ( " /andy/demo/flow/{} /artifacts/build/none" , run.number) ) .await.0 , StatusCode:: NOT_FOUND ) ;
076c7a7c Andy McCall 19 days ago ↶
928
929
930
let prov = format! ( " {link} /provenance" ) ;
931
assert_eq! ( get_bytes ( & h.app, & prov) .await.0 , StatusCode:: BAD_REQUEST , " no kept receipt, no provenance" ) ;
932
sqlx:: query( " UPDATE flow_jobs SET receipt_attest = 'a1b2', receipt_signer = 'test-1 runner' WHERE id = $1" )
933
.bind ( id)
934
.execute ( & h.app.db)
935
.await
936
.unwrap ( ) ;
937
let ( st, env) = get_bytes ( & h.app, & prov) .await;
938
assert_eq! ( st, StatusCode:: OK ) ;
939
let env: serde_json:: Value = serde_json:: from_slice( & env) .unwrap ( ) ;
940
assert_eq! ( env[ " payloadType" ] , " application/vnd.in-toto+json" ) ;
681db69e Andy McCall 19 days ago ↶
941
let unb64 = | t : & str | oxy_core:: dsse:: base64_decode( t) .expect ( " base64" ) ;
076c7a7c Andy McCall 19 days ago ↶
942
let payload = unb64 ( env[ " payload" ] .as_str ( ) .unwrap ( ) ) ;
943
let sig = unb64 ( env[ " signatures" ] [ 0 ] [ " sig" ] .as_str ( ) .unwrap ( ) ) ;
944
let signed = oxygenhub:: provenance:: pae( " application/vnd.in-toto+json" , & payload) ;
945
ring:: signature:: UnparsedPublicKey:: new( & ring:: signature:: ED25519 , h.app.hub_key.identity.ed25519_pk)
946
.verify ( & signed, & sig)
947
.expect ( " the hub's key signed the envelope" ) ;
948
let statement: serde_json:: Value = serde_json:: from_slice( & payload) .unwrap ( ) ;
949
use sha2:: Digest;
950
assert_eq! ( statement[ " subject" ] [ 0 ] [ " digest" ] [ " sha256" ] , hex:: encode( sha2:: Sha256:: digest( & body) ) ) ;
951
assert_eq! ( statement[ " predicateType" ] , " https://slsa.dev/provenance/v1" ) ;
952
assert_eq! ( statement[ " predicate" ] [ " buildDefinition" ] [ " externalParameters" ] [ " job" ] , " build" ) ;
953
assert_eq! ( statement[ " predicate" ] [ " runDetails" ] [ " byproducts" ] [ 0 ] [ " digest" ] [ " oxygen-attestation" ] , " a1b2" ) ;
954
let ( _ , pem) = get_bytes ( & h.app, " /.well-known/oxyhub-signing.pem" ) .await;
955
let pem = String :: from_utf8( pem) .unwrap ( ) ;
956
let der = unb64 ( pem.lines ( ) .filter ( | l | ! l.starts_with ( " -----" ) ) .collect:: < String> ( ) .as_str ( ) ) ;
957
assert_eq! ( & der[ der.len ( ) - 32 .. ] , & h.app.hub_key.identity.ed25519_pk[ .. ] , " the PEM holds the hub's key" ) ;
958
sqlx:: query( " UPDATE flow_jobs SET receipt_attest = '', receipt_signer = '' WHERE id = $1" ) .bind ( id) .execute ( & h.app.db) .await.unwrap ( ) ;
10708ab8 Andy McCall 1 month ago ↶
959
45977501 Andy McCall 25 days ago ↶
960
961
962
{
963
let name = RepoName:: parse( " andy/demo" ) .unwrap ( ) ;
964
let mut hosted = oxy_core:: repo:: Repo:: open( & h.app.repos.dir ( & name) ) .unwrap ( ) ;
965
hosted.config.set ( " core.gc-grace" , " 0" ) .unwrap ( ) ;
966
hosted.save_config ( ) .unwrap ( ) ;
967
}
968
let done = oxygenhub:: maintenance:: gc_all( & h.app) .await;
969
assert! ( done.iter ( ) .all ( | d | ! d.failed) , " {done:?}" ) ;
970
let ( st, got) = get_bytes ( & h.app, & link) .await;
971
assert_eq! ( ( st, got = = body) , ( StatusCode:: OK , true ) , " the artifact survives gc" ) ;
972
10708ab8 Andy McCall 1 month ago ↶
973
974
975
let form = | f : String| {
976
Request:: post( " /register" ) .header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) .body ( Body:: from( f) ) .unwrap ( )
977
} ;
978
let res = oxygenhub:: router( h.app.clone ( ) )
c8684d44 Andy McCall 20 days ago
979
.oneshot ( form ( " username=andy&email=a%40example.com&password=correct-horse-battery&setup=test-setup" .into ( ) ) )
10708ab8 Andy McCall 1 month ago ↶
980
.await
981
.unwrap ( ) ;
982
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
983
let with_cookie = | method : & str , path : & str , body : Option< String> | {
984
let mut req = Request:: builder( ) .method ( method) .uri ( path) .header ( header:: COOKIE , cookie.clone ( ) ) ;
985
if body.is_some ( ) {
986
req = req.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) ;
987
}
988
req.body ( body.map ( Body:: from) .unwrap_or_else ( Body:: empty) ) .unwrap ( )
989
} ;
990
let row = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
991
db:: releases:: create( & h.app.db, row.id, " v1" , " One" , " " , false , None ) .await.unwrap ( ) ;
992
let res = oxygenhub:: router( h.app.clone ( ) ) .oneshot ( with_cookie ( " GET" , & format! ( " /andy/demo/flow/{} " , run.number) , None ) ) .await.unwrap ( ) ;
993
let page = String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) ;
994
let attach = format! ( " /andy/demo/flow/{} /artifacts/build/tool/attach" , run.number) ;
995
assert! ( page.contains ( & attach) , " a writer is offered the release" ) ;
996
let i = page.find ( " name=\" _csrf\" value=\" " ) .unwrap ( ) + " name=\" _csrf\" value=\" " .len ( ) ;
997
let csrf = page[ i.. ] .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( ) ;
998
let release_id = db:: releases:: list( & h.app.db, row.id) .await.unwrap ( ) [ 0 ] .id;
999
let res = oxygenhub:: router( h.app.clone ( ) )
1000
.oneshot ( with_cookie ( " POST" , & attach, Some ( format! ( " _csrf={csrf} &release={release_id} " ) ) ) )
1001
.await
1002
.unwrap ( ) ;
1003
assert_eq! ( res.status ( ) , StatusCode:: SEE_OTHER ) ;
1004
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/releases/tag/v1" , None , None ) .await;
1005
assert! (
1006
page.contains ( " >tool</a>" ) & & page.contains ( & format! ( " built by <a href=\" /andy/demo/flow/{0} \" >run #{0} </a> build" , run.number) ) ,
1007
" {page}"
1008
) ;
1009
let ( _ , api) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/releases" , None , None ) .await;
1010
let api: serde_json:: Value = serde_json:: from_str( & api) .unwrap ( ) ;
1011
let asset = & api[ 0 ] [ " assets" ] [ 0 ] ;
1012
assert_eq! (
1013
( asset[ " name" ] .as_str ( ) , asset[ " built_by" ] [ " run" ] .as_i64 ( ) , asset[ " built_by" ] [ " job" ] .as_str ( ) ) ,
1014
( Some ( " tool" ) , Some ( run.number) , Some ( " build" ) )
1015
) ;
1016
assert! ( asset[ " built_by" ] [ " receipt" ] .is_null ( ) , " that receipt was not signed" ) ;
97fc47fe Andy McCall 19 days ago ↶
1017
assert! ( asset[ " built_by" ] [ " provenance_url" ] .is_null ( ) , " no signed receipt, no provenance" ) ;
1018
1019
1020
sqlx:: query( " UPDATE flow_jobs SET receipt_attest = 'a1b2', receipt_signer = 'earth-1' WHERE run_id = $1 AND name = 'build'" )
1021
.bind ( run.id)
1022
.execute ( & h.app.db)
1023
.await
1024
.unwrap ( ) ;
1025
let want = format! ( " /andy/demo/flow/{} /artifacts/build/tool/provenance" , run.number) ;
1026
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/releases/tag/v1" , None , None ) .await;
1027
assert! ( page.contains ( & format! ( " href=\" {want} \" " ) ) , " {page}" ) ;
1028
let ( _ , api) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/releases" , None , None ) .await;
1029
let api: serde_json:: Value = serde_json:: from_str( & api) .unwrap ( ) ;
1030
assert! ( api[ 0 ] [ " assets" ] [ 0 ] [ " built_by" ] [ " provenance_url" ] .as_str ( ) .unwrap ( ) .ends_with ( & want) ) ;
a3c8288b Andy McCall 1 month ago ↶
1031
}
df940d75 Andy McCall 1 month ago ↶
1032
d29070fa Andy McCall 1 month ago ↶
1033
1034
a36bab6b Andy McCall 25 days ago ↶
1035
async fn upload_report_file ( app : & Shared, job : i64 , name : & str , path : & str , bytes : & [u8 ]) -> ( StatusCode, String) {
1036
let req = Request:: post( format! ( " /api/v1/flow/jobs/{job} /reports/{name} /files" ) )
1037
.header ( header:: AUTHORIZATION , " Bearer runner-secret" )
1038
.header ( header:: CONTENT_TYPE , " application/octet-stream" )
1039
.header ( " X-Flow-Report-Path" , path)
1040
.body ( Body:: from( bytes.to_vec ( ) ) )
1041
.unwrap ( ) ;
1042
let res = oxygenhub:: router( app.clone ( ) ) .oneshot ( req) .await.unwrap ( ) ;
1043
let status = res.status ( ) ;
1044
( status, String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) )
1045
}
1046
1047
1048
1049
1050
# [ tokio ::test ( flavor = " multi_thread" ) ]
1051
async fn reports_are_kept_file_by_file_and_indexed ( ) {
1052
let h = hub ( ) .await;
1053
create ( & h.app, " andy" , " demo" ) .await;
1054
let url = format! ( " {} /andy/demo" , h.base) ;
1055
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1056
let c = client ( & url, work.path ( ) ) ;
1057
write (
1058
work.path ( ) ,
1059
" .flow/pipeline.kdl" ,
1060
& format! (
1061
" schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" perf\" {{ \n sandbox \" alpine\" \n exec \" make\" \n output {{ \n artifact \" tool\" {{ \n path \" /src/out/tool\" \n }} \n report \" frame-timing\" {{ \n path \" /src/out/report\" \n when \" always\" \n }} \n }} \n }} \n "
1062
) ,
1063
) ;
1064
c.snapshot ( " test" ) .unwrap ( ) ;
1065
c.commit ( " Measure frames" , & [ ] ) .unwrap ( ) ;
1066
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1067
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
1068
let a = claim ( & h.app) .await.unwrap ( ) ;
1069
let id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
1070
1071
let toml = b " kind = \" performance\" \n title = \" Frame timing\" \n summary = \" Median 16.4 ms\" \n \n [metrics]\n median_ms = { value = 16.4, unit = \" ms\" , better = \" lower\" , fail_above = 20 }\n \n [context]\n target_fps = 30\n " ;
1072
let page = b " <p>frames</p>" .to_vec ( ) ;
1073
let data: Vec< u8 > = ( 0 .. 100_000 u32 ) .map ( | i | ( i % 241 ) as u8 ) .collect ( ) ;
c8df33a6 Andy McCall 25 days ago ↶
1074
let ( page_bytes, data_bytes) = ( page.clone ( ) , data.clone ( ) ) ;
a36bab6b Andy McCall 25 days ago ↶
1075
let files: [ ( & str , & [ u8 ] ) ; 3 ] = [ ( " data/frames.csv" , & data) , ( " index.html" , & page) , ( " report.toml" , toml) ] ;
1076
1077
assert_eq! ( upload_report_file ( & h.app, id, " tool" , " report.toml" , toml) .await.0 , StatusCode:: BAD_REQUEST , " not a report" ) ;
1078
assert_eq! ( upload_report_file ( & h.app, id, " frame-timing" , " ../x" , toml) .await.0 , StatusCode:: BAD_REQUEST , " not a clean path" ) ;
1079
let mut listed = vec! [ ] ;
1080
for ( path, bytes) in files {
1081
let ( st, text) = upload_report_file ( & h.app, id, " frame-timing" , path, bytes) .await;
1082
assert_eq! ( st, StatusCode:: OK , " {text}" ) ;
1083
let blob = serde_json:: from_str:: < serde_json:: Value> ( & text) .unwrap ( ) [ " blob" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
1084
assert_eq! ( blob, oxy_core:: repo:: Repo:: blob_ref( bytes) .to_hex ( ) ) ;
1085
listed.push ( serde_json:: json! ( { " path" : path, " blob" : blob, " size" : bytes.len ( ) } ) ) ;
1086
}
1087
let mut missing = listed.clone ( ) ;
1088
missing.push ( serde_json:: json! ( { " path" : " assets/x.js" , " blob" : " ab" .repeat ( 32 ) , " size" : 1 } ) ) ;
1089
let ( st, text) =
1090
upload ( & h.app, id, " frame-timing" , & serde_json:: to_vec( & serde_json:: json! ( { " version" : 1 , " files" : missing } ) ) .unwrap ( ) ) .await;
1091
assert_eq! ( st, StatusCode:: BAD_REQUEST , " a file never uploaded: {text}" ) ;
1092
assert! ( text.contains ( " assets/x.js" ) , " {text}" ) ;
1093
let manifest = serde_json:: to_vec( & serde_json:: json! ( { " version" : 1 , " files" : listed } ) ) .unwrap ( ) ;
1094
let ( st, text) = upload ( & h.app, id, " frame-timing" , & manifest) .await;
1095
assert_eq! ( st, StatusCode:: OK , " {text}" ) ;
1096
1097
let ( kind, title, summary, files_n, bytes) : ( String , String , String , i64 , i64 ) =
3b1bc27f Andy McCall 20 days ago
1098
sqlx:: query_as( " SELECT kind, title, summary, files, bytes FROM reports WHERE job_id = $1 AND name = 'frame-timing'" )
a36bab6b Andy McCall 25 days ago ↶
1099
.bind ( id)
1100
.fetch_one ( & h.app.db)
1101
.await
1102
.unwrap ( ) ;
1103
assert_eq! ( ( kind.as_str ( ) , title.as_str ( ) , summary.as_str ( ) ) , ( " performance" , " Frame timing" , " Median 16.4 ms" ) ) ;
1104
assert_eq! ( ( files_n, bytes) , ( 3 , ( data.len ( ) + page.len ( ) + toml.len ( ) ) as i64 ) ) ;
3b1bc27f Andy McCall 20 days ago
1105
let rid: i64 = sqlx:: query_scalar( " SELECT id FROM reports WHERE job_id = $1" ) .bind ( id) .fetch_one ( & h.app.db) .await.unwrap ( ) ;
a36bab6b Andy McCall 25 days ago ↶
1106
let m = oxygenhub:: reports:: metrics( & h.app.db, rid) .await.unwrap ( ) ;
1107
assert_eq! ( ( m[ 0 ] .key.as_str ( ) , m[ 0 ] .value, m[ 0 ] .fail_above) , ( " median_ms" , 16. 4 , Some ( 20. 0 ) ) ) ;
1108
let log = db:: flow:: log( & h.app.db, id, 0 , 1000 ) .await.unwrap ( ) ;
1109
assert! ( log.iter ( ) .any ( | r | r.text.contains ( " report frame-timing: performance \" Frame timing\" , 1 metric" ) ) , " {log:?}" ) ;
c8df33a6 Andy McCall 25 days ago ↶
1110
1111
1112
let checks = db:: hooks:: checks_for_state( & h.app.db, run.repo_id, & run.state) .await.unwrap ( ) ;
1113
let c = checks.iter ( ) .find ( | c | c.name = = " report/frame-timing/median_ms" ) .expect ( " a check per threshold" ) ;
1114
assert_eq! ( c.status, " passed" , " {c:?}" ) ;
1115
assert! ( c.summary.contains ( " 16.4 ms" ) & & c.url.contains ( " /reports/frame-timing?id=" ) , " {c:?}" ) ;
1116
1117
1118
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo" , None , None ) .await;
1119
assert! ( page.contains ( " >Reports</a>" ) , " the tab" ) ;
1120
let ( st, page) = call ( & h.app, " GET" , " /andy/demo/reports" , None , None ) .await;
1121
assert! ( st = = StatusCode:: OK & & page.contains ( " Frame timing" ) & & page.contains ( " median_ms" ) , " {st}" ) ;
1122
let ( st, page) = call ( & h.app, " GET" , " /andy/demo/reports/frame-timing" , None , None ) .await;
1123
assert! ( st = = StatusCode:: OK & & page.contains ( " fails above 20" ) & & page.contains ( " sandbox=\" allow-scripts\" " ) , " {st}" ) ;
1124
let src = page.split ( " class=\" rep-frame\" src=\" " ) .nth ( 1 ) .unwrap ( ) .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( ) ;
1125
assert! ( src.ends_with ( " /index.html" ) , " {src}" ) ;
1126
let res = oxygenhub:: router( h.app.clone ( ) ) .oneshot ( Request:: get( & src) .body ( Body:: empty( ) ) .unwrap ( ) ) .await.unwrap ( ) ;
1127
assert_eq! ( res.status ( ) , StatusCode:: OK ) ;
1128
let csp = res.headers ( ) [ header:: CONTENT_SECURITY_POLICY ] .to_str ( ) .unwrap ( ) .to_string ( ) ;
1129
assert! ( csp.starts_with ( " sandbox allow-scripts;" ) & & csp.contains ( " default-src 'none'" ) , " {csp}" ) ;
d74ce22f Andy McCall 20 days ago ↶
1130
1131
assert! (
1132
res.headers ( ) .get ( header:: X_FRAME_OPTIONS ) .is_none ( ) & & res.headers ( ) .get ( header:: CONTENT_SECURITY_POLICY_REPORT_ONLY ) .is_none ( )
1133
) ;
1134
assert_eq! ( res.headers ( ) [ header:: X_CONTENT_TYPE_OPTIONS ] , " nosniff" ) ;
c8df33a6 Andy McCall 25 days ago ↶
1135
assert_eq! ( res.headers ( ) [ header:: ACCESS_CONTROL_ALLOW_ORIGIN ] , " *" ) ;
f3e5e2bc Andy McCall 25 days ago ↶
1136
1137
let served = res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) .to_vec ( ) ;
1138
assert! ( served.starts_with ( & page_bytes) , " the report's page" ) ;
1139
assert! ( String :: from_utf8_lossy( & served) .contains ( " /static/report-frame.js" ) , " with the frame script" ) ;
1140
let ( st, _ ) = get_bytes ( & h.app, " /static/report.css" ) .await;
1141
assert_eq! ( st, StatusCode:: OK , " the stylesheet a report may link" ) ;
c8df33a6 Andy McCall 25 days ago ↶
1142
let data = src.replace ( " /index.html" , " /data/frames.csv" ) ;
1143
assert_eq! ( get_bytes ( & h.app, & data) .await, ( StatusCode:: OK , data_bytes.clone ( ) ) ) ;
1144
95b525b0 Andy McCall 24 days ago ↶
1145
1146
1147
let first: i64 = sqlx:: query_scalar( " SELECT id FROM reports WHERE name = 'frame-timing'" ) .fetch_one ( & h.app.db) .await.unwrap ( ) ;
1148
let second: i64 = sqlx:: query_scalar(
1149
" INSERT INTO reports (repo_id, name, kind, title, summary, context, run_id, job_id, run_number, job, bookmark, state, manifest, files, bytes, created_ms) \
3b1bc27f Andy McCall 20 days ago
1150
SELECT repo_id, name, kind, title, 'Median 18.0 ms', context, run_id, NULL, 2, job, bookmark, state, manifest, files, bytes, created_ms + 1000 FROM reports WHERE id = $1 RETURNING id" ,
95b525b0 Andy McCall 24 days ago ↶
1151
)
1152
.bind ( first)
1153
.fetch_one ( & h.app.db)
1154
.await
1155
.unwrap ( ) ;
1156
sqlx:: query(
3b1bc27f Andy McCall 20 days ago
1157
" INSERT INTO report_metrics (report_id, key, value, unit, better, fail_above) VALUES ($1, 'median_ms', 18.0, 'ms', 'lower', 20)" ,
95b525b0 Andy McCall 24 days ago ↶
1158
)
1159
.bind ( second)
1160
.execute ( & h.app.db)
1161
.await
1162
.unwrap ( ) ;
1163
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/reports/frame-timing" , None , None ) .await;
1164
assert_eq! ( page.matches ( " data-tl-stop" ) .count ( ) , 2 , " a stop per run" ) ;
1165
assert! ( page.contains ( " Median 18.0 ms" ) , " the newest run is shown" ) ;
1166
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/demo/reports/frame-timing?id={first} &compare={second} &play=3" ) , None , None ) .await;
1167
assert! ( page.contains ( " This run against run #2" ) & & page.contains ( " -1.6" ) , " every metric against the other run" ) ;
1168
let stop = page.split ( " data-tl-stop" ) .next ( ) .unwrap_or ( " " ) .rsplit ( " href=" ) .next ( ) .unwrap_or ( " " ) .to_string ( ) ;
1169
assert! ( page.contains ( " data-play=\" 3\" " ) , " playing" ) ;
1170
assert! ( stop.contains ( " play" ) & & stop.contains ( ' 3' ) , " playing carries on from stop to stop: {stop}" ) ;
1171
assert! ( page.contains ( " data-report-runs=" ) , " the frame interface knows the runs" ) ;
fac6853c Andy McCall 8 days ago ↶
1172
1173
1174
1175
let n = run.number;
1176
let ( st, body) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/demo/flow/runs/{n} /jobs/perf/report" ) , None , None ) .await;
1177
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1178
let r: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
1179
assert_eq! ( ( r[ " id" ] .as_i64 ( ) , r[ " name" ] .as_str ( ) , r[ " kind" ] .as_str ( ) ) , ( Some ( first) , Some ( " frame-timing" ) , Some ( " performance" ) ) ) ;
1180
assert_eq! ( ( r[ " metrics" ] [ 0 ] [ " key" ] .as_str ( ) , r[ " metrics" ] [ 0 ] [ " fail_above" ] .as_f64 ( ) ) , ( Some ( " median_ms" ) , Some ( 20. 0 ) ) ) ;
1181
assert_eq! ( r[ " context" ] [ " target_fps" ] , 30 ) ;
1182
let paths: Vec< & str > = r[ " manifest" ] .as_array ( ) .unwrap ( ) .iter ( ) .map ( | f | f[ " path" ] .as_str ( ) .unwrap ( ) ) .collect ( ) ;
1183
assert_eq! ( paths, [ " data/frames.csv" , " index.html" , " report.toml" ] ) ;
1184
assert! ( r[ " url" ] .as_str ( ) .unwrap ( ) .ends_with ( & format! ( " /andy/demo/reports/frame-timing?id={first} " ) ) ) ;
1185
let ( st, body) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/demo/flow/runs/{n} /jobs/perf/reports" ) , None , None ) .await;
1186
assert! ( st = = StatusCode:: OK & & body.contains ( " \" frame-timing\" " ) & & ! body.contains ( " manifest" ) , " {st} {body}" ) ;
1187
let res = oxygenhub:: router( h.app.clone ( ) )
1188
.oneshot (
1189
Request:: get( format! ( " /api/v1/repos/andy/demo/flow/runs/{n} /jobs/perf/report/data/frames.csv" ) ) .body ( Body:: empty( ) ) .unwrap ( ) ,
1190
)
1191
.await
1192
.unwrap ( ) ;
1193
assert_eq! ( res.status ( ) , StatusCode:: OK ) ;
1194
assert_eq! ( res.headers ( ) [ header:: CONTENT_TYPE ] , " text/csv; charset=utf-8" ) ;
1195
assert_eq! ( res.headers ( ) [ header:: CONTENT_SECURITY_POLICY ] , " sandbox; default-src 'none'" , " a page in it runs nothing on the hub" ) ;
1196
assert_eq! ( res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) .to_vec ( ) , data_bytes) ;
1197
let toml_path = format! ( " /api/v1/repos/andy/demo/flow/runs/{n} /jobs/perf/report/report.toml?name=frame-timing" ) ;
1198
assert_eq! ( get_bytes ( & h.app, & toml_path) .await, ( StatusCode:: OK , toml.to_vec ( ) ) ) ;
1199
let ( st, body) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/demo/flow/runs/{n} /jobs/perf/report/nope.csv" ) , None , None ) .await;
1200
assert! ( st = = StatusCode:: NOT_FOUND & & body.contains ( " has no file nope.csv" ) & & body.contains ( " report.toml" ) , " {st} {body}" ) ;
1201
let ( st, body) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/demo/flow/runs/{n} /jobs/perf/report?name=other" ) , None , None ) .await;
1202
assert! ( st = = StatusCode:: NOT_FOUND & & body.contains ( " it published frame-timing" ) , " {st} {body}" ) ;
1203
let ( st, body) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/demo/flow/runs/{n} /jobs/nojob/report" ) , None , None ) .await;
1204
assert! ( st = = StatusCode:: NOT_FOUND & & body.contains ( " its jobs: perf" ) , " {st} {body}" ) ;
1205
1206
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/reports" , None , None ) .await;
1207
let latest: Vec< serde_json:: Value> = serde_json:: from_str( & body) .unwrap ( ) ;
1208
assert_eq! ( ( latest.len ( ) , latest[ 0 ] [ " id" ] .as_i64 ( ) , latest[ 0 ] [ " run" ] .as_i64 ( ) ) , ( 1 , Some ( second) , Some ( 2 ) ) ) ;
1209
assert_eq! ( latest[ 0 ] [ " metrics" ] [ 0 ] [ " value" ] .as_f64 ( ) , Some ( 18. 0 ) ) ;
1210
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/reports/frame-timing" , None , None ) .await;
1211
assert_eq! ( serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " id" ] .as_i64 ( ) , Some ( second) , " newest on its bookmark" ) ;
1212
let ( _ , body) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/demo/reports/frame-timing?id={first} " ) , None , None ) .await;
1213
assert_eq! ( serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " summary" ] , " Median 16.4 ms" ) ;
1214
let page_path = format! ( " /api/v1/repos/andy/demo/reports/frame-timing/files/index.html?id={first} " ) ;
1215
assert_eq! ( get_bytes ( & h.app, & page_path) .await, ( StatusCode:: OK , page_bytes.clone ( ) ) , " as published, no frame script" ) ;
1216
let ( st, body) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/reports/frame-timing?bookmark=elsewhere" , None , None ) .await;
1217
assert! ( st = = StatusCode:: NOT_FOUND & & body.contains ( " never ran on elsewhere" ) , " {st} {body}" ) ;
1218
let ( st, _ ) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/reports/nothing" , None , None ) .await;
1219
assert_eq! ( st, StatusCode:: NOT_FOUND ) ;
1220
1221
let phases = std:: collections:: BTreeMap:: from( [ ( " run" .to_string ( ) , 1200 u64 ) , ( " image" .to_string ( ) , 300 ) ] ) ;
1222
db:: flow:: set_phases( & h.app.db, id, & phases) .await.unwrap ( ) ;
1223
let ( _ , body) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/demo/flow/runs/{n} " ) , None , None ) .await;
1224
let v: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
1225
assert! ( v[ " jobs" ] [ 0 ] [ " queued_ms" ] .as_i64 ( ) .is_some ( ) , " {body}" ) ;
1226
assert_eq! ( v[ " jobs" ] [ 0 ] [ " phases" ] , serde_json:: json! ( [ { " name" : " image" , " ms" : 300 } , { " name" : " run" , " ms" : 1200 } ] ) ) ;
c8df33a6 Andy McCall 25 days ago ↶
1227
let forged = src.replacen ( " /files/" , " /files/9" , 1 ) ;
1228
assert_eq! ( get_bytes ( & h.app, & forged) .await.0 , StatusCode:: NOT_FOUND ) ;
1229
assert_eq! ( get_bytes ( & h.app, & src.replace ( " /index.html" , " /secret.txt" ) ) .await.0 , StatusCode:: NOT_FOUND ) ;
a36bab6b Andy McCall 25 days ago ↶
1230
let _ = run;
1231
1232
1233
{
1234
let name = RepoName:: parse( " andy/demo" ) .unwrap ( ) ;
1235
let mut hosted = oxy_core:: repo:: Repo:: open( & h.app.repos.dir ( & name) ) .unwrap ( ) ;
1236
hosted.config.set ( " core.gc-grace" , " 0" ) .unwrap ( ) ;
1237
hosted.save_config ( ) .unwrap ( ) ;
1238
}
1239
let done = oxygenhub:: maintenance:: gc_all( & h.app) .await;
1240
assert! ( done.iter ( ) .all ( | d | ! d.failed) , " {done:?}" ) ;
1241
let hosted = h.app.repos.open ( & RepoName:: parse( " andy/demo" ) .unwrap ( ) ) .unwrap ( ) ;
1242
for ( path, bytes) in files {
1243
assert_eq! ( hosted.read_blob ( & oxy_core:: repo:: Repo:: blob_ref( bytes) ) .unwrap ( ) , bytes, " {path} survives gc" ) ;
1244
}
1245
}
1246
d29070fa Andy McCall 1 month ago ↶
1247
# [ tokio ::test ( flavor = " multi_thread" ) ]
1248
async fn a_failing_job_keeps_an_artifact_outside_its_receipt ( ) {
1249
let h = hub ( ) .await;
1250
create ( & h.app, " andy" , " demo" ) .await;
1251
let url = format! ( " {} /andy/demo" , h.base) ;
1252
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1253
let c = client ( & url, work.path ( ) ) ;
1254
write (
1255
work.path ( ) ,
1256
" .flow/pipeline.kdl" ,
1257
& format! (
336c953a Andy McCall 1 month ago ↶
1258
" schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" test\" {{ \n sandbox \" alpine\" \n exec \" make\" \" test\" \n output {{ \n artifact \" junit\" {{ \n path \" /src/out/junit.xml\" \n when \" always\" \n }} \n }} \n }} \n "
d29070fa Andy McCall 1 month ago ↶
1259
) ,
1260
) ;
1261
c.snapshot ( " test" ) .unwrap ( ) ;
1262
c.commit ( " Test" , & [ ] ) .unwrap ( ) ;
1263
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1264
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
1265
let a = claim ( & h.app) .await.unwrap ( ) ;
336c953a Andy McCall 1 month ago ↶
1266
assert_eq! ( a[ " job" ] [ " outputs" ] [ 0 ] [ " always" ] , true , " the runner is told to keep it from a failing job" ) ;
d29070fa Andy McCall 1 month ago ↶
1267
let id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
0b203bbf Andy McCall 1 month ago ↶
1268
1269
let xml = br # "<testsuites><testsuite name="pytest"><testcase classname="tests.test_io" name="test_read" time="0.5"/><testcase classname="tests.test_io" name="test_write" time="1.5"><failure message="disk full">E OSError: disk full</failure></testcase></testsuite></testsuites>"# ;
1270
let ( st, text) = upload ( & h.app, id, " junit" , xml) .await;
d29070fa Andy McCall 1 month ago ↶
1271
assert_eq! ( st, StatusCode:: OK , " {text}" ) ;
1272
let blob = serde_json:: from_str:: < serde_json:: Value> ( & text) .unwrap ( ) [ " blob" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
1273
let mut r = receipt ( " test" , a[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) , 1 ) ;
1274
r[ " outputs" ] = serde_json:: json! ( [ blob] ) ;
1275
assert_eq! (
1276
finish ( & h.app, id, serde_json:: json! ( { " exit_code" : 1 , " duration_ms" : 700 , " receipt" : r.clone ( ) } ) ) .await,
1277
StatusCode:: BAD_REQUEST ,
1278
" a failing receipt names no outputs"
1279
) ;
1280
r[ " outputs" ] = serde_json:: json! ( [ ] ) ;
04683947 Andy McCall 9 days ago ↶
1281
1282
let phases = serde_json:: json! ( { " image" : 9000 , " prepare" : 1200 , " run" : 3400 , " finish" : 500 } ) ;
1283
assert_eq! (
1284
finish ( & h.app, id, serde_json:: json! ( { " exit_code" : 1 , " duration_ms" : 700 , " receipt" : r, " phases" : phases } ) ) .await,
1285
StatusCode:: NO_CONTENT
1286
) ;
d29070fa Andy McCall 1 month ago ↶
1287
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/demo/flow/{} " , run.number) , None , None ) .await;
04683947 Andy McCall 9 days ago ↶
1288
let at = page.find ( " Where its time went: " ) .expect ( " the job's phases" ) ;
1289
let line = & page[ at.. at + 120 ] ;
1290
let order: Vec< usize > = [ " image" , " prepare" , " run" , " finish" ] .iter ( ) .map ( | p | line.find ( p) .expect ( p) ) .collect ( ) ;
1291
assert! ( order.windows ( 2 ) .all ( | w | w[ 0 ] < w[ 1 ] ) , " in the order they happened: {line}" ) ;
d29070fa Andy McCall 1 month ago ↶
1292
assert! ( page.contains ( & format! ( " /andy/demo/flow/{} /artifacts/test/junit" , run.number) ) , " listed" ) ;
1293
assert! ( page.contains ( " not in its receipt" ) , " marked" ) ;
0b203bbf Andy McCall 1 month ago ↶
1294
assert! ( page.contains ( " 2 tests: 1 failed" ) , " counts beside the job" ) ;
1295
assert! ( page.contains ( " tests.test_io::test_write" ) & & page.contains ( " disk full" ) & & page.contains ( " OSError" ) , " the failure" ) ;
1296
assert! ( page.contains ( " tests: 2 tests: 1 failed in 2.0 s" ) , " said in the job's log" ) ;
d29070fa Andy McCall 1 month ago ↶
1297
}
1298
df940d75 Andy McCall 1 month ago ↶
1299
1300
1301
1302
# [ tokio ::test ( flavor = " multi_thread" ) ]
1303
async fn a_release_is_audited_from_authors_to_receipts ( ) {
1304
use oxy_core:: hash:: { hash_object, ChangeHash, ObjectType} ;
1305
use oxy_core:: identity:: SecretKey;
1306
use oxy_core:: objects:: StateObject;
1307
use oxy_core:: receipt:: SignedReceipt;
1308
let h = hub ( ) .await;
1309
create ( & h.app, " andy" , " demo" ) .await;
1310
let url = format! ( " {} /andy/demo" , h.base) ;
1311
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1312
let mut c = client ( & url, work.path ( ) ) ;
1313
c.create_identity ( " Andy" , " [email protected] " , oxy_core:: repo:: IdentityScope:: Repo) .unwrap ( ) ;
1314
write (
1315
work.path ( ) ,
1316
" .flow/pipeline.kdl" ,
1317
& format! (
1318
" schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" build\" {{ \n sandbox \" alpine\" \n exec \" make\" \n output {{ \n artifact \" tool\" {{ \n path \" /src/out/tool\" \n }} \n }} \n }} \n "
1319
) ,
1320
) ;
1321
c.snapshot ( " test" ) .unwrap ( ) ;
1322
c.commit ( " Build a tool" , & [ ] ) .unwrap ( ) ;
1323
let heads = c.bookmark_heads ( ) .unwrap ( ) ;
1324
c.tag_create ( " v1" , & heads, " first" ) .unwrap ( ) ;
1325
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1326
1327
1328
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
1329
let a = claim ( & h.app) .await.unwrap ( ) ;
1330
let id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
1331
let body: Vec< u8 > = b " #!/bin/sh\n echo tool\n " .to_vec ( ) ;
1332
let ( _ , text) = upload ( & h.app, id, " tool" , & body) .await;
1333
let blob = serde_json:: from_str:: < serde_json:: Value> ( & text) .unwrap ( ) [ " blob" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
1334
let mut r = receipt ( " build" , a[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) , 0 ) ;
1335
r[ " outputs" ] = serde_json:: json! ( [ blob] ) ;
1336
let heads_run: Vec< ChangeHash> = run.head_list ( ) .into_iter ( ) .filter_map ( ChangeHash:: from_hex) .collect ( ) ;
1337
let state = hash_object ( ObjectType:: State, & StateObject:: new( heads_run) .encode ( ) ) ;
1338
let repo_id = h.app.repos.open ( & RepoName:: parse( " andy/demo" ) .unwrap ( ) ) .unwrap ( ) .id ( ) .unwrap ( ) ;
1339
let sk = SecretKey:: generate( ) ;
1340
let runner = sk.identity ( " test-1 (OxygenFlow runner)" , " " , 1 ) ;
1341
let fr: flow_core:: Receipt = serde_json:: from_value( r.clone ( ) ) .unwrap ( ) ;
1342
let signed = hex:: encode( SignedReceipt:: sign( fr.to_oxy ( ) , repo_id, state, & format! ( " {url} /flow/1" ) , & runner, & sk) .encode ( ) ) ;
1343
let done = serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 900 , " receipt" : r, " signed_receipt" : signed } ) ;
1344
assert_eq! ( finish ( & h.app, id, done) .await, StatusCode:: NO_CONTENT ) ;
1345
1346
1347
let row = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
1348
let rel = db:: releases:: create( & h.app.db, row.id, " v1" , " One" , " " , false , None ) .await.unwrap ( ) ;
1349
oxygenhub:: web:: releases:: store_asset( & h.app, rel.id, " tool" , " " , & body) .await.unwrap ( ) ;
1350
oxygenhub:: web:: releases:: store_asset( & h.app, rel.id, " notes.txt" , " " , b " by hand\n " ) .await.unwrap ( ) ;
1351
1352
let ( st, page) = call ( & h.app, " GET" , " /andy/demo/releases/trust/v1" , None , None ) .await;
1353
assert_eq! ( st, StatusCode:: OK ) ;
1354
for want in [
1355
" Tag v1 is signed" ,
1356
" The one change in the release is authenticated" ,
1357
" tool: signed receipt from run #1 build" ,
1358
" a runner key of this hub" ,
edcef275 Andy McCall 1 month ago ↶
1359
" The tagged state is one change" ,
df940d75 Andy McCall 1 month ago ↶
1360
" tool: built from the tagged state" ,
1361
" tool: pinned environment" ,
1362
" notes.txt: no build found" ,
1363
" <svg class=\" trust-graph\" " ,
bf5101d3 Andy McCall 1 month ago ↶
1364
1365
1366
" Tag signed by" ,
1367
" 1 of 1 changes signed or sealed" ,
1368
" 1 of 2 assets built and attested by OxygenFlow" ,
1369
" attested build" ,
1370
" uploaded by hand" ,
1371
" receipt verified" ,
1372
" class=\" shackle\" " ,
1373
" <title>notes.txt:" ,
df940d75 Andy McCall 1 month ago ↶
1374
] {
1375
assert! ( page.contains ( want) , " missing {want:?}" ) ;
1376
}
bf5101d3 Andy McCall 1 month ago ↶
1377
let svg = & page[ page.find ( " <svg class=\" trust-graph\" " ) .unwrap ( ) .. ] ;
1378
assert! ( svg[ .. 200 ] .contains ( " width=\" " ) & & svg[ .. 200 ] .contains ( " height=\" " ) , " the graph has its own size: {}" , & svg[ .. 200 ] ) ;
df940d75 Andy McCall 1 month ago ↶
1379
let ( _ , releases) = call ( & h.app, " GET" , " /andy/demo/releases" , None , None ) .await;
1380
assert! ( releases.contains ( " /andy/demo/releases/trust/v1" ) , " the release links its audit" ) ;
1381
1382
1383
let ( st, json) = call ( & h.app, " GET" , " /andy/demo/releases/trust/v1?format=json" , None , None ) .await;
1384
assert_eq! ( st, StatusCode:: OK ) ;
1385
let t: serde_json:: Value = serde_json:: from_str( & json) .unwrap ( ) ;
1386
assert_eq! ( ( t[ " tag" ] .as_str ( ) , t[ " tag_signature" ] .as_str ( ) , t[ " changes" ] .as_i64 ( ) ) , ( Some ( " v1" ) , Some ( " valid" ) , Some ( 1 ) ) ) ;
1387
assert_eq! ( t[ " authors" ] [ 0 ] [ " who" ] [ " name" ] , " Andy" ) ;
1388
let tool = t[ " assets" ] .as_array ( ) .unwrap ( ) .iter ( ) .find ( | a | a[ " name" ] = = " tool" ) .unwrap ( ) ;
1389
let hex_receipt = tool[ " build" ] [ " signed_receipt" ] .as_str ( ) .unwrap ( ) ;
1390
let sr = SignedReceipt:: decode( & hex:: decode( hex_receipt) .unwrap ( ) ) .unwrap ( ) ;
1391
sr.verify ( ) .unwrap ( ) ;
1392
assert_eq! ( sr.receipt.outputs[ 0 ] .to_hex ( ) , blob) ;
1393
let audit = t[ " audit" ] .as_array ( ) .unwrap ( ) ;
1394
assert! ( audit.iter ( ) .any ( | c | c[ " verdict" ] = = " warn" & & c[ " title" ] .as_str ( ) .unwrap ( ) .starts_with ( " notes.txt" ) ) ) ;
1395
assert! ( ! audit.iter ( ) .any ( | c | c[ " verdict" ] = = " fail" ) , " {audit:?}" ) ;
1396
1397
let ( st, api) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/releases/trust/v1" , None , None ) .await;
1398
assert_eq! ( st, StatusCode:: OK ) ;
1399
assert_eq! ( serde_json:: from_str:: < serde_json:: Value> ( & api) .unwrap ( ) [ " assets" ] .as_array ( ) .unwrap ( ) .len ( ) , 2 ) ;
1400
assert_eq! ( call ( & h.app, " GET" , " /api/v1/repos/andy/demo/releases/trust/v9" , None , None ) .await.0 , StatusCode:: NOT_FOUND ) ;
1401
assert_eq! ( call ( & h.app, " GET" , " /andy/demo/releases/trust/v9" , None , None ) .await.0 , StatusCode:: NOT_FOUND ) ;
1402
}
b6b7e4cf Andy McCall 1 month ago ↶
1403
1404
1405
1406
1407
# [ tokio ::test ( flavor = " multi_thread" ) ]
1408
async fn runners_register_show_their_history_and_are_drained_and_removed ( ) {
1409
use oxy_core:: identity:: SecretKey;
1410
let h = hub ( ) .await;
1411
create ( & h.app, " andy" , " demo" ) .await;
1412
1413
let res = oxygenhub:: router( h.app.clone ( ) )
1414
.oneshot (
1415
Request:: post( " /register" )
1416
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
c8684d44 Andy McCall 20 days ago
1417
.body ( Body:: from( " username=carol&email=carol%40example.com&password=correct-horse-battery&setup=test-setup" ) )
b6b7e4cf Andy McCall 1 month ago ↶
1418
.unwrap ( ) ,
1419
)
1420
.await
1421
.unwrap ( ) ;
1422
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
1423
let admin = db:: accounts:: account_by_name( & h.app.db, " carol" ) .await.unwrap ( ) .unwrap ( ) ;
1424
db:: accounts:: set_admin( & h.app.db, admin.id, true ) .await.unwrap ( ) ;
1425
let web = | method : & 'static str , path : String, body : Option< String> | {
1426
let ( app, cookie) = ( h.app.clone ( ) , cookie.clone ( ) ) ;
1427
async move {
1428
let mut req = Request:: builder( ) .method ( method) .uri ( path) .header ( header:: COOKIE , cookie) ;
1429
if body.is_some ( ) {
1430
req = req.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) ;
1431
}
1432
let res = oxygenhub:: router( app) .oneshot ( req.body ( body.map ( Body:: from) .unwrap_or_else ( Body:: empty) ) .unwrap ( ) ) .await.unwrap ( ) ;
1433
let st = res.status ( ) ;
1434
( st, String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) )
1435
}
1436
} ;
1437
let csrf_of = | page : & str | {
1438
let i = page.find ( " name=\" _csrf\" value=\" " ) .unwrap ( ) + " name=\" _csrf\" value=\" " .len ( ) ;
1439
page[ i.. ] .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( )
1440
} ;
1441
let ( st, page) = web ( " GET" , " /admin/runners" .into ( ) , None ) .await;
1442
assert_eq! ( st, StatusCode:: OK ) ;
1443
assert! ( page.contains ( " No runners yet" ) ) ;
1444
assert_eq! ( call ( & h.app, " GET" , " /admin/runners" , None , None ) .await.0 , StatusCode:: NOT_FOUND , " administrators only" ) ;
1445
1446
1447
let ( _ , page) = web ( " POST" , " /admin/runners/token" .into ( ) , Some ( format! ( " _csrf={} " , csrf_of ( & page) ) ) ) .await;
1448
let i = page.find ( " --token " ) .unwrap ( ) + " --token " .len ( ) ;
1449
let token: String = page[ i.. ] .chars ( ) .take_while ( | c | ! c.is_whitespace ( ) ) .collect ( ) ;
1450
assert! ( token.starts_with ( " flowreg_" ) , " {token}" ) ;
1451
1452
1453
let sk = SecretKey:: generate( ) ;
1454
let identity = sk.identity ( " plex (OxygenFlow runner)" , " " , 1 ) ;
1455
let info = serde_json:: json! ( { " arch" : " x86_64" , " os" : " linux" , " version" : " 0.1.0" , " labels" : [ " big-disk" ] , " max_jobs" : 2 , " memory_mib" : 16384 , " cpus" : 4 } ) ;
1456
let reg = serde_json:: json! ( { " token" : token, " name" : " plex" , " identity" : hex:: encode( identity.encode ( ) ) , " info" : info } ) ;
1457
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await;
1458
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1459
let credential = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " credential" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
1460
assert! ( credential.starts_with ( " flowrunner_" ) ) ;
1461
assert_eq! (
1462
call ( & h.app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await.0 ,
1463
StatusCode:: FORBIDDEN ,
1464
" used once"
1465
) ;
1466
1467
1468
let claim_as = | who : String| {
1469
let app = h.app.clone ( ) ;
1470
let info = info.clone ( ) ;
1471
async move {
1472
let body = serde_json:: json! ( { " runner" : " ignored" , " runtimes" : [ " microvm" ] , " info" : info } ) .to_string ( ) ;
1473
call ( & app, " POST" , " /api/v1/flow/claim" , Some ( & who) , Some ( body) ) .await
1474
}
1475
} ;
1476
assert_eq! ( claim_as ( credential.clone ( ) ) .await.0 , StatusCode:: NO_CONTENT , " nothing queued" ) ;
1477
assert_eq! ( claim ( & h.app) .await, None ) ;
1478
let ( _ , page) = web ( " GET" , " /admin/runners" .into ( ) , None ) .await;
1479
assert! (
1480
page.contains ( " >plex</strong>" ) & & page.contains ( " st-online" ) & & page.contains ( " x86_64 · linux" ) & & page.contains ( " big-disk" ) ,
1481
" {page}"
1482
) ;
1483
assert! ( page.contains ( " >test-1</strong>" ) & & page.contains ( " hub.toml" ) , " the configured runner, by its claimed name" ) ;
1484
let plex = db:: runners:: list( & h.app.db) .await.unwrap ( ) .into_iter ( ) .find ( | r | r.name = = " plex" ) .unwrap ( ) ;
1485
assert_eq! ( plex.key, identity.hash ( ) .to_hex ( ) , " bound at registration" ) ;
1486
1487
1488
let url = format! ( " {} /andy/demo" , h.base) ;
1489
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1490
let c = client ( & url, work.path ( ) ) ;
1491
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
1492
c.snapshot ( " test" ) .unwrap ( ) ;
1493
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
1494
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1495
runs_of ( & h.app, " andy" , " demo" ) .await;
1496
let ( st, body) = claim_as ( credential.clone ( ) ) .await;
1497
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1498
let job: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
1ec29691 Andy McCall 8 days ago ↶
1499
1500
let kept: String =
1501
sqlx:: query_scalar( " SELECT runner_version FROM flow_jobs WHERE runner_id = $1" ) .bind ( plex.id) .fetch_one ( & h.app.db) .await.unwrap ( ) ;
1502
assert_eq! ( kept, " 0.1.0" ) ;
b6b7e4cf Andy McCall 1 month ago ↶
1503
let ( _ , page) = web ( " GET" , format! ( " /admin/runners/{} " , plex.id) , None ) .await;
1504
assert! (
1505
page.contains ( " andy/demo" ) & & page.contains ( job[ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) ) & & page.contains ( " Running now</dt><dd>1 job" ) ,
1506
" {page}"
1507
) ;
b66e70b7 Andy McCall 1 month ago ↶
1508
1509
let ( _ , page) = web ( " GET" , " /admin/runners" .into ( ) , None ) .await;
e9b9619c Andy McCall 19 days ago ↶
1510
assert! ( page.contains ( " id=\" queue\" " ) & & page.contains ( " >plex</a></td>" ) , " {page}" ) ;
1511
1512
1513
assert! ( page.contains ( " running 1 of " ) & & page.contains ( " /admin/runners?runner=plex#queue" ) , " {page}" ) ;
1514
let ( _ , only) = web ( " GET" , " /admin/runners?runner=plex" .into ( ) , None ) .await;
1515
assert! ( only.contains ( " >plex</a></td>" ) & & only.contains ( " of " ) & & only.contains ( " >all</a>" ) , " {only}" ) ;
1516
let ( _ , none) = web ( " GET" , " /admin/runners?runner=nobody&status=running" .into ( ) , None ) .await;
1517
assert! ( none.contains ( " Nothing matches." ) , " {none}" ) ;
1518
1519
sqlx:: query( " UPDATE flow_runners SET last_seen_ms = 0 WHERE id = $1" ) .bind ( plex.id) .execute ( & h.app.db) .await.unwrap ( ) ;
1520
let ( _ , page) = web ( " GET" , " /admin/runners" .into ( ) , None ) .await;
1521
assert! ( page.contains ( " st-offline" ) , " quiet for long: offline" ) ;
1522
assert_eq! ( call ( & h.app, " POST" , " /api/v1/flow/runners/heartbeat" , Some ( & credential) , None ) .await.0 , StatusCode:: NO_CONTENT ) ;
1523
assert_eq! ( call ( & h.app, " POST" , " /api/v1/flow/runners/heartbeat" , Some ( " flowrunner_nope" ) , None ) .await.0 , StatusCode:: UNAUTHORIZED ) ;
1524
let plex_now = db:: runners:: list( & h.app.db) .await.unwrap ( ) .into_iter ( ) .find ( | r | r.name = = " plex" ) .unwrap ( ) ;
1525
assert_eq! ( plex_now.connectivity ( ) , " online" ) ;
b66e70b7 Andy McCall 1 month ago ↶
1526
let ( _ , tokens) = web ( " GET" , " /settings/tokens" .into ( ) , None ) .await;
1527
let ( _ , made) = web ( " POST" , " /settings/tokens" .into ( ) , Some ( format! ( " _csrf={} &name=admin&scopes=user%3Aread" , csrf_of ( & tokens) ) ) ) .await;
1528
let i = made.find ( " oxyhub_" ) .unwrap ( ) ;
1529
let admin_token: String = made[ i.. ] .chars ( ) .take_while ( | c | c.is_ascii_alphanumeric ( ) | | * c = = ' _' ) .collect ( ) ;
1530
let ( st, body) = call ( & h.app, " GET" , " /api/v1/runners" , Some ( & admin_token) , None ) .await;
1531
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1532
let listed: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
1533
let plex_row = listed.as_array ( ) .unwrap ( ) .iter ( ) .find ( | r | r[ " name" ] = = " plex" ) .unwrap ( ) ;
1534
assert_eq! ( ( plex_row[ " running" ] .as_i64 ( ) , plex_row[ " arch" ] .as_str ( ) ) , ( Some ( 1 ) , Some ( " x86_64" ) ) ) ;
7f5f6c05 Andy McCall 1 month ago ↶
1535
1536
let ( _ , page) = web ( " GET" , format! ( " /admin/runners/{} " , plex.id) , None ) .await;
1537
let ( st, _ ) = web (
1538
" POST" ,
1539
format! ( " /admin/runners/{} /limits" , plex.id) ,
1540
Some ( format! ( " _csrf={} &labels=gpu%2C+plex&timeout_minutes=5" , csrf_of ( & page) ) ) ,
1541
)
1542
.await;
1543
assert_eq! ( st, StatusCode:: SEE_OTHER ) ;
1544
let ( _ , page) = web ( " GET" , format! ( " /admin/runners/{} " , plex.id) , None ) .await;
1545
assert! ( page.contains ( " >gpu<" ) & & page.contains ( " 5 min for its jobs" ) , " {page}" ) ;
1546
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/runners" , Some ( & admin_token) , None ) .await;
1547
let listed: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
1548
let plex_row = listed.as_array ( ) .unwrap ( ) .iter ( ) .find ( | r | r[ " name" ] = = " plex" ) .unwrap ( ) ;
1549
assert! ( plex_row[ " labels" ] .as_array ( ) .unwrap ( ) .iter ( ) .any ( | l | l = = " gpu" ) , " {plex_row}" ) ;
1550
let ( st, _ ) = web (
1551
" POST" ,
1552
format! ( " /admin/runners/{} /limits" , plex.id) ,
1553
Some ( format! ( " _csrf={} &labels=bad%20label%21&timeout_minutes=" , csrf_of ( & page) ) ) ,
1554
)
1555
.await;
1556
assert_eq! ( st, StatusCode:: BAD_REQUEST , " a label with a space or punctuation is refused" ) ;
b66e70b7 Andy McCall 1 month ago ↶
1557
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/runners/queue" , Some ( & admin_token) , None ) .await;
1558
let queue: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
1559
assert! ( queue.as_array ( ) .unwrap ( ) .iter ( ) .any ( | q | q[ " status" ] = = " running" & & q[ " runner" ] = = " plex" ) , " {queue}" ) ;
1560
let ( _ , body) = call ( & h.app, " GET" , & format! ( " /api/v1/runners/{} /jobs" , plex.id) , Some ( & admin_token) , None ) .await;
1561
let jobs: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
1562
assert_eq! ( jobs[ 0 ] [ " repo" ] , " andy/demo" , " {jobs}" ) ;
1563
assert_eq! ( call ( & h.app, " GET" , " /api/v1/runners" , None , None ) .await.0 , StatusCode:: NOT_FOUND , " administrators only" ) ;
b6b7e4cf Andy McCall 1 month ago ↶
1564
1565
1566
let ( _ , page) = web ( " GET" , " /admin/runners" .into ( ) , None ) .await;
1567
let csrf = csrf_of ( & page) ;
1568
web ( " POST" , format! ( " /admin/runners/{} /state" , plex.id) , Some ( format! ( " _csrf={csrf} &state=draining" ) ) ) .await;
1569
assert_eq! ( claim_as ( credential.clone ( ) ) .await.0 , StatusCode:: NO_CONTENT , " a draining runner takes nothing" ) ;
1570
let ( _ , page) = web ( " POST" , format! ( " /admin/runners/{} /remove" , plex.id) , Some ( format! ( " _csrf={csrf} " ) ) ) .await;
1571
assert! ( page.contains ( " is running jobs: drain it first" ) , " {page}" ) ;
1572
1573
let id = job[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
1574
let done = call (
1575
& h.app,
1576
" POST" ,
1577
& format! ( " /api/v1/flow/jobs/{id} /finish" ) ,
1578
Some ( & credential) ,
1579
Some ( r # "{"exit_code":0,"duration_ms":5}"# .into ( ) ) ,
1580
)
1581
.await;
1582
assert_eq! ( done.0 , StatusCode:: NO_CONTENT ) ;
1583
web ( " POST" , format! ( " /admin/runners/{} /state" , plex.id) , Some ( format! ( " _csrf={csrf} &state=disabled" ) ) ) .await;
1584
assert_eq! ( claim_as ( credential.clone ( ) ) .await.0 , StatusCode:: FORBIDDEN ) ;
1585
let ( _ , page) = web ( " POST" , format! ( " /admin/runners/{} /remove" , plex.id) , Some ( format! ( " _csrf={csrf} " ) ) ) .await;
1586
assert! ( page.contains ( " Removed plex." ) , " {page}" ) ;
1587
assert_eq! ( claim_as ( credential) .await.0 , StatusCode:: UNAUTHORIZED ) ;
1588
}
74be9796 Andy McCall 1 month ago ↶
1589
1590
1591
1592
# [ tokio ::test ( flavor = " multi_thread" ) ]
1593
async fn scoped_runners_take_only_their_scope ( ) {
1594
use oxy_core:: identity:: SecretKey;
1595
let h = hub ( ) .await;
1596
create ( & h.app, " andy" , " demo" ) .await;
1597
create ( & h.app, " andy" , " other" ) .await;
1598
let register = | scope : db:: runners:: Scope, name : & 'static str | {
1599
let app = h.app.clone ( ) ;
1600
async move {
1601
let token = format! ( " flowreg_{name} " ) ;
1602
db:: runners:: add_registration( & app.db, & blake3:: hash( token.as_bytes ( ) ) .to_hex ( ) , " admin" , & scope) .await.unwrap ( ) ;
1603
let identity = SecretKey:: generate( ) .identity ( name, " " , 1 ) ;
1604
let info = serde_json:: json! ( { " arch" : " x86_64" , " os" : " linux" , " version" : " 0.1.0" , " labels" : [ ] , " max_jobs" : 1 } ) ;
1605
let reg = serde_json:: json! ( { " token" : token, " name" : name, " identity" : hex:: encode( identity.encode ( ) ) , " info" : info } ) ;
1606
let ( st, body) = call ( & app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await;
1607
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1608
serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " credential" ] .as_str ( ) .unwrap ( ) .to_string ( )
1609
}
1610
} ;
1611
let demo = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
1612
let only_demo = register ( db:: runners:: Scope { owner: " andy" .into ( ) , repo_id: Some ( demo.id) } , " demo-runner" ) .await;
1613
let carols = register ( db:: runners:: Scope { owner: " carol" .into ( ) , repo_id: None } , " carol-runner" ) .await;
1614
let andys = register ( db:: runners:: Scope { owner: " andy" .into ( ) , repo_id: None } , " andy-runner" ) .await;
1615
let claim_as = | who : String| {
1616
let app = h.app.clone ( ) ;
1617
async move {
1618
let body = serde_json:: json! ( { " runner" : " x" , " runtimes" : [ " microvm" ] } ) .to_string ( ) ;
1619
let ( st, body) = call ( & app, " POST" , " /api/v1/flow/claim" , Some ( & who) , Some ( body) ) .await;
1620
( st = = StatusCode:: OK ) .then ( | | serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " repo" ] .as_str ( ) .unwrap ( ) .to_string ( ) )
1621
}
1622
} ;
1623
1624
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1625
let c = client ( & format! ( " {} /andy/other" , h.base) , work.path ( ) ) ;
1626
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
1627
c.snapshot ( " t" ) .unwrap ( ) ;
1628
c.commit ( " pipeline" , & [ ] ) .unwrap ( ) ;
1629
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1630
assert! ( ! runs_of ( & h.app, " andy" , " other" ) .await.is_empty ( ) ) ;
1631
assert_eq! ( claim_as ( only_demo.clone ( ) ) .await, None , " nothing of andy/demo is queued" ) ;
1632
assert_eq! ( claim_as ( carols) .await, None , " nothing of carol's" ) ;
1633
assert_eq! ( claim_as ( andys) .await.as_deref ( ) , Some ( " andy/other" ) , " the owner's runner takes its repositories' jobs" ) ;
1634
1635
assert_eq! ( claim ( & h.app) .await.unwrap ( ) [ " repo" ] , " andy/other" ) ;
1636
assert_eq! ( claim_as ( only_demo) .await, None , " still nothing of its own" ) ;
1637
}
7974adb0 Andy McCall 1 month ago ↶
1638
5714347a Andy McCall 11 days ago ↶
1639
1640
1641
1642
# [ tokio ::test ( flavor = " multi_thread" ) ]
1643
async fn a_user_registers_a_runner_for_their_own_repositories ( ) {
1644
use oxy_core:: identity:: SecretKey;
1645
let h = hub ( ) .await;
1646
create ( & h.app, " andy" , " demo" ) .await;
1647
let andy = db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " " , " salt" , false ) .await.unwrap ( ) ;
1648
let pat = " oxyhub_phone_test_token" ;
1649
db:: accounts:: create_token( & h.app.db, andy.id, " phone" , & oxygenhub:: auth:: token_hash( pat) , " repo:read,repo:write" , " " , None )
1650
.await
1651
.unwrap ( ) ;
1652
let read_only = " oxyhub_read_only_token" ;
1653
db:: accounts:: create_token( & h.app.db, andy.id, " ro" , & oxygenhub:: auth:: token_hash( read_only) , " repo:read" , " " , None ) .await.unwrap ( ) ;
1654
assert_eq! ( call ( & h.app, " POST" , " /api/v1/user/runner-token" , Some ( read_only) , None ) .await.0 , StatusCode:: FORBIDDEN ) ;
1655
assert_eq! ( call ( & h.app, " POST" , " /api/v1/user/runner-token" , None , None ) .await.0 , StatusCode:: UNAUTHORIZED ) ;
1656
let ( st, body) = call ( & h.app, " POST" , " /api/v1/user/runner-token" , Some ( pat) , None ) .await;
1657
assert_eq! ( st, StatusCode:: CREATED , " {body}" ) ;
1658
let got: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
1659
let token = got[ " token" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
1660
assert! ( token.starts_with ( " flowreg_" ) & & got[ " owner" ] = = " andy" , " {body}" ) ;
1661
let identity = SecretKey:: generate( ) .identity ( " andy's phone" , " " , 1 ) ;
1662
let info = serde_json:: json! ( { " arch" : " aarch64" , " os" : " android" , " version" : " 0.3.1" , " labels" : [ " android" ] , " max_jobs" : 1 } ) ;
1663
let reg = serde_json:: json! ( { " token" : token, " name" : " pixel" , " identity" : hex:: encode( identity.encode ( ) ) , " info" : info } ) ;
1664
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await;
1665
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1666
let phone = db:: runners:: list( & h.app.db) .await.unwrap ( ) .into_iter ( ) .find ( | r | r.name = = " pixel" ) .unwrap ( ) ;
1667
assert_eq! ( ( phone.scope_owner.as_str ( ) , phone.scope_repo_id) , ( " andy" , None ) , " andy's repositories, all of them, and no one else's" ) ;
1668
}
1669
7974adb0 Andy McCall 1 month ago ↶
1670
1671
1672
1673
# [ tokio ::test ( flavor = " multi_thread" ) ]
1674
async fn repository_admins_add_and_remove_their_own_runners ( ) {
1675
use oxy_core:: identity:: SecretKey;
1676
let h = hub ( ) .await;
1677
let res = oxygenhub:: router( h.app.clone ( ) )
1678
.oneshot (
1679
Request:: post( " /register" )
1680
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
c8684d44 Andy McCall 20 days ago
1681
.body ( Body:: from( " username=andy&email=andy%40example.com&password=correct-horse-battery&setup=test-setup" ) )
7974adb0 Andy McCall 1 month ago ↶
1682
.unwrap ( ) ,
1683
)
1684
.await
1685
.unwrap ( ) ;
1686
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
1687
create ( & h.app, " andy" , " demo" ) .await;
1688
let demo = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
1689
let web = | method : & 'static str , path : String, body : Option< String> | {
1690
let ( app, cookie) = ( h.app.clone ( ) , cookie.clone ( ) ) ;
1691
async move {
1692
let mut req = Request:: builder( ) .method ( method) .uri ( path) .header ( header:: COOKIE , cookie) ;
1693
if body.is_some ( ) {
1694
req = req.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) ;
1695
}
1696
let res = oxygenhub:: router( app) .oneshot ( req.body ( body.map ( Body:: from) .unwrap_or_else ( Body:: empty) ) .unwrap ( ) ) .await.unwrap ( ) ;
1697
let st = res.status ( ) ;
1698
( st, String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) )
1699
}
1700
} ;
1701
let csrf_of = | page : & str | {
1702
let i = page.find ( " name=\" _csrf\" value=\" " ) .unwrap ( ) + " name=\" _csrf\" value=\" " .len ( ) ;
1703
page[ i.. ] .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( )
1704
} ;
c7067c7b Andy McCall 16 days ago
1705
let ( st, page) = web ( " GET" , " /andy/demo/settings/runners" .into ( ) , None ) .await;
7974adb0 Andy McCall 1 month ago ↶
1706
assert_eq! ( st, StatusCode:: OK ) ;
c7067c7b Andy McCall 16 days ago
1707
assert! ( page.contains ( " <h2>Runners</h2>" ) , " the owner sees the page" ) ;
7974adb0 Andy McCall 1 month ago ↶
1708
let ( _ , page) = web ( " POST" , " /andy/demo/settings/runners/token" .into ( ) , Some ( format! ( " _csrf={} " , csrf_of ( & page) ) ) ) .await;
1709
let i = page.find ( " --token " ) .unwrap ( ) + " --token " .len ( ) ;
1710
let token: String = page[ i.. ] .chars ( ) .take_while ( | c | ! c.is_whitespace ( ) ) .collect ( ) ;
1711
let identity = SecretKey:: generate( ) .identity ( " mine" , " " , 1 ) ;
1712
let info = serde_json:: json! ( { " arch" : " x86_64" , " os" : " linux" , " version" : " 0.1.0" , " labels" : [ ] , " max_jobs" : 1 } ) ;
1713
let reg = serde_json:: json! ( { " token" : token, " name" : " mine" , " identity" : hex:: encode( identity.encode ( ) ) , " info" : info } ) ;
1714
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await;
1715
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1716
let mine = db:: runners:: list( & h.app.db) .await.unwrap ( ) .into_iter ( ) .find ( | r | r.name = = " mine" ) .unwrap ( ) ;
1717
assert_eq! ( mine.scope ( ) , db:: runners:: Scope { owner: " andy" .into ( ) , repo_id: Some ( demo.id) } ) ;
1718
1719
claim ( & h.app) .await;
1720
let site = db:: runners:: list( & h.app.db) .await.unwrap ( ) .into_iter ( ) .find ( | r | r.name = = " test-1" ) .unwrap ( ) ;
c7067c7b Andy McCall 16 days ago
1721
let ( _ , page) = web ( " GET" , " /andy/demo/settings/runners" .into ( ) , None ) .await;
7974adb0 Andy McCall 1 month ago ↶
1722
assert! ( page.contains ( " <strong>mine</strong>" ) & & page.contains ( " this repository" ) , " {page}" ) ;
1723
assert! ( page.contains ( " <strong>test-1</strong>" ) & & page.contains ( " the whole site" ) , " {page}" ) ;
1724
let csrf = csrf_of ( & page) ;
1725
let ( st, _ ) = web ( " POST" , format! ( " /andy/demo/settings/runners/{} /remove" , site.id) , Some ( format! ( " _csrf={csrf} " ) ) ) .await;
1726
assert_eq! ( st, StatusCode:: NOT_FOUND , " not the repository's own" ) ;
c7067c7b Andy McCall 16 days ago
1727
let ( st, _ ) = web ( " POST" , format! ( " /andy/demo/settings/runners/{} /remove" , mine.id) , Some ( format! ( " _csrf={csrf} " ) ) ) .await;
1728
assert_eq! ( st, StatusCode:: SEE_OTHER , " back to the Runners page" ) ;
1729
let ( _ , page) = web ( " GET" , " /andy/demo/settings/runners?notice=x" .into ( ) , None ) .await;
1730
assert! ( ! page.contains ( " <strong>mine</strong>" ) , " {page}" ) ;
7974adb0 Andy McCall 1 month ago ↶
1731
assert! ( db:: runners:: get( & h.app.db, mine.id) .await.unwrap ( ) .is_none ( ) ) ;
1732
1733
1734
let ( _ , page) = web ( " GET" , " /orgs/new" .into ( ) , None ) .await;
1735
web ( " POST" , " /orgs/new" .into ( ) , Some ( format! ( " _csrf={} &name=acme&display_name=Acme" , csrf_of ( & page) ) ) ) .await;
1736
let ( st, page) = web ( " GET" , " /orgs/acme/settings" .into ( ) , None ) .await;
1737
assert_eq! ( st, StatusCode:: OK ) ;
1738
let ( _ , page) = web ( " POST" , " /orgs/acme/settings/runners/token" .into ( ) , Some ( format! ( " _csrf={} " , csrf_of ( & page) ) ) ) .await;
1739
let i = page.find ( " --token " ) .unwrap ( ) + " --token " .len ( ) ;
1740
let token: String = page[ i.. ] .chars ( ) .take_while ( | c | ! c.is_whitespace ( ) ) .collect ( ) ;
1741
let identity = SecretKey:: generate( ) .identity ( " acme-runner" , " " , 1 ) ;
1742
let reg = serde_json:: json! ( { " token" : token, " name" : " acme-runner" , " identity" : hex:: encode( identity.encode ( ) ) , " info" : info } ) ;
1743
assert_eq! ( call ( & h.app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await.0 , StatusCode:: OK ) ;
1744
let acme = db:: runners:: list( & h.app.db) .await.unwrap ( ) .into_iter ( ) .find ( | r | r.name = = " acme-runner" ) .unwrap ( ) ;
1745
assert_eq! ( acme.scope ( ) , db:: runners:: Scope { owner: " acme" .into ( ) , repo_id: None } ) ;
1746
let ( _ , page) = web ( " GET" , " /orgs/acme/settings" .into ( ) , None ) .await;
1747
assert! ( page.contains ( " <strong>acme-runner</strong>" ) & & page.contains ( " the organisation's repositories" ) , " {page}" ) ;
1748
let ( _ , page) = web ( " POST" , format! ( " /orgs/acme/settings/runners/{} /remove" , acme.id) , Some ( format! ( " _csrf={} " , csrf_of ( & page) ) ) ) .await;
1749
assert! ( page.contains ( " acme-runner removed" ) , " {page}" ) ;
1750
}
c4c87f48 Andy McCall 1 month ago ↶
1751
1752
1753
1754
# [ tokio ::test ( flavor = " multi_thread" ) ]
1755
async fn jobs_that_ask_for_labels_wait_for_a_runner_with_them ( ) {
1756
use oxy_core:: identity:: SecretKey;
1757
let h = hub ( ) .await;
1758
create ( & h.app, " andy" , " demo" ) .await;
1759
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1760
let c = client ( & format! ( " {} /andy/demo" , h.base) , work.path ( ) ) ;
1761
let cross = " job \" cross\" {\n sandbox \" alpine\" \n runs-on \" riscv64\" \n exec \" true\" \n }\n " ;
1762
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( cross) ) ;
1763
c.snapshot ( " t" ) .unwrap ( ) ;
1764
c.commit ( " pipeline" , & [ ] ) .unwrap ( ) ;
1765
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1766
assert! ( ! runs_of ( & h.app, " andy" , " demo" ) .await.is_empty ( ) ) ;
1767
1768
let mut taken = Vec :: new( ) ;
1769
while let Some ( j) = claim ( & h.app) .await {
1770
taken.push ( j[ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) .to_string ( ) ) ;
1771
}
1772
assert! ( ! taken.contains ( & " cross" .to_string ( ) ) & & taken.contains ( & " lint" .to_string ( ) ) , " {taken:?}" ) ;
1773
let queue = db:: flow:: queue( & h.app.db) .await.unwrap ( ) ;
1774
let waiting = queue.iter ( ) .find ( | q | q.job = = " cross" ) .unwrap ( ) ;
1775
assert_eq! ( ( waiting.status.as_str ( ) , waiting.needs ( ) ) , ( " queued" , vec! [ " riscv64" .to_string ( ) ] ) ) ;
1776
1777
let token = " flowreg_riscv" ;
1778
db:: runners:: add_registration( & h.app.db, & blake3:: hash( token.as_bytes ( ) ) .to_hex ( ) , " admin" , & Default :: default( ) ) .await.unwrap ( ) ;
1779
let identity = SecretKey:: generate( ) .identity ( " visionfive" , " " , 1 ) ;
1780
let info = serde_json:: json! ( { " arch" : " riscv64" , " os" : " linux" , " version" : " 0.1.0" , " labels" : [ " riscv64" ] , " max_jobs" : 1 } ) ;
1781
let reg = serde_json:: json! ( { " token" : token, " name" : " visionfive" , " identity" : hex:: encode( identity.encode ( ) ) , " info" : info } ) ;
1782
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await;
1783
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1784
let credential = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " credential" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
1785
let claim_body = serde_json:: json! ( { " runner" : " x" , " runtimes" : [ " microvm" ] , " info" : info } ) .to_string ( ) ;
1786
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( & credential) , Some ( claim_body) ) .await;
1787
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1788
assert_eq! ( serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " job" ] [ " name" ] , " cross" ) ;
1789
}
070a7b64 Andy McCall 1 month ago ↶
1790
1791
1792
1793
1794
1795
# [ tokio ::test ( flavor = " multi_thread" ) ]
1796
async fn a_job_no_runner_reports_is_ended_after_the_time_limit ( ) {
1797
let h = hub ( ) .await;
1798
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
1799
create ( & h.app, " andy" , " lost" ) .await;
1800
let url = format! ( " {} /andy/lost" , h.base) ;
1801
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1802
let c = client ( & url, work.path ( ) ) ;
1803
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
1804
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
1805
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
1806
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1807
assert_eq! ( runs_of ( & h.app, " andy" , " lost" ) .await.len ( ) , 1 ) ;
1808
let job = claim ( & h.app) .await.expect ( " a job to claim" ) ;
1809
let id = job[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
1810
1811
1812
assert_eq! ( oxygenhub:: flow:: reap_lost_jobs( & h.app) .await.unwrap ( ) , 0 ) ;
1813
1814
let old = db:: now_ms( ) - 24 * 3600 * 1000 ;
3b1bc27f Andy McCall 20 days ago
1815
sqlx:: query( " UPDATE flow_jobs SET started_ms = $1 WHERE id = $2" ) .bind ( old) .bind ( id) .execute ( & h.app.db) .await.unwrap ( ) ;
070a7b64 Andy McCall 1 month ago ↶
1816
assert_eq! ( oxygenhub:: flow:: reap_lost_jobs( & h.app) .await.unwrap ( ) , 1 ) ;
1817
let j = db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) ;
1818
assert_eq! ( j.status, " error" ) ;
1819
assert! ( j.error.starts_with ( " lost:" ) , " {}" , j.error) ;
1820
let run = & runs_of ( & h.app, " andy" , " lost" ) .await[ 0 ] ;
1821
let jobs = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) ;
1822
assert! ( jobs.iter ( ) .all ( | j | j.status ! = " running" ) , " nothing left running" ) ;
1823
1824
assert_ne! ( finish ( & h.app, id, serde_json:: json! ( { " exit_code" : 0 } ) ) .await, StatusCode:: NO_CONTENT ) ;
1825
assert_eq! ( db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) .status, " error" ) ;
1826
}
b0612e7d Andy McCall 1 month ago ↶
1827
1828
1829
1830
# [ tokio ::test ( flavor = " multi_thread" ) ]
1831
async fn a_runner_is_offered_only_jobs_whose_image_it_holds ( ) {
1832
let h = hub ( ) .await;
1833
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
1834
create ( & h.app, " andy" , " images" ) .await;
1835
let url = format! ( " {} /andy/images" , h.base) ;
1836
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1837
let c = client ( & url, work.path ( ) ) ;
1838
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
1839
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
1840
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
1841
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1842
assert_eq! ( runs_of ( & h.app, " andy" , " images" ) .await.len ( ) , 1 ) ;
1843
let claim_with = | images : Vec< String> | {
1844
let body = serde_json:: json! ( { " runner" : " test-1" , " runtimes" : [ " microvm" ] , " images" : images } ) .to_string ( ) ;
1845
let app = h.app.clone ( ) ;
1846
async move {
1847
let ( st, body) = call ( & app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( body) ) .await;
1848
( st, body)
1849
}
1850
} ;
1851
let ( st, _ ) = claim_with ( vec! [ " ab" .repeat ( 32 ) ] ) .await;
1852
assert_eq! ( st, StatusCode:: NO_CONTENT , " no job for a runner without the image" ) ;
1853
let jobs = db:: flow:: jobs( & h.app.db, runs_of ( & h.app, " andy" , " images" ) .await[ 0 ] .id) .await.unwrap ( ) ;
1854
let rootfs: serde_json:: Value = serde_json:: from_str( & jobs[ 0 ] .sandbox) .unwrap ( ) ;
1855
let rootfs = rootfs[ " rootfs" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
1856
let ( st, body) = claim_with ( vec! [ " ab" .repeat ( 32 ) , rootfs] ) .await;
1857
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1858
}
eef677c4 Andy McCall 1 month ago ↶
1859
1860
1861
1862
1863
# [ tokio ::test ( flavor = " multi_thread" ) ]
1864
async fn a_job_its_runner_no_longer_holds_is_ended_at_its_next_claim ( ) {
1865
let h = hub ( ) .await;
1866
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
1867
create ( & h.app, " andy" , " held" ) .await;
1868
let url = format! ( " {} /andy/held" , h.base) ;
1869
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1870
let c = client ( & url, work.path ( ) ) ;
1871
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
1872
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
1873
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
1874
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1875
assert_eq! ( runs_of ( & h.app, " andy" , " held" ) .await.len ( ) , 1 ) ;
1876
let id = claim ( & h.app) .await.expect ( " a job to claim" ) [ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
1877
let claim_holding = | holding : Vec< i64 > | {
1878
let body = serde_json:: json! ( { " runner" : " test-1" , " runtimes" : [ " microvm" ] , " holding" : holding } ) .to_string ( ) ;
1879
let app = h.app.clone ( ) ;
1880
async move { call ( & app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( body) ) .await.0 }
1881
} ;
1882
claim_holding ( vec! [ id] ) .await;
1883
assert_eq! ( db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) .status, " running" , " a held job stays" ) ;
1884
claim_holding ( vec! [ ] ) .await;
1885
let j = db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) ;
1886
assert_eq! ( j.status, " error" ) ;
1887
assert! ( j.error.starts_with ( " lost:" ) & & j.error.contains ( " no longer holds" ) , " {}" , j.error) ;
1888
let run = & runs_of ( & h.app, " andy" , " held" ) .await[ 0 ] ;
1889
assert! ( db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) .iter ( ) .all ( | j | j.status ! = " running" ) , " the run settles" ) ;
1890
}
d7ce41ac Andy McCall 1 month ago ↶
1891
1892
1893
1894
1895
# [ tokio ::test ( flavor = " multi_thread" ) ]
1896
async fn a_scoped_job_runs_only_when_a_push_touches_its_paths ( ) {
1897
let h = hub ( ) .await;
1898
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
1899
create ( & h.app, " andy" , " scoped" ) .await;
1900
let url = format! ( " {} /andy/scoped" , h.base) ;
1901
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1902
let c = client ( & url, work.path ( ) ) ;
1903
let extra = " job \" dist\" {\n sandbox \" alpine\" \n scope \" Cargo.toml\" \" packaging/**\" \n exec \" true\" \n }\n " ;
1904
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( extra) ) ;
1905
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
1906
write ( work.path ( ) , " Cargo.toml" , " version = \" 0.1.0\" \n " ) ;
1907
c.commit ( " Start" , & [ ] ) .unwrap ( ) ;
1908
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1909
let names = | app : Shared, n : usize | async move {
1910
let mut runs = Vec :: new( ) ;
1911
for _ in 0 .. 100 {
1912
runs = runs_of ( & app, " andy" , " scoped" ) .await;
1913
if runs.len ( ) > = n {
1914
break ;
1915
}
1916
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
1917
}
1918
assert_eq! ( runs.len ( ) , n, " runs" ) ;
1919
let mut j: Vec< String> = db:: flow:: jobs( & app.db, runs[ 0 ] .id) .await.unwrap ( ) .into_iter ( ) .map ( | j | j.name ) .collect ( ) ;
1920
j.sort ( ) ;
745d225d Andy McCall 1 month ago ↶
1921
3b1bc27f Andy McCall 20 days ago
1922
sqlx:: query( " UPDATE flow_runs SET status = 'passed' WHERE id = $1" ) .bind ( runs[ 0 ] .id) .execute ( & app.db) .await.unwrap ( ) ;
d7ce41ac Andy McCall 1 month ago ↶
1923
j
1924
} ;
1925
assert_eq! ( names ( h.app.clone ( ) , 1 ) .await, [ " build" , " dist" , " lint" , " test" ] , " the first run has every job" ) ;
1926
write ( work.path ( ) , " test.sh" , " echo still ok\n " ) ;
1927
c.commit ( " Edit the test" , & [ ] ) .unwrap ( ) ;
1928
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1929
assert_eq! ( names ( h.app.clone ( ) , 2 ) .await, [ " build" , " lint" , " test" ] , " dist is out of scope" ) ;
1930
write ( work.path ( ) , " Cargo.toml" , " version = \" 0.1.1\" \n " ) ;
1931
c.commit ( " Release 0.1.1" , & [ ] ) .unwrap ( ) ;
1932
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1933
assert_eq! ( names ( h.app.clone ( ) , 3 ) .await, [ " build" , " dist" , " lint" , " test" ] , " a version bump runs dist" ) ;
1934
}
5dead1f4 Andy McCall 1 month ago ↶
1935
367f34b8 Andy McCall 16 days ago ↶
1936
1937
1938
1939
# [ tokio ::test ( flavor = " multi_thread" ) ]
1940
async fn a_push_can_skip_its_run ( ) {
1941
let h = hub ( ) .await;
1942
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
1943
create ( & h.app, " andy" , " skip" ) .await;
1944
let url = format! ( " {} /andy/skip" , h.base) ;
1945
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1946
let c = client ( & url, work.path ( ) ) ;
1947
let extra = " job \" dist\" {\n sandbox \" alpine\" \n scope \" Cargo.toml\" \n exec \" true\" \n }\n " ;
1948
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( extra) ) ;
1949
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
1950
write ( work.path ( ) , " Cargo.toml" , " version = \" 0.1.0\" \n " ) ;
1951
c.commit ( " Start" , & [ ] ) .unwrap ( ) ;
1952
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1953
let newest = | app : Shared, n : usize | async move {
1954
for _ in 0 .. 100 {
1955
let runs = runs_of ( & app, " andy" , " skip" ) .await;
1956
if runs.len ( ) > = n {
1957
assert_eq! ( runs.len ( ) , n, " runs" ) ;
1958
let r = runs[ 0 ] .clone ( ) ;
1959
let mut jobs: Vec< String> = db:: flow:: jobs( & app.db, r.id) .await.unwrap ( ) .into_iter ( ) .map ( | j | j.name ) .collect ( ) ;
1960
jobs.sort ( ) ;
1961
if r.status ! = " skipped" {
1962
sqlx:: query( " UPDATE flow_runs SET status = 'passed' WHERE id = $1" ) .bind ( r.id) .execute ( & app.db) .await.unwrap ( ) ;
1963
}
1964
return ( r, jobs) ;
1965
}
1966
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
1967
}
1968
panic! ( " no run {n}" ) ;
1969
} ;
1970
assert_eq! ( newest ( h.app.clone ( ) , 1 ) .await.1 , [ " build" , " dist" , " lint" , " test" ] ) ;
1971
1972
1973
write ( work.path ( ) , " Cargo.toml" , " version = \" 0.1.1\" \n " ) ;
1974
c.commit ( " Bump the version [Skip CI]" , & [ ] ) .unwrap ( ) ;
1975
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1976
let ( run, jobs) = newest ( h.app.clone ( ) , 2 ) .await;
1977
assert_eq! ( run.status, " skipped" ) ;
1978
assert! ( jobs.is_empty ( ) , " {jobs:?}" ) ;
1979
assert! ( run.error.starts_with ( " [skip ci] in change " ) & & run.error.contains ( " , pushed by " ) , " {}" , run.error) ;
1980
assert! ( run.finished_ms.is_some ( ) ) ;
1981
let ( _ , page) = call ( & h.app, " GET" , " /andy/skip/flow/2" , None , None ) .await;
1982
assert! ( page.contains ( " the push asked to skip this run" ) , " {page}" ) ;
1983
1984
1985
1986
write ( work.path ( ) , " test.sh" , " echo still ok\n " ) ;
1987
c.commit ( " Edit the test" , & [ ] ) .unwrap ( ) ;
1988
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1989
assert_eq! ( newest ( h.app.clone ( ) , 3 ) .await.1 , [ " build" , " dist" , " lint" , " test" ] , " the skipped bump still runs dist" ) ;
1990
1991
1992
let row = db:: repos:: get( & h.app.db, " andy" , " skip" ) .await.unwrap ( ) .unwrap ( ) ;
1993
db:: repos:: set_flow_skip( & h.app.db, row.id, " others" ) .await.unwrap ( ) ;
1994
write ( work.path ( ) , " test.sh" , " echo ok again\n " ) ;
1995
c.commit ( " Docs only [skip flow]" , & [ ] ) .unwrap ( ) ;
1996
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1997
let ( run, jobs) = newest ( h.app.clone ( ) , 4 ) .await;
1998
assert_ne! ( run.status, " skipped" ) ;
1999
assert_eq! ( jobs, [ " build" , " lint" , " test" ] ) ;
2000
let row = db:: repos:: get( & h.app.db, " andy" , " skip" ) .await.unwrap ( ) .unwrap ( ) ;
2001
assert! ( oxygenhub:: flow:: skip_allowed( & row, " feature" ) & & ! oxygenhub:: flow:: skip_allowed( & row, " main" ) ) ;
2002
db:: repos:: set_flow_skip( & h.app.db, row.id, " never" ) .await.unwrap ( ) ;
2003
let row = db:: repos:: get( & h.app.db, " andy" , " skip" ) .await.unwrap ( ) .unwrap ( ) ;
2004
assert! ( ! oxygenhub:: flow:: skip_allowed( & row, " feature" ) ) ;
2005
}
2006
5588bb36 Andy McCall 15 days ago ↶
2007
2008
2009
2010
# [ tokio ::test ( flavor = " multi_thread" ) ]
2011
async fn a_push_reuses_a_pass_on_the_same_state ( ) {
2012
let h = hub ( ) .await;
2013
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
2014
create ( & h.app, " andy" , " reuse" ) .await;
2015
let url = format! ( " {} /andy/reuse" , h.base) ;
2016
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2017
let c = client ( & url, work.path ( ) ) ;
2018
let gated = " job \" deploy\" {\n sandbox \" alpine\" \n depends-on \" test\" \n approval {\n by \" maintain\" \n }\n exec \" true\" \n }\n " ;
2019
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( gated) ) ;
2020
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2021
c.commit ( " Start" , & [ ] ) .unwrap ( ) ;
2022
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2023
let first = runs_of ( & h.app, " andy" , " reuse" ) .await.remove ( 0 ) ;
2024
2025
sqlx:: query( " UPDATE flow_jobs SET status = 'passed', exit_code = 0, receipt_hash = 'r-' || name WHERE run_id = $1" )
2026
.bind ( first.id)
2027
.execute ( & h.app.db)
2028
.await
2029
.unwrap ( ) ;
2030
sqlx:: query( " UPDATE flow_runs SET status = 'passed' WHERE id = $1" ) .bind ( first.id) .execute ( & h.app.db) .await.unwrap ( ) ;
2031
let heads: Vec< oxy_core:: hash:: ChangeHash> = first.head_list ( ) .into_iter ( ) .filter_map ( oxy_core:: hash:: ChangeHash:: from_hex) .collect ( ) ;
2032
let row = db:: repos:: get( & h.app.db, " andy" , " reuse" ) .await.unwrap ( ) .unwrap ( ) ;
2033
assert! ( row.flow_reuse, " on by default" ) ;
2034
let again = | row : db:: repos:: RepoRow| {
2035
let ( app, state, heads) = ( h.app.clone ( ) , first.state.clone ( ) , heads.clone ( ) ) ;
2036
async move {
2037
let n =
2038
oxygenhub:: flow:: start_run_scoped( & app, & row, " main" , & state, & heads, " andy" , " push" , None ) .await.unwrap ( ) .expect ( " a run" ) ;
2039
let run = db:: flow:: run( & app.db, row.id, n) .await.unwrap ( ) .unwrap ( ) ;
2040
let jobs = db:: flow:: jobs( & app.db, run.id) .await.unwrap ( ) ;
2041
( run, jobs)
2042
}
2043
} ;
2044
let ( run, jobs) = again ( row.clone ( ) ) .await;
2045
let by = | name : & str | jobs.iter ( ) .find ( | j | j.name = = name ) .unwrap ( ) .clone ( ) ;
2046
for name in [ " build" , " test" , " lint" ] {
2047
let j = by ( name) ;
2048
assert_eq! ( ( j.status.as_str ( ) , j.reused_from ( ) ) , ( " passed" , Some ( first.number) ) , " {name}: {j:?}" ) ;
2049
assert_eq! ( j.receipt_hash, format! ( " r-{name} " ) , " the pass's receipt stands for it" ) ;
2050
}
2051
let deploy = by ( " deploy" ) ;
2052
assert! ( deploy.status = = " queued" & & deploy.reused_from ( ) .is_none ( ) , " a gated job runs again: {deploy:?}" ) ;
2053
assert_eq! ( run.status, " queued" , " the run waits for the gated job" ) ;
2054
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/reuse/flow/{} " , run.number) , None , None ) .await;
2055
assert! ( page.contains ( " data-reused" ) & & page.contains ( & format! ( " /andy/reuse/flow/{} " , first.number) ) , " {page}" ) ;
2056
2057
let checks = db:: hooks:: checks_for_state( & h.app.db, row.id, & first.state) .await.unwrap ( ) ;
2058
assert! (
2059
checks.iter ( ) .any ( | c | c.name = = " flow/build" & & c.status = = " passed" & & c.summary.contains ( " reused" ) ) ,
2060
" a reused pass reports its check: {checks:?}"
2061
) ;
2062
2063
2064
db:: repos:: set_flow_reuse( & h.app.db, row.id, false ) .await.unwrap ( ) ;
2065
let row = db:: repos:: get( & h.app.db, " andy" , " reuse" ) .await.unwrap ( ) .unwrap ( ) ;
2066
let ( _ , jobs) = again ( row) .await;
2067
assert! ( jobs.iter ( ) .all ( | j | j.status = = " queued" & & j.reused_from ( ) .is_none ( ) ) , " {jobs:?}" ) ;
2068
}
2069
42cd46c9 Andy McCall 9 days ago ↶
2070
2071
2072
2073
# [ tokio ::test ( flavor = " multi_thread" ) ]
2074
async fn a_fork_reuses_what_upstream_proved ( ) {
2075
let h = hub ( ) .await;
2076
for who in [ " andy" , " bob" ] {
2077
db:: accounts:: create_account( & h.app.db, who, & format! ( " {who} @example.com" ) , " x" , " salt" , true ) .await.unwrap ( ) ;
2078
}
2079
create ( & h.app, " andy" , " game" ) .await;
2080
create ( & h.app, " bob" , " game" ) .await;
2081
let up = db:: repos:: get( & h.app.db, " andy" , " game" ) .await.unwrap ( ) .unwrap ( ) ;
2082
let fork = db:: repos:: get( & h.app.db, " bob" , " game" ) .await.unwrap ( ) .unwrap ( ) ;
2083
db:: repos:: set_forked_from( & h.app.db, fork.id, up.id) .await.unwrap ( ) ;
2084
2085
sqlx:: query( " UPDATE repositories SET flow = 'on' WHERE id = $1" ) .bind ( fork.id) .execute ( & h.app.db) .await.unwrap ( ) ;
2086
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2087
let mut c = client ( & format! ( " {} /andy/game" , h.base) , work.path ( ) ) ;
2088
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
2089
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2090
c.commit ( " Start" , & [ ] ) .unwrap ( ) ;
2091
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2092
let first = runs_of ( & h.app, " andy" , " game" ) .await.remove ( 0 ) ;
06b75635 Andy McCall 8 days ago ↶
2093
2094
2095
{
2096
use oxy_core:: hash:: { hash_object, ChangeHash, ObjectType} ;
2097
use oxy_core:: identity:: SecretKey;
2098
use oxy_core:: objects:: StateObject;
2099
use oxy_core:: receipt:: SignedReceipt;
2100
let heads: Vec< ChangeHash> = first.head_list ( ) .into_iter ( ) .filter_map ( ChangeHash:: from_hex) .collect ( ) ;
2101
let state = hash_object ( ObjectType:: State, & StateObject:: new( heads) .encode ( ) ) ;
2102
let repo_id = h.app.repos.open ( & RepoName:: parse( " andy/game" ) .unwrap ( ) ) .unwrap ( ) .id ( ) .unwrap ( ) ;
2103
let sk = SecretKey:: generate( ) ;
2104
let runner = sk.identity ( " test-1 (OxygenFlow runner)" , " " , 1 ) ;
2105
let run_url = format! ( " {} /andy/game/flow/{} " , h.base, first.number) ;
2106
while let Some ( j) = claim ( & h.app) .await {
2107
let r = receipt ( j[ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) , j[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) , 0 ) ;
2108
let parsed: flow_core:: Receipt = serde_json:: from_value( r.clone ( ) ) .unwrap ( ) ;
2109
let signed = hex:: encode( SignedReceipt:: sign( parsed.to_oxy ( ) , repo_id, state, & run_url, & runner, & sk) .encode ( ) ) ;
2110
let body = serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 5 , " receipt" : r, " signed_receipt" : signed } ) ;
2111
assert_eq! ( finish ( & h.app, j[ " job_id" ] .as_i64 ( ) .unwrap ( ) , body) .await, StatusCode:: NO_CONTENT ) ;
2112
}
2113
}
2114
let up_jobs = db:: flow:: jobs( & h.app.db, first.id) .await.unwrap ( ) ;
2115
assert! ( up_jobs.iter ( ) .all ( | j | j.status = = " passed" & & j.receipt_attest.len ( ) = = 64 ) , " {up_jobs:?}" ) ;
42cd46c9 Andy McCall 9 days ago ↶
2116
2117
c.config.set ( " remote.origin.url" , & format! ( " {} /bob/game" , h.base) ) .unwrap ( ) ;
2118
c.save_config ( ) .unwrap ( ) ;
2119
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2120
let run = runs_of ( & h.app, " bob" , " game" ) .await.remove ( 0 ) ;
2121
assert_eq! ( run.state, first.state) ;
2122
let jobs = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) ;
2123
for j in & jobs {
2124
assert_eq! ( ( j.status.as_str ( ) , j.reused_from ( ) , j.reused_repo ( ) ) , ( " passed" , Some ( first.number) , Some ( " andy/game" ) ) , " {j:?}" ) ;
06b75635 Andy McCall 8 days ago ↶
2125
let from = up_jobs.iter ( ) .find ( | u | u.name = = j.name ) .unwrap ( ) ;
2126
assert_eq! ( j.receipt_attest, from.receipt_attest, " upstream's signed receipt stands for it" ) ;
42cd46c9 Andy McCall 9 days ago ↶
2127
}
2128
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /bob/game/flow/{} " , run.number) , None , None ) .await;
2129
assert! ( page.contains ( & format! ( " href=\" /andy/game/flow/{} \" >andy/game run #{} </a>" , first.number, first.number) ) , " {page}" ) ;
2130
let checks = db:: hooks:: checks_for_state( & h.app.db, fork.id, & run.state) .await.unwrap ( ) ;
2131
assert! ( checks.iter ( ) .any ( | c | c.summary.contains ( " passed in andy/game run #" ) ) , " {checks:?}" ) ;
06b75635 Andy McCall 8 days ago ↶
2132
2133
2134
2135
let lint = up_jobs.iter ( ) .find ( | j | j.name = = " lint" ) .unwrap ( ) ;
2136
sqlx:: query( " UPDATE flow_jobs SET receipt_attest = $1 WHERE id = $2" )
2137
.bind ( " ab" .repeat ( 32 ) )
2138
.bind ( lint.id)
2139
.execute ( & h.app.db)
2140
.await
2141
.unwrap ( ) ;
2142
create ( & h.app, " bob" , " game2" ) .await;
2143
let fork2 = db:: repos:: get( & h.app.db, " bob" , " game2" ) .await.unwrap ( ) .unwrap ( ) ;
2144
db:: repos:: set_forked_from( & h.app.db, fork2.id, up.id) .await.unwrap ( ) ;
2145
sqlx:: query( " UPDATE repositories SET flow = 'on' WHERE id = $1" ) .bind ( fork2.id) .execute ( & h.app.db) .await.unwrap ( ) ;
2146
c.config.set ( " remote.origin.url" , & format! ( " {} /bob/game2" , h.base) ) .unwrap ( ) ;
2147
c.save_config ( ) .unwrap ( ) ;
2148
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2149
let jobs2 = db:: flow:: jobs( & h.app.db, runs_of ( & h.app, " bob" , " game2" ) .await.remove ( 0 ) .id) .await.unwrap ( ) ;
2150
let status = | name : & str | jobs2.iter ( ) .find ( | j | j.name = = name ) .map ( | j | ( j.status.clone ( ) , j.reused_repo ( ) .map ( String :: from) ) ) .unwrap ( ) ;
2151
assert_eq! ( status ( " lint" ) , ( " queued" .to_string ( ) , None ) , " {jobs2:?}" ) ;
2152
assert_eq! ( status ( " build" ) , ( " passed" .to_string ( ) , Some ( " andy/game" .to_string ( ) ) ) , " {jobs2:?}" ) ;
2153
sqlx:: query( " UPDATE flow_jobs SET receipt_attest = $1 WHERE id = $2" )
2154
.bind ( & lint.receipt_attest)
2155
.bind ( lint.id)
2156
.execute ( & h.app.db)
2157
.await
2158
.unwrap ( ) ;
42cd46c9 Andy McCall 9 days ago ↶
2159
91c306aa Andy McCall 8 days ago ↶
2160
2161
2162
for who in [ " carol" ] {
2163
db:: accounts:: create_account( & h.app.db, who, & format! ( " {who} @example.com" ) , " x" , " salt" , false ) .await.unwrap ( ) ;
2164
}
2165
let push_to = | owner : & 'static str , name : & 'static str | {
2166
let ( app, base) = ( h.app.clone ( ) , h.base.clone ( ) ) ;
2167
let path = work.path ( ) .to_path_buf ( ) ;
2168
async move {
2169
create ( & app, owner, name) .await;
2170
let r = db:: repos:: get( & app.db, owner, name) .await.unwrap ( ) .unwrap ( ) ;
2171
sqlx:: query( " UPDATE repositories SET flow = 'on' WHERE id = $1" ) .bind ( r.id) .execute ( & app.db) .await.unwrap ( ) ;
2172
let mut c = Repo:: open( & path) .unwrap ( ) ;
2173
c.config.set ( " remote.origin.url" , & format! ( " {base} /{owner} /{name} " ) ) .unwrap ( ) ;
2174
c.save_config ( ) .unwrap ( ) ;
2175
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2176
let run = runs_of ( & app, owner, name) .await.remove ( 0 ) ;
2177
db:: flow:: jobs( & app.db, run.id) .await.unwrap ( )
2178
}
2179
} ;
2180
let jobs = push_to ( " carol" , " tetris-clone" ) .await;
2181
assert! ( jobs.iter ( ) .all ( | j | j.status = = " passed" & & j.reused_repo ( ) .is_some ( ) ) , " {jobs:?}" ) ;
2182
2183
2184
2185
sqlx:: query( " UPDATE repositories SET visibility = 'private' WHERE owner IN ('andy', 'bob')" ) .execute ( & h.app.db) .await.unwrap ( ) ;
2186
sqlx:: query( " UPDATE repositories SET visibility = 'private' WHERE owner = 'carol'" ) .execute ( & h.app.db) .await.unwrap ( ) ;
2187
sqlx:: query( " DELETE FROM flow_jobs WHERE run_id IN (SELECT id FROM flow_runs WHERE repo_id IN (SELECT id FROM repositories WHERE owner = 'carol'))" )
2188
.execute ( & h.app.db)
2189
.await
2190
.unwrap ( ) ;
2191
let jobs = push_to ( " carol" , " another" ) .await;
42cd46c9 Andy McCall 9 days ago ↶
2192
assert! ( jobs.iter ( ) .all ( | j | j.status = = " queued" & & j.reused_from ( ) .is_none ( ) ) , " {jobs:?}" ) ;
2193
}
2194
367f34b8 Andy McCall 16 days ago ↶
2195
# [ test ]
2196
fn skip_markers_are_the_forges_spellings_in_any_case ( ) {
2197
use oxygenhub:: flow:: skip_marker;
2198
assert_eq! ( skip_marker ( " Fix typo [skip flow]" ) , Some ( " [skip flow]" ) ) ;
2199
assert_eq! ( skip_marker ( " [CI SKIP] docs" ) , Some ( " [ci skip]" ) ) ;
2200
assert_eq! ( skip_marker ( " Docs\n \n [Skip CI]" ) , Some ( " [skip ci]" ) ) ;
2201
assert_eq! ( skip_marker ( " skip ci without brackets" ) , None ) ;
3b3a51af Andy McCall 15 days ago ↶
2202
2203
assert_eq! ( skip_marker ( " Document the `[skip flow]` marker" ) , None ) ;
2204
assert_eq! ( skip_marker ( " Example:\n \n ```\n git commit -m 'x [skip ci]'\n ```\n " ) , None ) ;
2205
assert_eq! ( skip_marker ( " `code` then [skip ci]" ) , Some ( " [skip ci]" ) ) ;
2206
assert_eq! ( skip_marker ( " an odd ` backtick [skip ci]" ) , Some ( " [skip ci]" ) ) ;
367f34b8 Andy McCall 16 days ago ↶
2207
}
2208
745d225d Andy McCall 1 month ago ↶
2209
2210
2211
2212
# [ tokio ::test ( flavor = " multi_thread" ) ]
2213
async fn runs_are_cancelled_by_hand_or_by_a_newer_push ( ) {
2214
let h = hub ( ) .await;
2215
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
2216
create ( & h.app, " andy" , " stop" ) .await;
2217
let url = format! ( " {} /andy/stop" , h.base) ;
2218
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2219
let c = client ( & url, work.path ( ) ) ;
2220
let extra = " job \" dist\" {\n sandbox \" alpine\" \n scope \" Cargo.toml\" \n exec \" true\" \n }\n " ;
2221
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( extra) ) ;
2222
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2223
write ( work.path ( ) , " Cargo.toml" , " version = \" 0.1.0\" \n " ) ;
2224
c.commit ( " Start" , & [ ] ) .unwrap ( ) ;
2225
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2226
let wait_runs = | app : Shared, n : usize | async move {
2227
for _ in 0 .. 100 {
2228
let runs = runs_of ( & app, " andy" , " stop" ) .await;
2229
if runs.len ( ) > = n {
2230
return runs;
2231
}
2232
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
2233
}
2234
panic! ( " no run {n}" ) ;
2235
} ;
2236
let first = wait_runs ( h.app.clone ( ) , 1 ) .await;
3b1bc27f Andy McCall 20 days ago
2237
sqlx:: query( " UPDATE flow_runs SET status = 'passed' WHERE id = $1" ) .bind ( first[ 0 ] .id) .execute ( & h.app.db) .await.unwrap ( ) ;
2238
sqlx:: query( " UPDATE flow_jobs SET status = 'passed' WHERE run_id = $1" ) .bind ( first[ 0 ] .id) .execute ( & h.app.db) .await.unwrap ( ) ;
745d225d Andy McCall 1 month ago ↶
2239
2240
2241
write ( work.path ( ) , " Cargo.toml" , " version = \" 0.1.1\" \n " ) ;
2242
c.commit ( " Release 0.1.1" , & [ ] ) .unwrap ( ) ;
2243
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2244
wait_runs ( h.app.clone ( ) , 2 ) .await;
2245
let job = claim ( & h.app) .await.expect ( " a job of run 2" ) ;
2246
let id = job[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
2247
write ( work.path ( ) , " test.sh" , " echo still ok\n " ) ;
2248
c.commit ( " Edit the test" , & [ ] ) .unwrap ( ) ;
2249
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2250
let runs = wait_runs ( h.app.clone ( ) , 3 ) .await;
2251
let third = & runs[ 0 ] ;
2252
let mut second = runs[ 1 ] .clone ( ) ;
2253
for _ in 0 .. 100 {
2254
second = db:: flow:: run_by_id( & h.app.db, runs[ 1 ] .id) .await.unwrap ( ) .unwrap ( ) ;
2255
if ! second.live ( ) {
2256
break ;
2257
}
2258
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
2259
}
2260
assert_eq! ( second.status, " cancelled" , " a newer push supersedes a run still going" ) ;
2261
let jobs = db:: flow:: jobs( & h.app.db, second.id) .await.unwrap ( ) ;
2262
assert! ( jobs.iter ( ) .all ( | j | j.status = = " cancelled" & & j.error = = " superseded by run #3" ) , " {jobs:?}" ) ;
2263
let mut names: Vec< String> = db:: flow:: jobs( & h.app.db, third.id) .await.unwrap ( ) .into_iter ( ) .map ( | j | j.name ) .collect ( ) ;
2264
names.sort ( ) ;
2265
assert_eq! ( names, [ " build" , " dist" , " lint" , " test" ] , " run 3 also checks what run 2 would have: the version bump" ) ;
2266
let mut checks = vec! [ ] ;
2267
for _ in 0 .. 100 {
2268
checks = db:: hooks:: checks_for_state( & h.app.db, second.repo_id, & second.state) .await.unwrap ( ) ;
2269
if checks.iter ( ) .all ( | c | c.status = = " cancelled" ) {
2270
break ;
2271
}
2272
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
2273
}
2274
assert! ( checks.iter ( ) .all ( | c | c.status = = " cancelled" ) , " {checks:?}" ) ;
2275
2276
let ( st, _ ) =
2277
call ( & h.app, " POST" , & format! ( " /api/v1/flow/jobs/{id} /log" ) , Some ( " runner-secret" ) , Some ( r # "{"records":[]}"# .into ( ) ) ) .await;
2278
assert_eq! ( st, StatusCode:: CONFLICT ) ;
2279
2280
2281
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/stop/flow/runs/3/cancel" , Some ( " site" ) , None ) .await;
2282
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
2283
let third = db:: flow:: run_by_id( & h.app.db, third.id) .await.unwrap ( ) .unwrap ( ) ;
2284
assert_eq! ( third.status, " cancelled" ) ;
2285
assert! ( db:: flow:: jobs( & h.app.db, third.id) .await.unwrap ( ) .iter ( ) .all ( | j | j.error = = " cancelled by andy" ) ) ;
2286
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/stop/flow/runs/3/cancel" , Some ( " site" ) , None ) .await;
2287
assert! ( st = = StatusCode:: CONFLICT & & body.contains ( " already cancelled" ) , " {st} {body}" ) ;
2288
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/repos/andy/stop/flow/runs/3/cancel" , None , None ) .await;
2289
assert_ne! ( st, StatusCode:: OK , " anonymous callers cannot cancel" ) ;
2290
let ( _ , page) = call ( & h.app, " GET" , " /andy/stop/flow/3" , None , None ) .await;
2291
assert! ( page.contains ( " ⊘" ) & & ! page.contains ( " CANCEL</button>" ) , " a cancelled run shows so, with no button" ) ;
2292
}
2293
5dead1f4 Andy McCall 1 month ago ↶
2294
2295
2296
2297
# [ tokio ::test ( flavor = " multi_thread" ) ]
2298
async fn a_job_runs_under_its_own_time_limit_within_the_sites ( ) {
2299
let h = hub ( ) .await;
2300
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
2301
db:: accounts:: set_setting( & h.app.db, oxygenhub:: flow:: SETTING_DEFAULT , " 30" ) .await.unwrap ( ) ;
2302
db:: accounts:: set_setting( & h.app.db, oxygenhub:: flow:: SETTING_MAX , " 120" ) .await.unwrap ( ) ;
2303
create ( & h.app, " andy" , " limits" ) .await;
2304
let url = format! ( " {} /andy/limits" , h.base) ;
2305
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2306
let c = client ( & url, work.path ( ) ) ;
2307
let extra = " job \" slow\" {\n sandbox \" alpine\" \n timeout \" 90m\" \n exec \" true\" \n }\n job \" huge\" {\n sandbox \" alpine\" \n timeout \" 5h\" \n exec \" true\" \n }\n " ;
2308
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( extra) ) ;
2309
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2310
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
2311
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2312
assert_eq! ( runs_of ( & h.app, " andy" , " limits" ) .await.len ( ) , 1 ) ;
2313
let mut limits = std:: collections:: HashMap:: new( ) ;
2314
let mut ids = std:: collections:: HashMap:: new( ) ;
2315
while let Some ( j) = claim ( & h.app) .await {
2316
let name = j[ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
2317
limits.insert ( name.clone ( ) , j[ " timeout_s" ] .as_u64 ( ) .unwrap ( ) ) ;
2318
ids.insert ( name, j[ " job_id" ] .as_i64 ( ) .unwrap ( ) ) ;
2319
}
2320
assert_eq! ( limits[ " slow" ] , 5400 , " its own" ) ;
2321
assert_eq! ( limits[ " huge" ] , 7200 , " cut to the site's maximum" ) ;
2322
assert_eq! ( limits[ " build" ] , 1800 , " the site's default" ) ;
2323
2324
let started = db:: now_ms( ) - 40 * 60 * 1000 ;
3b1bc27f Andy McCall 20 days ago
2325
sqlx:: query( " UPDATE flow_jobs SET started_ms = $1" ) .bind ( started) .execute ( & h.app.db) .await.unwrap ( ) ;
5dead1f4 Andy McCall 1 month ago ↶
2326
oxygenhub:: flow:: reap_lost_jobs( & h.app) .await.unwrap ( ) ;
2327
let status = | id : i64 | {
2328
let app = h.app.clone ( ) ;
2329
async move { db:: flow:: job( & app.db, id) .await.unwrap ( ) .unwrap ( ) .status }
2330
} ;
2331
assert_eq! ( status ( ids[ " build" ] ) .await, " error" , " past its 30 minutes" ) ;
2332
assert_eq! ( status ( ids[ " slow" ] ) .await, " running" , " within its 90" ) ;
2333
assert_eq! ( status ( ids[ " huge" ] ) .await, " running" , " within its 120" ) ;
2334
}
c889911b Andy McCall 1 month ago ↶
2335
2336
2337
2338
2339
# [ tokio ::test ( flavor = " multi_thread" ) ]
2340
async fn on_decides_which_events_start_a_job ( ) {
2341
let h = hub ( ) .await;
2342
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
2343
create ( & h.app, " andy" , " events" ) .await;
2344
let url = format! ( " {} /andy/events" , h.base) ;
2345
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2346
let c = client ( & url, work.path ( ) ) ;
2347
let text = format! (
2348
" on {{ \n push \" main\" \n manual\n }} \n {} job \" dist\" {{ \n sandbox \" alpine\" \n on {{ \n tag \" v*\" \n }} \n exec \" true\" \n }} \n " ,
2349
pipeline ( " " )
2350
) ;
2351
2352
let text = text.replacen (
2353
" on {\n push \" main\" \n manual\n }\n schema-version 1\n " ,
2354
" schema-version 1\n on {\n push \" main\" \n manual\n }\n " ,
2355
1 ,
2356
) ;
2357
write ( work.path ( ) , " .flow/pipeline.kdl" , & text) ;
2358
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2359
c.commit ( " Start" , & [ ] ) .unwrap ( ) ;
2360
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2361
let jobs_of_latest = | app : Shared, n : usize | async move {
2362
let mut runs = Vec :: new( ) ;
2363
for _ in 0 .. 100 {
2364
runs = runs_of ( & app, " andy" , " events" ) .await;
2365
if runs.len ( ) > = n {
2366
break ;
2367
}
2368
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
2369
}
2370
assert_eq! ( runs.len ( ) , n, " runs" ) ;
2371
let mut j: Vec< String> = db:: flow:: jobs( & app.db, runs[ 0 ] .id) .await.unwrap ( ) .into_iter ( ) .map ( | j | j.name ) .collect ( ) ;
2372
j.sort ( ) ;
2373
( runs[ 0 ] .clone ( ) , j)
2374
} ;
2375
let ( _ , j) = jobs_of_latest ( h.app.clone ( ) , 1 ) .await;
2376
assert_eq! ( j, [ " build" , " lint" , " test" ] , " a push to main: no dist" ) ;
2377
2378
let heads = c.bookmark_heads ( ) .unwrap ( ) ;
2379
c.tag_create ( " v1.0.0" , & heads, " first" ) .unwrap ( ) ;
2380
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2381
let ( run, j) = jobs_of_latest ( h.app.clone ( ) , 2 ) .await;
2382
assert_eq! ( j, [ " dist" ] , " a tag runs only what names it" ) ;
2383
assert_eq! ( run.trigger, " tag:v1.0.0" ) ;
2384
2385
let row = db:: repos:: get( & h.app.db, " andy" , " events" ) .await.unwrap ( ) .unwrap ( ) ;
2386
db:: flow:: finish(
2387
& h.app.db,
2388
db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) [ 0 ] .id,
2389
db:: flow:: Outcome { status: " passed" , exit_code: Some ( 0 ) , error: " " , receipt: " " , receipt_hash: " " } ,
2390
)
2391
.await
2392
.unwrap ( ) ;
2393
let run = db:: flow:: run_by_id( & h.app.db, run.id) .await.unwrap ( ) .unwrap ( ) ;
2394
oxygenhub:: flow:: rerun( & h.app, & row, & run, " andy" ) .await.unwrap ( ) ;
2395
let ( _ , j) = jobs_of_latest ( h.app.clone ( ) , 3 ) .await;
2396
assert_eq! ( j, [ " dist" ] , " the rerun is the tag's run again" ) ;
2397
2398
c.bookmark_create ( " develop" , None ) .unwrap ( ) ;
2399
assert! ( c.push_bookmark ( " origin" , " develop" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2400
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 300 ) ) .await;
2401
assert_eq! ( runs_of ( & h.app, " andy" , " events" ) .await.len ( ) , 3 , " no run for develop" ) ;
348f8c4b Andy McCall 1 month ago ↶
2402
}
2403
2404
2405
2406
# [ tokio ::test ( flavor = " multi_thread" ) ]
2407
async fn scheduled_jobs_run_when_due ( ) {
2408
let h = hub ( ) .await;
2409
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
2410
create ( & h.app, " andy" , " nightly" ) .await;
2411
let url = format! ( " {} /andy/nightly" , h.base) ;
2412
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2413
let c = client ( & url, work.path ( ) ) ;
2414
2415
let extra = " job \" nightly\" {\n sandbox \" alpine\" \n on {\n schedule \" 17 4 29 2 *\" \n }\n exec \" true\" \n }\n " ;
2416
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( extra) ) ;
2417
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2418
c.commit ( " Start" , & [ ] ) .unwrap ( ) ;
2419
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2420
assert_eq! ( runs_of ( & h.app, " andy" , " nightly" ) .await.len ( ) , 1 ) ;
2421
let first = runs_of ( & h.app, " andy" , " nightly" ) .await;
2422
let names: Vec< String> = db:: flow:: jobs( & h.app.db, first[ 0 ] .id) .await.unwrap ( ) .into_iter ( ) .map ( | j | j.name ) .collect ( ) ;
2423
assert! ( ! names.contains ( & " nightly" .to_string ( ) ) , " a push does not run a scheduled job" ) ;
2424
use flow_core:: pipeline:: When;
2425
let not_yet = When { minute: 16 , hour: 4 , day: 29 , month: 2 , weekday: 4 } ;
2426
assert_eq! ( oxygenhub:: flow:: run_due( & h.app, not_yet) .await, 0 ) ;
2427
let due = When { minute: 17 , .. not_yet } ;
2428
assert_eq! ( oxygenhub:: flow:: run_due( & h.app, due) .await, 1 ) ;
2429
let runs = runs_of ( & h.app, " andy" , " nightly" ) .await;
2430
assert_eq! ( runs.len ( ) , 2 ) ;
2431
assert_eq! ( runs[ 0 ] .trigger, " schedule" ) ;
2432
let names: Vec< String> = db:: flow:: jobs( & h.app.db, runs[ 0 ] .id) .await.unwrap ( ) .into_iter ( ) .map ( | j | j.name ) .collect ( ) ;
2433
assert_eq! ( names, [ " nightly" ] ) ;
2434
assert_eq! ( runs[ 0 ] .bookmark, " main" , " on the default bookmark" ) ;
5f3fa151 Andy McCall 10 days ago ↶
2435
2436
2437
write ( work.path ( ) , " .flow/pipeline.kdl" , " schema-version 1\n " ) ;
2438
c.commit ( " No jobs" , & [ ] ) .unwrap ( ) ;
2439
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2440
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 300 ) ) .await;
2441
let runs = runs_of ( & h.app, " andy" , " nightly" ) .await;
2442
assert_eq! ( runs.len ( ) , 3 ) ;
2443
assert! ( runs[ 0 ] .error.contains ( " at least one job" ) , " {}" , runs[ 0 ] .error) ;
2444
assert_eq! ( oxygenhub:: flow:: run_due( & h.app, due) .await, 0 ) ;
2445
assert_eq! ( oxygenhub:: flow:: run_due( & h.app, not_yet) .await, 0 ) ;
2446
assert_eq! ( runs_of ( & h.app, " andy" , " nightly" ) .await.len ( ) , 3 , " no scheduled runs of a broken pipeline" ) ;
c889911b Andy McCall 1 month ago ↶
2447
}
2a83f530 Andy McCall 1 month ago ↶
2448
2449
# [ tokio ::test ( flavor = " multi_thread" ) ]
2450
async fn variables_from_three_levels_reach_a_job_inner_overriding_outer ( ) {
2451
use oxygenhub:: levels:: Scope;
2452
use oxygenhub:: variables;
2453
let h = hub ( ) .await;
2454
let app = & h.app;
2455
2456
let res = oxygenhub:: router( app.clone ( ) )
2457
.oneshot (
2458
Request:: post( " /register" )
2459
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
c8684d44 Andy McCall 20 days ago
2460
.body ( Body:: from( " username=andy&email=a%40example.com&password=correct-horse-battery&setup=test-setup" ) )
2a83f530 Andy McCall 1 month ago ↶
2461
.unwrap ( ) ,
2462
)
2463
.await
2464
.unwrap ( ) ;
2465
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
2466
let andy = db:: accounts:: account_by_name( & app.db, " andy" ) .await.unwrap ( ) .unwrap ( ) ;
2467
db:: accounts:: create_org( & app.db, " acme" , " Acme" , andy.id) .await.unwrap ( ) ;
2468
create ( app, " acme" , " app" ) .await;
2469
let page = | path : & str , form : Option< String> | {
2470
let mut req =
2471
Request:: builder( ) .method ( if form.is_some ( ) { " POST" } else { " GET" } ) .uri ( path) .header ( header:: COOKIE , cookie.clone ( ) ) ;
2472
if form.is_some ( ) {
2473
req = req.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) ;
2474
}
2475
let req = req.body ( Body:: from( form.unwrap_or_default ( ) ) ) .unwrap ( ) ;
c7067c7b Andy McCall 16 days ago
2476
let ( app, cookie) = ( app.clone ( ) , cookie.clone ( ) ) ;
2a83f530 Andy McCall 1 month ago ↶
2477
async move {
c7067c7b Andy McCall 16 days ago
2478
let mut res = oxygenhub:: router( app.clone ( ) ) .oneshot ( req) .await.unwrap ( ) ;
2479
2480
let back =
2481
res.headers ( ) .get ( header:: LOCATION ) .and_then ( | v | v.to_str ( ) .ok ( ) ) .filter ( | l | l.contains ( " /settings/" ) ) .map ( String :: from) ;
2482
if let Some ( to) = back {
2483
let req = Request:: get( to) .header ( header:: COOKIE , cookie) .body ( Body:: empty( ) ) .unwrap ( ) ;
2484
res = oxygenhub:: router( app) .oneshot ( req) .await.unwrap ( ) ;
2485
}
2a83f530 Andy McCall 1 month ago ↶
2486
let status = res.status ( ) ;
2487
let body = String :: from_utf8( res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) .to_vec ( ) ) .unwrap ( ) ;
2488
( status, body)
2489
}
2490
} ;
2491
let csrf_of = | page : & str | {
2492
let i = page.find ( " name=\" _csrf\" value=\" " ) .expect ( " csrf field" ) + " name=\" _csrf\" value=\" " .len ( ) ;
2493
page[ i.. ] .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( )
2494
} ;
2495
2496
2497
2498
variables:: set( & app.db, Scope:: Site, " CC" , " gcc" ) .await.unwrap ( ) ;
2499
variables:: set( & app.db, Scope:: Site, " LANG" , " C" ) .await.unwrap ( ) ;
2500
let org = db:: accounts:: org_by_name( & app.db, " acme" ) .await.unwrap ( ) .unwrap ( ) ;
2501
variables:: set( & app.db, Scope:: Org( org.id) , " CC" , " clang" ) .await.unwrap ( ) ;
c7067c7b Andy McCall 16 days ago
2502
let ( st, settings) = page ( " /acme/app/settings/variables" , None ) .await;
2a83f530 Andy McCall 1 month ago ↶
2503
assert_eq! ( st, StatusCode:: OK , " {settings}" ) ;
2504
assert! ( settings.contains ( " Inherited" ) & & settings.contains ( " the acme organisation" ) & & settings.contains ( " overridden" ) , " {settings}" ) ;
2505
let csrf = csrf_of ( & settings) ;
2506
let ( _ , body) = page ( " /acme/app/settings/variables" , Some ( format! ( " _csrf={csrf} &action=set&name=RUST_LOG&value=debug" ) ) ) .await;
2507
assert! ( body.contains ( " RUST_LOG set." ) & & body.contains ( " <code>debug</code>" ) , " {body}" ) ;
2508
let ( _ , body) = page ( " /acme/app/settings/variables" , Some ( format! ( " _csrf={csrf} &action=set&name=FLOW_REPO&value=x" ) ) ) .await;
2509
assert! ( body.contains ( " not FLOW_" ) , " {body}" ) ;
2510
let repo = db:: repos:: get( & app.db, " acme" , " app" ) .await.unwrap ( ) .unwrap ( ) ;
2511
db:: repos:: set_flow( & app.db, repo.id, " on" ) .await.unwrap ( ) ;
2512
let env = variables:: env_for_repo( & app.db, & repo) .await.unwrap ( ) ;
2513
assert_eq! (
2514
env.iter ( ) .map ( | ( k , v ) | ( k.as_str ( ) , v.as_str ( ) ) ) .collect:: < Vec< _ > > ( ) ,
2515
[ ( " CC" , " clang" ) , ( " LANG" , " C" ) , ( " RUST_LOG" , " debug" ) ] ,
2516
" the organisation's CC wins over the site's"
2517
) ;
2518
2519
2520
let url = format! ( " {} /acme/app" , h.base) ;
2521
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2522
let c = client ( & url, work.path ( ) ) ;
2523
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
2524
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2525
c.snapshot ( " test" ) .unwrap ( ) ;
2526
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
2527
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2528
let runs = runs_of ( app, " acme" , " app" ) .await;
2529
assert_eq! ( runs.len ( ) , 1 , " {runs:?}" ) ;
2530
let got = claim ( app) .await.unwrap_or_else ( | | panic! ( " a job: {:?}" , runs[ 0 ] ) ) ;
2531
assert_eq! ( got[ " env" ] , serde_json:: json! ( { " CC" : " clang" , " LANG" : " C" , " RUST_LOG" : " debug" } ) , " {got}" ) ;
2532
2533
737de0fa Andy McCall 15 days ago ↶
2534
let ( _ , admin) = page ( " /admin/variables" , None ) .await;
2a83f530 Andy McCall 1 month ago ↶
2535
assert! ( admin.contains ( " Variables" ) & & admin.contains ( " <code>gcc</code>" ) , " {admin}" ) ;
2536
let ( _ , body) = page ( " /admin/variables" , Some ( format! ( " _csrf={} &action=delete&name=LANG" , csrf_of ( & admin) ) ) ) .await;
2537
assert! ( body.contains ( " LANG removed." ) , " {body}" ) ;
2538
let got = claim ( app) .await.expect ( " another job" ) ;
2539
assert_eq! ( got[ " env" ] , serde_json:: json! ( { " CC" : " clang" , " RUST_LOG" : " debug" } ) , " {got}" ) ;
2540
}
4ac01a73 Andy McCall 1 month ago ↶
2541
2542
# [ tokio ::test ( flavor = " multi_thread" ) ]
2543
async fn secrets_a_job_declares_come_with_the_claim_and_never_reach_the_log ( ) {
2544
let h = hub ( ) .await;
2545
let app = & h.app;
2546
let res = oxygenhub:: router( app.clone ( ) )
2547
.oneshot (
2548
Request:: post( " /register" )
2549
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
c8684d44 Andy McCall 20 days ago
2550
.body ( Body:: from( " username=andy&email=a%40example.com&password=correct-horse-battery&setup=test-setup" ) )
4ac01a73 Andy McCall 1 month ago ↶
2551
.unwrap ( ) ,
2552
)
2553
.await
2554
.unwrap ( ) ;
2555
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
2556
let page = | path : & str , form : Option< String> | {
2557
let mut req =
2558
Request:: builder( ) .method ( if form.is_some ( ) { " POST" } else { " GET" } ) .uri ( path) .header ( header:: COOKIE , cookie.clone ( ) ) ;
2559
if form.is_some ( ) {
2560
req = req.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) ;
2561
}
2562
let req = req.body ( Body:: from( form.unwrap_or_default ( ) ) ) .unwrap ( ) ;
c7067c7b Andy McCall 16 days ago
2563
let ( app, cookie) = ( app.clone ( ) , cookie.clone ( ) ) ;
4ac01a73 Andy McCall 1 month ago ↶
2564
async move {
c7067c7b Andy McCall 16 days ago
2565
let mut res = oxygenhub:: router( app.clone ( ) ) .oneshot ( req) .await.unwrap ( ) ;
2566
2567
let back =
2568
res.headers ( ) .get ( header:: LOCATION ) .and_then ( | v | v.to_str ( ) .ok ( ) ) .filter ( | l | l.contains ( " /settings/" ) ) .map ( String :: from) ;
2569
if let Some ( to) = back {
2570
let req = Request:: get( to) .header ( header:: COOKIE , cookie) .body ( Body:: empty( ) ) .unwrap ( ) ;
2571
res = oxygenhub:: router( app) .oneshot ( req) .await.unwrap ( ) ;
2572
}
4ac01a73 Andy McCall 1 month ago ↶
2573
let status = res.status ( ) ;
2574
let body = String :: from_utf8( res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) .to_vec ( ) ) .unwrap ( ) ;
2575
( status, body)
2576
}
2577
} ;
2578
let csrf_of = | page : & str | {
2579
let i = page.find ( " name=\" _csrf\" value=\" " ) .expect ( " csrf field" ) + " name=\" _csrf\" value=\" " .len ( ) ;
2580
page[ i.. ] .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( )
2581
} ;
2582
create ( app, " andy" , " demo" ) .await;
2583
2584
2585
let ( _ , admin) = page ( " /admin" , None ) .await;
2586
let csrf = csrf_of ( & admin) ;
2587
let ( _ , body) = page ( " /admin/secrets" , Some ( format! ( " _csrf={csrf} &action=set&name=deploy-token&value=site-level-value" ) ) ) .await;
2588
assert! ( body.contains ( " deploy-token set." ) & & ! body.contains ( " site-level-value" ) , " {body}" ) ;
2589
let ( _ , body) = page ( " /admin/secrets" , Some ( format! ( " _csrf={csrf} &action=set&name=bad+name&value=x" ) ) ) .await;
2590
assert! ( body.contains ( " name is 1 to 64 of" ) , " {body}" ) ;
c7067c7b Andy McCall 16 days ago
2591
let ( _ , settings) = page ( " /andy/demo/settings/secrets" , None ) .await;
4ac01a73 Andy McCall 1 month ago ↶
2592
assert! (
2593
settings.contains ( " Secrets" ) & & settings.contains ( " <code>deploy-token</code>" ) & & settings.contains ( " the site</td>" ) ,
2594
" {settings}"
2595
) ;
2596
let csrf_r = csrf_of ( & settings) ;
2597
let ( _ , body) =
2598
page ( " /andy/demo/settings/secrets" , Some ( format! ( " _csrf={csrf_r} &action=set&name=deploy-token&value=hunter22-repo-value" ) ) ) .await;
2599
assert! ( body.contains ( " deploy-token set." ) & & body.contains ( " overridden" ) & & ! body.contains ( " hunter22" ) , " {body}" ) ;
2600
let repo = db:: repos:: get( & app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
2601
assert! (
2602
! String :: from_utf8_lossy( & oxygenhub:: secrets:: list( & app.db, oxygenhub:: levels:: Scope:: Repo( repo.id) ) .await.unwrap ( ) [ 0 ] .value)
2603
.contains ( " hunter22" ) ,
2604
" sealed at rest"
2605
) ;
2606
2607
2608
let url = format! ( " {} /andy/demo" , h.base) ;
2609
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2610
let c = client ( & url, work.path ( ) ) ;
2611
write (
2612
work.path ( ) ,
2613
" .flow/pipeline.kdl" ,
2614
& pipeline ( " job \" deploy\" {\n sandbox \" alpine\" \n secret \" deploy-token\" {\n mount-path \" /run/secrets/token\" \n }\n secret \" missing\" {\n mount-path \" /run/secrets/missing\" \n }\n exec \" deploy\" \n }\n " ) ,
2615
) ;
2616
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2617
c.snapshot ( " test" ) .unwrap ( ) ;
2618
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
2619
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2620
let runs = runs_of ( app, " andy" , " demo" ) .await;
2621
let jobs = match runs.first ( ) {
2622
Some ( r) => db:: flow:: jobs( & app.db, r.id) .await.unwrap ( ) .iter ( ) .map ( | j | format! ( " {} :{} " , j.name, j.status) ) .collect:: < Vec< _ > > ( ) ,
2623
None => vec! [ ] ,
2624
} ;
2625
let mut deploy = None ;
2626
for _ in 0 .. 50 {
2627
let Some ( got) = claim ( app) .await else {
2628
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 100 ) ) .await;
2629
continue ;
2630
} ;
2631
if got[ " job" ] [ " name" ] = = " deploy" {
2632
deploy = Some ( got) ;
2633
break ;
2634
}
2635
assert_eq! ( got[ " secrets" ] , serde_json:: Value:: Null, " {got}" ) ;
2636
}
2637
let got = deploy.unwrap_or_else ( | | panic! ( " the deploy job: runs {runs:?} jobs {jobs:?}" ) ) ;
2638
assert_eq! (
2639
got[ " secrets" ] ,
2640
serde_json:: json! ( [ { " name" : " deploy-token" , " path" : " /run/secrets/token" , " value" : " hunter22-repo-value" } ] ) ,
2641
" the repository's value, the missing one left out: {got}"
2642
) ;
2643
let id = got[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
2644
let log = db:: flow:: log( & app.db, id, 0 , 100 ) .await.unwrap ( ) ;
2645
assert! ( log.iter ( ) .any ( | l | l.stream = = " flow" & & l.text.contains ( " secret \" missing\" is not set" ) ) , " {log:?}" ) ;
2646
2647
2648
let records = serde_json:: json! ( { " records" : [
2649
{ " t" : 1 , " stream" : " out" , " text" : " token=hunter22-repo-value" } ,
2650
{ " t" : 2 , " stream" : " err" , " text" : " b64 aHVudGVyMjItcmVwby12YWx1ZQ== end" } ,
2651
] } ) ;
2652
let ( st, _ ) = call ( app, " POST" , & format! ( " /api/v1/flow/jobs/{id} /log" ) , Some ( " runner-secret" ) , Some ( records.to_string ( ) ) ) .await;
2653
assert_eq! ( st, StatusCode:: NO_CONTENT ) ;
2654
let log = db:: flow:: log( & app.db, id, 0 , 100 ) .await.unwrap ( ) ;
2655
let texts: Vec< & str > = log.iter ( ) .map ( | l | l.text.as_str ( ) ) .collect ( ) ;
2656
assert! ( texts.contains ( & " token=***" ) & & texts.contains ( & " b64 *** end" ) , " {texts:?}" ) ;
2657
assert! ( ! texts.iter ( ) .any ( | t | t.contains ( " hunter22" ) ) , " {texts:?}" ) ;
2658
}
d1727c57 Andy McCall 1 month ago ↶
2659
2660
# [ tokio ::test ( flavor = " multi_thread" ) ]
2661
async fn a_job_s_identity_is_a_signed_token_the_discovery_document_lets_anyone_verify ( ) {
2662
let h = hub ( ) .await;
2663
let app = & h.app;
2664
let ( st, disc) = call ( app, " GET" , " /.well-known/openid-configuration" , None , None ) .await;
2665
assert_eq! ( st, StatusCode:: OK , " {disc}" ) ;
2666
let disc: serde_json:: Value = serde_json:: from_str( & disc) .unwrap ( ) ;
2667
assert_eq! ( disc[ " issuer" ] , h.base.as_str ( ) ) ;
2668
assert_eq! ( disc[ " jwks_uri" ] , format! ( " {} /.well-known/jwks.json" , h.base) ) ;
2669
let ( st, jwks) = call ( app, " GET" , " /.well-known/jwks.json" , None , None ) .await;
2670
assert_eq! ( st, StatusCode:: OK , " {jwks}" ) ;
2671
let jwks: serde_json:: Value = serde_json:: from_str( & jwks) .unwrap ( ) ;
2672
assert_eq! ( jwks[ " keys" ] [ 0 ] [ " alg" ] , " RS256" ) ;
2673
let kid = jwks[ " keys" ] [ 0 ] [ " kid" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
2674
2675
create ( app, " andy" , " demo" ) .await;
2676
let url = format! ( " {} /andy/demo" , h.base) ;
2677
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2678
let c = client ( & url, work.path ( ) ) ;
2679
write (
2680
work.path ( ) ,
2681
" .flow/pipeline.kdl" ,
2682
& pipeline ( " job \" deploy\" {\n sandbox \" alpine\" \n identity \" sts.amazonaws.com\" {\n mount-path \" /run/secrets/aws-token\" \n }\n exec \" deploy\" \n }\n " ) ,
2683
) ;
2684
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2685
c.snapshot ( " test" ) .unwrap ( ) ;
2686
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
2687
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2688
let mut deploy = None ;
2689
for _ in 0 .. 50 {
2690
let Some ( got) = claim ( app) .await else {
2691
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 100 ) ) .await;
2692
continue ;
2693
} ;
2694
if got[ " job" ] [ " name" ] = = " deploy" {
2695
deploy = Some ( got) ;
2696
break ;
2697
}
2698
}
2699
let got = deploy.expect ( " the deploy job" ) ;
2700
let secrets = got[ " secrets" ] .as_array ( ) .expect ( " secrets" ) ;
2701
assert_eq! ( secrets.len ( ) , 1 ) ;
2702
assert_eq! (
2703
( secrets[ 0 ] [ " name" ] .as_str ( ) , secrets[ 0 ] [ " path" ] .as_str ( ) ) ,
2704
( Some ( " identity:sts.amazonaws.com" ) , Some ( " /run/secrets/aws-token" ) )
2705
) ;
2706
let token = secrets[ 0 ] [ " value" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
2707
let issuer = app.oidc ( ) .await.unwrap ( ) ;
2708
assert_eq! ( issuer.kid, kid, " the key the JWKS publishes signed it" ) ;
2709
let claims = issuer.verify ( & token, " sts.amazonaws.com" , db:: now_ms( ) ) .expect ( " verifies" ) ;
2710
assert_eq! ( claims[ " sub" ] , " repo:andy/demo:bookmark:main" ) ;
2711
assert_eq! ( claims[ " repository" ] , " andy/demo" ) ;
2712
assert_eq! ( claims[ " job" ] , " deploy" ) ;
2713
assert_eq! ( claims[ " run_number" ] , 1 ) ;
2714
assert_eq! ( claims[ " runner" ] , " test-1" ) ;
2715
assert_eq! ( claims[ " iss" ] , h.base.as_str ( ) ) ;
2716
assert! ( claims[ " state" ] .as_str ( ) .unwrap ( ) .len ( ) = = 64 ) ;
2717
2718
let id = got[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
2719
let records = serde_json:: json! ( { " records" : [ { " t" : 1 , " stream" : " out" , " text" : format! ( " TOKEN={token} end" ) } ] } ) ;
2720
let ( st, _ ) = call ( app, " POST" , & format! ( " /api/v1/flow/jobs/{id} /log" ) , Some ( " runner-secret" ) , Some ( records.to_string ( ) ) ) .await;
2721
assert_eq! ( st, StatusCode:: NO_CONTENT ) ;
2722
let log = db:: flow:: log( & app.db, id, 0 , 100 ) .await.unwrap ( ) ;
2723
assert! ( log.iter ( ) .any ( | l | l.text = = " TOKEN=*** end" ) , " {log:?}" ) ;
2724
}
efd29a00 Andy McCall 1 month ago ↶
2725
2726
# [ tokio ::test ( flavor = " multi_thread" ) ]
2727
async fn jobs_are_offered_only_to_runners_of_their_architecture ( ) {
2728
let h = hub ( ) .await;
2729
let app = & h.app;
2730
create ( app, " andy" , " demo" ) .await;
2731
let url = format! ( " {} /andy/demo" , h.base) ;
2732
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2733
let c = client ( & url, work.path ( ) ) ;
2734
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " sandbox \" board\" {\n rootfs \" cas://blob/{ROOTFS}\" \n arch \" riscv64\" \n }\n job \" flash\" {\n sandbox \" board\" \n exec \" flash\" \n }\n " ) .replace ( " {ROOTFS}" , & " ab" .repeat ( 32 ) ) ) ;
2735
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2736
c.snapshot ( " test" ) .unwrap ( ) ;
2737
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
2738
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2739
let claim_as = | arch : & 'static str , archs : serde_json:: Value| {
2740
let app = app.clone ( ) ;
2741
async move {
2742
let body = serde_json:: json! ( {
2743
" runner" : format! ( " {arch} -1" ) , " runtimes" : [ " microvm" ] ,
2744
" info" : { " arch" : arch, " os" : " linux" , " version" : " t" , " labels" : [ ] , " max_jobs" : 1 } ,
2745
" image_archs" : archs,
2746
} ) ;
2747
let ( st, body) = call ( & app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( body.to_string ( ) ) ) .await;
2748
match st {
2749
StatusCode:: NO_CONTENT => None ,
2750
StatusCode:: OK => {
2751
Some ( serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) .to_string ( ) )
2752
}
2753
other => panic! ( " claim: {other} {body}" ) ,
2754
}
2755
}
2756
} ;
2757
2758
2759
let rootfs = pipeline ( " " ) .split ( " cas://blob/" ) .nth ( 1 ) .unwrap ( ) [ .. 64 ] .to_string ( ) ;
2760
for _ in 0 .. 50 {
2761
if ! runs_of ( app, " andy" , " demo" ) .await.is_empty ( ) {
2762
break ;
2763
}
2764
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 100 ) ) .await;
2765
}
2766
let runs = runs_of ( app, " andy" , " demo" ) .await;
2767
assert_eq! ( runs.len ( ) , 1 , " {runs:?}" ) ;
2768
let mut taken = vec! [ ] ;
2769
while let Some ( j) = claim_as ( " x86_64" , serde_json:: json! ( { rootfs.clone ( ) : " x86_64" } ) ) .await {
2770
taken.push ( j) ;
2771
}
2772
assert_eq! ( taken, [ " build" , " lint" ] , " test waits for build; flash is for riscv64" ) ;
2773
2774
assert_eq! ( claim_as ( " riscv64" , serde_json:: json! ( { } ) ) .await.as_deref ( ) , Some ( " flash" ) ) ;
2775
assert_eq! ( claim_as ( " riscv64" , serde_json:: json! ( { } ) ) .await, None ) ;
2776
let ( _ , queue) = call ( app, " GET" , " /admin/runners" , None , None ) .await;
2777
let _ = queue;
2778
}
4d2e44d7 Andy McCall 1 month ago ↶
2779
2780
2781
2782
2783
2784
# [ tokio ::test ( flavor = " multi_thread" ) ]
2785
async fn a_declined_job_goes_to_another_runner_or_ends_unplaceable ( ) {
2786
let h = hub ( ) .await;
2787
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
2788
create ( & h.app, " andy" , " declines" ) .await;
2789
let url = format! ( " {} /andy/declines" , h.base) ;
2790
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2791
let c = client ( & url, work.path ( ) ) ;
2792
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
2793
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2794
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
2795
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2796
let run = runs_of ( & h.app, " andy" , " declines" ) .await[ 0 ] .clone ( ) ;
2797
let claim_as = | token : & 'static str , name : & 'static str | {
2798
let app = h.app.clone ( ) ;
2799
async move {
2800
let body = serde_json:: json! ( { " runner" : name, " runtimes" : [ " microvm" ] } ) .to_string ( ) ;
2801
let ( st, body) = call ( & app, " POST" , " /api/v1/flow/claim" , Some ( token) , Some ( body) ) .await;
2802
( st = = StatusCode:: OK ) .then ( | | serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " job_id" ] .as_i64 ( ) .unwrap ( ) )
2803
}
2804
} ;
2805
let decline = | token : & 'static str , id : i64 , reason : & 'static str | {
2806
let app = h.app.clone ( ) ;
2807
async move {
2808
let body = serde_json:: json! ( { " reason" : reason } ) .to_string ( ) ;
2809
call ( & app, " POST" , & format! ( " /api/v1/flow/jobs/{id} /decline" ) , Some ( token) , Some ( body) ) .await.0
2810
}
2811
} ;
2812
let job = claim_as ( " runner-secret" , " earth-1" ) .await.expect ( " a job" ) ;
2813
2814
assert_eq! ( decline ( " runner-two" , job, " not mine" ) .await, StatusCode:: CONFLICT , " only the runner it runs on may decline it" ) ;
2815
assert_eq! ( decline ( " runner-secret" , job, " the MicroVM tier needs /dev/kvm" ) .await, StatusCode:: NO_CONTENT ) ;
2816
let j = db:: flow:: job( & h.app.db, job) .await.unwrap ( ) .unwrap ( ) ;
2817
assert_eq! ( ( j.status.as_str ( ) , j.runner.as_str ( ) ) , ( " queued" , " " ) ) ;
2818
assert_eq! ( j.waiting, " declined by earth-1: the MicroVM tier needs /dev/kvm; waiting for another runner" ) ;
2819
2820
let next = claim_as ( " runner-secret" , " earth-1" ) .await.expect ( " the other ready job" ) ;
2821
assert_ne! ( next, job) ;
2822
let page = call ( & h.app, " GET" , & format! ( " /andy/declines/flow/{} " , run.number) , None , None ) .await.1 ;
2823
assert! ( page.contains ( " Waiting: declined by earth-1: the MicroVM tier needs /dev/kvm" ) , " {page}" ) ;
2824
assert_eq! ( claim_as ( " runner-two" , " plex-1" ) .await, Some ( job) , " another runner takes it" ) ;
2825
assert_eq! ( decline ( " runner-two" , job, " rootfs 36fc4e323804 is not on this runner" ) .await, StatusCode:: NO_CONTENT ) ;
2826
assert_eq! ( claim_as ( " runner-three" , " narvi" ) .await, Some ( job) ) ;
2827
assert_eq! ( decline ( " runner-three" , job, " rootfs 36fc4e323804 is for x86_64; this runner is riscv64" ) .await, StatusCode:: NO_CONTENT ) ;
2828
let j = db:: flow:: job( & h.app.db, job) .await.unwrap ( ) .unwrap ( ) ;
2829
assert_eq! ( j.status, " error" ) ;
2830
assert_eq! (
2831
j.error,
2832
" unplaceable: declined by 3 runners: earth-1: the MicroVM tier needs /dev/kvm; plex-1: rootfs 36fc4e323804 is not on this runner; narvi: rootfs 36fc4e323804 is for x86_64; this runner is riscv64"
2833
) ;
2834
let r = db:: runners:: list( & h.app.db) .await.unwrap ( ) .into_iter ( ) .find ( | r | r.name = = " plex-1" ) .unwrap ( ) ;
2835
let declined = db:: flow:: declines_of_runner( & h.app.db, r.id, 10 ) .await.unwrap ( ) ;
2836
assert_eq! ( declined.len ( ) , 1 ) ;
2837
assert_eq! ( ( declined[ 0 ] .job_id, declined[ 0 ] .reason.as_str ( ) ) , ( job, " rootfs 36fc4e323804 is not on this runner" ) ) ;
2838
assert_eq! ( r.tier_list ( ) , vec! [ " microvm" ] ) ;
2839
}
554ccacf Andy McCall 1 month ago ↶
2840
ab4df6aa Andy McCall 11 days ago ↶
2841
2842
2843
2844
2845
# [ tokio ::test ( flavor = " multi_thread" ) ]
2846
async fn a_drained_job_goes_back_ahead_of_new_work ( ) {
2847
let h = hub ( ) .await;
2848
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
2849
create ( & h.app, " andy" , " drains" ) .await;
2850
let url = format! ( " {} /andy/drains" , h.base) ;
2851
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2852
let c = client ( & url, work.path ( ) ) ;
e4203e14 Andy McCall 10 days ago ↶
2853
let bulk = " job \" bulk\" {\n sandbox \" alpine\" \n priority \" low\" \n checkpoint \" /src/state\" \n exec \" true\" \n }\n " ;
ab4df6aa Andy McCall 11 days ago ↶
2854
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( bulk) ) ;
2855
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2856
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
2857
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2858
runs_of ( & h.app, " andy" , " drains" ) .await;
2859
let claim_as = | token : & 'static str , name : & 'static str | {
2860
let app = h.app.clone ( ) ;
2861
async move {
2862
let body = serde_json:: json! ( { " runner" : name, " runtimes" : [ " microvm" ] } ) .to_string ( ) ;
2863
let ( st, body) = call ( & app, " POST" , " /api/v1/flow/claim" , Some ( token) , Some ( body) ) .await;
2864
( st = = StatusCode:: OK ) .then ( | | serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " job_id" ] .as_i64 ( ) .unwrap ( ) )
2865
}
2866
} ;
2867
let release = | token : & 'static str , id : i64 , reason : & 'static str | {
2868
let app = h.app.clone ( ) ;
2869
async move {
2870
let body = serde_json:: json! ( { " reason" : reason } ) .to_string ( ) ;
2871
call ( & app, " POST" , & format! ( " /api/v1/flow/jobs/{id} /release" ) , Some ( token) , Some ( body) ) .await.0
2872
}
2873
} ;
2874
let name_of = | id : i64 | {
2875
let app = h.app.clone ( ) ;
2876
async move { db:: flow:: job( & app.db, id) .await.unwrap ( ) .unwrap ( ) .name }
2877
} ;
2878
let build = claim_as ( " runner-secret" , " earth-1" ) .await.unwrap ( ) ;
2879
let lint = claim_as ( " runner-two" , " plex-1" ) .await.unwrap ( ) ;
2880
let low = claim_as ( " runner-three" , " phone" ) .await.unwrap ( ) ;
2881
assert_eq! ( ( name_of ( build) .await, name_of ( lint) .await, name_of ( low) .await) , ( " build" .into ( ) , " lint" .into ( ) , " bulk" .into ( ) ) ) ;
2882
2883
let line = serde_json:: json! ( { " records" : [ { " t" : 1200 , " stream" : " stdout" , " text" : " step 1 of 30" } ] } ) .to_string ( ) ;
2884
call ( & h.app, " POST" , & format! ( " /api/v1/flow/jobs/{low} /log" ) , Some ( " runner-three" ) , Some ( line) ) .await;
2885
assert_eq! ( finish ( & h.app, build, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 } ) ) .await, StatusCode:: NO_CONTENT ) ;
e4203e14 Andy McCall 10 days ago ↶
2886
2887
let ( st, body) = call ( & h.app, " POST" , & format! ( " /api/v1/flow/jobs/{lint} /checkpoint" ) , Some ( " runner-two" ) , Some ( " x" .into ( ) ) ) .await;
2888
assert_eq! ( st, StatusCode:: BAD_REQUEST , " lint declares no checkpoint: {body}" ) ;
2889
let ( st, body) =
2890
call ( & h.app, " POST" , & format! ( " /api/v1/flow/jobs/{low} /checkpoint" ) , Some ( " runner-three" ) , Some ( " step 32" .into ( ) ) ) .await;
2891
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
2892
let saved = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " blob" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
ab4df6aa Andy McCall 11 days ago ↶
2893
assert_eq! ( release ( " runner-two" , low, " off Wi-Fi" ) .await, StatusCode:: CONFLICT , " only the runner that holds it" ) ;
2894
assert_eq! ( release ( " runner-three" , low, " off Wi-Fi" ) .await, StatusCode:: NO_CONTENT ) ;
2895
let j = db:: flow:: job( & h.app.db, low) .await.unwrap ( ) .unwrap ( ) ;
2896
assert_eq! ( ( j.status.as_str ( ) , j.runner.as_str ( ) , j.handbacks) , ( " queued" , " " , 1 ) ) ;
2897
let log: Vec< String> = db:: flow:: log( & h.app.db, low, 0 , 50 ) .await.unwrap ( ) .into_iter ( ) .map ( | l | l.text ) .collect ( ) ;
2898
assert_eq! ( log, [ " step 1 of 30" , " ── handed back by phone: off Wi-Fi; queued again ahead of new work" ] ) ;
2899
assert! ( db:: flow:: declines_of_job( & h.app.db, low) .await.unwrap ( ) .is_empty ( ) , " not a decline" ) ;
2900
2901
e4203e14 Andy McCall 10 days ago ↶
2902
let body = serde_json:: json! ( { " runner" : " phone" , " runtimes" : [ " microvm" ] } ) .to_string ( ) ;
2903
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-three" ) , Some ( body) ) .await;
2904
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
2905
let again: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
2906
assert_eq! ( ( again[ " job_id" ] .as_i64 ( ) , again[ " checkpoint" ] .as_str ( ) ) , ( Some ( low) , Some ( saved.as_str ( ) ) ) , " it resumes: {again}" ) ;
2907
let ( st, bytes) = call ( & h.app, " GET" , & format! ( " /api/v1/flow/jobs/{low} /checkpoint" ) , Some ( " runner-three" ) , None ) .await;
2908
assert_eq! ( ( st, bytes.as_str ( ) ) , ( StatusCode:: OK , " step 32" ) ) ;
2909
assert! ( db:: flow:: artifacts( & h.app.db, low) .await.unwrap ( ) .is_empty ( ) , " a checkpoint is no artifact of the job" ) ;
ab4df6aa Andy McCall 11 days ago ↶
2910
assert_eq! ( claim_as ( " runner-secret" , " earth-1" ) .await.map ( | id | id ! = low ) , Some ( true ) , " then new work" ) ;
2911
2912
2913
let phone = db:: runners:: list( & h.app.db) .await.unwrap ( ) .into_iter ( ) .find ( | r | r.name = = " phone" ) .unwrap ( ) ;
2914
sqlx:: query( " UPDATE flow_runners SET last_seen_ms = $1 WHERE id = $2" )
2915
.bind ( db:: now_ms( ) - oxygenhub:: flow:: LEASE_MS - 1000 )
2916
.bind ( phone.id)
2917
.execute ( & h.app.db)
2918
.await
2919
.unwrap ( ) ;
2920
assert_eq! ( oxygenhub:: flow:: requeue_lapsed( & h.app) .await.unwrap ( ) , 1 ) ;
2921
let j = db:: flow:: job( & h.app.db, low) .await.unwrap ( ) .unwrap ( ) ;
2922
assert_eq! ( ( j.status.as_str ( ) , j.handbacks) , ( " queued" , 2 ) ) ;
2923
let log: Vec< String> = db:: flow:: log( & h.app.db, low, 0 , 50 ) .await.unwrap ( ) .into_iter ( ) .map ( | l | l.text ) .collect ( ) ;
2924
assert_eq! ( log.last ( ) .unwrap ( ) , " ── the runner phone has not been heard from for 3 minutes; queued again ahead of new work" ) ;
2925
assert_eq! ( claim_as ( " runner-two" , " plex-1" ) .await, Some ( low) , " another runner takes it" ) ;
2926
}
2927
3b650652 Andy McCall 11 days ago ↶
2928
2929
2930
2931
# [ tokio ::test ( flavor = " multi_thread" ) ]
2932
async fn a_claim_carries_the_job_s_expected_time ( ) {
2933
let h = hub ( ) .await;
2934
create ( & h.app, " andy" , " eta" ) .await;
2935
let url = format! ( " {} /andy/eta" , h.base) ;
2936
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2937
let c = client ( & url, work.path ( ) ) ;
2938
let one = " schema-version 1\n sandbox \" alpine\" {\n rootfs \" cas://blob/ROOTFS\" \n }\n job \" build\" {\n sandbox \" alpine\" \n exec \" make\" \n }\n " .replace ( " ROOTFS" , ROOTFS ) ;
2939
write ( work.path ( ) , " .flow/pipeline.kdl" , & one) ;
2940
c.commit ( " One job" , & [ ] ) .unwrap ( ) ;
2941
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2942
runs_of ( & h.app, " andy" , " eta" ) .await;
2943
let first = claim ( & h.app) .await.expect ( " a job" ) ;
2944
assert! ( first.get ( " expected_ms" ) .is_none ( ) , " never passed: no estimate, {first}" ) ;
2945
let id = first[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
2946
assert_eq! ( finish ( & h.app, id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 } ) ) .await, StatusCode:: NO_CONTENT ) ;
2947
2948
sqlx:: query( " UPDATE flow_jobs SET started_ms = finished_ms - 42000 WHERE id = $1" ) .bind ( id) .execute ( & h.app.db) .await.unwrap ( ) ;
2949
write ( work.path ( ) , " README" , " again\n " ) ;
2950
c.commit ( " Again" , & [ ] ) .unwrap ( ) ;
2951
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2952
let mut next = None ;
2953
for _ in 0 .. 100 {
2954
if let Some ( j) = claim ( & h.app) .await {
2955
next = Some ( j) ;
2956
break ;
2957
}
2958
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
2959
}
2960
let next = next.expect ( " the second run's job" ) ;
2961
assert_eq! ( next[ " expected_ms" ] , 42_000 , " {next}" ) ;
2962
}
2963
554ccacf Andy McCall 1 month ago ↶
2964
2965
2966
2967
# [ tokio ::test ( flavor = " multi_thread" ) ]
2968
async fn a_coverage_artifact_becomes_a_coverage_report ( ) {
2969
let h = hub ( ) .await;
2970
create ( & h.app, " andy" , " cov" ) .await;
2971
let url = format! ( " {} /andy/cov" , h.base) ;
2972
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2973
let c = client ( & url, work.path ( ) ) ;
2974
write (
2975
work.path ( ) ,
2976
" .flow/pipeline.kdl" ,
2977
& format! (
2978
" schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" coverage\" {{ \n sandbox \" alpine\" \n exec \" make\" \" coverage\" \n output {{ \n artifact \" coverage\" {{ \n path \" /src/lcov.info\" \n }} \n }} \n }} \n "
2979
) ,
2980
) ;
2981
write ( work.path ( ) , " src/lib.rs" , " pub fn add(a: u32, b: u32) -> u32 {\n a + b\n }\n \n pub fn unused() {\n todo!()\n }\n " ) ;
2982
c.snapshot ( " test" ) .unwrap ( ) ;
2983
c.commit ( " Code and coverage" , & [ ] ) .unwrap ( ) ;
2984
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2985
let run = runs_of ( & h.app, " andy" , " cov" ) .await.remove ( 0 ) ;
c8df33a6 Andy McCall 25 days ago ↶
2986
assert! ( ! call ( & h.app, " GET" , " /andy/cov" , None , None ) .await.1. contains ( " >Reports</a>" ) , " no tab before a report" ) ;
554ccacf Andy McCall 1 month ago ↶
2987
let id = claim ( & h.app) .await.unwrap ( ) [ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
2988
2989
let ( st, text) = upload ( & h.app, id, " coverage" , b " this is not a report" ) .await;
2990
assert_eq! ( st, StatusCode:: OK , " a bad report never fails the upload: {text}" ) ;
2991
let lcov = " TN:\n SF:/src/src/lib.rs\n FN:1,add\n FN:5,unused\n FNDA:3,add\n FNDA:0,unused\n DA:1,3\n DA:2,3\n DA:5,0\n DA:6,0\n end_of_record\n \
2992
SF:/src/src/gone.rs\n DA:1,1\n end_of_record\n SF:/usr/lib/rustlib/x.rs\n DA:1,1\n end_of_record\n " ;
2993
let ( st, text) = upload ( & h.app, id, " coverage" , lcov.as_bytes ( ) ) .await;
2994
assert_eq! ( st, StatusCode:: OK , " {text}" ) ;
2995
let log: Vec< String> = db:: flow:: log( & h.app.db, id, 0 , 50 ) .await.unwrap ( ) .into_iter ( ) .map ( | l | l.text ) .collect ( ) ;
2996
assert! ( log.iter ( ) .any ( | l | l.contains ( " coverage: this is not an LCOV report" ) ) , " {log:?}" ) ;
2997
assert! (
2998
log.iter ( ) .any ( | l | l = = " coverage: 50.0% of lines (2 of 4) in 1 files (2 named files are not in this state and were left out)" ) ,
2999
" {log:?}"
3000
) ;
3001
3002
let ( st, page) = call ( & h.app, " GET" , " /andy/cov/coverage" , None , None ) .await;
3003
assert_eq! ( st, StatusCode:: OK ) ;
c987c865 Andy McCall 1 month ago ↶
3004
assert! (
c8df33a6 Andy McCall 25 days ago ↶
3005
page.contains ( " >Reports</a>" )
c987c865 Andy McCall 1 month ago ↶
3006
& & page.contains ( " 50.0%" )
3007
& & page.contains ( " <strong>2</strong> of <strong>4</strong> relevant lines covered" ) ,
3008
" {page}"
3009
) ;
554ccacf Andy McCall 1 month ago ↶
3010
assert! ( page.contains ( " 1 of 2" ) & & page.contains ( " /andy/cov/coverage/" ) & & page.contains ( " /file/src/lib.rs" ) , " {page}" ) ;
3011
let report = crate_report ( & h.app, run.id) .await;
3012
let ( st, file) = call ( & h.app, " GET" , & format! ( " /andy/cov/coverage/{report} /file/src/lib.rs" ) , None , None ) .await;
3013
assert_eq! ( st, StatusCode:: OK , " {file}" ) ;
3014
assert_eq! ( file.matches ( " class=\" cov-hit\" " ) .count ( ) , 2 , " {file}" ) ;
3015
assert_eq! ( file.matches ( " class=\" cov-miss\" " ) .count ( ) , 2 , " {file}" ) ;
3016
assert! ( file.contains ( " 3×" ) , " {file}" ) ;
3017
assert_eq! ( call ( & h.app, " GET" , & format! ( " /andy/cov/coverage/{report} /file/src/gone.rs" ) , None , None ) .await.0 , StatusCode:: NOT_FOUND ) ;
3018
let ( _ , badge) = call ( & h.app, " GET" , " /andy/cov/coverage/badge.svg" , None , None ) .await;
3019
assert! ( badge.contains ( " coverage: 50.0%" ) , " {badge}" ) ;
3020
let ( _ , run_page) = call ( & h.app, " GET" , & format! ( " /andy/cov/flow/{} " , run.number) , None , None ) .await;
3021
assert! ( run_page.contains ( " coverage 50.0%" ) , " {run_page}" ) ;
617c2840 Andy McCall 1 month ago ↶
3022
3023
let ( _ , list) = call ( & h.app, " GET" , " /api/v1/repos/andy/cov/coverage" , None , None ) .await;
3024
let list: serde_json:: Value = serde_json:: from_str( & list) .unwrap ( ) ;
3025
assert_eq! ( ( list[ 0 ] [ " percent" ] .as_f64 ( ) , list[ 0 ] [ " format" ] .as_str ( ) ) , ( Some ( 50. 0 ) , Some ( " lcov" ) ) , " {list}" ) ;
3026
let ( _ , one) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/cov/coverage/{report} " ) , None , None ) .await;
3027
let one: serde_json:: Value = serde_json:: from_str( & one) .unwrap ( ) ;
3028
assert_eq! ( ( one[ " files" ] [ 0 ] [ " path" ] .as_str ( ) , one[ " files" ] [ 0 ] [ " lines_hit" ] .as_i64 ( ) ) , ( Some ( " src/lib.rs" ) , Some ( 2 ) ) , " {one}" ) ;
7edaff01 Andy McCall 1 month ago ↶
3029
3030
3031
3032
let row = db:: repos:: get( & h.app.db, " andy" , " cov" ) .await.unwrap ( ) .unwrap ( ) ;
3033
let check = | checks : Vec< db:: hooks:: Check> | checks.into_iter ( ) .find ( | c | c.name = = " coverage" ) .expect ( " a coverage check" ) ;
3034
let c = check ( db:: hooks:: checks_for_state( & h.app.db, row.id, & run.state) .await.unwrap ( ) ) ;
3035
assert_eq! ( ( c.status.as_str ( ) , c.summary.as_str ( ) ) , ( " passed" , " 50.0%" ) ) ;
3036
assert! ( c.url.ends_with ( & format! ( " /andy/cov/coverage?report={report} " ) ) , " {}" , c.url) ;
3037
oxygenhub:: coverage:: set_rules( & h.app.db, row.id, oxygenhub:: coverage:: Rules { min: Some ( 60. 0 ) , max_drop: None } ) .await.unwrap ( ) ;
3038
assert_eq! ( upload ( & h.app, id, " coverage" , lcov.as_bytes ( ) ) .await.0 , StatusCode:: OK ) ;
3039
let c = check ( db:: hooks:: checks_for_state( & h.app.db, row.id, & run.state) .await.unwrap ( ) ) ;
3040
assert_eq! ( ( c.status.as_str ( ) , c.summary.as_str ( ) ) , ( " failed" , " 50.0%: below the minimum of 60%" ) ) ;
554ccacf Andy McCall 1 month ago ↶
3041
}
3042
3043
async fn crate_report ( app : & Shared, run_id : i64 ) -> i64 {
3044
oxygenhub:: coverage:: for_run( & app.db, run_id) .await.unwrap ( ) [ 0 ] .id
3045
}
36e5a9c1 Andy McCall 1 month ago ↶
3046
3047
3048
3049
# [ tokio ::test ( flavor = " multi_thread" ) ]
3050
async fn junit_reports_make_a_tests_tab ( ) {
3051
let h = hub ( ) .await;
3052
create ( & h.app, " andy" , " t" ) .await;
3053
let url = format! ( " {} /andy/t" , h.base) ;
3054
let work = tempfile:: tempdir( ) .unwrap ( ) ;
3055
let c = client ( & url, work.path ( ) ) ;
3056
write (
3057
work.path ( ) ,
3058
" .flow/pipeline.kdl" ,
3059
& format! (
3060
" schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" test\" {{ \n sandbox \" alpine\" \n exec \" make\" \" test\" \n output {{ \n artifact \" junit\" {{ \n path \" /src/junit.xml\" \n }} \n }} \n }} \n "
3061
) ,
3062
) ;
c8df33a6 Andy McCall 25 days ago ↶
3063
assert! ( ! call ( & h.app, " GET" , " /andy/t" , None , None ) .await.1. contains ( " >Reports</a>" ) ) ;
36e5a9c1 Andy McCall 1 month ago ↶
3064
let case = | name : & str , ok : bool , secs : & str | {
3065
if ok {
3066
format! ( r # "<testcase classname="t.suite" name="{name} " time="{secs} "/>"# )
3067
} else {
3068
format! (
3069
r # "<testcase classname="t.suite" name="{name} " time="{secs} "><failure message="{name} broke">AssertionError</failure></testcase>"#
3070
)
3071
}
3072
} ;
3073
let runs = [
3074
vec! [ case ( " a" , true , " 0.1" ) , case ( " b" , true , " 2.5" ) , case ( " c" , true , " 0.2" ) ] ,
3075
vec! [ case ( " a" , true , " 0.1" ) , case ( " b" , false , " 2.6" ) , case ( " c" , true , " 0.2" ) , case ( " d" , true , " 0.3" ) ] ,
3076
vec! [ case ( " a" , true , " 0.1" ) , case ( " b" , true , " 2.4" ) , case ( " c" , true , " 0.2" ) , case ( " d" , true , " 0.3" ) ] ,
3077
] ;
3078
for ( i, cases) in runs.iter ( ) .enumerate ( ) {
3079
write ( work.path ( ) , " n.txt" , & i.to_string ( ) ) ;
3080
c.snapshot ( " test" ) .unwrap ( ) ;
3081
c.commit ( & format! ( " Run {i} " ) , & [ ] ) .unwrap ( ) ;
3082
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
3083
3084
let mut a = None ;
3085
for _ in 0 .. 100 {
3086
a = claim ( & h.app) .await;
3087
if a.is_some ( ) {
3088
break ;
3089
}
3090
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
3091
}
3092
let a = a.expect ( " a job to claim" ) ;
3093
let id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
3094
let xml = format! ( " <testsuites><testsuite name=\" t\" >{} </testsuite></testsuites>" , cases.concat ( ) ) ;
3095
assert_eq! ( upload ( & h.app, id, " junit" , xml.as_bytes ( ) ) .await.0 , StatusCode:: OK ) ;
3096
let failed = cases.iter ( ) .any ( | c | c.contains ( " <failure" ) ) ;
3097
let mut r = receipt ( " test" , a[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) , failed as u32 ) ;
3098
if ! failed {
3099
r[ " outputs" ] = serde_json:: json! ( [ oxy_core:: repo:: Repo:: blob_ref( xml.as_bytes ( ) ) .to_hex ( ) ] ) ;
3100
}
3101
let code = failed as i32 ;
3102
assert_eq! (
3103
finish ( & h.app, id, serde_json:: json! ( { " exit_code" : code, " duration_ms" : 700 , " receipt" : r } ) ) .await,
3104
StatusCode:: NO_CONTENT
3105
) ;
3106
}
3107
let ( st, page) = call ( & h.app, " GET" , " /andy/t/tests" , None , None ) .await;
3108
assert_eq! ( st, StatusCode:: OK , " {page}" ) ;
3109
if let Some ( out) = std:: env:: var_os( " OXYHUB_DUMP_TESTS_PAGE" ) {
3110
std:: fs:: write( out, & page) .unwrap ( ) ;
3111
}
c8df33a6 Andy McCall 25 days ago ↶
3112
assert! ( page.contains ( " >Reports</a>" ) , " the tab" ) ;
36e5a9c1 Andy McCall 1 month ago ↶
3113
assert! ( page.contains ( " 100%" ) & & page.contains ( " of 4 tests passed" ) , " the latest run passes" ) ;
3114
assert! ( page.contains ( " 1 fixed" ) & & page.contains ( " t.suite::b" ) , " b fixed against the run before" ) ;
3115
assert! ( page.contains ( " 1 ✕" ) & & page.contains ( " 2 ✓" ) , " b is flaky: {page}" ) ;
3116
assert! ( page.contains ( " <svg class=\" cov-chart\" " ) , " charts" ) ;
3117
3118
let slow = page.find ( " Slowest tests" ) .unwrap ( ) ;
3119
assert! ( page[ slow.. ] .find ( " t.suite::b" ) .unwrap ( ) < page[ slow.. ] .find ( " t.suite::d" ) .unwrap ( ) ) ;
3120
3121
let row = db:: repos:: get( & h.app.db, " andy" , " t" ) .await.unwrap ( ) .unwrap ( ) ;
3122
let second = oxygenhub:: junit:: on_bookmark( & h.app.db, row.id, " main" , 10 ) .await.unwrap ( ) [ 1 ] .id;
3123
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/t/tests?report={second} " ) , None , None ) .await;
3124
assert! ( page.contains ( " 1 failing" ) & & page.contains ( " Newly failing" ) & & page.contains ( " b broke" ) , " {page}" ) ;
3125
assert! ( page.contains ( " 1 new tests" ) , " d is new" ) ;
617c2840 Andy McCall 1 month ago ↶
3126
3127
3128
3129
let ( st, list) = call ( & h.app, " GET" , " /api/v1/repos/andy/t/tests" , None , None ) .await;
3130
assert_eq! ( st, StatusCode:: OK , " {list}" ) ;
3131
let list: serde_json:: Value = serde_json:: from_str( & list) .unwrap ( ) ;
3132
assert_eq! ( list.as_array ( ) .unwrap ( ) .len ( ) , 3 ) ;
3133
assert_eq! ( ( list[ 0 ] [ " tests" ] .as_i64 ( ) , list[ 1 ] [ " failed" ] .as_i64 ( ) ) , ( Some ( 4 ) , Some ( 1 ) ) ) ;
3134
let ( _ , one) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/t/tests/{second} ?status=failed" ) , None , None ) .await;
3135
let one: serde_json:: Value = serde_json:: from_str( & one) .unwrap ( ) ;
3136
assert_eq! ( one[ " cases" ] .as_array ( ) .unwrap ( ) .len ( ) , 1 ) ;
3137
assert_eq! ( ( one[ " cases" ] [ 0 ] [ " name" ] .as_str ( ) , one[ " cases" ] [ 0 ] [ " message" ] .as_str ( ) ) , ( Some ( " b" ) , Some ( " b broke" ) ) ) ;
3138
let ( _ , all) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/t/tests/{second} " ) , None , None ) .await;
3139
assert_eq! ( serde_json:: from_str:: < serde_json:: Value> ( & all) .unwrap ( ) [ " cases" ] .as_array ( ) .unwrap ( ) .len ( ) , 4 ) ;
3140
assert_eq! ( call ( & h.app, " GET" , " /api/v1/repos/andy/t/tests/99999" , None , None ) .await.0 , StatusCode:: NOT_FOUND ) ;
3d5c825a Andy McCall 1 month ago ↶
3141
3142
3143
let reports = oxygenhub:: junit:: on_bookmark( & h.app.db, row.id, " main" , 10 ) .await.unwrap ( ) ;
3144
let check = | state : String| {
3145
let app = h.app.clone ( ) ;
3146
async move { db:: hooks:: checks_for_state( & app.db, row.id, & state) .await.unwrap ( ) .into_iter ( ) .find ( | c | c.name = = " tests" ) .unwrap ( ) }
3147
} ;
3148
let c = check ( reports[ 1 ] .state.clone ( ) ) .await;
3149
assert_eq! ( ( c.status.as_str ( ) , c.summary.as_str ( ) ) , ( " failed" , " 1 failing; 1 new against main: t.suite::b" ) ) ;
3150
let c = check ( reports[ 0 ] .state.clone ( ) ) .await;
3151
assert_eq! ( ( c.status.as_str ( ) , c.summary.as_str ( ) ) , ( " passed" , " 4 passed" ) ) ;
3152
3153
3154
let ( _ , tab) = call ( & h.app, " GET" , " /andy/t/tests" , None , None ) .await;
3155
assert! (
3156
tab.contains ( " /andy/t/tests/case?classname=t.suite&name=b" ) | | tab.contains ( " /andy/t/tests/case?classname=t.suite&name=b" ) ,
3157
" linked"
3158
) ;
3159
let ( st, hist) = call ( & h.app, " GET" , " /andy/t/tests/case?classname=t.suite&name=b" , None , None ) .await;
3160
assert_eq! ( st, StatusCode:: OK , " {hist}" ) ;
3161
assert! ( hist.contains ( " 2 passed" ) & & hist.contains ( " 1 failed" ) & & hist.contains ( " b broke" ) , " {hist}" ) ;
3162
assert_eq! ( call ( & h.app, " GET" , " /andy/t/tests/case?classname=t.suite&name=nope" , None , None ) .await.0 , StatusCode:: NOT_FOUND ) ;
36e5a9c1 Andy McCall 1 month ago ↶
3163
}
67040b15 Andy McCall 25 days ago ↶
3164
3165
3166
3167
# [ tokio ::test ( flavor = " multi_thread" ) ]
3168
async fn a_job_no_runner_can_take_says_why ( ) {
3169
let h = hub ( ) .await;
3170
create ( & h.app, " andy" , " demo" ) .await;
3171
let url = format! ( " {} /andy/demo" , h.base) ;
3172
let work = tempfile:: tempdir( ) .unwrap ( ) ;
3173
let c = client ( & url, work.path ( ) ) ;
3174
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
3175
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
3176
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
3177
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
3178
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
3179
3180
3181
let r = db:: runners:: configured( & h.app.db, & blake3:: hash( b " elsewhere" ) .to_hex ( ) , " earth-9" ) .await.unwrap ( ) ;
3182
db:: runners:: seen( & h.app.db, r.id, None , None ) .await.unwrap ( ) ;
3183
let other = " ab" .repeat ( 32 ) ;
92a797b2 Andy McCall 25 days ago ↶
3184
db:: runners:: set_offer( & h.app.db, r.id, & [ " microvm" .to_string ( ) ] , Some ( & [ other] ) , false , " p" ) .await.unwrap ( ) ;
67040b15 Andy McCall 25 days ago ↶
3185
oxygenhub:: flow:: explain_waits( & h.app) .await.unwrap ( ) ;
3186
3187
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/flow/runs/1" , Some ( " site" ) , None ) .await;
3188
let v: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
3189
let build = v[ " jobs" ] .as_array ( ) .unwrap ( ) .iter ( ) .find ( | j | j[ " name" ] = = " build" ) .unwrap ( ) .clone ( ) ;
3190
let why = build[ " waiting" ] .as_str ( ) .unwrap_or_default ( ) .to_string ( ) ;
3191
assert! ( why.contains ( " earth-9 does not hold image" ) & & why.contains ( & ROOTFS [ .. 12 ] ) , " {body}" ) ;
3192
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/flow/1" , None , None ) .await;
3193
assert! ( page.contains ( " data-waiting" ) & & page.contains ( " does not hold image" ) , " the run page says it" ) ;
13b27cb2 Andy McCall 18 days ago ↶
3194
3195
3196
3197
let mut summary = String :: new( ) ;
3198
for _ in 0 .. 40 {
3199
let checks = db:: hooks:: checks_for_state( & h.app.db, run.repo_id, & run.state) .await.unwrap ( ) ;
3200
summary = checks.iter ( ) .find ( | c | c.name = = " flow/build" ) .unwrap ( ) .summary.clone ( ) ;
3201
if summary.contains ( " does not hold image" ) {
3202
break ;
3203
}
3204
let build = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) .into_iter ( ) .find ( | j | j.name = = " build" ) .unwrap ( ) ;
3205
db:: flow:: set_waiting( & h.app.db, build.id, " " ) .await.unwrap ( ) ;
3206
oxygenhub:: flow:: explain_waits( & h.app) .await.unwrap ( ) ;
3207
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
3208
}
3209
assert! ( summary.contains ( " does not hold image" ) , " {summary}" ) ;
67040b15 Andy McCall 25 days ago ↶
3210
}
92a797b2 Andy McCall 25 days ago ↶
3211
58cbf424 Andy McCall 18 days ago ↶
3212
3213
3214
3215
# [ tokio ::test ( flavor = " multi_thread" ) ]
3216
async fn a_job_waiting_for_a_busy_runner_says_so ( ) {
3217
let h = hub ( ) .await;
3218
create ( & h.app, " andy" , " demo" ) .await;
3219
let url = format! ( " {} /andy/demo" , h.base) ;
3220
let work = tempfile:: tempdir( ) .unwrap ( ) ;
3221
let c = client ( & url, work.path ( ) ) ;
3222
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
3223
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
3224
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
3225
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
3226
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
3227
3228
let r = db:: runners:: configured( & h.app.db, & blake3:: hash( b " busy" ) .to_hex ( ) , " earth-9" ) .await.unwrap ( ) ;
3229
db:: runners:: seen( & h.app.db, r.id, None , None ) .await.unwrap ( ) ;
3230
db:: runners:: set_offer( & h.app.db, r.id, & [ " microvm" .to_string ( ) ] , Some ( & [ ROOTFS .to_string ( ) ] ) , false , " p" ) .await.unwrap ( ) ;
3231
sqlx:: query(
3232
" UPDATE flow_jobs SET status = 'running', runner_id = $1, runner = 'earth-9', started_ms = $2 WHERE run_id = $3 AND name = 'lint'" ,
3233
)
3234
.bind ( r.id)
3235
.bind ( db:: now_ms( ) )
3236
.bind ( run.id)
3237
.execute ( & h.app.db)
3238
.await
3239
.unwrap ( ) ;
3240
oxygenhub:: flow:: explain_waits( & h.app) .await.unwrap ( ) ;
3241
let jobs = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) ;
3242
let build = jobs.iter ( ) .find ( | j | j.name = = " build" ) .unwrap ( ) ;
3243
assert! ( build.waiting.contains ( " the runners that can take it are busy: earth-9 busy (1 of 1)" ) , " {}" , build.waiting) ;
3244
3245
sqlx:: query( " UPDATE flow_runners SET max_jobs = 2 WHERE id = $1" ) .bind ( r.id) .execute ( & h.app.db) .await.unwrap ( ) ;
3246
oxygenhub:: flow:: explain_waits( & h.app) .await.unwrap ( ) ;
3247
let build = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) .into_iter ( ) .find ( | j | j.name = = " build" ) .unwrap ( ) ;
5d66f27c Andy McCall 9 days ago ↶
3248
assert! ( build.waiting.contains ( " earth-9 has a free slot (1 of 2) but not the memory, CPUs or disk for it now" ) , " {}" , build.waiting) ;
58cbf424 Andy McCall 18 days ago ↶
3249
}
3250
794ff94f Andy McCall 11 days ago ↶
3251
3252
3253
# [ tokio ::test ( flavor = " multi_thread" ) ]
3254
async fn an_expedited_run_moves_the_queue_at_once ( ) {
3255
let h = hub ( ) .await;
3256
for name in [ " first" , " urgent" ] {
3257
create ( & h.app, " andy" , name) .await;
3258
let work = tempfile:: tempdir( ) .unwrap ( ) ;
3259
let c = client ( & format! ( " {} /andy/{name} " , h.base) , work.path ( ) ) ;
3260
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
3261
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
3262
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
3263
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
3264
3265
assert_eq! ( runs_of ( & h.app, " andy" , name) .await.len ( ) , 1 ) ;
3266
}
3267
3268
let r = db:: runners:: configured( & h.app.db, & blake3:: hash( b " busy" ) .to_hex ( ) , " earth-9" ) .await.unwrap ( ) ;
3269
db:: runners:: seen( & h.app.db, r.id, None , None ) .await.unwrap ( ) ;
3270
db:: runners:: set_offer( & h.app.db, r.id, & [ " microvm" .to_string ( ) ] , Some ( & [ ROOTFS .to_string ( ) ] ) , false , " p" ) .await.unwrap ( ) ;
3271
let first = runs_of ( & h.app, " andy" , " first" ) .await.remove ( 0 ) ;
3272
sqlx:: query(
3273
" UPDATE flow_jobs SET status = 'running', runner_id = $1, runner = 'earth-9', started_ms = $2 WHERE run_id = $3 AND name = 'lint'" ,
3274
)
3275
.bind ( r.id)
3276
.bind ( db:: now_ms( ) )
3277
.bind ( first.id)
3278
.execute ( & h.app.db)
3279
.await
3280
.unwrap ( ) ;
3281
oxygenhub:: flow:: explain_waits( & h.app) .await.unwrap ( ) ;
3282
let db = h.app.db.clone ( ) ;
3283
let build = | run : i64 | {
3284
let db = db.clone ( ) ;
3285
async move { db:: flow:: jobs( & db, run) .await.unwrap ( ) .into_iter ( ) .find ( | j | j.name = = " build" ) .unwrap ( ) }
3286
} ;
3287
let urgent = runs_of ( & h.app, " andy" , " urgent" ) .await.remove ( 0 ) ;
3288
assert! ( build ( urgent.id) .await.waiting.contains ( " 1 ready job comes first" ) , " {}" , build ( urgent.id) .await.waiting) ;
3289
3290
3291
let res = oxygenhub:: router( h.app.clone ( ) )
3292
.oneshot (
3293
Request:: post( " /register" )
3294
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
3295
.body ( Body:: from( " username=andy&email=a%40example.com&password=correct-horse-battery&setup=test-setup" ) )
3296
.unwrap ( ) ,
3297
)
3298
.await
3299
.unwrap ( ) ;
3300
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
3301
let page = oxygenhub:: router( h.app.clone ( ) )
3302
.oneshot ( Request:: get( " /andy/urgent/flow/1" ) .header ( header:: COOKIE , cookie.clone ( ) ) .body ( Body:: empty( ) ) .unwrap ( ) )
3303
.await
3304
.unwrap ( ) ;
3305
let page = String :: from_utf8_lossy( & page.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) ;
3306
let i = page.find ( " name=\" _csrf\" value=\" " ) .unwrap ( ) + " name=\" _csrf\" value=\" " .len ( ) ;
3307
let csrf = page[ i.. ] .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( ) ;
3308
let res = oxygenhub:: router( h.app.clone ( ) )
3309
.oneshot (
3310
Request:: post( " /andy/urgent/flow/1/expedite" )
3311
.header ( header:: COOKIE , cookie.clone ( ) )
3312
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
3313
.body ( Body:: from( format! ( " _csrf={csrf} &reason=incident" ) ) )
3314
.unwrap ( ) ,
3315
)
3316
.await
3317
.unwrap ( ) ;
3318
assert! ( res.status ( ) .is_redirection ( ) , " {}" , res.status ( ) ) ;
3319
assert! ( ! build ( urgent.id) .await.waiting.contains ( " come" ) , " first now: {}" , build ( urgent.id) .await.waiting) ;
3320
let w = build ( first.id) .await.waiting;
3321
assert! ( w.contains ( " 2 ready jobs come first (after expedited run andy/urgent #1)" ) , " {w}" ) ;
b1860d60 Andy McCall 8 days ago ↶
3322
3323
3324
3325
let andy = db:: accounts:: account_by_name( & h.app.db, " andy" ) .await.unwrap ( ) .unwrap ( ) ;
3326
let ( token, hash) = oxygenhub:: auth:: new_token( ) ;
3327
db:: accounts:: create_token( & h.app.db, andy.id, " phone" , & hash, " repo:read,repo:write" , " " , None ) .await.unwrap ( ) ;
3328
let url = format! ( " /api/v1/repos/andy/first/flow/runs/{} /expedite" , first.number) ;
3329
let ( st, body) = call ( & h.app, " POST" , & url, Some ( & token) , Some ( r # "{"reason":" "}"# .into ( ) ) ) .await;
3330
assert_eq! ( st, StatusCode:: CONFLICT , " {body}" ) ;
3331
assert! ( body.contains ( " Say why the run is urgent." ) , " {body}" ) ;
3332
let ( st, body) = call ( & h.app, " POST" , & url, Some ( & token) , Some ( r # "{"reason":"the fix for the outage"}"# .into ( ) ) ) .await;
3333
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
3334
let run = db:: flow:: run( & h.app.db, first.repo_id, first.number) .await.unwrap ( ) .unwrap ( ) ;
3335
assert! (
3336
run.priority = = " urgent" & & run.priority_note.contains ( " expedited by andy: the fix for the outage" ) ,
3337
" {} {}" ,
3338
run.priority,
3339
run.priority_note
3340
) ;
794ff94f Andy McCall 11 days ago ↶
3341
}
3342
92a797b2 Andy McCall 25 days ago ↶
3343
async fn put_bytes ( app : & Shared, path : & str , bearer : & str , bytes : Vec< u8 > ) -> ( StatusCode, String) {
3344
let req = Request:: put( path)
3345
.header ( header:: AUTHORIZATION , format! ( " Bearer {bearer} " ) )
3346
.header ( header:: CONTENT_TYPE , " application/octet-stream" )
3347
.body ( Body:: from( bytes) )
3348
.unwrap ( ) ;
3349
let res = oxygenhub:: router( app.clone ( ) ) .oneshot ( req) .await.unwrap ( ) ;
3350
let st = res.status ( ) ;
3351
( st, String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) )
3352
}
3353
3354
3355
3356
3357
951aed50 Andy McCall 9 days ago ↶
3358
3359
3360
92a797b2 Andy McCall 25 days ago ↶
3361
# [ tokio ::test ( flavor = " multi_thread" ) ]
951aed50 Andy McCall 9 days ago ↶
3362
async fn a_runner_offers_the_manifest_of_an_image_the_hub_lacks ( ) {
3363
let h = hub ( ) .await;
3364
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
3365
let image: Vec< u8 > = ( 0 .. 300_000 u32 ) .map ( | i | ( i % 241 ) as u8 ) .collect ( ) ;
3366
let mut chunks = std:: collections:: HashMap:: new( ) ;
3367
let m = jet:: images:: scan( & image[ .. ] , Some ( " x86_64" .into ( ) ) , | c , b | {
3368
chunks.insert ( c.hash ( ) .to_string ( ) , b.to_vec ( ) ) ;
3369
Ok ( ( ) )
3370
} )
3371
.unwrap ( ) ;
3372
let hash = m.hash.clone ( ) ;
3373
3374
let held = serde_json:: json! ( { " images" : [ hash.clone ( ) , " not-a-hash" ] } ) .to_string ( ) ;
3375
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/images/held" , Some ( " runner-secret" ) , Some ( held.clone ( ) ) ) .await;
3376
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
3377
assert_eq! ( serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " wanted" ] , serde_json:: json! ( [ hash.clone ( ) ] ) ) ;
3378
3379
assert_ne! ( call ( & h.app, " POST" , " /api/v1/flow/images/held" , Some ( " site" ) , Some ( held.clone ( ) ) ) .await.0 , StatusCode:: OK ) ;
3380
3381
let manifest = serde_json:: to_string( & m) .unwrap ( ) ;
3382
let wrong = format! ( " /api/v1/flow/images/{} /held" , " ab" .repeat ( 32 ) ) ;
3383
assert_eq! ( call ( & h.app, " PUT" , & wrong, Some ( " runner-secret" ) , Some ( manifest.clone ( ) ) ) .await.0 , StatusCode:: BAD_REQUEST ) ;
3384
let ( st, body) = call ( & h.app, " PUT" , & format! ( " /api/v1/flow/images/{hash} /held" ) , Some ( " runner-secret" ) , Some ( manifest.clone ( ) ) ) .await;
3385
assert_eq! ( st, StatusCode:: CREATED , " {body}" ) ;
3386
let kept = oxygenhub:: images:: kept( & h.app.db) .await.unwrap ( ) ;
3387
let row = kept.iter ( ) .find ( | k | k.hash = = hash ) .expect ( " recorded" ) ;
3388
assert_eq! ( row.source, " peer" ) ;
3389
let ( _ , body) = call ( & h.app, " POST" , " /api/v1/flow/images/held" , Some ( " runner-secret" ) , Some ( held) ) .await;
3390
assert! ( body.contains ( " \" wanted\" :[]" ) , " {body}" ) ;
3391
3392
let ( st, _ ) = call ( & h.app, " GET" , & format! ( " /api/v1/flow/images/{hash} " ) , Some ( " runner-secret" ) , None ) .await;
3393
assert_eq! ( st, StatusCode:: OK ) ;
3394
let first = m.needed ( ) [ 0 ] .to_string ( ) ;
3395
assert_ne! ( call ( & h.app, " GET" , & format! ( " /api/v1/flow/images/chunks/{first} " ) , Some ( " runner-secret" ) , None ) .await.0 , StatusCode:: OK ) ;
3396
let reach = oxygenhub:: images:: reach( & h.app) .await.unwrap ( ) ;
3397
assert! ( reach.peer_only ( & hash) ) ;
3398
3399
for c in m.needed ( ) {
3400
let ( st, body) = put_bytes ( & h.app, & format! ( " /api/v1/flow/images/chunks/{c} " ) , " site" , chunks[ c] .clone ( ) ) .await;
3401
assert_eq! ( st, StatusCode:: NO_CONTENT , " {body}" ) ;
3402
}
3403
let ( st, body) = call ( & h.app, " PUT" , & format! ( " /api/v1/flow/images/{hash} " ) , Some ( " site" ) , Some ( manifest) ) .await;
3404
assert_eq! ( st, StatusCode:: CREATED , " {body}" ) ;
3405
let kept = oxygenhub:: images:: kept( & h.app.db) .await.unwrap ( ) ;
3406
assert_eq! ( kept.iter ( ) .find ( | k | k.hash = = hash ) .unwrap ( ) .source, " hub" ) ;
3407
assert! ( ! oxygenhub:: images:: reach( & h.app) .await.unwrap ( ) .peer_only ( & hash) ) ;
3408
}
3409
3410
# [ tokio ::test ( flavor = " multi_thread" ) ]
92a797b2 Andy McCall 25 days ago ↶
3411
async fn images_are_sent_by_chunk_and_offered_to_runners_that_fetch ( ) {
3412
let h = hub ( ) .await;
3413
3414
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
3415
3416
let mut image: Vec< u8 > = ( 0 .. 400_000 u32 ) .map ( | i | ( i % 251 ) as u8 ) .collect ( ) ;
3417
image.extend ( std:: iter:: repeat_n( 0 u8 , 2 < < 20 ) ) ;
3418
image.extend ( ( 0 .. 100_000 u32 ) .map ( | i | ( i % 239 ) as u8 ) ) ;
3419
let mut chunks = std:: collections:: HashMap:: new( ) ;
3420
let m = jet:: images:: scan( & image[ .. ] , Some ( " x86_64" .into ( ) ) , | c , b | {
3421
chunks.insert ( c.hash ( ) .to_string ( ) , b.to_vec ( ) ) ;
3422
Ok ( ( ) )
3423
} )
3424
.unwrap ( ) ;
3425
let hash = m.hash.clone ( ) ;
3426
3427
3428
let ask = serde_json:: json! ( { " chunks" : m.needed ( ) } ) .to_string ( ) ;
3429
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/images/missing" , Some ( " site" ) , Some ( ask.clone ( ) ) ) .await;
3430
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
3431
let missing: Vec< String> = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " missing" ]
3432
.as_array ( )
3433
.unwrap ( )
3434
.iter ( )
3435
.map ( | v | v.as_str ( ) .unwrap ( ) .to_string ( ) )
3436
.collect ( ) ;
3437
assert_eq! ( missing.len ( ) , m.needed ( ) .len ( ) ) ;
3438
let manifest = serde_json:: to_string( & m) .unwrap ( ) ;
3439
let ( st, body) = call ( & h.app, " PUT" , & format! ( " /api/v1/flow/images/{hash} " ) , Some ( " site" ) , Some ( manifest.clone ( ) ) ) .await;
3440
assert! ( st = = StatusCode:: CONFLICT & & body.contains ( " was not sent" ) , " {st} {body}" ) ;
3441
3442
let first = & missing[ 0 ] ;
3443
assert_eq! ( put_bytes ( & h.app, & format! ( " /api/v1/flow/images/chunks/{first} " ) , " site" , vec! [ 1 , 2 , 3 ] ) .await.0 , StatusCode:: BAD_REQUEST ) ;
3444
assert_ne! (
3445
put_bytes ( & h.app, & format! ( " /api/v1/flow/images/chunks/{first} " ) , " runner-secret" , chunks[ first] .clone ( ) ) .await.0 ,
3446
StatusCode:: NO_CONTENT
3447
) ;
3448
for c in & missing {
3449
let ( st, body) = put_bytes ( & h.app, & format! ( " /api/v1/flow/images/chunks/{c} " ) , " site" , chunks[ c] .clone ( ) ) .await;
3450
assert_eq! ( st, StatusCode:: NO_CONTENT , " {body}" ) ;
3451
}
3452
let ( _ , body) = call ( & h.app, " POST" , " /api/v1/flow/images/missing" , Some ( " site" ) , Some ( ask) ) .await;
3453
assert! ( body.contains ( " \" missing\" :[]" ) , " {body}" ) ;
750579ec Andy McCall 24 days ago ↶
3454
3455
let many: Vec< String> = ( 0 .. 60_000 u32 ) .map ( | i | blake3:: hash( & i.to_le_bytes ( ) ) .to_hex ( ) .to_string ( ) ) .collect ( ) ;
3456
let ( st, body) =
3457
call ( & h.app, " POST" , " /api/v1/flow/images/missing" , Some ( " site" ) , Some ( serde_json:: json! ( { " chunks" : many } ) .to_string ( ) ) ) .await;
3458
assert_eq! ( st, StatusCode:: OK , " {}" , & body[ .. body.len ( ) .min ( 200 ) ] ) ;
3459
assert_eq! ( serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " missing" ] .as_array ( ) .unwrap ( ) .len ( ) , 60_000 ) ;
92a797b2 Andy McCall 25 days ago ↶
3460
3461
let mut wrong = m.clone ( ) ;
3462
wrong.hash = " ab" .repeat ( 32 ) ;
3463
let ( st, _ ) =
3464
call ( & h.app, " PUT" , & format! ( " /api/v1/flow/images/{} " , wrong.hash) , Some ( " site" ) , Some ( serde_json:: to_string( & wrong) .unwrap ( ) ) )
3465
.await;
3466
assert_eq! ( st, StatusCode:: CONFLICT , " chunks that make another image" ) ;
d9d0d314 Andy McCall 10 days ago ↶
3467
let ( st, body) = call ( & h.app, " PUT" , & format! ( " /api/v1/flow/images/{hash} " ) , Some ( " site" ) , Some ( manifest.clone ( ) ) ) .await;
3468
assert_eq! ( st, StatusCode:: CREATED , " {body}" ) ;
3469
3470
3471
3472
sqlx:: query( " DELETE FROM jet_images" ) .execute ( & h.app.db) .await.unwrap ( ) ;
3473
let gone = h.app.config.data.join ( " images/chunks" ) .join ( & missing[ 0 ] [ .. 2 ] ) .join ( & missing[ 0 ] ) ;
3474
let kept = std:: fs:: read( & gone) .unwrap ( ) ;
3475
std:: fs:: remove_file( & gone) .unwrap ( ) ;
92a797b2 Andy McCall 25 days ago ↶
3476
let ( st, body) = call ( & h.app, " PUT" , & format! ( " /api/v1/flow/images/{hash} " ) , Some ( " site" ) , Some ( manifest) ) .await;
3477
assert_eq! ( st, StatusCode:: CREATED , " {body}" ) ;
d9d0d314 Andy McCall 10 days ago ↶
3478
assert! ( oxygenhub:: images:: held( & h.app.db) .await.unwrap ( ) .contains ( & hash) ) ;
3479
std:: fs:: write( & gone, kept) .unwrap ( ) ;
92a797b2 Andy McCall 25 days ago ↶
3480
3481
3482
let ( st, body) = call ( & h.app, " GET" , & format! ( " /api/v1/flow/images/{hash} " ) , Some ( " runner-secret" ) , None ) .await;
3483
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
3484
let got: jet:: images:: Manifest = serde_json:: from_str( & body) .unwrap ( ) ;
3485
assert_eq! ( got, m) ;
3486
let dir = tempfile:: tempdir( ) .unwrap ( ) ;
3487
let out = dir.path ( ) .join ( " img.ext4" ) ;
3488
let mut fetched = Vec :: new( ) ;
3489
for c in got.needed ( ) {
3490
let req = Request:: get( format! ( " /api/v1/flow/images/chunks/{c} " ) )
3491
.header ( header:: AUTHORIZATION , " Bearer runner-secret" )
3492
.body ( Body:: empty( ) )
3493
.unwrap ( ) ;
3494
let res = oxygenhub:: router( h.app.clone ( ) ) .oneshot ( req) .await.unwrap ( ) ;
3495
assert_eq! ( res.status ( ) , StatusCode:: OK ) ;
3496
let bytes = res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) .to_vec ( ) ;
3497
fetched.push ( ( c.to_string ( ) , bytes) ) ;
3498
}
3499
jet:: images:: assemble( & got, & out, | c | Ok ( fetched.iter ( ) .find ( | ( h , _) | h = = c.hash ( ) ) .unwrap ( ) .1. clone ( ) ) ) .unwrap ( ) ;
3500
assert_eq! ( std:: fs:: read( & out) .unwrap ( ) , image) ;
3501
assert_eq! (
3502
call ( & h.app, " GET" , & format! ( " /api/v1/flow/images/{} " , " cd" .repeat ( 32 ) ) , Some ( " runner-secret" ) , None ) .await.0 ,
3503
StatusCode:: NOT_FOUND
3504
) ;
3505
assert_ne! ( call ( & h.app, " GET" , & format! ( " /api/v1/flow/images/{hash} " ) , None , None ) .await.0 , StatusCode:: OK , " not anonymous" ) ;
3506
3507
3508
3509
create ( & h.app, " andy" , " demo" ) .await;
3510
let url = format! ( " {} /andy/demo" , h.base) ;
3511
let work = tempfile:: tempdir( ) .unwrap ( ) ;
3512
let c = client ( & url, work.path ( ) ) ;
3513
write (
3514
work.path ( ) ,
3515
" .flow/pipeline.kdl" ,
3516
& format! ( " schema-version 1\n sandbox \" img\" {{ \n rootfs \" cas://blob/{hash} \" \n }} \n job \" build\" {{ \n sandbox \" img\" \n exec \" make\" \n }} \n " ) ,
3517
) ;
3518
c.commit ( " Build on the image" , & [ ] ) .unwrap ( ) ;
3519
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
3520
runs_of ( & h.app, " andy" , " demo" ) .await;
1c8276b9 Andy McCall 24 days ago ↶
3521
3522
let row = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
3523
let run1 = db:: flow:: run( & h.app.db, row.id, 1 ) .await.unwrap ( ) .unwrap ( ) ;
3524
let checks = checks_when ( & h.app, row.id, & run1.state, 1 ) .await;
3525
assert! ( checks.iter ( ) .all ( | c | c.name ! = " pipeline/images" ) , " {checks:?}" ) ;
92a797b2 Andy McCall 25 days ago ↶
3526
let ask = | fetches : bool | format! ( r # "{{ "runner":"test-1","runtimes":["microvm"],"images":[],"fetches_images":{fetches} }} "# ) ;
3527
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( ask ( false ) ) ) .await;
3528
assert_eq! ( st, StatusCode:: NO_CONTENT , " it lacks the image and does not fetch" ) ;
3529
oxygenhub:: flow:: explain_waits( & h.app) .await.unwrap ( ) ;
3530
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/flow/runs/1" , Some ( " site" ) , None ) .await;
3531
assert! ( body.contains ( " does not hold image" ) , " {body}" ) ;
3532
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( ask ( true ) ) ) .await;
3533
assert_eq! ( st, StatusCode:: OK , " it fetches, and the hub holds the image: {body}" ) ;
7829b7c7 Andy McCall 24 days ago ↶
3534
3535
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/flow/runs/1" , Some ( " site" ) , None ) .await;
3536
assert! ( ! body.contains ( " waiting" ) , " a running job keeps no waiting reason: {body}" ) ;
92a797b2 Andy McCall 25 days ago ↶
3537
}
40dd3f86 Andy McCall 25 days ago ↶
3538
3539
3540
3541
3542
# [ tokio ::test ( flavor = " multi_thread" ) ]
3543
async fn coverage_uploaded_from_a_workstation_says_so ( ) {
3544
let h = hub ( ) .await;
3545
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
3546
create ( & h.app, " andy" , " vk" ) .await;
3547
let url = format! ( " {} /andy/vk" , h.base) ;
3548
let work = tempfile:: tempdir( ) .unwrap ( ) ;
3549
let c = client ( & url, work.path ( ) ) ;
3550
write ( work.path ( ) , " src/game.c" , " int a;\n int b;\n int c;\n int d;\n " ) ;
3551
c.commit ( " Game" , & [ ] ) .unwrap ( ) ;
3552
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
3553
3554
let lcov = " SF:src/game.c\n DA:1,3\n DA:2,1\n DA:3,0\n DA:4,5\n end_of_record\n SF:reference/rom.asm\n DA:1,1\n end_of_record\n " ;
3555
let body = serde_json:: json! ( { " bookmark" : " main" , " content" : lcov } ) .to_string ( ) ;
3556
let ( st, out) = call ( & h.app, " POST" , " /api/v1/repos/andy/vk/coverage" , Some ( " site" ) , Some ( body) ) .await;
3557
assert_eq! ( st, StatusCode:: CREATED , " {out}" ) ;
3558
let v: serde_json:: Value = serde_json:: from_str( & out) .unwrap ( ) ;
3559
assert_eq! ( ( v[ " uploaded_by" ] .as_str ( ) , v[ " lines_hit" ] .as_i64 ( ) , v[ " lines_found" ] .as_i64 ( ) ) , ( Some ( " andy" ) , Some ( 3 ) , Some ( 4 ) ) ) ;
3560
assert_eq! ( v[ " dropped" ] , 1 , " a file outside the state is left out" ) ;
3561
assert_eq! ( v[ " bookmark" ] , " main" ) ;
3562
3563
let ( _ , page) = call ( & h.app, " GET" , " /andy/vk/coverage" , None , None ) .await;
3564
assert! ( page.contains ( " uploaded by andy" ) & & page.contains ( " 75.0%" ) , " the Coverage page says where it came from" ) ;
3565
let ( _ , page) = call ( & h.app, " GET" , " /andy/vk/reports" , None , None ) .await;
3566
assert! ( page.contains ( " uploaded by andy" ) , " and the Reports card" ) ;
3567
let row = db:: repos:: get( & h.app.db, " andy" , " vk" ) .await.unwrap ( ) .unwrap ( ) ;
3568
let state = v[ " state" ] .as_str ( ) .unwrap ( ) ;
3569
let checks = db:: hooks:: checks_for_state( & h.app.db, row.id, state) .await.unwrap ( ) ;
3570
assert! ( checks.iter ( ) .all ( | c | c.name ! = " coverage" ) , " an upload sets no check: {checks:?}" ) ;
3571
3572
3573
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/repos/andy/vk/coverage" , None , Some ( r # "{"content":"x"}"# .into ( ) ) ) .await;
3574
assert_ne! ( st, StatusCode:: CREATED ) ;
3575
let ( st, _ ) = call (
3576
& h.app,
3577
" POST" ,
3578
" /api/v1/repos/andy/vk/coverage" ,
3579
Some ( " site" ) ,
3580
Some ( r # "{"bookmark":"nope","content":"SF:a\nend_of_record\n"}"# .into ( ) ) ,
3581
)
3582
.await;
3583
assert_eq! ( st, StatusCode:: BAD_REQUEST ) ;
3584
}
2fb70d01 Andy McCall 24 days ago ↶
3585
3586
3587
# [ tokio ::test ( flavor = " multi_thread" ) ]
3588
async fn a_runners_facts_are_kept_from_its_claim ( ) {
3589
let h = hub ( ) .await;
3590
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
3591
let body = r # "{"runner":"test-1","runtimes":["microvm"],"facts":{"free_disk_mib":42000,"load_x100":112,"speed":100,"speed_class":3,"kernel":"7.0.0"}}"# ;
3592
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( body.into ( ) ) ) .await;
3593
assert_eq! ( st, StatusCode:: NO_CONTENT , " nothing to run, but the facts are kept" ) ;
3594
let ( st, out) = call ( & h.app, " GET" , " /api/v1/runners" , Some ( " site" ) , None ) .await;
3595
assert_eq! ( st, StatusCode:: OK , " {out}" ) ;
3596
let v: serde_json:: Value = serde_json:: from_str( & out) .unwrap ( ) ;
3597
let r = & v[ 0 ] ;
3598
assert_eq! (
3599
( r[ " speed" ] .as_i64 ( ) , r[ " speed_class" ] .as_i64 ( ) , r[ " free_disk_mib" ] .as_i64 ( ) , r[ " load_x100" ] .as_i64 ( ) ) ,
3600
( Some ( 100 ) , Some ( 3 ) , Some ( 42000 ) , Some ( 112 ) )
3601
) ;
3602
let row = db:: runners:: list( & h.app.db) .await.unwrap ( ) .remove ( 0 ) ;
3603
assert_eq! ( row.facts_line ( ) , " speed 100 (class 3) · load 1.12 · 41 GiB free" ) ;
3604
}
44a3d2eb Andy McCall 24 days ago ↶
3605
3606
3607
3608
# [ tokio ::test ( flavor = " multi_thread" ) ]
3609
async fn a_jobs_next_run_needs_the_disk_its_last_took ( ) {
3610
let h = hub ( ) .await;
3611
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
3612
create ( & h.app, " andy" , " demo" ) .await;
3613
let url = format! ( " {} /andy/demo" , h.base) ;
3614
let work = tempfile:: tempdir( ) .unwrap ( ) ;
3615
let c = client ( & url, work.path ( ) ) ;
3616
write (
3617
work.path ( ) ,
3618
" .flow/pipeline.kdl" ,
3619
& format! ( " schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" build\" {{ \n sandbox \" alpine\" \n exec \" make\" \n }} \n " ) ,
3620
) ;
3621
c.commit ( " Build" , & [ ] ) .unwrap ( ) ;
3622
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
3623
runs_of ( & h.app, " andy" , " demo" ) .await;
3624
let claim_with = | free : u64 | format! ( r # "{{ "runner":"test-1","runtimes":["microvm"],"facts":{{ "free_disk_mib":{free} }} }} "# ) ;
3625
3626
3627
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( claim_with ( 500 ) ) ) .await;
3628
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
3629
let id = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
3630
let done = serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 10 , " peak_disk_mib" : 8000 , " peak_memory_mib" : 900 } ) ;
3631
assert_eq! ( finish ( & h.app, id, done) .await, StatusCode:: NO_CONTENT ) ;
3632
let job = db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) ;
3633
assert_eq! ( ( job.peak_disk_mib, job.peak_memory_mib) , ( Some ( 8000 ) , Some ( 900 ) ) ) ;
3634
3635
3636
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( " {}" .into ( ) ) ) .await;
3637
assert_eq! ( st, StatusCode:: CREATED ) ;
3638
let repo = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
3639
let run2 = db:: flow:: run( & h.app.db, repo.id, 2 ) .await.unwrap ( ) .unwrap ( ) ;
3640
let next = db:: flow:: jobs( & h.app.db, run2.id) .await.unwrap ( ) .remove ( 0 ) ;
3641
assert_eq! ( next.need_disk_mib, Some ( 8000 + 2000 + 64 ) ) ;
3642
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( claim_with ( 9000 ) ) ) .await;
3643
assert_eq! ( st, StatusCode:: NO_CONTENT , " 9000 MiB free is not enough" ) ;
3644
oxygenhub:: flow:: explain_waits( & h.app) .await.unwrap ( ) ;
3645
let waiting = db:: flow:: job( & h.app.db, next.id) .await.unwrap ( ) .unwrap ( ) .waiting;
3646
assert! ( waiting.contains ( " 8.8 GiB free; the job needs ~9.8 GiB" ) , " {waiting}" ) ;
3647
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( claim_with ( 20_000 ) ) ) .await;
3648
assert_eq! ( st, StatusCode:: OK , " a roomier runner takes it: {body}" ) ;
3649
}
501d8b26 Andy McCall 24 days ago ↶
3650
3651
3652
3653
3654
# [ tokio ::test ( flavor = " multi_thread" ) ]
3655
async fn a_job_waits_briefly_for_a_runner_that_holds_its_image ( ) {
3656
let h = hub ( ) .await;
3657
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
3658
3659
db:: runners:: configured( & h.app.db, & blake3:: hash( b " other-secret" ) .to_hex ( ) , " earth-9" ) .await.unwrap ( ) ;
3660
let ask = | images : & str | {
3661
format! (
3662
r # "{{ "runner":"r","runtimes":["microvm"],"images":[{images} ],"fetches_images":true,"facts":{{ "speed":100,"speed_class":3,"free_disk_mib":50000}} }} "#
3663
)
3664
} ;
3665
let holder = ask ( & format! ( " \" {ROOTFS} \" " ) ) ;
3666
3667
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " other-secret" ) , Some ( holder.clone ( ) ) ) .await;
3668
assert_eq! ( st, StatusCode:: NO_CONTENT ) ;
3669
3b1bc27f Andy McCall 20 days ago
3670
sqlx:: query( " INSERT INTO jet_images (hash, size, arch, chunks, uploader, created_ms) VALUES ($1, 1, 'x86_64', 1, 'andy', 0)" )
501d8b26 Andy McCall 24 days ago ↶
3671
.bind ( ROOTFS )
3672
.execute ( & h.app.db)
3673
.await
3674
.unwrap ( ) ;
3675
3676
create ( & h.app, " andy" , " demo" ) .await;
3677
let url = format! ( " {} /andy/demo" , h.base) ;
3678
let work = tempfile:: tempdir( ) .unwrap ( ) ;
3679
let c = client ( & url, work.path ( ) ) ;
3680
write (
3681
work.path ( ) ,
3682
" .flow/pipeline.kdl" ,
3683
& format! ( " schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" build\" {{ \n sandbox \" alpine\" \n exec \" make\" \n }} \n " ) ,
3684
) ;
3685
c.commit ( " Build" , & [ ] ) .unwrap ( ) ;
3686
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
3687
runs_of ( & h.app, " andy" , " demo" ) .await;
3688
3689
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( ask ( " " ) ) ) .await;
3690
assert_eq! ( st, StatusCode:: NO_CONTENT , " held for the runner that holds the image" ) ;
3691
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " other-secret" ) , Some ( holder) ) .await;
3692
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
3693
let id = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
3694
let placed = db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) .placed;
3695
assert! ( placed.starts_with ( " holds the image, idle, speed 100 (class 3)" ) , " {placed}" ) ;
3696
assert! ( placed.contains ( " held for a better runner" ) , " {placed}" ) ;
3697
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/flow/runs/1" , Some ( " site" ) , None ) .await;
3698
assert! ( body.contains ( " \" placed\" :\" holds the image" ) , " {body}" ) ;
3699
}
20b8fe85 Andy McCall 24 days ago ↶
3700
4fa0da12 Andy McCall 23 days ago ↶
3701
3702
3703
# [ tokio ::test ( flavor = " multi_thread" ) ]
3704
async fn a_job_waits_briefly_for_the_runner_holding_its_build_cache ( ) {
3705
let h = hub ( ) .await;
3706
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
3707
db:: runners:: configured( & h.app.db, & blake3:: hash( b " other-secret" ) .to_hex ( ) , " earth-9" ) .await.unwrap ( ) ;
3708
let slot = flow_core:: api:: cache_slot( " andy/demo" , " build" ) ;
3709
let ask = | caches : & str | {
3710
format! (
3711
r # "{{ "runner":"r","runtimes":["microvm"],"images":["{ROOTFS} "],"fetches_images":true,"caches":[{caches} ],"facts":{{ "speed":100,"speed_class":3,"free_disk_mib":50000}} }} "#
3712
)
3713
} ;
3714
let holder = ask ( & format! ( " \" {slot} \" " ) ) ;
3715
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " other-secret" ) , Some ( holder.clone ( ) ) ) .await;
3716
assert_eq! ( st, StatusCode:: NO_CONTENT ) ;
3717
3718
create ( & h.app, " andy" , " demo" ) .await;
3719
let url = format! ( " {} /andy/demo" , h.base) ;
3720
let work = tempfile:: tempdir( ) .unwrap ( ) ;
3721
let c = client ( & url, work.path ( ) ) ;
3722
write (
3723
work.path ( ) ,
3724
" .flow/pipeline.kdl" ,
3725
& format! ( " schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" build\" {{ \n sandbox \" alpine\" \n exec \" make\" \n }} \n " ) ,
3726
) ;
3727
c.commit ( " Build" , & [ ] ) .unwrap ( ) ;
3728
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
3729
runs_of ( & h.app, " andy" , " demo" ) .await;
3730
3731
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( ask ( " " ) ) ) .await;
3732
assert_eq! ( st, StatusCode:: NO_CONTENT , " held for the runner holding the build cache" ) ;
3733
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " other-secret" ) , Some ( holder) ) .await;
3734
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
3735
let id = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
3736
let placed = db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) .placed;
3737
assert! ( placed.starts_with ( " holds its build cache, holds the image" ) , " {placed}" ) ;
3738
}
3739
c2be148c Andy McCall 22 days ago ↶
3740
3741
3742
# [ tokio ::test ( flavor = " multi_thread" ) ]
3743
async fn a_job_waits_briefly_for_the_fresher_build_cache ( ) {
3744
let h = hub ( ) .await;
3745
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
3746
db:: runners:: configured( & h.app.db, & blake3:: hash( b " other-secret" ) .to_hex ( ) , " earth-9" ) .await.unwrap ( ) ;
3747
let slot = flow_core:: api:: cache_slot( " andy/demo" , " build" ) ;
3748
let now = db:: now_ms( ) ;
3749
let ask = | kept : i64 | {
3750
format! (
3751
r # "{{ "runner":"r","runtimes":["microvm"],"images":["{ROOTFS} "],"fetches_images":true,"caches":["{slot} "],"cache_times":{{ "{slot} ":{kept} }} ,"facts":{{ "speed":100,"speed_class":3,"free_disk_mib":50000}} }} "#
3752
)
3753
} ;
3754
3755
let fresh = ask ( now - 60_000 ) ;
3756
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " other-secret" ) , Some ( fresh.clone ( ) ) ) .await;
3757
assert_eq! ( st, StatusCode:: NO_CONTENT ) ;
3758
3759
create ( & h.app, " andy" , " demo" ) .await;
3760
let url = format! ( " {} /andy/demo" , h.base) ;
3761
let work = tempfile:: tempdir( ) .unwrap ( ) ;
3762
let c = client ( & url, work.path ( ) ) ;
3763
write (
3764
work.path ( ) ,
3765
" .flow/pipeline.kdl" ,
3766
& format! ( " schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" build\" {{ \n sandbox \" alpine\" \n exec \" make\" \n }} \n " ) ,
3767
) ;
3768
c.commit ( " Build" , & [ ] ) .unwrap ( ) ;
3769
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
3770
runs_of ( & h.app, " andy" , " demo" ) .await;
3771
3772
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( ask ( now - 86_400_000 ) ) ) .await;
3773
assert_eq! ( st, StatusCode:: NO_CONTENT , " held for the fresher copy" ) ;
3774
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " other-secret" ) , Some ( fresh) ) .await;
3775
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
3776
let id = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
3777
let placed = db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) .placed;
3778
assert! ( placed.starts_with ( " holds its build cache (kept 1 min ago), holds the image" ) , " {placed}" ) ;
3779
}
3780
20b8fe85 Andy McCall 24 days ago ↶
3781
3782
3783
3784
# [ tokio ::test ( flavor = " multi_thread" ) ]
3785
async fn report_files_expire_under_retention ( ) {
3786
let h = hub ( ) .await;
3787
create ( & h.app, " andy" , " demo" ) .await;
3788
let repo = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
3789
let add = | bookmark : & 'static str , at : i64 | {
3790
let db = h.app.db.clone ( ) ;
3791
async move {
3792
let id: i64 = sqlx:: query_scalar(
3793
" INSERT INTO reports (repo_id, name, kind, title, summary, context, job, bookmark, state, files, bytes, created_ms) \
3b1bc27f Andy McCall 20 days ago
3794
VALUES ($1, 'perf', 'performance', 'Perf', '', '{}', 'perf', $2, 's', 1, 1, $3) RETURNING id" ,
20b8fe85 Andy McCall 24 days ago ↶
3795
)
3796
.bind ( repo.id)
3797
.bind ( bookmark)
3798
.bind ( at)
3799
.fetch_one ( & db)
3800
.await
3801
.unwrap ( ) ;
3b1bc27f Andy McCall 20 days ago
3802
sqlx:: query( " INSERT INTO report_files (report_id, path, blob, size) VALUES ($1, 'report.toml', $2, 1)" )
20b8fe85 Andy McCall 24 days ago ↶
3803
.bind ( id)
3804
.bind ( format! ( " {:064x}" , id) )
3805
.execute ( & db)
3806
.await
3807
.unwrap ( ) ;
3b1bc27f Andy McCall 20 days ago
3808
sqlx:: query( " INSERT INTO report_metrics (report_id, key, value, unit, better) VALUES ($1, 'ms', 1, '', '')" )
20b8fe85 Andy McCall 24 days ago ↶
3809
.bind ( id)
3810
.execute ( & db)
3811
.await
3812
.unwrap ( ) ;
3813
id
3814
}
3815
} ;
3816
let main_old = add ( " main" , 1 ) .await;
3817
add ( " main" , 2 ) .await;
3818
let mut feature = Vec :: new( ) ;
3819
for at in 0 .. 52 {
3820
feature.push ( add ( " feature" , 10 + at) .await) ;
3821
}
3822
assert_eq! ( oxygenhub:: reports:: expire( & h.app.db) .await.unwrap ( ) , 2 , " the two oldest feature runs" ) ;
3823
let files = | id : i64 | {
3824
let db = h.app.db.clone ( ) ;
3825
async move {
3b1bc27f Andy McCall 20 days ago
3826
sqlx:: query_scalar:: < _ , i64 > ( " SELECT COUNT(*) FROM report_files WHERE report_id = $1" ) .bind ( id) .fetch_one ( & db) .await.unwrap ( )
20b8fe85 Andy McCall 24 days ago ↶
3827
}
3828
} ;
3829
assert_eq! ( ( files ( feature[ 0 ] ) .await, files ( feature[ 1 ] ) .await, files ( feature[ 2 ] ) .await) , ( 0 , 0 , 1 ) ) ;
3830
assert_eq! ( files ( main_old) .await, 1 , " the default bookmark keeps every run" ) ;
3831
assert_eq! ( oxygenhub:: reports:: metrics( & h.app.db, feature[ 0 ] ) .await.unwrap ( ) .len ( ) , 1 , " metrics are kept" ) ;
3832
assert_eq! ( oxygenhub:: reports:: expire( & h.app.db) .await.unwrap ( ) , 0 , " once" ) ;
3833
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/demo/reports/perf?id={} " , feature[ 0 ] ) , None , None ) .await;
3834
assert! ( page.contains ( " files have expired" ) , " the page says so" ) ;
1c8276b9 Andy McCall 24 days ago ↶
3835
}
3836
3837
3838
3839
3840
# [ tokio ::test ( flavor = " multi_thread" ) ]
3841
async fn a_pipeline_pinning_an_image_nobody_holds_fails_a_check ( ) {
3842
let h = hub ( ) .await;
3843
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
3844
3845
let none = r # "{"runner":"test-1","runtimes":["microvm"],"images":[]}"# ;
3846
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( none.into ( ) ) ) .await;
3847
assert_eq! ( st, StatusCode:: NO_CONTENT ) ;
3848
create ( & h.app, " andy" , " demo" ) .await;
3849
let url = format! ( " {} /andy/demo" , h.base) ;
3850
let work = tempfile:: tempdir( ) .unwrap ( ) ;
3851
let c = client ( & url, work.path ( ) ) ;
3852
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
3853
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
3854
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
3855
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
3856
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
3857
let checks = checks_when ( & h.app, run.repo_id, & run.state, 4 ) .await;
3858
let check = checks.iter ( ) .find ( | c | c.name = = " pipeline/images" ) .expect ( " an images check" ) ;
3859
assert! ( check.status = = " failed" & & check.summary.contains ( & ROOTFS [ .. 12 ] ) & & check.summary.contains ( " flow rootfs push" ) , " {check:?}" ) ;
3860
3861
let holds = format! ( r # "{{ "runner":"test-1","runtimes":["microvm"],"images":["{ROOTFS} "]}} "# ) ;
3862
call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( holds) ) .await;
3863
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( " {}" .into ( ) ) ) .await;
3864
assert_eq! ( st, StatusCode:: CREATED ) ;
3865
let checks = db:: hooks:: checks_for_state( & h.app.db, run.repo_id, & run.state) .await.unwrap ( ) ;
3866
let check = checks.iter ( ) .find ( | c | c.name = = " pipeline/images" ) .unwrap ( ) ;
3867
assert_eq! ( check.status, " passed" , " {check:?}" ) ;
20b8fe85 Andy McCall 24 days ago ↶
3868
}
a46d15e7 Andy McCall 24 days ago ↶
3869
3870
3871
3872
3873
# [ tokio ::test ( flavor = " multi_thread" ) ]
3874
async fn the_tracker_names_peers_in_the_same_site_and_signs_a_ticket ( ) {
3875
let h = hub ( ) .await;
3876
let app = & h.app;
3877
let image = " ab" .repeat ( 32 ) ;
3878
let claim_as = | token : & 'static str , name : & 'static str , body : serde_json:: Value| async move {
3879
let mut b = serde_json:: json! ( { " runner" : name, " runtimes" : [ " microvm" ] } ) ;
3880
b.as_object_mut ( ) .unwrap ( ) .extend ( body.as_object ( ) .unwrap ( ) .clone ( ) ) ;
3881
let ( st, out) = call ( app, " POST" , " /api/v1/flow/claim" , Some ( token) , Some ( b.to_string ( ) ) ) .await;
3882
assert_eq! ( st, StatusCode:: NO_CONTENT , " {out}" ) ;
3883
} ;
3884
let announce = | token : & 'static str , state : & 'static str | {
3885
let body = serde_json:: json! ( { " image" : image, " state" : state } ) .to_string ( ) ;
3886
async move { call ( app, " POST" , " /api/v1/flow/swarm/announce" , Some ( token) , Some ( body) ) .await }
3887
} ;
3888
6adcd68b Andy McCall 24 days ago ↶
3889
claim_as (
3890
" runner-secret" ,
3891
" a" ,
3892
serde_json:: json! ( { " images" : [ image] , " swarm" : { " addr" : " 10.0.0.1:7401" , " site" : " lab" , " cert" : " 11" .repeat ( 32 ) } } ) ,
3893
)
3894
.await;
3895
claim_as (
3896
" runner-two" ,
3897
" b" ,
3898
serde_json:: json! ( { " images" : [ ] , " swarm" : { " addr" : " 10.0.0.2:7401" , " site" : " lab" , " cert" : " 22" .repeat ( 32 ) } } ) ,
3899
)
3900
.await;
3901
claim_as (
3902
" runner-three" ,
3903
" c" ,
3904
serde_json:: json! ( { " images" : [ ] , " swarm" : { " addr" : " 10.0.0.3:7401" , " site" : " lab" , " cert" : " 33" .repeat ( 32 ) } } ) ,
3905
)
3906
.await;
a46d15e7 Andy McCall 24 days ago ↶
3907
3908
let ( st, body) = announce ( " runner-two" , " fetching" ) .await;
3909
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
3910
let a: flow_core:: api:: Announced = serde_json:: from_str( & body) .unwrap ( ) ;
3911
let names: Vec< ( & str , bool ) > = a.peers.iter ( ) .map ( | p | ( p.addr.as_str ( ) , p.complete) ) .collect ( ) ;
3912
assert_eq! ( names, vec! [ ( " 10.0.0.1:7401" , true ) ] , " only A holds it so far" ) ;
6adcd68b Andy McCall 24 days ago ↶
3913
assert_eq! ( a.peers[ 0 ] .cert, " 11" .repeat ( 32 ) , " with the fingerprint peers check it by" ) ;
a46d15e7 Andy McCall 24 days ago ↶
3914
let t = jet:: swarm:: Ticket:: open( & a.ticket, & app.hub_key.identity, oxygenhub:: db:: now_ms( ) as u64 ) .unwrap ( ) ;
3915
assert_eq! ( ( t.name.as_str ( ) , t.site.as_str ( ) ) , ( " b" , " lab" ) ) ;
3916
3917
3918
let ( _ , body) = announce ( " runner-three" , " fetching" ) .await;
3919
let a: flow_core:: api:: Announced = serde_json:: from_str( & body) .unwrap ( ) ;
3920
let mut peers: Vec< ( & str , bool ) > = a.peers.iter ( ) .map ( | p | ( p.addr.as_str ( ) , p.complete) ) .collect ( ) ;
3921
peers.sort ( ) ;
3922
assert_eq! ( peers, vec! [ ( " 10.0.0.1:7401" , true ) , ( " 10.0.0.2:7401" , false ) ] ) ;
3923
3924
3925
announce ( " runner-two" , " done" ) .await;
6adcd68b Andy McCall 24 days ago ↶
3926
claim_as (
3927
" runner-three" ,
3928
" c" ,
3929
serde_json:: json! ( { " images" : [ ] , " swarm" : { " addr" : " 10.0.0.3:7401" , " site" : " elsewhere" , " cert" : " 33" .repeat ( 32 ) } } ) ,
3930
)
3931
.await;
a46d15e7 Andy McCall 24 days ago ↶
3932
let ( _ , body) = announce ( " runner-three" , " fetching" ) .await;
3933
let a: flow_core:: api:: Announced = serde_json:: from_str( & body) .unwrap ( ) ;
3934
assert! ( a.peers.is_empty ( ) , " {:?}" , a.peers) ;
6adcd68b Andy McCall 24 days ago ↶
3935
3936
3937
claim_as ( " runner-secret" , " a" , serde_json:: json! ( { " images" : [ image] , " swarm" : { " addr" : " 10.0.0.1:7401" , " site" : " elsewhere" } } ) )
3938
.await;
3939
let ( st, _ ) = announce ( " runner-secret" , " fetching" ) .await;
3940
assert_eq! ( st, StatusCode:: CONFLICT ) ;
a46d15e7 Andy McCall 24 days ago ↶
3941
3942
3943
claim_as ( " runner-two" , " b" , serde_json:: json! ( { " images" : [ ] } ) ) .await;
3944
let ( st, body) = announce ( " runner-two" , " fetching" ) .await;
3945
assert_eq! ( st, StatusCode:: CONFLICT ) ;
3946
assert! ( body.contains ( " [runner.swarm]" ) , " {body}" ) ;
3947
let ( st, _ ) =
3948
call ( app, " POST" , " /api/v1/flow/swarm/announce" , Some ( " runner-secret" ) , Some ( r # "{"image":"nope","state":"fetching"}"# .into ( ) ) )
3949
.await;
3950
assert_eq! ( st, StatusCode:: BAD_REQUEST ) ;
3951
let ( st, _ ) = call (
3952
app,
3953
" POST" ,
3954
" /api/v1/flow/swarm/announce" ,
3955
None ,
3956
Some ( serde_json:: json! ( { " image" : image, " state" : " fetching" } ) .to_string ( ) ) ,
3957
)
3958
.await;
3959
assert_eq! ( st, StatusCode:: UNAUTHORIZED ) ;
3960
}
6ca0b971 Andy McCall 24 days ago ↶
3961
3962
3963
# [ tokio ::test ( flavor = " multi_thread" ) ]
3964
async fn the_images_page_says_where_each_image_is_and_what_uses_it ( ) {
3965
let h = hub ( ) .await;
3966
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
3967
create ( & h.app, " andy" , " demo" ) .await;
3968
let work = tempfile:: tempdir( ) .unwrap ( ) ;
3969
let c = client ( & format! ( " {} /andy/demo" , h.base) , work.path ( ) ) ;
3970
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
3971
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
3972
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
3973
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
3974
3975
let held = " cd" .repeat ( 32 ) ;
3976
let kept = " ef" .repeat ( 32 ) ;
3977
let body = serde_json:: json! ( { " runner" : " earth-1" , " runtimes" : [ " wasm" ] , " images" : [ held] , " swarm" : { " addr" : " 10.0.0.1:7401" , " site" : " home" , " cert" : " 11" .repeat ( 32 ) } } ) ;
3978
let ( st, out) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( body.to_string ( ) ) ) .await;
3979
assert! ( st = = StatusCode:: NO_CONTENT | | st = = StatusCode:: OK , " {st} {out}" ) ;
3b1bc27f Andy McCall 20 days ago
3980
sqlx:: query( " INSERT INTO jet_images (hash, size, arch, chunks, uploader, created_ms) VALUES ($1, 1048576, 'x86_64', 9, 'andy', 1)" )
6ca0b971 Andy McCall 24 days ago ↶
3981
.bind ( & kept)
3982
.execute ( & h.app.db)
3983
.await
3984
.unwrap ( ) ;
3985
3986
let res = oxygenhub:: router( h.app.clone ( ) )
3987
.oneshot (
3988
Request:: post( " /register" )
3989
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
c8684d44 Andy McCall 20 days ago
3990
.body ( Body:: from( " username=carol&email=carol%40example.com&password=correct-horse-battery&setup=test-setup" ) )
6ca0b971 Andy McCall 24 days ago ↶
3991
.unwrap ( ) ,
3992
)
3993
.await
3994
.unwrap ( ) ;
3995
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
3996
let carol = db:: accounts:: account_by_name( & h.app.db, " carol" ) .await.unwrap ( ) .unwrap ( ) ;
3997
db:: accounts:: set_admin( & h.app.db, carol.id, true ) .await.unwrap ( ) ;
5d66f27c Andy McCall 9 days ago ↶
3998
3999
let lost = jet:: images:: scan( & [ 7 u8 ; 4096 ] [ .. ] , Some ( " x86_64" .into ( ) ) , | _, _| Ok ( ( ) ) ) .unwrap ( ) ;
4000
oxygenhub:: images:: record_peer( & h.app.db, & lost, " mars-1" ) .await.unwrap ( ) ;
6ca0b971 Andy McCall 24 days ago ↶
4001
let res = oxygenhub:: router( h.app.clone ( ) )
4002
.oneshot ( Request:: get( " /admin/runners/images" ) .header ( header:: COOKIE , cookie) .body ( Body:: empty( ) ) .unwrap ( ) )
4003
.await
4004
.unwrap ( ) ;
4005
assert_eq! ( res.status ( ) , StatusCode:: OK ) ;
4006
let page = String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) ;
4007
let row = | hash : & str | {
4008
let at = page.find ( & format! ( " title=\" {hash} \" " ) ) .unwrap_or_else ( | | panic! ( " {hash} is not on the page" ) ) ;
4009
let end = page[ at.. ] .find ( " </tr>" ) .unwrap ( ) ;
4010
page[ at.. at + end] .to_string ( )
4011
} ;
4012
let pinned = row ( ROOTFS ) ;
4013
assert! ( pinned.contains ( " nowhere" ) & & pinned.contains ( " andy/demo" ) & & pinned.contains ( " 3 jobs" ) , " {pinned}" ) ;
4014
let on_runner = row ( & held) ;
4015
assert! ( on_runner.contains ( " img-holds" ) & & on_runner.contains ( " earth-1" ) & & ! on_runner.contains ( " nowhere" ) , " {on_runner}" ) ;
4016
let on_hub = row ( & kept) ;
4017
assert! ( on_hub.contains ( " img-hub" ) & & on_hub.contains ( " 1.0 MiB" ) & & on_hub.contains ( " x86_64" ) , " {on_hub}" ) ;
5d66f27c Andy McCall 9 days ago ↶
4018
let peer_only = row ( & lost.hash) ;
4019
assert! ( peer_only.contains ( " img-peer" ) & & peer_only.contains ( " sent by mars-1" ) & & peer_only.contains ( " nowhere" ) , " {peer_only}" ) ;
6ca0b971 Andy McCall 24 days ago ↶
4020
assert! ( page.contains ( " earth-1 (home)" ) , " the swarm is named" ) ;
6246d1e5 Andy McCall 23 days ago ↶
4021
4022
assert! ( page.contains ( r # "{"images":"# ) & & page.contains ( r # ""scene":"swarm""# ) & & page.contains ( r # ""earth-1""# ) , " the swarm's data" ) ;
6ca0b971 Andy McCall 24 days ago ↶
4023
assert_eq! ( call ( & h.app, " GET" , " /admin/runners/images" , None , None ) .await.0 , StatusCode:: NOT_FOUND , " administrators only" ) ;
4024
}
f677d627 Andy McCall 23 days ago ↶
4025
4026
4027
4028
4029
# [ tokio ::test ( flavor = " multi_thread" ) ]
4030
async fn consumers_of_a_subrepo_are_indexed_with_how_far_behind_they_are ( ) {
4031
let h = hub ( ) .await;
e7158350 Andy McCall 23 days ago ↶
4032
4033
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
f677d627 Andy McCall 23 days ago ↶
4034
create ( & h.app, " andy" , " api" ) .await;
4035
create ( & h.app, " andy" , " game" ) .await;
4036
let api_work = tempfile:: tempdir( ) .unwrap ( ) ;
4037
let api = client ( & format! ( " {} /andy/api" , h.base) , api_work.path ( ) ) ;
4038
write ( api_work.path ( ) , " api.h" , " v1\n " ) ;
4039
api.snapshot ( " t" ) .unwrap ( ) ;
4040
api.commit ( " api v1" , & [ ] ) .unwrap ( ) ;
4041
assert! ( api.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4042
4043
let game_work = tempfile:: tempdir( ) .unwrap ( ) ;
4044
let game = client ( & format! ( " {} /andy/game" , h.base) , game_work.path ( ) ) ;
4045
game.submodule_add ( & format! ( " {} /andy/api" , h.base) , " deps/api" , Some ( " main" ) ) .unwrap ( ) ;
4046
write ( game_work.path ( ) , " game.c" , " int main;\n " ) ;
4047
game.snapshot ( " t" ) .unwrap ( ) ;
4048
game.commit ( " game on api v1" , & [ ] ) .unwrap ( ) ;
4049
assert! ( game.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4050
4051
let consumers = | | async {
4052
let ( st, body) = call ( & h.app, " GET" , " /api/v1/repos/andy/api/consumers" , Some ( " site" ) , None ) .await;
4053
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
4054
serde_json:: from_str:: < Vec< serde_json:: Value> > ( & body) .unwrap ( )
4055
} ;
4056
4057
4058
let settle = | want : & 'static ( dyn Fn( & [serde_json:: Value]) -> bool + Sync) | {
4059
let consumers = & consumers;
4060
async move {
4061
for _ in 0 .. 100 {
4062
let list = consumers ( ) .await;
4063
if want ( & list) {
4064
return list;
4065
}
4066
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
4067
}
4068
consumers ( ) .await
4069
}
4070
} ;
4071
let list = settle ( & | l| ! l.is_empty ( ) ) .await;
4072
assert_eq! ( list.len ( ) , 1 , " {list:?}" ) ;
4073
assert_eq! (
4074
( list[ 0 ] [ " name" ] .as_str ( ) , list[ 0 ] [ " path" ] .as_str ( ) , list[ 0 ] [ " bookmark" ] .as_str ( ) ) ,
4075
( Some ( " game" ) , Some ( " deps/api" ) , Some ( " main" ) )
4076
) ;
4077
assert_eq! ( list[ 0 ] [ " behind" ] , 0 ) ;
4078
4079
4080
write ( api_work.path ( ) , " api.c" , " v2\n " ) ;
4081
api.snapshot ( " t" ) .unwrap ( ) ;
4082
api.commit ( " api v2" , & [ ] ) .unwrap ( ) ;
4083
assert! ( api.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4084
assert_eq! ( settle ( & | l| l.first ( ) .is_some_and ( | c | c[ " behind" ] = = 1 ) ) .await[ 0 ] [ " behind" ] , 1 ) ;
4085
let ( _ , page) = call ( & h.app, " GET" , " /andy/api" , None , None ) .await;
c9aacc0e Andy McCall 23 days ago ↶
4086
assert! ( page.contains ( " Diffuses to" ) & & page.contains ( " andy/game" ) & & page.contains ( " 1 behind" ) , " the page shows it" ) ;
f677d627 Andy McCall 23 days ago ↶
4087
4088
4089
sqlx:: query( " UPDATE repositories SET visibility = 'private' WHERE name = 'game'" ) .execute ( & h.app.db) .await.unwrap ( ) ;
4090
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/repos/andy/api/consumers" , None , None ) .await;
4091
assert_eq! ( body.trim ( ) , " []" ) ;
4092
let ( _ , page) = call ( & h.app, " GET" , " /andy/api" , None , None ) .await;
4093
assert! ( ! page.contains ( " andy/game" ) ) ;
4094
sqlx:: query( " UPDATE repositories SET visibility = 'public' WHERE name = 'game'" ) .execute ( & h.app.db) .await.unwrap ( ) ;
e7158350 Andy McCall 23 days ago ↶
4095
1d634f69 Andy McCall 23 days ago ↶
4096
4097
4098
let game_row = db:: repos:: get( & h.app.db, " andy" , " game" ) .await.unwrap ( ) .unwrap ( ) ;
4099
let api_row = db:: repos:: get( & h.app.db, " andy" , " api" ) .await.unwrap ( ) .unwrap ( ) ;
4100
let api_heads = api.bookmark_heads ( ) .unwrap ( ) ;
4101
let bumped = oxygenhub:: diffusion:: bump( & h.app, & game_row, " main" , " deps/api" , & api_row, api_heads.clone ( ) ) .await.unwrap ( ) ;
4102
let oxygenhub:: diffusion:: Bumped:: Moved { bookmark, heads, .. } = bumped else { panic! ( " {bumped:?}" ) } ;
4103
assert_eq! ( bookmark, " diffusion/deps/api" ) ;
4104
game.fetch ( " origin" ) .unwrap ( ) ;
4105
let bump_state = game.load_state ( & heads) .unwrap ( ) .materialize ( ) ;
4106
let pin = match & bump_state.files[ " deps/api" ] .content {
4107
oxy_core:: state:: FileContent:: Opaque { blobs } => oxy_core:: subrepo:: Pin:: decode( & game.read_blob ( & blobs[ 0 ] ) .unwrap ( ) ) .unwrap ( ) ,
4108
other => panic! ( " {other:?}" ) ,
4109
} ;
4110
assert! (
4111
matches! ( & pin, oxy_core:: subrepo:: Pin:: Oxy { heads, .. } if { let mut a = heads.clone ( ) ; a.sort ( ) ; let mut b = api_heads.clone ( ) ; b.sort ( ) ; a = = b } ) ,
4112
" {pin:?}"
4113
) ;
4114
assert! ( bump_state.files.contains_key ( " game.c" ) , " the rest of the game is as it was" ) ;
9aaa88b2 Andy McCall 23 days ago ↶
4115
assert! ( matches! (
1d634f69 Andy McCall 23 days ago ↶
4116
oxygenhub:: diffusion:: bump( & h.app, & game_row, " main" , " deps/api" , & api_row, api_heads.clone ( ) ) .await.unwrap ( ) ,
9aaa88b2 Andy McCall 23 days ago ↶
4117
oxygenhub:: diffusion:: Bumped:: Already { .. }
4118
) ) ;
7a14bdae Andy McCall 23 days ago ↶
4119
4120
4121
1d634f69 Andy McCall 23 days ago ↶
4122
write ( api_work.path ( ) , " other.c" , " v3\n " ) ;
4123
api.snapshot ( " t" ) .unwrap ( ) ;
4124
let v3 = api.commit ( " api v3, on its own" , & [ ] ) .unwrap ( ) ;
4125
assert! ( api.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4126
let only_v3 = api.load_state ( & [ v3] ) .unwrap ( ) .heads ( ) .to_vec ( ) ;
7a14bdae Andy McCall 23 days ago ↶
4127
let combined = oxygenhub:: diffusion:: bump( & h.app, & game_row, " main" , " deps/api" , & api_row, only_v3) .await.unwrap ( ) ;
4128
let oxygenhub:: diffusion:: Bumped:: Moved { heads: ch, .. } = combined else { panic! ( " {combined:?}" ) } ;
4129
game.fetch ( " origin" ) .unwrap ( ) ;
4130
let st = game.load_state ( & ch) .unwrap ( ) .materialize ( ) ;
4131
let pin = match & st.files[ " deps/api" ] .content {
4132
oxy_core:: state:: FileContent:: Opaque { blobs } => oxy_core:: subrepo:: Pin:: decode( & game.read_blob ( & blobs[ 0 ] ) .unwrap ( ) ) .unwrap ( ) ,
4133
other => panic! ( " {other:?}" ) ,
4134
} ;
4135
let oxy_core:: subrepo:: Pin:: Oxy { heads: pinned, .. } = pin else { panic! ( ) } ;
4136
let all = api.load_state ( & pinned) .unwrap ( ) ;
4137
4138
assert! ( all.contains ( & v3) & & all.len ( ) = = 2 , " v1 and v3, {} changes" , all.len ( ) ) ;
1d634f69 Andy McCall 23 days ago ↶
4139
4140
let newer = api.bookmark_heads ( ) .unwrap ( ) ;
4141
let again = oxygenhub:: diffusion:: bump( & h.app, & game_row, " main" , " deps/api" , & api_row, newer) .await.unwrap ( ) ;
4142
assert! ( matches! ( again, oxygenhub:: diffusion:: Bumped:: Moved { ref bookmark, .. } if bookmark = = " diffusion/deps/api" ) , " {again:?}" ) ;
4143
e7158350 Andy McCall 23 days ago ↶
4144
4145
let put =
4146
| body : & 'static str , who : Option< & 'static str > | call ( & h.app, " PUT" , " /api/v1/repos/andy/game/diffusion" , who, Some ( body.into ( ) ) ) ;
4147
let ( st, body) = put ( r # "{"follow":"green"}"# , Some ( " site" ) ) .await;
4148
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
4149
assert! ( body.contains ( " \" follow\" :\" green\" " ) & & body.contains ( " \" from\" :\" setting\" " ) , " {body}" ) ;
4150
assert_eq! ( put ( r # "{"follow":"sometimes"}"# , Some ( " site" ) ) .await.0 , StatusCode:: BAD_REQUEST ) ;
4151
assert_ne! ( put ( r # "{"follow":"any"}"# , None ) .await.0 , StatusCode:: OK , " not without a token" ) ;
4152
let ( _ , body) = put ( r # "{"follow":"default"}"# , Some ( " site" ) ) .await;
4153
assert! ( body.contains ( " \" follow\" :\" off\" " ) & & body.contains ( " \" from\" :\" site\" " ) , " {body}" ) ;
f677d627 Andy McCall 23 days ago ↶
4154
4155
4156
game.submodule_remove ( " deps/api" ) .unwrap ( ) ;
4157
std:: fs:: remove_dir_all( game_work.path ( ) .join ( " deps/api" ) ) .unwrap ( ) ;
4158
game.snapshot ( " t" ) .unwrap ( ) ;
4159
game.commit ( " game without api" , & [ ] ) .unwrap ( ) ;
4160
assert! ( game.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4161
let left = settle ( & | l| l.is_empty ( ) ) .await;
4162
assert! ( left.is_empty ( ) , " {left:?}" ) ;
4163
}
a7e6e8e6 Andy McCall 23 days ago ↶
4164
4165
4166
4167
4168
# [ tokio ::test ( flavor = " multi_thread" ) ]
4169
async fn diffusion_proposes_a_bump_when_the_upstream_moves ( ) {
4170
let h = hub ( ) .await;
4171
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
4172
create ( & h.app, " andy" , " api" ) .await;
4173
create ( & h.app, " andy" , " game" ) .await;
180dc43a Andy McCall 23 days ago ↶
4174
4175
let seen: std:: sync:: Mutex< Vec< serde_json:: Value> > = std:: sync:: Mutex:: new( Vec :: new( ) ) ;
4176
let seen = & seen;
4177
let hub_app = & h.app;
4178
let pass_all = move | | async move {
4179
let h_app = hub_app;
a7e6e8e6 Andy McCall 23 days ago ↶
4180
for _ in 0 .. 20 {
180dc43a Andy McCall 23 days ago ↶
4181
let Some ( a) = claim ( h_app) .await else {
a7e6e8e6 Andy McCall 23 days ago ↶
4182
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 30 ) ) .await;
180dc43a Andy McCall 23 days ago ↶
4183
if ! jobs_left ( h_app) .await {
a7e6e8e6 Andy McCall 23 days ago ↶
4184
break ;
4185
}
4186
continue ;
4187
} ;
180dc43a Andy McCall 23 days ago ↶
4188
seen.lock ( ) .unwrap ( ) .push ( a.clone ( ) ) ;
a7e6e8e6 Andy McCall 23 days ago ↶
4189
let id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
4190
let ( name, ph) = ( a[ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) .to_string ( ) , a[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) .to_string ( ) ) ;
f7bcee49 Andy McCall 22 days ago ↶
4191
let st = finish ( h_app, id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 10 , " receipt" : receipt ( & name, & ph, 0 ) } ) ) .await;
4192
4193
4194
4195
if st = = StatusCode:: CONFLICT {
4196
let job = db:: flow:: job( & h_app.db, id) .await.unwrap ( ) .unwrap ( ) ;
4197
assert_eq! ( job.status, " cancelled" , " refused only because a newer run cancelled it" ) ;
4198
continue ;
4199
}
4200
assert_eq! ( st, StatusCode:: NO_CONTENT ) ;
a7e6e8e6 Andy McCall 23 days ago ↶
4201
}
4202
} ;
4203
let api_work = tempfile:: tempdir( ) .unwrap ( ) ;
4204
let api = client ( & format! ( " {} /andy/api" , h.base) , api_work.path ( ) ) ;
4205
write ( api_work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
4206
write ( api_work.path ( ) , " test.sh" , " echo ok\n " ) ;
4207
write ( api_work.path ( ) , " api.h" , " v1\n " ) ;
4208
api.snapshot ( " t" ) .unwrap ( ) ;
4209
api.commit ( " api v1" , & [ ] ) .unwrap ( ) ;
4210
assert! ( api.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4211
pass_all ( ) .await;
4212
4213
let game_work = tempfile:: tempdir( ) .unwrap ( ) ;
4214
let game = client ( & format! ( " {} /andy/game" , h.base) , game_work.path ( ) ) ;
4215
game.submodule_add ( & format! ( " {} /andy/api" , h.base) , " deps/api" , Some ( " main" ) ) .unwrap ( ) ;
4216
write ( game_work.path ( ) , " game.c" , " int main;\n " ) ;
f01cb865 Andy McCall 23 days ago ↶
4217
write ( game_work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
4218
write ( game_work.path ( ) , " test.sh" , " echo ok\n " ) ;
a7e6e8e6 Andy McCall 23 days ago ↶
4219
game.snapshot ( " t" ) .unwrap ( ) ;
4220
game.commit ( " game on api v1" , & [ ] ) .unwrap ( ) ;
4221
assert! ( game.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4222
let game_row = db:: repos:: get( & h.app.db, " andy" , " game" ) .await.unwrap ( ) .unwrap ( ) ;
4223
let bump = | | async {
4224
for _ in 0 .. 100 {
4225
let open = db:: proposals:: list( & h.app.db, game_row.id, & [ " open" ] ) .await.unwrap ( ) ;
4226
if let Some ( p) = open.into_iter ( ) .find ( | p | p.source_bookmark = = " diffusion/deps/api" ) {
4227
return Some ( p) ;
4228
}
4229
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
4230
}
4231
None
4232
} ;
4233
4234
4235
write ( api_work.path ( ) , " a2.c" , " v2\n " ) ;
4236
api.snapshot ( " t" ) .unwrap ( ) ;
4237
api.commit ( " api v2: faster sprites" , & [ ] ) .unwrap ( ) ;
4238
assert! ( api.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4239
pass_all ( ) .await;
4240
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 300 ) ) .await;
4241
assert! ( db:: proposals:: list( & h.app.db, game_row.id, & [ ] ) .await.unwrap ( ) .is_empty ( ) , " off proposes nothing" ) ;
4242
3776b70d Andy McCall 23 days ago ↶
4243
4244
a7e6e8e6 Andy McCall 23 days ago ↶
4245
let put = | body : & 'static str | call ( & h.app, " PUT" , " /api/v1/repos/andy/game/diffusion" , Some ( " site" ) , Some ( body.into ( ) ) ) ;
4246
assert_eq! ( put ( r # "{"follow":"any"}"# ) .await.0 , StatusCode:: OK ) ;
3776b70d Andy McCall 23 days ago ↶
4247
let first = bump ( ) .await.expect ( " caught up when turned on" ) ;
4248
assert! ( first.title.contains ( " 1 change" ) & & first.body.contains ( " api v2: faster sprites" ) , " {:?}" , first.title) ;
f01cb865 Andy McCall 23 days ago ↶
4249
4250
let mut ran = false ;
4251
for _ in 0 .. 100 {
4252
if runs_of ( & h.app, " andy" , " game" ) .await.iter ( ) .any ( | r | r.bookmark = = " diffusion/deps/api" ) {
4253
ran = true ;
4254
break ;
4255
}
4256
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
4257
}
4258
assert! ( ran, " a run of the bump" ) ;
4259
pass_all ( ) .await;
a7e6e8e6 Andy McCall 23 days ago ↶
4260
write ( api_work.path ( ) , " a3.c" , " v3\n " ) ;
4261
api.snapshot ( " t" ) .unwrap ( ) ;
4262
api.commit ( " api v3: save slots past ten" , & [ ] ) .unwrap ( ) ;
4263
assert! ( api.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
3776b70d Andy McCall 23 days ago ↶
4264
let mut p = first.clone ( ) ;
4265
for _ in 0 .. 100 {
4266
p = bump ( ) .await.unwrap ( ) ;
4267
if p.body.contains ( " api v3" ) {
4268
break ;
4269
}
4270
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
4271
}
4272
assert_eq! ( p.number, first.number, " the same proposal" ) ;
a7e6e8e6 Andy McCall 23 days ago ↶
4273
assert_eq! ( ( p.author.as_str ( ) , p.target.as_str ( ) ) , ( " diffusion" , " main" ) ) ;
4274
assert! ( p.title.contains ( " deps/api" ) & & p.title.contains ( " andy/api" ) & & p.title.contains ( " 2 changes" ) , " {}" , p.title) ;
4275
assert! ( p.body.contains ( " api v2: faster sprites" ) & & p.body.contains ( " api v3: save slots past ten" ) , " {}" , p.body) ;
4276
let labels = db:: issues:: proposal_labels( & h.app.db, p.id) .await.unwrap ( ) ;
4277
assert! ( labels.iter ( ) .any ( | l | l.name = = " dependencies" ) ) ;
4278
pass_all ( ) .await;
4279
4280
4281
4282
assert_eq! ( put ( r # "{"follow":"green"}"# ) .await.0 , StatusCode:: OK ) ;
4283
write ( api_work.path ( ) , " a4.c" , " v4\n " ) ;
4284
api.snapshot ( " t" ) .unwrap ( ) ;
4285
api.commit ( " api v4: rsm_set cheaper" , & [ ] ) .unwrap ( ) ;
4286
assert! ( api.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4287
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 300 ) ) .await;
4288
assert! ( ! bump ( ) .await.unwrap ( ) .body.contains ( " api v4" ) , " not before its run passes" ) ;
4289
pass_all ( ) .await;
4290
let mut updated = None ;
04683947 Andy McCall 9 days ago ↶
4291
4292
for _ in 0 .. 600 {
a7e6e8e6 Andy McCall 23 days ago ↶
4293
let p2 = bump ( ) .await.unwrap ( ) ;
4294
if p2.body.contains ( " api v4" ) {
4295
updated = Some ( p2) ;
4296
break ;
4297
}
4298
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
4299
}
4300
let p2 = updated.expect ( " brought up to date once the run passed" ) ;
4301
assert_eq! ( p2.number, p.number, " the same proposal" ) ;
4302
assert! ( p2.title.contains ( " 3 changes" ) , " {}" , p2.title) ;
4303
assert_eq! ( db:: proposals:: list( & h.app.db, game_row.id, & [ ] ) .await.unwrap ( ) .len ( ) , 1 , " never a pile of bumps" ) ;
3776b70d Andy McCall 23 days ago ↶
4304
757195ad Andy McCall 23 days ago ↶
4305
4306
4307
pass_all ( ) .await;
4308
let ( st, page) = call ( & h.app, " GET" , " /andy/api/diffusion" , None , None ) .await;
4309
assert_eq! ( st, StatusCode:: OK ) ;
c9aacc0e Andy McCall 23 days ago ↶
4310
assert! ( page.contains ( " Diffuses to" ) & & page.contains ( " andy/game" ) & & page.contains ( & format! ( " #{} " , p2.number) ) , " the bump is listed" ) ;
757195ad Andy McCall 23 days ago ↶
4311
assert! ( page.contains ( " df-green" ) , " a passing run of a checked upstream is green" ) ;
6246d1e5 Andy McCall 23 days ago ↶
4312
4313
assert! (
4314
page.contains ( r # ""scene":"ripples""# ) & & page.contains ( r # "["andy/game","# ) & & page.contains ( r # ""repo":"andy/api""# ) ,
4315
" the ripples' data"
4316
) ;
757195ad Andy McCall 23 days ago ↶
4317
let ( _ , page) = call ( & h.app, " GET" , " /andy/game/diffusion" , None , None ) .await;
c9aacc0e Andy McCall 23 days ago ↶
4318
assert! ( page.contains ( " Takes from" ) & & page.contains ( " andy/api" ) ) ;
757195ad Andy McCall 23 days ago ↶
4319
let ( _ , overview) = call ( & h.app, " GET" , " /andy/game" , None , None ) .await;
c9aacc0e Andy McCall 23 days ago ↶
4320
assert! ( overview.contains ( " Takes from" ) & & overview.contains ( " >Diffusion</a>" ) , " the tab and the panel" ) ;
eb2a3502 Andy McCall 23 days ago ↶
4321
4322
let ( st, body) = call ( & h.app, " GET" , " /api/v1/repos/andy/game/diffusion" , Some ( " site" ) , None ) .await;
4323
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
4324
let d: hub_api:: Diffusion = serde_json:: from_str( & body) .unwrap ( ) ;
4325
assert! ( d.diffuses_to.is_empty ( ) , " nothing pins the game" ) ;
4326
let pin = d.takes_from.first ( ) .expect ( " the game takes from the api" ) ;
4327
assert_eq! ( ( pin.upstream.as_str ( ) , pin.consumer.as_str ( ) ) , ( " andy/api" , " andy/game" ) ) ;
4328
assert_eq! ( pin.bump, Some ( p2.number) ) ;
4329
assert_eq! ( pin.upstream_state.tone, " green" , " {body}" ) ;
4330
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/repos/andy/api/diffusion" , Some ( " site" ) , None ) .await;
4331
let d: hub_api:: Diffusion = serde_json:: from_str( & body) .unwrap ( ) ;
4332
assert! ( d.takes_from.is_empty ( ) & & d.diffuses_to.iter ( ) .any ( | p | p.consumer = = " andy/game" ) , " {body}" ) ;
757195ad Andy McCall 23 days ago ↶
4333
let ( st, _ ) = call ( & h.app, " GET" , " /admin/diffusion" , None , None ) .await;
4334
assert_eq! ( st, StatusCode:: NOT_FOUND , " administrators only" ) ;
4335
4336
5d54cf74 Andy McCall 20 days ago ↶
4337
4338
4339
let game = oxygenhub:: db:: repos:: get( & h.app.db, " andy" , " game" ) .await.unwrap ( ) .unwrap ( ) ;
4340
oxygenhub:: diffusion:: fail_next_records( & h.app, game.id, 1 ) ;
757195ad Andy McCall 23 days ago ↶
4341
let ( st, body) =
4342
call ( & h.app, " POST" , & format! ( " /api/v1/repos/andy/game/proposals/{} /land" , p2.number) , Some ( " site" ) , Some ( " {}" .into ( ) ) ) .await;
4343
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
4344
let mut behind = serde_json:: Value:: Null;
4345
for _ in 0 .. 100 {
4346
let ( _ , list) = call ( & h.app, " GET" , " /api/v1/repos/andy/api/consumers" , Some ( " site" ) , None ) .await;
4347
let list: Vec< serde_json:: Value> = serde_json:: from_str( & list) .unwrap ( ) ;
4348
behind = list.iter ( ) .find ( | c | c[ " name" ] = = " game" ) .map ( | c | c[ " behind" ] .clone ( ) ) .unwrap_or_default ( ) ;
4349
if behind = = 0 {
4350
break ;
4351
}
4352
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
4353
}
4354
assert_eq! ( behind, 0 , " the index followed the landing" ) ;
5d54cf74 Andy McCall 20 days ago ↶
4355
assert_eq! ( oxygenhub:: diffusion:: forced_left( & h.app, game.id) , 0 , " the first write did fail" ) ;
0e04dfc2 Andy McCall 20 days ago ↶
4356
4357
4358
4359
let pins_of = | | async {
3b1bc27f Andy McCall 20 days ago
4360
sqlx:: query_scalar:: < _ , String> ( " SELECT heads FROM diffusion_pins WHERE consumer_id = $1" )
0e04dfc2 Andy McCall 20 days ago ↶
4361
.bind ( game.id)
4362
.fetch_one ( & h.app.db)
4363
.await
4364
.unwrap ( )
4365
} ;
4366
let right = pins_of ( ) .await;
3b1bc27f Andy McCall 20 days ago
4367
sqlx:: query( " UPDATE diffusion_pins SET heads = 'stale' WHERE consumer_id = $1" ) .bind ( game.id) .execute ( & h.app.db) .await.unwrap ( ) ;
4368
sqlx:: query( " DELETE FROM diffusion_indexed WHERE consumer_id = $1" ) .bind ( game.id) .execute ( & h.app.db) .await.unwrap ( ) ;
0e04dfc2 Andy McCall 20 days ago ↶
4369
let r = oxygenhub:: diffusion:: reconcile( & h.app) .await;
4370
assert! ( r.reindexed > = 1 , " {r:?}" ) ;
4371
assert_eq! ( pins_of ( ) .await, right, " the pin read again from the game's bookmark" ) ;
4372
assert_eq! ( oxygenhub:: diffusion:: reconcile( & h.app) .await.reindexed, 0 , " nothing left to mend" ) ;
757195ad Andy McCall 23 days ago ↶
4373
180dc43a Andy McCall 23 days ago ↶
4374
4375
4376
sqlx:: query( " UPDATE repositories SET visibility = 'private' WHERE name = 'api'" ) .execute ( & h.app.db) .await.unwrap ( ) ;
4377
seen.lock ( ) .unwrap ( ) .clear ( ) ;
4378
write ( api_work.path ( ) , " a6.c" , " v6\n " ) ;
4379
api.snapshot ( " t" ) .unwrap ( ) ;
4380
api.commit ( " api v6" , & [ ] ) .unwrap ( ) ;
4381
assert! ( api.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4fa0da12 Andy McCall 23 days ago ↶
4382
4383
4384
let mut bump_job = None ;
4385
for _ in 0 .. 100 {
4386
pass_all ( ) .await;
4387
bump_job = seen.lock ( ) .unwrap ( ) .iter ( ) .find ( | c | c[ " repo" ] = = " andy/game" & & c[ " bookmark" ] = = " diffusion/deps/api" ) .cloned ( ) ;
4388
if bump_job.is_some ( ) {
4389
break ;
4390
}
4391
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 100 ) ) .await;
4392
}
4393
let bump_job = bump_job.expect ( " the bump's job ran" ) ;
180dc43a Andy McCall 23 days ago ↶
4394
let subs: Vec< & str > = bump_job[ " subrepos" ] .as_array ( ) .unwrap ( ) .iter ( ) .filter_map ( | s | s[ " path" ] .as_str ( ) ) .collect ( ) ;
4395
assert_eq! ( subs, vec! [ " deps/api" ] , " {bump_job}" ) ;
4396
3776b70d Andy McCall 23 days ago ↶
4397
4398
4399
create ( & h.app, " andy" , " game2" ) .await;
4400
let g2_work = tempfile:: tempdir( ) .unwrap ( ) ;
4401
let g2 = client ( & format! ( " {} /andy/game2" , h.base) , g2_work.path ( ) ) ;
4402
g2.submodule_add ( & format! ( " {} /andy/api" , h.base) , " deps/api" , Some ( " main" ) ) .unwrap ( ) ;
4403
g2.snapshot ( " t" ) .unwrap ( ) ;
4404
g2.commit ( " game2 on api" , & [ ] ) .unwrap ( ) ;
4405
assert! ( g2.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4406
4407
write ( api_work.path ( ) , " a5.c" , " v5\n " ) ;
4408
api.snapshot ( " t" ) .unwrap ( ) ;
4409
api.commit ( " api v5" , & [ ] ) .unwrap ( ) ;
7e01a766 Andy McCall 20 days ago ↶
4410
let api_row = db:: repos:: get( & h.app.db, " andy" , " api" ) .await.unwrap ( ) .unwrap ( ) ;
4411
let runs_before = db:: flow:: runs( & h.app.db, api_row.id, 100 , None ) .await.unwrap ( ) .len ( ) ;
3776b70d Andy McCall 23 days ago ↶
4412
assert! ( api.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
7e01a766 Andy McCall 20 days ago ↶
4413
4414
4415
for _ in 0 .. 300 {
4416
if db:: flow:: runs( & h.app.db, api_row.id, 100 , None ) .await.unwrap ( ) .len ( ) > runs_before {
4417
break ;
4418
}
4419
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 20 ) ) .await;
4420
}
3776b70d Andy McCall 23 days ago ↶
4421
pass_all ( ) .await;
4422
let g2_row = db:: repos:: get( & h.app.db, " andy" , " game2" ) .await.unwrap ( ) .unwrap ( ) ;
4423
for _ in 0 .. 100 {
3b1bc27f Andy McCall 20 days ago
4424
if sqlx:: query_scalar:: < _ , i64 > ( " SELECT COUNT(*) FROM diffusion_pins WHERE consumer_id = $1" )
3776b70d Andy McCall 23 days ago ↶
4425
.bind ( g2_row.id)
4426
.fetch_one ( & h.app.db)
4427
.await
4428
.unwrap ( )
4429
> 0
4430
{
4431
break ;
4432
}
4433
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 30 ) ) .await;
4434
}
4435
assert! ( db:: proposals:: list( & h.app.db, g2_row.id, & [ ] ) .await.unwrap ( ) .is_empty ( ) , " off until turned on" ) ;
4436
let ( st, _ ) = call ( & h.app, " PUT" , " /api/v1/repos/andy/game2/diffusion" , Some ( " site" ) , Some ( r # "{"follow":"green"}"# .into ( ) ) ) .await;
4437
assert_eq! ( st, StatusCode:: OK ) ;
4438
let mut caught = None ;
5d54cf74 Andy McCall 20 days ago ↶
4439
4440
for _ in 0 .. 300 {
3776b70d Andy McCall 23 days ago ↶
4441
if let Some ( p) = db:: proposals:: list( & h.app.db, g2_row.id, & [ " open" ] ) .await.unwrap ( ) .into_iter ( ) .next ( ) {
4442
caught = Some ( p) ;
4443
break ;
4444
}
4445
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
4446
}
4447
let caught = caught.expect ( " bumped as soon as it was turned on" ) ;
4448
assert! ( caught.body.contains ( " api v5" ) & & caught.title.contains ( " 1 change" ) , " {:?} {:?}" , caught.title, caught.body) ;
0e04dfc2 Andy McCall 20 days ago ↶
4449
3b1bc27f Andy McCall 20 days ago
4450
sqlx:: query( " UPDATE proposals SET state = 'closed' WHERE id = $1" ) .bind ( caught.id) .execute ( & h.app.db) .await.unwrap ( ) ;
62cdf34d Andy McCall 20 days ago ↶
4451
4452
0e04dfc2 Andy McCall 20 days ago ↶
4453
let r = oxygenhub:: diffusion:: reconcile( & h.app) .await;
62cdf34d Andy McCall 20 days ago ↶
4454
assert_eq! ( r.caught_up, 0 , " the reconciler respects the closed bump: {r:?}" ) ;
a7e6e8e6 Andy McCall 23 days ago ↶
4455
}
4456
4457
4458
async fn jobs_left ( app : & Shared) -> bool {
4459
sqlx:: query_scalar:: < _ , i64 > ( " SELECT COUNT(*) FROM flow_jobs WHERE status IN ('queued', 'running')" )
4460
.fetch_one ( & app.db)
4461
.await
4462
.unwrap_or ( 0 )
4463
> 0
9aaa88b2 Andy McCall 23 days ago ↶
4464
}
4465
4466
4467
4468
# [ tokio ::test ( flavor = " multi_thread" ) ]
4469
async fn bumps_from_an_upstream_without_checks_are_marked_unverified ( ) {
4470
let h = hub ( ) .await;
4471
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
4472
create ( & h.app, " andy" , " lib" ) .await;
4473
create ( & h.app, " andy" , " app" ) .await;
4474
let lib_work = tempfile:: tempdir( ) .unwrap ( ) ;
4475
let lib = client ( & format! ( " {} /andy/lib" , h.base) , lib_work.path ( ) ) ;
4476
write ( lib_work.path ( ) , " lib.h" , " v1\n " ) ;
4477
lib.snapshot ( " t" ) .unwrap ( ) ;
4478
lib.commit ( " lib v1" , & [ ] ) .unwrap ( ) ;
4479
assert! ( lib.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4480
let app_work = tempfile:: tempdir( ) .unwrap ( ) ;
4481
let app = client ( & format! ( " {} /andy/app" , h.base) , app_work.path ( ) ) ;
4482
app.submodule_add ( & format! ( " {} /andy/lib" , h.base) , " deps/lib" , Some ( " main" ) ) .unwrap ( ) ;
c4e432da Andy McCall 23 days ago ↶
4483
4484
let modules = std:: fs:: read_to_string( app_work.path ( ) .join ( " .oxymodules" ) ) .unwrap ( ) ;
4485
write ( app_work.path ( ) , " .oxymodules" , & format! ( " {modules} watch = lib.c\n " ) ) ;
9aaa88b2 Andy McCall 23 days ago ↶
4486
app.snapshot ( " t" ) .unwrap ( ) ;
4487
app.commit ( " app on lib" , & [ ] ) .unwrap ( ) ;
4488
assert! ( app.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4489
let app_row = db:: repos:: get( & h.app.db, " andy" , " app" ) .await.unwrap ( ) .unwrap ( ) ;
4490
for _ in 0 .. 100 {
3b1bc27f Andy McCall 20 days ago
4491
if sqlx:: query_scalar:: < _ , i64 > ( " SELECT COUNT(*) FROM diffusion_pins WHERE consumer_id = $1" )
9aaa88b2 Andy McCall 23 days ago ↶
4492
.bind ( app_row.id)
4493
.fetch_one ( & h.app.db)
4494
.await
4495
.unwrap ( )
4496
> 0
4497
{
4498
break ;
4499
}
4500
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 30 ) ) .await;
4501
}
4502
let ( st, _ ) = call ( & h.app, " PUT" , " /api/v1/repos/andy/app/diffusion" , Some ( " site" ) , Some ( r # "{"follow":"green"}"# .into ( ) ) ) .await;
4503
assert_eq! ( st, StatusCode:: OK ) ;
4504
write ( lib_work.path ( ) , " lib.c" , " v2\n " ) ;
4505
lib.snapshot ( " t" ) .unwrap ( ) ;
4506
lib.commit ( " lib v2" , & [ ] ) .unwrap ( ) ;
4507
assert! ( lib.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4508
let mut found = None ;
4509
for _ in 0 .. 100 {
4510
if let Some ( p) = db:: proposals:: list( & h.app.db, app_row.id, & [ " open" ] ) .await.unwrap ( ) .into_iter ( ) .next ( ) {
4511
found = Some ( p) ;
4512
break ;
4513
}
4514
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
4515
}
4516
let p = found.expect ( " a bump although the upstream never goes green" ) ;
4517
assert! ( p.body.contains ( " Upstream unverified" ) , " {}" , p.body) ;
c4e432da Andy McCall 23 days ago ↶
4518
assert! ( p.body.contains ( " Touches what you watch" ) & & p.body.contains ( " touches `lib.c`" ) & & p.body.contains ( " ⚑" ) , " {}" , p.body) ;
9aaa88b2 Andy McCall 23 days ago ↶
4519
let mut labels = Vec :: new( ) ;
4520
for _ in 0 .. 100 {
4521
labels = db:: issues:: proposal_labels( & h.app.db, p.id) .await.unwrap ( ) .into_iter ( ) .map ( | l | l.name ) .collect:: < Vec< String> > ( ) ;
4522
if labels.len ( ) = = 2 {
4523
break ;
4524
}
4525
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 30 ) ) .await;
4526
}
4527
assert! ( labels.contains ( & " upstream unverified" .to_string ( ) ) & & labels.contains ( & " dependencies" .to_string ( ) ) , " {labels:?}" ) ;
a7e6e8e6 Andy McCall 23 days ago ↶
4528
}
d943160d Andy McCall 23 days ago ↶
4529
4530
4531
4532
# [ tokio ::test ( flavor = " multi_thread" ) ]
4533
async fn a_moved_target_runs_its_open_proposals_again ( ) {
4534
let h = hub ( ) .await;
4535
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
4536
create ( & h.app, " andy" , " demo" ) .await;
4537
let work = tempfile:: tempdir( ) .unwrap ( ) ;
4538
let mut c = client ( & format! ( " {} /andy/demo" , h.base) , work.path ( ) ) ;
4539
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
4540
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
4541
c.snapshot ( " t" ) .unwrap ( ) ;
4542
let base = c.commit ( " base" , & [ ] ) .unwrap ( ) ;
4543
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4544
4545
for ( b, file) in [ ( " fix-a" , " a.txt" ) , ( " fix-b" , " b.txt" ) ] {
4546
c.bookmark_create ( b, Some ( & [ base] ) ) .unwrap ( ) ;
4547
c.bookmark_switch ( b, true ) .unwrap ( ) ;
4548
write ( work.path ( ) , file, " x\n " ) ;
4549
c.snapshot ( " t" ) .unwrap ( ) ;
4550
c.commit ( & format! ( " add {file} " ) , & [ ] ) .unwrap ( ) ;
4551
assert! ( c.push_bookmark ( " origin" , b, false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4552
let body = serde_json:: json! ( { " title" : b, " target" : " main" , " source" : b } ) .to_string ( ) ;
4553
let ( st, out) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/proposals" , Some ( " site" ) , Some ( body) ) .await;
4554
assert_eq! ( st, StatusCode:: CREATED , " {out}" ) ;
4555
}
4556
let row = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
4557
let props = db:: proposals:: list( & h.app.db, row.id, & [ " open" ] ) .await.unwrap ( ) ;
4558
let a = props.iter ( ) .find ( | p | p.source_bookmark = = " fix-a" ) .unwrap ( ) .number;
4559
let b = props.iter ( ) .find ( | p | p.source_bookmark = = " fix-b" ) .unwrap ( ) .number;
4560
4561
let ( st, out) = call ( & h.app, " POST" , & format! ( " /api/v1/repos/andy/demo/proposals/{a} /land" ) , Some ( " site" ) , Some ( " {}" .into ( ) ) ) .await;
4562
assert_eq! ( st, StatusCode:: OK , " {out}" ) ;
4563
let label = format! ( " proposal/{b} " ) ;
4564
let mut found = None ;
4565
for _ in 0 .. 100 {
4566
if let Some ( r) = runs_of ( & h.app, " andy" , " demo" ) .await.into_iter ( ) .find ( | r | r.bookmark = = label ) {
4567
found = Some ( r) ;
4568
break ;
4569
}
4570
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
4571
}
4572
let run = found.expect ( " a run of fix-b's new projected state" ) ;
4573
let projected = db:: proposals:: get( & h.app.db, row.id, b) .await.unwrap ( ) .unwrap ( ) ;
4574
assert_eq! ( ( run.trigger.as_str ( ) , run.state.as_str ( ) ) , ( " proposal" , projected.projected_state.as_str ( ) ) , " on its projected state" ) ;
4575
4576
let before = runs_of ( & h.app, " andy" , " demo" ) .await.len ( ) ;
4577
oxygenhub:: web:: proposals:: settle_after_push( & h.app, & row, " main" , " andy" ) .await;
4578
assert_eq! ( runs_of ( & h.app, " andy" , " demo" ) .await.len ( ) , before, " not again for the same state" ) ;
4579
}
93501da2 Andy McCall 22 days ago ↶
4580
4581
4582
4583
async fn send_image ( app : & Shared, image : & [u8 ]) -> ( String, serde_json:: Value) {
4584
let mut chunks = std:: collections:: HashMap:: new( ) ;
4585
let m = jet:: images:: scan( image, None , | c , b | {
4586
chunks.insert ( c.hash ( ) .to_string ( ) , b.to_vec ( ) ) ;
4587
Ok ( ( ) )
4588
} )
4589
.unwrap ( ) ;
4590
for c in m.needed ( ) {
4591
let ( st, body) = put_bytes ( app, & format! ( " /api/v1/flow/images/chunks/{c} " ) , " site" , chunks[ c] .clone ( ) ) .await;
4592
assert_eq! ( st, StatusCode:: NO_CONTENT , " {body}" ) ;
4593
}
4594
let ( st, body) =
4595
call ( app, " PUT" , & format! ( " /api/v1/flow/images/{} " , m.hash) , Some ( " site" ) , Some ( serde_json:: to_string( & m) .unwrap ( ) ) ) .await;
4596
assert_eq! ( st, StatusCode:: CREATED , " {body}" ) ;
4597
( m.hash.clone ( ) , serde_json:: from_str( & body) .unwrap ( ) )
4598
}
4599
4600
4601
4602
4603
4604
# [ tokio ::test ( flavor = " multi_thread" ) ]
4605
async fn an_image_names_its_own_architecture_and_a_pass_teaches_the_rest ( ) {
4606
use oxy_core:: identity:: SecretKey;
4607
let h = hub ( ) .await;
4608
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
4609
4610
let dir = tempfile:: tempdir( ) .unwrap ( ) ;
4611
let tree = dir.path ( ) .join ( " tree" ) ;
4612
std:: fs:: create_dir_all( tree.join ( " bin" ) ) .unwrap ( ) ;
4613
let mut elf = vec! [ 0 u8 ; 64 ] ;
4614
elf[ .. 7 ] .copy_from_slice ( b " \x7f ELF\x02 \x01 \x01 " ) ;
4615
elf[ 18 .. 20 ] .copy_from_slice ( & 0x3E u16 .to_le_bytes ( ) ) ;
4616
std:: fs:: write( tree.join ( " bin/sh" ) , & elf) .unwrap ( ) ;
4617
let img = dir.path ( ) .join ( " rootfs.ext4" ) ;
4618
let made = std:: process:: Command:: new( " mkfs.ext4" ) .args ( [ " -q" , " -F" , " -d" ] ) .arg ( & tree) .arg ( & img) .arg ( " 8M" ) .status ( ) ;
4619
if ! made.is_ok_and ( | s | s.success ( ) ) {
4620
eprintln! ( " mkfs.ext4 not found: skipped" ) ;
4621
return ;
4622
}
4623
let ( x86, answer) = send_image ( & h.app, & std:: fs:: read( & img) .unwrap ( ) ) .await;
4624
assert_eq! ( answer[ " arch" ] , " x86_64" , " read from its shell: {answer}" ) ;
4625
let ( _ , body) = call ( & h.app, " GET" , & format! ( " /api/v1/flow/images/{x86} " ) , Some ( " runner-secret" ) , None ) .await;
4626
assert_eq! ( serde_json:: from_str:: < jet:: images:: Manifest> ( & body) .unwrap ( ) .arch.as_deref ( ) , Some ( " x86_64" ) , " runners fetch it named" ) ;
4627
4628
let blob: Vec< u8 > = ( 0 .. 300_000 u32 ) .map ( | i | ( i % 241 ) as u8 ) .collect ( ) ;
4629
let ( unread, answer) = send_image ( & h.app, & blob) .await;
4630
assert! ( answer[ " arch" ] .is_null ( ) , " {answer}" ) ;
4631
let archs = db:: flow:: image_archs( & h.app.db) .await.unwrap ( ) ;
4632
assert_eq! ( ( archs.get ( & x86) .map ( String :: as_str) , archs.get ( & unread) ) , ( Some ( " x86_64" ) , None ) ) ;
4633
4634
sqlx:: query( " UPDATE jet_images SET arch = ''" ) .execute ( & h.app.db) .await.unwrap ( ) ;
4635
sqlx:: query( " DELETE FROM flow_images" ) .execute ( & h.app.db) .await.unwrap ( ) ;
4636
assert_eq! ( oxygenhub:: images:: learn_archs( & h.app) .await.unwrap ( ) , 1 ) ;
4637
assert_eq! ( db:: flow:: image_archs( & h.app.db) .await.unwrap ( ) .get ( & x86) .map ( String :: as_str) , Some ( " x86_64" ) ) ;
4638
4639
4640
create ( & h.app, " andy" , " arch" ) .await;
4641
let work = tempfile:: tempdir( ) .unwrap ( ) ;
4642
let c = client ( & format! ( " {} /andy/arch" , h.base) , work.path ( ) ) ;
4643
let kdl = format! (
4644
" schema-version 1\n sandbox \" unread\" {{ \n rootfs \" cas://blob/{unread} \" \n }} \n sandbox \" x86\" {{ \n rootfs \" cas://blob/{x86} \" \n }} \n \
4645
job \" first\" {{ \n sandbox \" unread\" \n exec \" make\" \n }} \n job \" second\" {{ \n sandbox \" x86\" \n exec \" make\" \n depends-on \" first\" \n }} \n \
4646
job \" third\" {{ \n sandbox \" unread\" \n exec \" make\" \n depends-on \" first\" \n }} \n "
4647
) ;
4648
write ( work.path ( ) , " .flow/pipeline.kdl" , & kdl) ;
4649
c.commit ( " Jobs on two images" , & [ ] ) .unwrap ( ) ;
4650
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4651
runs_of ( & h.app, " andy" , " arch" ) .await;
4652
4653
4654
let x86_info = serde_json:: json! ( { " arch" : " x86_64" , " os" : " linux" , " version" : " 0.1.0" , " labels" : [ ] , " max_jobs" : 1 } ) ;
4655
let x86_claim =
4656
serde_json:: json! ( { " runner" : " earth" , " runtimes" : [ " microvm" ] , " images" : [ ] , " fetches_images" : true , " info" : x86_info } ) ;
4657
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( x86_claim.to_string ( ) ) ) .await;
4658
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
4659
let first: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
4660
assert_eq! ( first[ " job" ] [ " name" ] , " first" ) ;
4661
let ph = first[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) ;
4662
let id = first[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
4663
assert_eq! (
4664
finish ( & h.app, id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 , " receipt" : receipt ( " first" , ph, 0 ) } ) ) .await,
4665
StatusCode:: NO_CONTENT
4666
) ;
4667
assert_eq! ( db:: flow:: image_archs( & h.app.db) .await.unwrap ( ) .get ( & unread) .map ( String :: as_str) , Some ( " x86_64" ) , " learned" ) ;
4668
4669
4670
let token = " flowreg_narvi" ;
4671
db:: runners:: add_registration( & h.app.db, & blake3:: hash( token.as_bytes ( ) ) .to_hex ( ) , " admin" , & Default :: default( ) ) .await.unwrap ( ) ;
4672
let identity = SecretKey:: generate( ) .identity ( " narvi" , " " , 1 ) ;
4673
let info = serde_json:: json! ( { " arch" : " riscv64" , " os" : " linux" , " version" : " 0.1.0" , " labels" : [ ] , " max_jobs" : 1 } ) ;
4674
let reg = serde_json:: json! ( { " token" : token, " name" : " narvi" , " identity" : hex:: encode( identity.encode ( ) ) , " info" : info } ) ;
4675
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await;
4676
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
4677
let credential = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " credential" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
4678
let narvi =
4679
serde_json:: json! ( { " runner" : " narvi" , " runtimes" : [ " microvm" ] , " images" : [ ] , " fetches_images" : true , " info" : info } ) .to_string ( ) ;
4680
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( & credential) , Some ( narvi) ) .await;
4681
assert_eq! ( st, StatusCode:: NO_CONTENT , " the second job's image is x86_64 and the third's learned so: {body}" ) ;
4682
for _ in 0 .. 2 {
4683
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( x86_claim.to_string ( ) ) ) .await;
4684
assert_eq! ( st, StatusCode:: OK , " both go to x86_64: {body}" ) ;
4685
}
4686
}
f4d6ef41 Andy McCall 22 days ago ↶
4687
4688
4689
4690
4691
4692
# [ tokio ::test ( flavor = " multi_thread" ) ]
4693
async fn a_consumer_that_chose_auto_lands_green_bumps_itself ( ) {
4694
let h = hub ( ) .await;
4695
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
4696
create ( & h.app, " andy" , " api" ) .await;
4697
create ( & h.app, " andy" , " game" ) .await;
4698
let hub_app = & h.app;
4699
4700
let pass_all = move | | async move {
4701
for _ in 0 .. 20 {
4702
let Some ( a) = claim ( hub_app) .await else {
4703
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 30 ) ) .await;
4704
if ! jobs_left ( hub_app) .await {
4705
break ;
4706
}
4707
continue ;
4708
} ;
4709
let id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
4710
let ( name, ph) = ( a[ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) .to_string ( ) , a[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) .to_string ( ) ) ;
4711
assert_eq! (
4712
finish ( hub_app, id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 10 , " receipt" : receipt ( & name, & ph, 0 ) } ) ) .await,
4713
StatusCode:: NO_CONTENT
4714
) ;
4715
}
4716
} ;
4717
let api_work = tempfile:: tempdir( ) .unwrap ( ) ;
4718
let api = client ( & format! ( " {} /andy/api" , h.base) , api_work.path ( ) ) ;
4719
write ( api_work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
4720
write ( api_work.path ( ) , " test.sh" , " echo ok\n " ) ;
4721
api.snapshot ( " t" ) .unwrap ( ) ;
4722
api.commit ( " api v1" , & [ ] ) .unwrap ( ) ;
4723
assert! ( api.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4724
pass_all ( ) .await;
4725
let game_work = tempfile:: tempdir( ) .unwrap ( ) ;
4726
let game = client ( & format! ( " {} /andy/game" , h.base) , game_work.path ( ) ) ;
4727
game.submodule_add ( & format! ( " {} /andy/api" , h.base) , " deps/api" , Some ( " main" ) ) .unwrap ( ) ;
4728
write ( game_work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
4729
write ( game_work.path ( ) , " test.sh" , " echo ok\n " ) ;
4730
game.snapshot ( " t" ) .unwrap ( ) ;
4731
game.commit ( " game on api v1" , & [ ] ) .unwrap ( ) ;
4732
assert! ( game.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4733
pass_all ( ) .await;
4734
let game_row = db:: repos:: get( & h.app.db, " andy" , " game" ) .await.unwrap ( ) .unwrap ( ) ;
4735
for _ in 0 .. 100 {
3b1bc27f Andy McCall 20 days ago
4736
let n: i64 = sqlx:: query_scalar( " SELECT COUNT(*) FROM diffusion_pins WHERE consumer_id = $1" )
f4d6ef41 Andy McCall 22 days ago ↶
4737
.bind ( game_row.id)
4738
.fetch_one ( & h.app.db)
4739
.await
4740
.unwrap ( ) ;
4741
if n > 0 {
4742
break ;
4743
}
4744
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 30 ) ) .await;
4745
}
4746
let ( st, body) =
4747
call ( & h.app, " PUT" , " /api/v1/repos/andy/game/diffusion" , Some ( " site" ) , Some ( r # "{"follow":"green","land":"auto"}"# .into ( ) ) ) .await;
4748
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
4749
assert! ( body.contains ( r # ""land":"auto""# ) , " {body}" ) ;
4750
let ( st, _ ) =
4751
call ( & h.app, " PUT" , " /api/v1/repos/andy/game/diffusion" , Some ( " site" ) , Some ( r # "{"follow":"green","land":"soon"}"# .into ( ) ) ) .await;
4752
assert_eq! ( st, StatusCode:: BAD_REQUEST ) ;
4753
4754
4755
4756
write ( api_work.path ( ) , " a2.c" , " v2\n " ) ;
4757
api.snapshot ( " t" ) .unwrap ( ) ;
4758
api.commit ( " api v2: faster sprites" , & [ ] ) .unwrap ( ) ;
4759
assert! ( api.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4760
let mut landed = None ;
4761
for _ in 0 .. 100 {
4762
pass_all ( ) .await;
4763
let all = db:: proposals:: list( & h.app.db, game_row.id, & [ ] ) .await.unwrap ( ) ;
4764
if let Some ( p) = all.into_iter ( ) .find ( | p | p.source_bookmark = = " diffusion/deps/api" & & p.state = = " landed" ) {
4765
landed = Some ( p) ;
4766
break ;
4767
}
4768
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
4769
}
4770
let p = landed.expect ( " landed by itself once both halves were green" ) ;
4771
assert! ( p.body.contains ( " lands its bumps itself" ) , " {}" , p.body) ;
a2ef53c5 Andy McCall 20 days ago ↶
4772
4773
let mut events = vec! [ ] ;
4774
for _ in 0 .. 100 {
4775
events = db:: proposals:: events( & h.app.db, p.id) .await.unwrap ( ) ;
4776
if events.iter ( ) .any ( | e | e.event = = " landed" ) {
4777
break ;
4778
}
4779
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 20 ) ) .await;
4780
}
f4d6ef41 Andy McCall 22 days ago ↶
4781
let landing = events.iter ( ) .find ( | e | e.event = = " landed" ) .expect ( " a landing on the timeline" ) ;
4782
assert_eq! ( landing.actor, " diffusion" ) ;
4783
assert! ( landing.detail.contains ( " Diffusion, both halves green" ) , " {}" , landing.detail) ;
4784
4785
let mut behind = serde_json:: Value:: Null;
4786
for _ in 0 .. 100 {
4787
let ( _ , list) = call ( & h.app, " GET" , " /api/v1/repos/andy/api/consumers" , Some ( " site" ) , None ) .await;
4788
let list: Vec< serde_json:: Value> = serde_json:: from_str( & list) .unwrap ( ) ;
4789
behind = list.iter ( ) .find ( | c | c[ " name" ] = = " game" ) .map ( | c | c[ " behind" ] .clone ( ) ) .unwrap_or_default ( ) ;
4790
if behind = = 0 {
4791
break ;
4792
}
4793
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
4794
}
4795
assert_eq! ( behind, 0 , " main moved to the bump" ) ;
4796
4797
4798
4799
create ( & h.app, " andy" , " lib" ) .await;
4800
create ( & h.app, " andy" , " app" ) .await;
4801
let lib_work = tempfile:: tempdir( ) .unwrap ( ) ;
4802
let lib = client ( & format! ( " {} /andy/lib" , h.base) , lib_work.path ( ) ) ;
4803
write ( lib_work.path ( ) , " lib.h" , " v1\n " ) ;
4804
lib.snapshot ( " t" ) .unwrap ( ) ;
4805
lib.commit ( " lib v1" , & [ ] ) .unwrap ( ) ;
4806
assert! ( lib.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4807
let app_work = tempfile:: tempdir( ) .unwrap ( ) ;
4808
let app = client ( & format! ( " {} /andy/app" , h.base) , app_work.path ( ) ) ;
4809
app.submodule_add ( & format! ( " {} /andy/lib" , h.base) , " deps/lib" , Some ( " main" ) ) .unwrap ( ) ;
4810
let modules = std:: fs:: read_to_string( app_work.path ( ) .join ( " .oxymodules" ) ) .unwrap ( ) ;
4811
write ( app_work.path ( ) , " .oxymodules" , & format! ( " {modules} follow = green\n land = auto\n " ) ) ;
4812
write ( app_work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
4813
write ( app_work.path ( ) , " test.sh" , " echo ok\n " ) ;
4814
app.snapshot ( " t" ) .unwrap ( ) ;
4815
app.commit ( " app on lib" , & [ ] ) .unwrap ( ) ;
4816
assert! ( app.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4817
pass_all ( ) .await;
4818
let app_row = db:: repos:: get( & h.app.db, " andy" , " app" ) .await.unwrap ( ) .unwrap ( ) ;
4819
for _ in 0 .. 100 {
3b1bc27f Andy McCall 20 days ago
4820
let n: i64 = sqlx:: query_scalar( " SELECT COUNT(*) FROM diffusion_pins WHERE consumer_id = $1" )
f4d6ef41 Andy McCall 22 days ago ↶
4821
.bind ( app_row.id)
4822
.fetch_one ( & h.app.db)
4823
.await
4824
.unwrap ( ) ;
4825
if n > 0 {
4826
break ;
4827
}
4828
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 30 ) ) .await;
4829
}
4830
write ( lib_work.path ( ) , " lib.c" , " v2\n " ) ;
4831
lib.snapshot ( " t" ) .unwrap ( ) ;
4832
lib.commit ( " lib v2" , & [ ] ) .unwrap ( ) ;
4833
assert! ( lib.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4834
let mut passed = false ;
4835
for _ in 0 .. 100 {
4836
pass_all ( ) .await;
4837
let s: Option< String> = sqlx:: query_scalar(
3b1bc27f Andy McCall 20 days ago
4838
" SELECT status FROM flow_runs WHERE repo_id = $1 AND bookmark = 'diffusion/deps/lib' ORDER BY id DESC LIMIT 1" ,
f4d6ef41 Andy McCall 22 days ago ↶
4839
)
4840
.bind ( app_row.id)
4841
.fetch_optional ( & h.app.db)
4842
.await
4843
.unwrap ( ) ;
4844
if s.as_deref ( ) = = Some ( " passed" ) {
4845
passed = true ;
4846
break ;
4847
}
4848
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
4849
}
4850
assert! ( passed, " the app's run on the bump passed" ) ;
4851
assert_eq! ( oxygenhub:: diffusion:: auto_land( & h.app, & app_row, " deps/lib" ) .await, Err ( oxygenhub:: diffusion:: Held:: Unverified) ) ;
4852
let open = db:: proposals:: list( & h.app.db, app_row.id, & [ " open" ] ) .await.unwrap ( ) ;
4853
assert! ( open.iter ( ) .any ( | p | p.source_bookmark = = " diffusion/deps/lib" ) , " amber waits for a person" ) ;
4854
}
e8fa5369 Andy McCall 22 days ago ↶
4855
4856
4857
4858
4859
4860
# [ tokio ::test ( flavor = " multi_thread" ) ]
4861
async fn one_job_reruns_with_what_depends_on_it ( ) {
4862
let h = hub ( ) .await;
4863
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
4864
create ( & h.app, " andy" , " flaky" ) .await;
4865
let work = tempfile:: tempdir( ) .unwrap ( ) ;
4866
let c = client ( & format! ( " {} /andy/flaky" , h.base) , work.path ( ) ) ;
4867
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
4868
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
4869
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
4870
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4871
let run = runs_of ( & h.app, " andy" , " flaky" ) .await[ 0 ] .clone ( ) ;
4872
4873
let ( st, _ ) =
4874
call ( & h.app, " POST" , & format! ( " /api/v1/repos/andy/flaky/flow/runs/{} /jobs/test/rerun" , run.number) , Some ( " site" ) , None ) .await;
4875
assert_eq! ( st, StatusCode:: CONFLICT , " not while the run is live" ) ;
4876
while let Some ( j) = claim ( & h.app) .await {
4877
let id = j[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
4878
let ( name, ph) = ( j[ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) .to_string ( ) , j[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) .to_string ( ) ) ;
4879
let exit = if name = = " test" { 1 } else { 0 } ;
d9b3b846 Andy McCall 22 days ago ↶
4880
let log = serde_json:: json! ( { " records" : [
4881
{ " t" : 1 , " stream" : " out" , " text" : format! ( " {name} ran" ) } ,
4882
{ " t" : 2 , " stream" : " out" , " text" : " ::group::compile" } ,
4883
{ " t" : 5 , " stream" : " out" , " text" : " cc a.c" } ,
4884
{ " t" : 1502 , " stream" : " out" , " text" : " ::endgroup::" } ,
4885
] } ) ;
e8fa5369 Andy McCall 22 days ago ↶
4886
call ( & h.app, " POST" , & format! ( " /api/v1/flow/jobs/{id} /log" ) , Some ( " runner-secret" ) , Some ( log.to_string ( ) ) ) .await;
4887
assert_eq! (
4888
finish ( & h.app, id, serde_json:: json! ( { " exit_code" : exit, " duration_ms" : 10 , " receipt" : receipt ( & name, & ph, exit as u32 ) } ) )
4889
.await,
4890
StatusCode:: NO_CONTENT
4891
) ;
4892
}
d9b3b846 Andy McCall 22 days ago ↶
4893
4894
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/flaky/flow/{} " , run.number) , None , None ) .await;
4895
assert! ( page.contains ( r # "<details class="log-group" data-from="2">"# ) , " {page}" ) ;
4896
assert! ( page.contains ( r # "compile <span class="took">1.5s</span></summary>"# ) & & ! page.contains ( " ::endgroup::" ) , " {page}" ) ;
e8fa5369 Andy McCall 22 days ago ↶
4897
let status = | n : & 'static str | {
4898
let ( app, run_id) = ( h.app.clone ( ) , run.id) ;
4899
async move { db:: flow:: jobs( & app.db, run_id) .await.unwrap ( ) .into_iter ( ) .find ( | j | j.name = = n ) .unwrap ( ) }
4900
} ;
4901
assert_eq! ( db:: flow:: run_by_id( & h.app.db, run.id) .await.unwrap ( ) .unwrap ( ) .status, " failed" ) ;
4902
let failed_test = status ( " test" ) .await;
4903
4904
4905
let ( st, body) =
4906
call ( & h.app, " POST" , & format! ( " /api/v1/repos/andy/flaky/flow/runs/{} /jobs/test/rerun" , run.number) , Some ( " site" ) , None ) .await;
4907
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
4908
assert! ( body.contains ( r # ""queued":1"# ) , " {body}" ) ;
4909
assert_eq! (
4910
( status ( " build" ) .await.status, status ( " lint" ) .await.status, status ( " test" ) .await.status) ,
4911
( " passed" .into ( ) , " passed" .into ( ) , " queued" .into ( ) )
4912
) ;
4913
assert! ( db:: flow:: log( & h.app.db, failed_test.id, 0 , 100 ) .await.unwrap ( ) .is_empty ( ) , " its old log is gone" ) ;
4914
assert! ( db:: flow:: run_by_id( & h.app.db, run.id) .await.unwrap ( ) .unwrap ( ) .live ( ) ) ;
4915
let checks = checks_when ( & h.app, run.repo_id, & run.state, 3 ) .await;
4916
assert_eq! ( checks.iter ( ) .find ( | c | c.name = = " flow/test" ) .map ( | c | c.status.as_str ( ) ) , Some ( " queued" ) ) ;
4917
let j = claim ( & h.app) .await.expect ( " the test, queued again" ) ;
4918
assert_eq! ( j[ " job" ] [ " name" ] , " test" ) ;
4919
let ph = j[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
4920
let id = j[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
4921
assert_eq! (
4922
finish ( & h.app, id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 10 , " receipt" : receipt ( " test" , & ph, 0 ) } ) ) .await,
4923
StatusCode:: NO_CONTENT
4924
) ;
4925
assert_eq! ( db:: flow:: run_by_id( & h.app.db, run.id) .await.unwrap ( ) .unwrap ( ) .status, " passed" , " the run is whole again" ) ;
bff9106e Andy McCall 22 days ago ↶
4926
4927
let andy_id = db:: accounts:: account_by_name( & h.app.db, " andy" ) .await.unwrap ( ) .unwrap ( ) .id;
4928
let kinds: Vec< String> = db:: notify:: inbox( & h.app.db, andy_id, 20 ) .await.unwrap ( ) .into_iter ( ) .map ( | n | n.kind ) .collect ( ) ;
4929
assert! ( kinds.contains ( & " flow.failed" .to_string ( ) ) & & kinds.contains ( & " flow.fixed" .to_string ( ) ) , " {kinds:?}" ) ;
e8fa5369 Andy McCall 22 days ago ↶
4930
4931
4932
let ( _ , body) =
4933
call ( & h.app, " POST" , & format! ( " /api/v1/repos/andy/flaky/flow/runs/{} /jobs/build/rerun" , run.number) , Some ( " site" ) , None ) .await;
4934
assert! ( body.contains ( r # ""queued":2"# ) , " {body}" ) ;
4935
assert_eq! (
4936
( status ( " build" ) .await.status, status ( " lint" ) .await.status, status ( " test" ) .await.status) ,
4937
( " queued" .into ( ) , " passed" .into ( ) , " queued" .into ( ) )
4938
) ;
4939
let ( st, _ ) =
4940
call ( & h.app, " POST" , & format! ( " /api/v1/repos/andy/flaky/flow/runs/{} /jobs/deploy/rerun" , run.number) , Some ( " site" ) , None ) .await;
4941
assert_eq! ( st, StatusCode:: CONFLICT , " live again, and no such job either way" ) ;
4942
}
4a4400c5 Andy McCall 22 days ago ↶
4943
4944
4945
4946
4947
4948
# [ tokio ::test ( flavor = " multi_thread" ) ]
4949
async fn failed_runs_are_told_and_an_erroring_runner_is_drained ( ) {
4950
let h = hub ( ) .await;
4951
let andy = db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
4952
let bob = db:: accounts:: create_account( & h.app.db, " bob" , " [email protected] " , " x" , " salt" , false ) .await.unwrap ( ) ;
4953
create ( & h.app, " andy" , " alerts" ) .await;
4954
let row = db:: repos:: get( & h.app.db, " andy" , " alerts" ) .await.unwrap ( ) .unwrap ( ) ;
4955
db:: notify:: set_watch( & h.app.db, bob.id, row.id, true ) .await.unwrap ( ) ;
4956
let work = tempfile:: tempdir( ) .unwrap ( ) ;
4957
let c = client ( & format! ( " {} /andy/alerts" , h.base) , work.path ( ) ) ;
4958
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
4959
write ( work.path ( ) , " test.sh" , " exit 1\n " ) ;
4960
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
4961
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
4962
let run = runs_of ( & h.app, " andy" , " alerts" ) .await[ 0 ] .clone ( ) ;
3b1bc27f Andy McCall 20 days ago
4963
sqlx:: query( " UPDATE flow_runs SET actor = 'andy' WHERE id = $1" ) .bind ( run.id) .execute ( & h.app.db) .await.unwrap ( ) ;
4a4400c5 Andy McCall 22 days ago ↶
4964
while let Some ( j) = claim ( & h.app) .await {
4965
let id = j[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
4966
let ( name, ph) = ( j[ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) .to_string ( ) , j[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) .to_string ( ) ) ;
4967
let exit = if name = = " test" { 1 } else { 0 } ;
4968
finish ( & h.app, id, serde_json:: json! ( { " exit_code" : exit, " duration_ms" : 10 , " receipt" : receipt ( & name, & ph, exit as u32 ) } ) ) .await;
4969
}
4970
let told = | id : i64 | {
4971
let db = h.app.db.clone ( ) ;
4972
async move { db:: notify:: inbox( & db, id, 20 ) .await.unwrap ( ) .into_iter ( ) .filter ( | n | n.kind = = " flow.failed" ) .collect:: < Vec< _ > > ( ) }
4973
} ;
4974
let ( for_andy, for_bob) = ( told ( andy.id) .await, told ( bob.id) .await) ;
4975
assert_eq! ( ( for_andy.len ( ) , for_bob.len ( ) ) , ( 1 , 1 ) , " the pusher and a watcher" ) ;
4976
assert! ( for_andy[ 0 ] .title.contains ( & format! ( " run #{} on main failed (test)" , run.number) ) , " {}" , for_andy[ 0 ] .title) ;
4977
assert_eq! ( for_andy[ 0 ] .url, format! ( " /andy/alerts/flow/{} " , run.number) ) ;
4978
4979
4980
let runner_id = | | {
4981
let db = h.app.db.clone ( ) ;
4982
async move { sqlx:: query_scalar:: < _ , i64 > ( " SELECT id FROM flow_runners WHERE name = 'test-1'" ) .fetch_one ( & db) .await.unwrap ( ) }
4983
} ;
4984
let mut errors = 0 ;
4985
while errors < 5 {
4986
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/repos/andy/alerts/flow/runs" , Some ( " site" ) , Some ( " {}" .into ( ) ) ) .await;
4987
assert_eq! ( st, StatusCode:: CREATED ) ;
4988
while let Some ( j) = claim ( & h.app) .await {
4989
let id = j[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
4990
let st = finish (
4991
& h.app,
4992
id,
4993
serde_json:: json! ( { " exit_code" : null, " duration_ms" : 10 , " error" : " the microVM did not start: /dev/kvm busy" } ) ,
4994
)
4995
.await;
4996
assert_eq! ( st, StatusCode:: NO_CONTENT ) ;
4997
errors + = 1 ;
4998
if errors = = 5 {
4999
break ;
5000
}
5001
}
5002
}
5003
let state = | id : i64 | {
5004
let db = h.app.db.clone ( ) ;
5005
async move { db:: runners:: get( & db, id) .await.unwrap ( ) .unwrap ( ) .state }
5006
} ;
5007
let id = runner_id ( ) .await;
5008
assert_eq! ( state ( id) .await, " draining" ) ;
5009
let drained: Vec< _ > =
5010
db:: notify:: inbox( & h.app.db, andy.id, 50 ) .await.unwrap ( ) .into_iter ( ) .filter ( | n | n.kind = = " flow.runner_drained" ) .collect ( ) ;
5011
assert_eq! ( drained.len ( ) , 1 , " the administrator is told" ) ;
5012
assert! ( drained[ 0 ] .title.contains ( " /dev/kvm busy" ) , " {}" , drained[ 0 ] .title) ;
5013
assert! (
5014
! told ( bob.id) .await.is_empty ( )
5015
& & db:: notify:: inbox( & h.app.db, bob.id, 50 ) .await.unwrap ( ) .iter ( ) .all ( | n | n.kind ! = " flow.runner_drained" )
5016
) ;
5017
5018
db:: runners:: set_state( & h.app.db, id, " active" ) .await.unwrap ( ) ;
5019
assert_eq! ( db:: runners:: erroring( & h.app.db, id, 5 ) .await.unwrap ( ) , None ) ;
f7bcee49 Andy McCall 22 days ago ↶
5020
}
5021
5022
5023
5024
# [ tokio ::test ( flavor = " multi_thread" ) ]
5025
async fn a_landed_proposal_runs_its_targets_pipeline ( ) {
5026
let h = hub ( ) .await;
5027
create ( & h.app, " andy" , " demo" ) .await;
5028
let url = format! ( " {} /andy/demo" , h.base) ;
5029
let work = tempfile:: tempdir( ) .unwrap ( ) ;
5030
let mut c = client ( & url, work.path ( ) ) ;
5031
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
5032
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
5033
c.snapshot ( " test" ) .unwrap ( ) ;
5034
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
5035
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
5036
assert_eq! ( runs_of ( & h.app, " andy" , " demo" ) .await.len ( ) , 1 ) ;
5037
c.bookmark_create ( " feature" , None ) .unwrap ( ) ;
5038
c.bookmark_switch ( " feature" , false ) .unwrap ( ) ;
5039
write ( work.path ( ) , " b.txt" , " more\n " ) ;
5040
c.snapshot ( " test" ) .unwrap ( ) ;
5041
c.commit ( " More" , & [ ] ) .unwrap ( ) ;
5042
assert! ( c.push_bookmark ( " origin" , " feature" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
5043
let runs_before =
5044
db:: flow:: runs( & h.app.db, db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) .id, 10 , None ) .await.unwrap ( ) .len ( ) ;
5045
5046
let andy = db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
5047
let row = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
5048
db:: proposals:: create(
5049
& h.app.db,
5050
db:: proposals:: NewProposal {
5051
repo_id: row.id,
5052
title: " More" ,
5053
body: " " ,
5054
author_id: Some ( andy.id) ,
5055
author: " andy" ,
5056
draft: false ,
5057
target: " main" ,
5058
source_kind: " bookmark" ,
5059
source_repo_id: Some ( row.id) ,
5060
source_bookmark: " feature" ,
5061
source_changes: " " ,
5062
} ,
5063
)
5064
.await
5065
.unwrap ( ) ;
5066
let repo = RepoName:: parse( " andy/demo" ) .unwrap ( ) ;
2a8b6216 Andy McCall 18 days ago ↶
5067
let state = oxygenhub:: web:: proposals:: land_as( & h.app, & repo, & row, & andy, 1 , " web" , false , None ) .await.ok ( ) .expect ( " landed" ) ;
f7bcee49 Andy McCall 22 days ago ↶
5068
let mut runs = vec! [ ] ;
5069
for _ in 0 .. 100 {
5070
runs = db:: flow:: runs( & h.app.db, row.id, 10 , None ) .await.unwrap ( ) ;
5071
if runs.len ( ) > runs_before {
5072
break ;
5073
}
5074
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
5075
}
5076
let landed = runs.iter ( ) .find ( | r | r.bookmark = = " main" & & r.state = = state ) .expect ( " a run of the landed state on main" ) ;
5077
assert_eq! ( ( landed.trigger.as_str ( ) , landed.actor.as_str ( ) ) , ( " push" , " andy" ) ) ;
4a4400c5 Andy McCall 22 days ago ↶
5078
}
1870e4fa Andy McCall 20 days ago ↶
5079
5080
5081
5082
# [ tokio ::test ( flavor = " multi_thread" ) ]
5083
async fn owners_run_flow_on_their_own_runners_only ( ) {
5084
use oxy_core:: identity:: SecretKey;
5085
let h = hub ( ) .await;
5086
let register = | name : & 'static str | {
5087
let app = h.app.clone ( ) ;
5088
async move {
c8684d44 Andy McCall 20 days ago
5089
let body = format! ( " username={name} &email={name} %40example.com&password=correct-horse-battery&setup=test-setup" ) ;
1870e4fa Andy McCall 20 days ago ↶
5090
let res = oxygenhub:: router( app)
5091
.oneshot (
5092
Request:: post( " /register" )
5093
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
5094
.body ( Body:: from( body) )
5095
.unwrap ( ) ,
5096
)
5097
.await
5098
.unwrap ( ) ;
5099
res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( )
5100
}
5101
} ;
5102
register ( " andy" ) .await;
5103
let bob = register ( " bob" ) .await;
5104
create ( & h.app, " bob" , " game" ) .await;
5105
let web = | method : & 'static str , path : & str , body : Option< String> | {
5106
let ( app, cookie, path) = ( h.app.clone ( ) , bob.clone ( ) , path.to_string ( ) ) ;
5107
async move {
c7067c7b Andy McCall 16 days ago
5108
let mut req = Request:: builder( ) .method ( method) .uri ( path) .header ( header:: COOKIE , cookie.clone ( ) ) ;
1870e4fa Andy McCall 20 days ago ↶
5109
if body.is_some ( ) {
5110
req = req.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) ;
5111
}
c7067c7b Andy McCall 16 days ago
5112
let req = req.body ( body.map ( Body:: from) .unwrap_or_else ( Body:: empty) ) .unwrap ( ) ;
5113
let mut res = oxygenhub:: router( app.clone ( ) ) .oneshot ( req) .await.unwrap ( ) ;
5114
5115
let back =
5116
res.headers ( ) .get ( header:: LOCATION ) .and_then ( | v | v.to_str ( ) .ok ( ) ) .filter ( | l | l.contains ( " /settings/" ) ) .map ( String :: from) ;
5117
if let Some ( to) = back {
5118
let req = Request:: get( to) .header ( header:: COOKIE , cookie) .body ( Body:: empty( ) ) .unwrap ( ) ;
5119
res = oxygenhub:: router( app) .oneshot ( req) .await.unwrap ( ) ;
5120
}
1870e4fa Andy McCall 20 days ago ↶
5121
String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( )
5122
}
5123
} ;
5124
let csrf_of = | page : & str | {
5125
let i = page.find ( " name=\" _csrf\" value=\" " ) .unwrap ( ) + " name=\" _csrf\" value=\" " .len ( ) ;
5126
page[ i.. ] .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( )
5127
} ;
5128
c7067c7b Andy McCall 16 days ago
5129
let page = web ( " GET" , " /bob/game/settings/flow" , None ) .await;
1870e4fa Andy McCall 20 days ago ↶
5130
assert! ( page.contains ( " register a runner for this repository" ) & & ! page.contains ( " value=\" own\" " ) , " {page}" ) ;
5131
let page = web ( " POST" , " /bob/game/settings/flow" , Some ( format! ( " _csrf={} &flow=own" , csrf_of ( & page) ) ) ) .await;
5132
assert! ( page.contains ( " Register a runner for this repository" ) , " {page}" ) ;
5133
5134
let page = web ( " GET" , " /bob/game/settings" , None ) .await;
5135
let page = web ( " POST" , " /bob/game/settings/runners/token" , Some ( format! ( " _csrf={} " , csrf_of ( & page) ) ) ) .await;
5136
let i = page.find ( " --token " ) .unwrap ( ) + " --token " .len ( ) ;
5137
let token: String = page[ i.. ] .chars ( ) .take_while ( | c | ! c.is_whitespace ( ) ) .collect ( ) ;
5138
let info = serde_json:: json! ( { " arch" : " x86_64" , " os" : " linux" , " version" : " 0.1.0" , " labels" : [ ] , " max_jobs" : 1 } ) ;
5139
let identity = SecretKey:: generate( ) .identity ( " bobs" , " " , 1 ) ;
5140
let reg = serde_json:: json! ( { " token" : token, " name" : " bobs" , " identity" : hex:: encode( identity.encode ( ) ) , " info" : info } ) ;
5141
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await;
5142
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
5143
let credential = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " credential" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
5144
c7067c7b Andy McCall 16 days ago
5145
let page = web ( " GET" , " /bob/game/settings/flow" , None ) .await;
1870e4fa Andy McCall 20 days ago ↶
5146
assert! ( page.contains ( " value=\" own\" " ) & & ! page.contains ( " value=\" on\" " ) , " {page}" ) ;
5147
let page = web ( " POST" , " /bob/game/settings/flow" , Some ( format! ( " _csrf={} &flow=on" , csrf_of ( & page) ) ) ) .await;
5148
assert! ( page.contains ( " Only site administrators let the site's runners" ) , " {page}" ) ;
5149
let page = web ( " POST" , " /bob/game/settings/flow" , Some ( format! ( " _csrf={} &flow=own" , csrf_of ( & page) ) ) ) .await;
5150
assert! ( page.contains ( " value=\" own\" checked" ) & & page.contains ( " start runs" ) , " {page}" ) ;
5151
assert_eq! ( db:: repos:: get( & h.app.db, " bob" , " game" ) .await.unwrap ( ) .unwrap ( ) .flow, " own" ) ;
5152
5153
let work = tempfile:: tempdir( ) .unwrap ( ) ;
5154
let c = client ( & format! ( " {} /bob/game" , h.base) , work.path ( ) ) ;
5155
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
5156
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
5157
c.snapshot ( " t" ) .unwrap ( ) ;
5158
c.commit ( " A pipeline" , & [ ] ) .unwrap ( ) ;
5159
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
5160
assert_eq! ( runs_of ( & h.app, " bob" , " game" ) .await.len ( ) , 1 , " a run started" ) ;
5161
assert! ( claim ( & h.app) .await.is_none ( ) , " the site's runner is not offered it" ) ;
5162
let body = serde_json:: json! ( { " runner" : " ignored" , " runtimes" : [ " microvm" ] , " info" : info } ) .to_string ( ) ;
5163
let ( st, job) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( & credential) , Some ( body) ) .await;
5164
assert_eq! ( st, StatusCode:: OK , " {job}" ) ;
5165
assert_eq! ( serde_json:: from_str:: < serde_json:: Value> ( & job) .unwrap ( ) [ " repo" ] , " bob/game" ) ;
5166
}
cd437c4f Andy McCall 20 days ago ↶
5167
5168
# [ tokio ::test ( flavor = " multi_thread" ) ]
5169
async fn a_job_takes_the_artifact_its_producer_kept ( ) {
5170
let h = hub ( ) .await;
5171
create ( & h.app, " andy" , " demo" ) .await;
5172
let url = format! ( " {} /andy/demo" , h.base) ;
5173
let work = tempfile:: tempdir( ) .unwrap ( ) ;
5174
let c = client ( & url, work.path ( ) ) ;
5175
write (
5176
work.path ( ) ,
5177
" .flow/pipeline.kdl" ,
5178
& format! (
5179
" schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" plan\" {{ \n sandbox \" alpine\" \n exec \" plan\" \n output {{ \n artifact \" tfplan\" {{ \n path \" /src/out/tfplan\" \n }} \n }} \n }} \n job \" apply\" {{ \n sandbox \" alpine\" \n depends-on \" plan\" \n input {{ \n artifact \" tfplan\" from \" plan\" \n path \" /src/tfplan\" \n }} \n exec \" apply\" \n }} \n "
5180
) ,
5181
) ;
5182
c.snapshot ( " test" ) .unwrap ( ) ;
5183
c.commit ( " Plan, then apply that plan" , & [ ] ) .unwrap ( ) ;
5184
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
5185
5186
5187
let runs = runs_of ( & h.app, " andy" , " demo" ) .await;
5188
assert_eq! ( runs.len ( ) , 1 , " a run" ) ;
5189
let jobs = db:: flow:: jobs( & h.app.db, runs[ 0 ] .id) .await.unwrap ( ) ;
5190
assert! ( ! jobs.is_empty ( ) , " jobs: {:?}" , runs[ 0 ] ) ;
5191
let p = claim ( & h.app)
5192
.await
5193
.unwrap_or_else ( | | panic! ( " nothing to claim: {:?}" , jobs.iter ( ) .map ( | j | ( & j.name, & j.status, & j.error) ) .collect:: < Vec< _ > > ( ) ) ) ;
5194
assert_eq! ( p[ " job" ] [ " name" ] , " plan" ) ;
5195
let plan_id = p[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
5196
let plan = b " the reviewed plan" .to_vec ( ) ;
5197
let ( st, text) = upload ( & h.app, plan_id, " tfplan" , & plan) .await;
5198
assert_eq! ( st, StatusCode:: OK , " {text}" ) ;
5199
let blob = oxy_core:: repo:: Repo:: blob_ref( & plan) .to_hex ( ) ;
5200
let mut r = receipt ( " plan" , p[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) , 0 ) ;
5201
r[ " outputs" ] = serde_json:: json! ( [ blob] ) ;
5202
assert_eq! (
5203
finish ( & h.app, plan_id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 , " receipt" : r } ) ) .await,
5204
StatusCode:: NO_CONTENT
5205
) ;
5206
5207
5208
5209
let a = claim ( & h.app) .await.unwrap ( ) ;
5210
assert_eq! ( a[ " job" ] [ " name" ] , " apply" ) ;
5211
let apply_id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
5212
assert_eq! (
5213
a[ " inputs" ] ,
5214
serde_json:: json! ( [ { " artifact" : " tfplan" , " from" : " plan" , " path" : " /src/tfplan" , " blob" : blob } ] ) ,
5215
" the claim names the producer's blob"
5216
) ;
5217
let req = Request:: get( format! ( " /api/v1/flow/jobs/{apply_id} /inputs/0" ) ) .header ( header:: AUTHORIZATION , " Bearer runner-secret" ) ;
5218
let res = oxygenhub:: router( h.app.clone ( ) ) .oneshot ( req.body ( Body:: empty( ) ) .unwrap ( ) ) .await.unwrap ( ) ;
5219
assert_eq! ( res.status ( ) , StatusCode:: OK ) ;
5220
assert_eq! ( res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) .to_vec ( ) , plan) ;
5221
let ( st, _ ) = call ( & h.app, " GET" , & format! ( " /api/v1/flow/jobs/{apply_id} /inputs/1" ) , Some ( " runner-secret" ) , None ) .await;
5222
assert_eq! ( st, StatusCode:: NOT_FOUND , " no second input" ) ;
5223
let ( st, _ ) = call ( & h.app, " GET" , & format! ( " /api/v1/flow/jobs/{apply_id} /inputs/0" ) , None , None ) .await;
5224
assert_eq! ( st, StatusCode:: UNAUTHORIZED , " runners only" ) ;
5225
let ( st, _ ) = call ( & h.app, " GET" , & format! ( " /api/v1/flow/jobs/{plan_id} /inputs/0" ) , Some ( " runner-secret" ) , None ) .await;
5226
assert_eq! ( st, StatusCode:: CONFLICT , " only a running job's" ) ;
5227
}
417184c7 Andy McCall 20 days ago ↶
5228
5229
# [ tokio ::test ( flavor = " multi_thread" ) ]
5230
async fn a_job_waits_for_approval_and_is_approved_rejected_or_expires ( ) {
5231
let h = hub ( ) .await;
5232
let andy = db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
5233
let bob = db:: accounts:: create_account( & h.app.db, " bob" , " [email protected] " , " x" , " salt" , false ) .await.unwrap ( ) ;
5234
let carol = db:: accounts:: create_account( & h.app.db, " carol" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
5235
create ( & h.app, " andy" , " gated" ) .await;
5236
let row = db:: repos:: get( & h.app.db, " andy" , " gated" ) .await.unwrap ( ) .unwrap ( ) ;
5237
let work = tempfile:: tempdir( ) .unwrap ( ) ;
5238
let c = client ( & format! ( " {} /andy/gated" , h.base) , work.path ( ) ) ;
5239
write (
5240
work.path ( ) ,
5241
" .flow/pipeline.kdl" ,
5242
& format! (
47e37402 Andy McCall 20 days ago ↶
5243
" schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" plan\" {{ \n sandbox \" alpine\" \n exec \" plan\" \n }} \n job \" apply\" {{ \n sandbox \" alpine\" \n depends-on \" plan\" \n approval {{ \n by \" maintain\" \n others-only #true\n sealed #false\n }} \n environment \" prod\" \n identity \" https://sts.example\" {{ \n mount-path \" /run/secrets/cloud\" \n }} \n exec \" apply\" \n }} \n "
417184c7 Andy McCall 20 days ago ↶
5244
) ,
5245
) ;
5246
let mut n = 0 ;
5247
let mut push = | | {
5248
n + = 1 ;
5249
write ( work.path ( ) , " n.txt" , & format! ( " {n} \n " ) ) ;
5250
c.snapshot ( " test" ) .unwrap ( ) ;
5251
c.commit ( & format! ( " Push {n} " ) , & [ ] ) .unwrap ( ) ;
5252
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
5253
} ;
5254
5255
let plan_passes = | number : i64 | {
5256
let app = h.app.clone ( ) ;
5257
let row = row.clone ( ) ;
5258
async move {
5259
let mut run = None ;
5260
for _ in 0 .. 100 {
5261
if let Some ( r) = db:: flow:: run( & app.db, row.id, number) .await.unwrap ( ) {
5262
run = Some ( r) ;
5263
break ;
5264
}
5265
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 30 ) ) .await;
5266
}
5267
let run = run.expect ( " the run" ) ;
3b1bc27f Andy McCall 20 days ago
5268
sqlx:: query( " UPDATE flow_runs SET actor = 'andy' WHERE id = $1" ) .bind ( run.id) .execute ( & app.db) .await.unwrap ( ) ;
417184c7 Andy McCall 20 days ago ↶
5269
let j = claim ( & app) .await.expect ( " plan is claimed" ) ;
5270
assert_eq! ( j[ " job" ] [ " name" ] , " plan" ) ;
5271
let ( id, ph) = ( j[ " job_id" ] .as_i64 ( ) .unwrap ( ) , j[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) .to_string ( ) ) ;
5272
finish ( & app, id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 10 , " receipt" : receipt ( " plan" , & ph, 0 ) } ) ) .await;
5273
db:: flow:: run( & app.db, row.id, number) .await.unwrap ( ) .unwrap ( )
5274
}
5275
} ;
5276
5277
5278
push ( ) ;
5279
let run = runs_of ( & h.app, " andy" , " gated" ) .await[ 0 ] .clone ( ) ;
7e01a766 Andy McCall 20 days ago ↶
5280
33d98147 Andy McCall 16 days ago
5281
let heard = | topic : String| match h.app.bus.live.since ( 0 , & [ topic] ) {
7e01a766 Andy McCall 20 days ago ↶
5282
oxygenhub:: events:: Replay:: Events( e) => e.into_iter ( ) .map ( | e | e.kind ) .collect:: < Vec< _ > > ( ) ,
5283
oxygenhub:: events:: Replay:: Reload => vec! [ " reload" .into ( ) ] ,
5284
} ;
5285
assert! ( heard ( oxygenhub:: events:: topic:: repo( row.id) ) .contains ( & " push" .to_string ( ) ) , " the push is on the repository's topic" ) ;
3b1bc27f Andy McCall 20 days ago
5286
sqlx:: query( " UPDATE flow_runs SET actor = 'andy' WHERE id = $1" ) .bind ( run.id) .execute ( & h.app.db) .await.unwrap ( ) ;
05f1a85a Andy McCall 18 days ago ↶
5287
let early = oxygenhub:: flow:: decide( & h.app, & row, & run, " apply" , & carol, true , None ) .await.unwrap_err ( ) ;
417184c7 Andy McCall 20 days ago ↶
5288
assert! ( early.contains ( " waits for plan first" ) , " {early}" ) ;
5289
let run = plan_passes ( run.number) .await;
5290
assert! ( claim ( & h.app) .await.is_none ( ) , " apply waits for a person, so no runner gets it" ) ;
7e01a766 Andy McCall 20 days ago ↶
5291
assert! ( heard ( oxygenhub:: events:: topic:: run( run.id) ) .contains ( & " jobs" .to_string ( ) ) , " the run's topic hears its jobs" ) ;
dc59d1b5 Andy McCall 20 days ago ↶
5292
5293
5294
let told = | id : i64 | {
5295
let db = h.app.db.clone ( ) ;
5296
async move { db:: notify:: inbox( & db, id, 20 ) .await.unwrap ( ) .into_iter ( ) .filter ( | n | n.kind = = " flow.approval" ) .count ( ) }
5297
} ;
5298
assert_eq! ( ( told ( carol.id) .await, told ( andy.id) .await, told ( bob.id) .await) , ( 1 , 0 , 0 ) ) ;
417184c7 Andy McCall 20 days ago ↶
5299
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/gated/flow/{} " , run.number) , None , None ) .await;
5300
assert! ( page.contains ( " waits for approval by someone with the <em>maintain</em> role, other than andy" ) , " the run page says so" ) ;
b1860d60 Andy McCall 8 days ago ↶
5301
5302
let ( _ , v) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/gated/flow/runs/{} " , run.number) , None , None ) .await;
5303
let v: serde_json:: Value = serde_json:: from_str( & v) .unwrap ( ) ;
5304
let apply = v[ " jobs" ] .as_array ( ) .unwrap ( ) .iter ( ) .find ( | j | j[ " name" ] = = " apply" ) .unwrap ( ) .clone ( ) ;
5305
assert_eq! ( apply[ " approval" ] , " waiting" , " {v}" ) ;
5306
assert! ( v[ " jobs" ] .as_array ( ) .unwrap ( ) .iter ( ) .filter ( | j | j[ " name" ] ! = " apply" ) .all ( | j | j.get ( " approval" ) .is_none ( ) ) , " {v}" ) ;
05f1a85a Andy McCall 18 days ago ↶
5307
let own = oxygenhub:: flow:: decide( & h.app, & row, & run, " apply" , & andy, true , None ) .await.unwrap_err ( ) ;
417184c7 Andy McCall 20 days ago ↶
5308
assert! ( own.contains ( " someone other than andy" ) , " {own}" ) ;
05f1a85a Andy McCall 18 days ago ↶
5309
let no_role = oxygenhub:: flow:: decide( & h.app, & row, & run, " apply" , & bob, true , None ) .await.unwrap_err ( ) ;
417184c7 Andy McCall 20 days ago ↶
5310
assert! ( no_role.contains ( " takes the maintain role" ) , " {no_role}" ) ;
05f1a85a Andy McCall 18 days ago ↶
5311
assert_eq! ( oxygenhub:: flow:: decide( & h.app, & row, & run, " apply" , & carol, true , None ) .await.unwrap ( ) , " apply: approved by carol" ) ;
dc59d1b5 Andy McCall 20 days ago ↶
5312
assert_eq! ( told ( carol.id) .await, 1 , " told once" ) ;
417184c7 Andy McCall 20 days ago ↶
5313
let j = claim ( & h.app) .await.expect ( " approved, it runs" ) ;
5314
assert_eq! ( j[ " job" ] [ " name" ] , " apply" ) ;
b644e175 Andy McCall 20 days ago ↶
5315
5316
let token =
5317
j[ " secrets" ] .as_array ( ) .unwrap ( ) .iter ( ) .find ( | s | s[ " path" ] = = " /run/secrets/cloud" ) .unwrap ( ) [ " value" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
5318
let claims = h.app.oidc ( ) .await.unwrap ( ) .verify ( & token, " https://sts.example" , db:: now_ms( ) ) .expect ( " a valid token" ) ;
5319
assert_eq! ( claims[ " environment" ] , " prod" ) ;
417184c7 Andy McCall 20 days ago ↶
5320
let ( id, ph) = ( j[ " job_id" ] .as_i64 ( ) .unwrap ( ) , j[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) .to_string ( ) ) ;
5321
finish ( & h.app, id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 10 , " receipt" : receipt ( " apply" , & ph, 0 ) } ) ) .await;
5322
let audit = db:: accounts:: audit_rows( & h.app.db, 50 ) .await.unwrap ( ) ;
5323
assert! ( audit.iter ( ) .any ( | r | r.action = = " flow.approved" & & r.actor = = " carol" & & r.target.ends_with ( " #1 apply" ) ) , " audited" ) ;
5324
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/gated/flow/{} " , run.number) , None , None ) .await;
5325
assert! ( page.contains ( " approved by carol" ) , " the run says who" ) ;
b644e175 Andy McCall 20 days ago ↶
5326
let ( _ , page) = call ( & h.app, " GET" , " /andy/gated/deployments" , None , None ) .await;
5327
assert! ( page.contains ( " <strong>prod</strong>" ) & & page.contains ( " >#1</a>" ) & & page.contains ( " <td>carol</td>" ) , " the Deployments page" ) ;
5328
let ( _ , list) = call ( & h.app, " GET" , " /api/v1/repos/andy/gated/deployments" , Some ( " site" ) , None ) .await;
5329
let list: serde_json:: Value = serde_json:: from_str( & list) .unwrap ( ) ;
5330
assert_eq! ( ( list[ 0 ] [ " environment" ] .as_str ( ) , list[ 0 ] [ " approved_by" ] .as_str ( ) ) , ( Some ( " prod" ) , Some ( " carol" ) ) ) ;
417184c7 Andy McCall 20 days ago ↶
5331
05f1a85a Andy McCall 18 days ago ↶
5332
5333
5334
417184c7 Andy McCall 20 days ago ↶
5335
push ( ) ;
5336
let run = plan_passes ( 2 ) .await;
05f1a85a Andy McCall 18 days ago ↶
5337
let ( secret, hash) = oxygenhub:: auth:: new_token( ) ;
5338
let tok = db:: accounts:: create_token( & h.app.db, carol.id, " claude-code" , & hash, " repo:write" , " " , None ) .await.unwrap ( ) ;
5339
db:: accounts:: set_token_agent( & h.app.db, carol.id, tok.id, true ) .await.unwrap ( ) ;
4a68f106 Andy McCall 18 days ago ↶
5340
5341
05f1a85a Andy McCall 18 days ago ↶
5342
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/gated/flow/runs/2/jobs/apply/reject" , Some ( & secret) , Some ( " {}" .into ( ) ) ) .await;
4a68f106 Andy McCall 18 days ago ↶
5343
assert_eq! ( st, StatusCode:: CONFLICT , " {body}" ) ;
5344
assert! ( body.contains ( " decided by a person here" ) & & body.contains ( " Settings › OxygenFlow" ) , " {body}" ) ;
5345
let gated = db:: repos:: get( & h.app.db, " andy" , " gated" ) .await.unwrap ( ) .unwrap ( ) ;
5346
db:: repos:: set_agents_decide_gates( & h.app.db, gated.id, true ) .await.unwrap ( ) ;
5347
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/gated/flow/runs/2/jobs/apply/reject" , Some ( & secret) , Some ( " {}" .into ( ) ) ) .await;
05f1a85a Andy McCall 18 days ago ↶
5348
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
5349
assert! ( body.contains ( " rejected by carol via claude-code" ) , " {body}" ) ;
417184c7 Andy McCall 20 days ago ↶
5350
let jobs = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) ;
5351
let apply = jobs.iter ( ) .find ( | j | j.name = = " apply" ) .unwrap ( ) ;
05f1a85a Andy McCall 18 days ago ↶
5352
assert_eq! ( ( apply.status.as_str ( ) , apply.error.as_str ( ) ) , ( " cancelled" , " rejected by carol via claude-code" ) ) ;
5353
assert_eq! ( ( apply.approved_by.as_str ( ) , apply.approved_via.as_str ( ) ) , ( " carol" , " claude-code" ) ) ;
5354
let audit = db:: accounts:: audit_rows( & h.app.db, 50 ) .await.unwrap ( ) ;
5355
assert! (
5356
audit.iter ( ) .any ( | r | r.action = = " flow.rejected" & & r.actor = = " carol" & & r.detail.contains ( " via claude-code" ) ) ,
5357
" audited with the agent"
5358
) ;
417184c7 Andy McCall 20 days ago ↶
5359
assert! ( claim ( & h.app) .await.is_none ( ) ) ;
5360
5361
5362
push ( ) ;
5363
let run = plan_passes ( 3 ) .await;
5364
assert_eq! ( oxygenhub:: flow:: expire_approvals( & h.app, db:: now_ms( ) + 3_600_000 ) .await.unwrap ( ) , 0 , " not yet" ) ;
5365
assert_eq! ( oxygenhub:: flow:: expire_approvals( & h.app, db:: now_ms( ) + 25 * 3_600_000 ) .await.unwrap ( ) , 1 ) ;
5366
let jobs = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) ;
5367
let apply = jobs.iter ( ) .find ( | j | j.name = = " apply" ) .unwrap ( ) ;
5368
assert_eq! ( ( apply.status.as_str ( ) , apply.error.as_str ( ) ) , ( " cancelled" , " no approval within 24h" ) ) ;
5369
}
67be8ac0 Andy McCall 20 days ago ↶
5370
5371
# [ tokio ::test ( flavor = " multi_thread" ) ]
5372
async fn live_streams_only_topics_the_viewer_may_hear_and_replays ( ) {
5373
let h = hub ( ) .await;
5374
create ( & h.app, " andy" , " open" ) .await;
5375
h.app.repos.create ( & RepoName:: parse( " andy/shut" ) .unwrap ( ) , None ) .unwrap ( ) ;
5376
db:: repos:: create( & h.app.db, " andy" , " shut" , " " , hub_api:: Visibility:: Private) .await.unwrap ( ) ;
5377
let open = db:: repos:: get( & h.app.db, " andy" , " open" ) .await.unwrap ( ) .unwrap ( ) ;
5378
let shut = db:: repos:: get( & h.app.db, " andy" , " shut" ) .await.unwrap ( ) .unwrap ( ) ;
33d98147 Andy McCall 16 days ago
5379
let live = & h.app.bus.live;
67be8ac0 Andy McCall 20 days ago ↶
5380
let first = live.publish ( & oxygenhub:: events:: topic:: repo( open.id) , " push" ) ;
5381
live.publish ( & oxygenhub:: events:: topic:: repo( shut.id) , " push" ) ;
5382
live.publish ( & oxygenhub:: events:: topic:: proposal( open.id, 3 ) , " review" ) ;
5383
5384
5385
let read = | path : String, last : Option< u64 > | {
5386
let app = h.app.clone ( ) ;
5387
async move {
5388
let mut req = Request:: get( & path) ;
5389
if let Some ( l) = last {
5390
req = req.header ( " last-event-id" , l.to_string ( ) ) ;
5391
}
5392
let res = oxygenhub:: router( app) .oneshot ( req.body ( Body:: empty( ) ) .unwrap ( ) ) .await.unwrap ( ) ;
5393
let status = res.status ( ) ;
5394
let mut body = res.into_body ( ) ;
5395
let mut text = String :: new( ) ;
5396
while let Ok ( Some ( Ok ( frame) ) ) = tokio:: time:: timeout( std:: time:: Duration:: from_millis( 300 ) , body.frame ( ) ) .await {
5397
if let Some ( d) = frame.data_ref ( ) {
5398
text.push_str ( & String :: from_utf8_lossy( d) ) ;
5399
}
5400
}
5401
( status, text)
5402
}
5403
} ;
5404
let topics = format! ( " /live?topics=repo:{} ,repo:{} /proposal:3" , open.id, open.id) ;
5405
5406
let ( st, text) = read ( topics.clone ( ) , Some ( first.id) ) .await;
5407
assert_eq! ( st, StatusCode:: OK ) ;
5408
assert! ( text.contains ( " event: change" ) & & text.contains ( & format! ( " \" topic\" :\" repo:{} /proposal:3\" " , open.id) ) , " {text}" ) ;
5409
assert! ( ! text.contains ( & format! ( " \" topic\" :\" repo:{} \" " , shut.id) ) , " never another topic: {text}" ) ;
5410
assert! ( ! text.contains ( & format! ( " id: {} \n " , first.id) ) , " not what was already seen: {text}" ) ;
5411
5412
let ( st, _ ) = read ( format! ( " /live?topics=repo:{} ,repo:{} " , open.id, shut.id) , None ) .await;
5413
assert_eq! ( st, StatusCode:: FORBIDDEN ) ;
5414
assert_eq! ( read ( " /live?topics=user:1" .into ( ) , None ) .await.0 , StatusCode:: FORBIDDEN ) ;
5415
assert_eq! ( read ( " /live?topics=nonsense" .into ( ) , None ) .await.0 , StatusCode:: BAD_REQUEST ) ;
5416
assert_eq! ( read ( " /live?topics=" .into ( ) , None ) .await.0 , StatusCode:: BAD_REQUEST ) ;
5417
5418
for _ in 0 .. oxygenhub:: events:: REPLAY {
5419
live.publish ( " repo:999999" , " push" ) ;
5420
}
5421
let ( _ , text) = read ( topics, Some ( first.id) ) .await;
5422
assert! ( text.contains ( " event: reload" ) , " {text}" ) ;
5423
}
47e37402 Andy McCall 20 days ago ↶
5424
5425
# [ tokio ::test ( flavor = " multi_thread" ) ]
5426
async fn a_sealed_approval_releases_exactly_the_reviewed_plan ( ) {
5427
use oxy_core:: approval:: { SignedApproval, Statement} ;
5428
use oxy_core:: hash:: ObjectRef;
5429
use oxy_core:: identity:: SecretKey;
5430
let h = hub ( ) .await;
5431
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
5432
let carol = db:: accounts:: create_account( & h.app.db, " carol" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
5433
let sk = SecretKey:: generate( ) ;
5434
let carol_id = sk.identity ( " Carol" , " [email protected] " , 1 ) ;
5435
db:: accounts:: link_identity( & h.app.db, carol.id, & carol_id.hash ( ) .to_hex ( ) , " Carol" , " [email protected] " , & carol_id.encode ( ) )
5436
.await
5437
.unwrap ( ) ;
5438
create ( & h.app, " andy" , " infra" ) .await;
5439
let row = db:: repos:: get( & h.app.db, " andy" , " infra" ) .await.unwrap ( ) .unwrap ( ) ;
5440
let work = tempfile:: tempdir( ) .unwrap ( ) ;
5441
let c = client ( & format! ( " {} /andy/infra" , h.base) , work.path ( ) ) ;
5442
write (
5443
work.path ( ) ,
5444
" .flow/pipeline.kdl" ,
5445
& format! (
5446
" schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" plan\" {{ \n sandbox \" alpine\" \n exec \" plan\" \n output {{ \n artifact \" tfplan\" {{ \n path \" /src/out/tfplan\" \n }} \n }} \n }} \n job \" apply\" {{ \n sandbox \" alpine\" \n depends-on \" plan\" \n environment \" prod\" \n approval\n input {{ \n artifact \" tfplan\" from \" plan\" \n path \" /src/tfplan\" \n }} \n identity \" https://sts.example\" {{ \n mount-path \" /run/secrets/cloud\" \n }} \n exec \" apply\" \n }} \n "
5447
) ,
5448
) ;
5449
c.snapshot ( " test" ) .unwrap ( ) ;
5450
c.commit ( " Plan, then apply" , & [ ] ) .unwrap ( ) ;
5451
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
5452
let run = runs_of ( & h.app, " andy" , " infra" ) .await[ 0 ] .clone ( ) ;
5453
let p = claim ( & h.app) .await.unwrap ( ) ;
5454
let plan_id = p[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
5455
let plan = b " the reviewed plan" .to_vec ( ) ;
5456
upload ( & h.app, plan_id, " tfplan" , & plan) .await;
5457
let mut r = receipt ( " plan" , p[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) , 0 ) ;
5458
r[ " outputs" ] = serde_json:: json! ( [ oxy_core:: repo:: Repo:: blob_ref( & plan) .to_hex ( ) ] ) ;
5459
finish ( & h.app, plan_id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 10 , " receipt" : r } ) ) .await;
5460
let run = db:: flow:: run( & h.app.db, row.id, run.number) .await.unwrap ( ) .unwrap ( ) ;
5461
5462
05f1a85a Andy McCall 18 days ago ↶
5463
let plain = oxygenhub:: flow:: decide( & h.app, & row, & run, " apply" , & carol, true , None ) .await.unwrap_err ( ) ;
47e37402 Andy McCall 20 days ago ↶
5464
assert! ( plain.contains ( & format! ( " oxy seal --flow {} /andy/infra/flow/{} --job apply" , h.base, run.number) ) , " {plain}" ) ;
5465
5466
let ( st, body) =
5467
call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/infra/flow/runs/{} /jobs/apply/approval" , run.number) , Some ( " site" ) , None ) .await;
5468
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
5469
let v: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
5470
assert_eq! ( v[ " environment" ] , " prod" ) ;
5471
assert_eq! ( v[ " inputs" ] , serde_json:: json! ( [ oxy_core:: repo:: Repo:: blob_ref( & plan) .to_hex ( ) ] ) , " the reviewed plan" ) ;
5472
let mut repo = [ 0 u8 ; 16 ] ;
5473
repo.copy_from_slice ( & hex:: decode( v[ " repo" ] .as_str ( ) .unwrap ( ) ) .unwrap ( ) ) ;
5474
let oref = | k : & str | ObjectRef:: from_hex( v[ k] .as_str ( ) .unwrap ( ) ) .unwrap ( ) ;
5475
let statement = Statement {
5476
repo,
5477
run: v[ " run" ] .as_str ( ) .unwrap ( ) .into ( ) ,
5478
job: " apply" .into ( ) ,
5479
job_hash: oref ( " job_hash" ) ,
5480
state: oref ( " state" ) ,
5481
inputs: v[ " inputs" ] .as_array ( ) .unwrap ( ) .iter ( ) .map ( | i | ObjectRef:: from_hex( i.as_str ( ) .unwrap ( ) ) .unwrap ( ) ) .collect ( ) ,
5482
environment: " prod" .into ( ) ,
5483
approve: true ,
5484
} ;
5485
let seal = | s : & Statement, key : & SecretKey, id : & oxy_core:: identity:: Identity| {
5486
let body = serde_json:: json! ( { " signed" : hex:: encode( SignedApproval:: sign( s, id, key) .encode ( ) ) } ) .to_string ( ) ;
5487
let app = h.app.clone ( ) ;
5488
let path = format! ( " /api/v1/repos/andy/infra/flow/runs/{} /jobs/apply/seal" , run.number) ;
5489
async move { call ( & app, " POST" , & path, Some ( " site" ) , Some ( body) ) .await }
5490
} ;
5491
5492
let other_plan = Statement { inputs: vec! [ ObjectRef( [ 9 ; 32 ] ) ] , .. statement.clone ( ) } ;
5493
let ( st, body) = seal ( & other_plan, & sk, & carol_id) .await;
5494
assert! ( st = = StatusCode:: CONFLICT & & body.contains ( " approves something else" ) , " {st} {body}" ) ;
5495
let agent = SecretKey:: generate( ) ;
5496
let agent_id = agent.identity ( " Claude Code" , " [email protected] " , 1 ) ;
5497
let ( st, body) = seal ( & statement, & agent, & agent_id) .await;
5498
assert! ( st = = StatusCode:: CONFLICT & & body.contains ( " not linked to an account" ) , " {st} {body}" ) ;
5499
assert! ( claim ( & h.app) .await.is_none ( ) , " still waiting" ) ;
5500
317d776c Andy McCall 18 days ago ↶
5501
5502
let forged = oxy_core:: identity:: Identity { previous: Some ( carol_id.hash ( ) ) , .. agent_id.clone ( ) } ;
5503
let ( st, body) = seal ( & statement, & agent, & forged) .await;
5504
assert! ( st = = StatusCode:: CONFLICT & & body.contains ( " not linked" ) , " {st} {body}" ) ;
5505
5506
5507
5508
5509
let renewed = carol_id.successor ( vec! [ ] , None , 2 ) ;
5510
let ( st, body) = seal ( & statement, & sk, & renewed) .await;
47e37402 Andy McCall 20 days ago ↶
5511
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
317d776c Andy McCall 18 days ago ↶
5512
assert! (
5513
db:: accounts:: identity_by_key( & h.app.db, & renewed.hash ( ) .to_hex ( ) ) .await.unwrap ( ) .is_some_and ( | l | l.account_id = = carol.id ) ,
5514
" linked now"
5515
) ;
47e37402 Andy McCall 20 days ago ↶
5516
assert! ( body.contains ( " approved by carol, sealed" ) , " {body}" ) ;
5517
let name = oxygenhub:: repos:: RepoName:: parse( " andy/infra" ) .unwrap ( ) ;
5518
let state = statement.state;
5519
let kept = h.app.repos.read ( & name, move | r, _ | Ok ( r.store ( ) .approvals_for ( & state) ? ) ) .await.unwrap ( ) ;
5520
assert_eq! ( kept.len ( ) , 1 , " the approval is in the repository" ) ;
5521
let j = claim ( & h.app) .await.expect ( " released" ) ;
5522
assert_eq! ( j[ " job" ] [ " name" ] , " apply" ) ;
5523
let token =
5524
j[ " secrets" ] .as_array ( ) .unwrap ( ) .iter ( ) .find ( | s | s[ " path" ] = = " /run/secrets/cloud" ) .unwrap ( ) [ " value" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
5525
let claims = h.app.oidc ( ) .await.unwrap ( ) .verify ( & token, " https://sts.example" , db:: now_ms( ) ) .unwrap ( ) ;
5526
assert_eq! ( ( claims[ " environment" ] .as_str ( ) , claims[ " approved_by" ] .as_str ( ) ) , ( Some ( " prod" ) , Some ( " carol" ) ) ) ;
5527
assert_eq! ( claims[ " approval" ] .as_str ( ) , Some ( kept[ 0 ] .to_hex ( ) .as_str ( ) ) ) ;
5528
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/infra/flow/{} " , run.number) , None , None ) .await;
5529
assert! ( page.contains ( " sealed with their key" ) , " the run page says so" ) ;
5530
}
55673c65 Andy McCall 19 days ago ↶
5531
5532
5533
5534
# [ tokio ::test ( flavor = " multi_thread" ) ]
5535
async fn repositories_take_turns_on_the_runners ( ) {
5536
let h = hub ( ) .await;
5537
for name in [ " busy" , " quiet" ] {
5538
create ( & h.app, " andy" , name) .await;
5539
let work = tempfile:: tempdir( ) .unwrap ( ) ;
5540
let c = client ( & format! ( " {} /andy/{name} " , h.base) , work.path ( ) ) ;
5541
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
5542
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
5543
c.snapshot ( " test" ) .unwrap ( ) ;
5544
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
5545
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
5546
assert_eq! ( runs_of ( & h.app, " andy" , name) .await.len ( ) , 1 ) ;
5547
}
5548
5549
5550
let order: Vec< String> = {
5551
let mut out = Vec :: new( ) ;
5552
for _ in 0 .. 3 {
5553
let c = claim ( & h.app) .await.expect ( " a job to claim" ) ;
5554
out.push ( c[ " repo" ] .as_str ( ) .unwrap ( ) .to_string ( ) ) ;
5555
}
5556
out
5557
} ;
5558
assert_eq! ( order, [ " andy/busy" , " andy/quiet" , " andy/busy" ] ) ;
5559
}
107eaba5 Andy McCall 19 days ago ↶
5560
53a0fb2e Andy McCall 19 days ago ↶
5561
5562
5563
5564
5565
# [ tokio ::test ( flavor = " multi_thread" ) ]
5566
async fn a_pipeline_lowers_its_jobs_and_expediting_still_wins ( ) {
5567
let h = hub ( ) .await;
5568
for ( name, extra) in [ ( " nightly" , " priority \" low\" \n " ) , ( " everyday" , " " ) ] {
5569
create ( & h.app, " andy" , name) .await;
5570
let work = tempfile:: tempdir( ) .unwrap ( ) ;
5571
let c = client ( & format! ( " {} /andy/{name} " , h.base) , work.path ( ) ) ;
5572
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( extra) ) ;
5573
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
5574
c.snapshot ( " test" ) .unwrap ( ) ;
5575
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
5576
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
5577
assert_eq! ( runs_of ( & h.app, " andy" , name) .await.len ( ) , 1 ) ;
5578
}
5579
let nightly = db:: repos:: get( & h.app.db, " andy" , " nightly" ) .await.unwrap ( ) .unwrap ( ) ;
5580
db:: repos:: set_flow_priority( & h.app.db, nightly.id, " high" ) .await.unwrap ( ) ;
5581
let q = db:: flow:: queue( & h.app.db) .await.unwrap ( ) ;
5582
assert! (
5583
q.iter ( ) .filter ( | e | e.name = = " nightly" ) .all ( | e | e.priority ( ) = = " low" & & e.priority_note = = " pipeline: low" ) ,
5584
" {:?}" ,
5585
q.iter ( ) .map ( | e | ( & e.name, e.priority ( ) , & e.priority_note) ) .collect:: < Vec< _ > > ( )
5586
) ;
5587
let repo_of = | c : serde_json:: Value| c[ " repo" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
5588
5589
5590
assert_eq! ( repo_of ( claim ( & h.app) .await.unwrap ( ) ) , " andy/everyday" ) ;
5591
assert_eq! ( repo_of ( claim ( & h.app) .await.unwrap ( ) ) , " andy/everyday" ) ;
5592
let run = & runs_of ( & h.app, " andy" , " nightly" ) .await[ 0 ] ;
5593
assert! ( db:: flow:: expedite( & h.app.db, run.id, " expedited by andy: needed now" ) .await.unwrap ( ) ) ;
5594
assert_eq! ( repo_of ( claim ( & h.app) .await.unwrap ( ) ) , " andy/nightly" , " an expedited run is urgent whatever its jobs say" ) ;
5595
}
5596
107eaba5 Andy McCall 19 days ago ↶
5597
5598
5599
# [ tokio ::test ( flavor = " multi_thread" ) ]
5600
async fn priority_then_turns_and_an_expedited_run_first ( ) {
5601
let h = hub ( ) .await;
5602
for name in [ " early" , " later" ] {
5603
create ( & h.app, " andy" , name) .await;
5604
let work = tempfile:: tempdir( ) .unwrap ( ) ;
5605
let c = client ( & format! ( " {} /andy/{name} " , h.base) , work.path ( ) ) ;
5606
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
5607
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
5608
c.snapshot ( " test" ) .unwrap ( ) ;
5609
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
5610
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
5611
assert_eq! ( runs_of ( & h.app, " andy" , name) .await.len ( ) , 1 ) ;
5612
}
5613
let early = db:: repos:: get( & h.app.db, " andy" , " early" ) .await.unwrap ( ) .unwrap ( ) ;
5614
db:: repos:: set_flow_priority( & h.app.db, early.id, " low" ) .await.unwrap ( ) ;
5615
5616
let repo_of = | c : serde_json:: Value| c[ " repo" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
5617
assert_eq! ( repo_of ( claim ( & h.app) .await.unwrap ( ) ) , " andy/later" ) ;
5618
assert_eq! ( repo_of ( claim ( & h.app) .await.unwrap ( ) ) , " andy/later" ) ;
5619
5620
let run = & runs_of ( & h.app, " andy" , " early" ) .await[ 0 ] ;
5621
assert! ( db:: flow:: expedite( & h.app.db, run.id, " expedited by andy: an incident" ) .await.unwrap ( ) ) ;
76f93eae Andy McCall 19 days ago ↶
5622
5623
assert_eq! ( db:: flow:: expedited_live( & h.app.db, " andy" ) .await.unwrap ( ) , 1 ) ;
5624
assert_eq! ( db:: flow:: expedited_live( & h.app.db, " someone-else" ) .await.unwrap ( ) , 0 ) ;
107eaba5 Andy McCall 19 days ago ↶
5625
assert_eq! ( repo_of ( claim ( & h.app) .await.unwrap ( ) ) , " andy/early" ) ;
5626
let q = db:: flow:: queue( & h.app.db) .await.unwrap ( ) ;
5627
assert! (
5628
q.iter ( ) .any ( | e | e.name = = " early" & & e.priority ( ) = = " urgent" & & e.priority_note.contains ( " an incident" ) ) ,
5629
" {:?}" ,
5630
q.iter ( ) .map ( | e | ( & e.name, e.priority ( ) ) ) .collect:: < Vec< _ > > ( )
5631
) ;
5632
5633
db:: accounts:: set_setting( & h.app.db, " flow_priority" , " high" ) .await.unwrap ( ) ;
5634
let q = db:: flow:: queue( & h.app.db) .await.unwrap ( ) ;
5635
assert! ( q.iter ( ) .filter ( | e | e.name = = " later" ) .all ( | e | e.priority ( ) = = " high" ) ) ;
5636
}
026a76d2 Andy McCall 19 days ago ↶
5637
5638
5639
5640
5641
5642
5643
# [ tokio ::test ( flavor = " multi_thread" ) ]
5644
async fn a_running_job_gets_a_short_certificate_for_a_fresh_key ( ) {
5645
use oxy_core:: cert:: { certify_challenge, SignedCertificate} ;
5646
use oxy_core:: identity:: { Scope, SecretKey} ;
5647
let h = hub ( ) .await;
5648
create ( & h.app, " andy" , " demo" ) .await;
5649
let work = tempfile:: tempdir( ) .unwrap ( ) ;
5650
let c = client ( & format! ( " {} /andy/demo" , h.base) , work.path ( ) ) ;
5651
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
5652
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
5653
c.snapshot ( " test" ) .unwrap ( ) ;
5654
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
5655
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
5656
assert_eq! ( runs_of ( & h.app, " andy" , " demo" ) .await.len ( ) , 1 ) ;
5657
let job = claim ( & h.app) .await.expect ( " a job" ) ;
5658
let token = job[ " token" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
5659
let job_id = job[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
5660
5661
let sk = SecretKey:: generate( ) ;
5662
let id = sk.identity ( " job" , " [email protected] " , 1 ) ;
5663
let ask = | credential : & str , signed_for : & str | {
5664
let proof = sk.sign_challenge ( & id.hash ( ) , & certify_challenge ( signed_for) ) .unwrap ( ) ;
5665
let body = serde_json:: json! ( {
5666
" identity" : oxy_core:: dsse:: base64( & id.encode ( ) ) ,
5667
" proof" : oxy_core:: dsse:: base64( & oxy_core:: cbor:: encode( & proof.to_value ( ) ) ) ,
5668
" scope" : " changes,refs" ,
5669
} ) ;
5670
let ( credential, app) = ( credential.to_string ( ) , h.app.clone ( ) ) ;
5671
async move { call ( & app, " POST" , " /api/v1/keyless/certify" , Some ( & credential) , Some ( body.to_string ( ) ) ) .await }
5672
} ;
5673
let ( st, body) = ask ( & token, & token) .await;
5674
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
5675
let v: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
5676
let sc = SignedCertificate:: decode( & oxy_core:: dsse:: base64_decode( v[ " certificate" ] .as_str ( ) .unwrap ( ) ) .unwrap ( ) ) .unwrap ( ) ;
5677
assert! ( sc.verify ( & h.app.hub_key.identity) .is_ok ( ) , " signed with the hub's key" ) ;
5678
assert_eq! ( sc.cert.key_id ( ) , id.hash ( ) ) ;
5679
assert_eq! ( sc.cert.subject, " repo:andy/demo:bookmark:main" ) ;
5680
assert_eq! ( sc.cert.claim ( " repository" ) , Some ( " andy/demo" ) ) ;
5681
assert! ( sc.cert.claim ( " job" ) .is_some ( ) & & sc.cert.claim ( " run" ) = = Some ( " 1" ) ) ;
5682
assert_eq! ( sc.cert.scope, Scope:: CHANGES | Scope:: REFS ) ;
5683
assert! ( sc.cert.expires - sc.cert.not_before < = 10 * 60_000 + 30_000 , " ten minutes" ) ;
5684
5685
5686
let ( st, body) = ask ( & token, " flowjob_someone_else" ) .await;
5687
assert_eq! ( st, StatusCode:: UNAUTHORIZED , " {body}" ) ;
5688
assert! ( body.contains ( " the proof does not show" ) , " {body}" ) ;
5689
5690
5691
5692
let issuer = h.app.oidc ( ) .await.unwrap ( ) ;
5693
let now = oxygenhub:: db:: now_ms( ) ;
5694
let aud = oxygenhub:: keyless:: audience( h.app.config.base_url ( ) ) ;
5695
let good = issuer.mint_for ( h.app.config.base_url ( ) , " agent:claude@andy" , & aud, now, 600 ) ;
5696
let ( st, body) = ask ( & good, & good) .await;
5697
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
5698
assert! ( body.contains ( " \" subject\" :\" agent:claude@andy\" " ) , " {body}" ) ;
5699
let other = issuer.mint_for ( h.app.config.base_url ( ) , " agent:claude@andy" , " sts.amazonaws.com" , now, 600 ) ;
5700
let ( st, _ ) = ask ( & other, & other) .await;
5701
assert_eq! ( st, StatusCode:: UNAUTHORIZED ) ;
5702
5703
5704
sqlx:: query( " UPDATE flow_jobs SET status = 'passed' WHERE id = $1" ) .bind ( job_id) .execute ( & h.app.db) .await.unwrap ( ) ;
5705
let ( st, body) = ask ( & token, & token) .await;
5706
assert_eq! ( st, StatusCode:: UNAUTHORIZED , " {body}" ) ;
5707
}
2c31915f Andy McCall 19 days ago ↶
5708
5709
5710
5711
5712
5713
# [ tokio ::test ( flavor = " multi_thread" ) ]
5714
async fn a_receipt_signed_by_a_certified_key_is_kept ( ) {
5715
use oxy_core:: cert:: { certify_challenge, SignedCertificate} ;
5716
use oxy_core:: hash:: { hash_object, ChangeHash, ObjectType} ;
5717
use oxy_core:: identity:: SecretKey;
5718
use oxy_core:: objects:: StateObject;
5719
use oxy_core:: receipt:: SignedReceipt;
5720
let h = hub ( ) .await;
5721
create ( & h.app, " andy" , " demo" ) .await;
5722
let url = format! ( " {} /andy/demo" , h.base) ;
5723
let work = tempfile:: tempdir( ) .unwrap ( ) ;
5724
let c = client ( & url, work.path ( ) ) ;
5725
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
5726
c.snapshot ( " test" ) .unwrap ( ) ;
5727
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
5728
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
5729
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
5730
let heads: Vec< ChangeHash> = run.head_list ( ) .into_iter ( ) .filter_map ( ChangeHash:: from_hex) .collect ( ) ;
5731
let state = hash_object ( ObjectType:: State, & StateObject:: new( heads.clone ( ) ) .encode ( ) ) ;
5732
let repo_id = h.app.repos.open ( & RepoName:: parse( " andy/demo" ) .unwrap ( ) ) .unwrap ( ) .id ( ) .unwrap ( ) ;
5733
5734
5735
5736
let certify = | token : String| {
5737
let app = h.app.clone ( ) ;
5738
async move {
5739
let sk = SecretKey:: generate( ) ;
5740
let id = sk.identity ( " test-1 job" , " " , 1 ) ;
5741
let proof = sk.sign_challenge ( & id.hash ( ) , & certify_challenge ( & token) ) .unwrap ( ) ;
5742
let body = serde_json:: json! ( {
5743
" identity" : oxy_core:: dsse:: base64( & id.encode ( ) ) ,
5744
" proof" : oxy_core:: dsse:: base64( & oxy_core:: cbor:: encode( & proof.to_value ( ) ) ) ,
5745
" scope" : " seal" ,
5746
} ) ;
5747
let ( st, body) = call ( & app, " POST" , " /api/v1/keyless/certify" , Some ( & token) , Some ( body.to_string ( ) ) ) .await;
5748
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
5749
let v: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
5750
let cert = SignedCertificate:: decode( & oxy_core:: dsse:: base64_decode( v[ " certificate" ] .as_str ( ) .unwrap ( ) ) .unwrap ( ) ) .unwrap ( ) ;
5751
( sk, id, cert)
5752
}
5753
} ;
5754
let sign = | r : & serde_json:: Value,
5755
sk: & SecretKey,
5756
id: & oxy_core:: identity:: Identity,
5757
cert: SignedCertificate,
5758
issuer: oxy_core:: identity:: Identity| {
5759
let r: flow_core:: Receipt = serde_json:: from_value( r.clone ( ) ) .unwrap ( ) ;
5760
let run_url = format! ( " {url} /flow/{} " , run.number) ;
5761
hex:: encode( SignedReceipt:: sign( r.to_oxy ( ) , repo_id, state, & run_url, id, sk) .with_certificate ( cert, issuer) .encode ( ) )
5762
} ;
5763
let hub_id = h.app.hub_key.identity.clone ( ) ;
5764
5765
let a = claim ( & h.app) .await.unwrap ( ) ;
5766
let b = claim ( & h.app) .await.unwrap ( ) ;
5767
let ph = a[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
5768
let ( a_name, b_name) = ( a[ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) .to_string ( ) , b[ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) .to_string ( ) ) ;
5769
let ( a_id, b_id) = ( a[ " job_id" ] .as_i64 ( ) .unwrap ( ) , b[ " job_id" ] .as_i64 ( ) .unwrap ( ) ) ;
5770
let ( sk_a, id_a, cert_a) = certify ( a[ " token" ] .as_str ( ) .unwrap ( ) .to_string ( ) ) .await;
5771
5772
5773
let rb = receipt ( & b_name, & ph, 0 ) ;
5774
let body = serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 300 , " receipt" : rb, " signed_receipt" : sign ( & rb, & sk_a, & id_a, cert_a.clone ( ) , hub_id.clone ( ) ) } ) ;
5775
assert_eq! ( finish ( & h.app, b_id, body) .await, StatusCode:: NO_CONTENT ) ;
5776
5777
let ra = receipt ( & a_name, & ph, 0 ) ;
5778
let body = serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 , " receipt" : ra, " signed_receipt" : sign ( & ra, & sk_a, & id_a, cert_a, hub_id.clone ( ) ) } ) ;
5779
assert_eq! ( finish ( & h.app, a_id, body) .await, StatusCode:: NO_CONTENT ) ;
5780
5781
let jobs = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) ;
5782
let job = | n : & str | jobs.iter ( ) .find ( | j | j.name = = n ) .unwrap ( ) .clone ( ) ;
5783
assert_eq! ( job ( & a_name) .receipt_attest.len ( ) , 64 , " {}" , job ( & a_name) .receipt_note) ;
5784
assert! ( job ( & a_name) .receipt_signer.ends_with ( " (short-lived key)" ) , " {}" , job ( & a_name) .receipt_signer) ;
5785
assert! ( job ( & b_name) .receipt_attest.is_empty ( ) ) ;
5786
assert! ( job ( & b_name) .receipt_note.contains ( " another run or job" ) , " {}" , job ( & b_name) .receipt_note) ;
5787
5788
let kept = h.app.repos.open ( & RepoName:: parse( " andy/demo" ) .unwrap ( ) ) .unwrap ( ) .receipts ( & heads) .unwrap ( ) ;
5789
assert_eq! ( kept.len ( ) , 1 ) ;
5790
assert! ( kept[ 0 ] .problem.is_none ( ) & & kept[ 0 ] .signed.certificate.is_some ( ) ) ;
5791
5792
5793
let t = claim ( & h.app) .await.unwrap ( ) ;
5794
let t_name = t[ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
5795
let fake_sk = SecretKey:: generate( ) ;
5796
let fake = fake_sk.identity ( " not the hub" , " " , 1 ) ;
5797
let ( sk_t , id_t , mut cert_t ) = certify ( t[ " token" ] .as_str ( ) .unwrap ( ) .to_string ( ) ) .await;
5798
cert_t .cert.issuer = fake.hash ( ) ;
5799
let cert_t = SignedCertificate:: sign( cert_t .cert, & fake, & fake_sk) ;
5800
let rt = receipt ( & t_name, & ph, 0 ) ;
5801
let body =
5802
serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 500 , " receipt" : rt, " signed_receipt" : sign ( & rt, & sk_t , & id_t , cert_t , fake) } ) ;
5803
assert_eq! ( finish ( & h.app, t[ " job_id" ] .as_i64 ( ) .unwrap ( ) , body) .await, StatusCode:: NO_CONTENT ) ;
5804
let jobs = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) ;
5805
let t_job = jobs.iter ( ) .find ( | j | j.name = = t_name ) .unwrap ( ) ;
5806
assert! ( t_job.receipt_note.contains ( " not issued by this hub" ) , " {}" , t_job.receipt_note) ;
5807
}
3eb491e9 Andy McCall 19 days ago ↶
5808
24472413 Andy McCall 19 days ago ↶
5809
5810
5811
5812
5813
# [ tokio ::test ( flavor = " multi_thread" ) ]
5814
async fn a_release_built_by_a_certified_key_is_attested ( ) {
5815
use oxy_core:: cert:: { certify_challenge, SignedCertificate} ;
5816
use oxy_core:: hash:: { hash_object, ChangeHash, ObjectType} ;
5817
use oxy_core:: identity:: SecretKey;
5818
use oxy_core:: objects:: StateObject;
5819
use oxy_core:: receipt:: SignedReceipt;
5820
let h = hub ( ) .await;
5821
create ( & h.app, " andy" , " demo" ) .await;
5822
let url = format! ( " {} /andy/demo" , h.base) ;
5823
let work = tempfile:: tempdir( ) .unwrap ( ) ;
5824
let mut c = client ( & url, work.path ( ) ) ;
5825
c.create_identity ( " Andy" , " [email protected] " , oxy_core:: repo:: IdentityScope:: Repo) .unwrap ( ) ;
5826
write (
5827
work.path ( ) ,
5828
" .flow/pipeline.kdl" ,
5829
& format! (
5830
" schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" build\" {{ \n sandbox \" alpine\" \n exec \" make\" \n output {{ \n artifact \" tool\" {{ \n path \" /src/out/tool\" \n }} \n }} \n }} \n "
5831
) ,
5832
) ;
5833
c.snapshot ( " test" ) .unwrap ( ) ;
5834
c.commit ( " Build a tool" , & [ ] ) .unwrap ( ) ;
5835
let heads = c.bookmark_heads ( ) .unwrap ( ) ;
5836
c.tag_create ( " v1" , & heads, " first" ) .unwrap ( ) ;
5837
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
5838
5839
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
5840
let a = claim ( & h.app) .await.unwrap ( ) ;
5841
let id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
5842
let token = a[ " token" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
5843
let body: Vec< u8 > = b " #!/bin/sh\n echo tool\n " .to_vec ( ) ;
5844
let ( _ , text) = upload ( & h.app, id, " tool" , & body) .await;
5845
let blob = serde_json:: from_str:: < serde_json:: Value> ( & text) .unwrap ( ) [ " blob" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
5846
let mut r = receipt ( " build" , a[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) , 0 ) ;
5847
r[ " outputs" ] = serde_json:: json! ( [ blob] ) ;
5848
let heads_run: Vec< ChangeHash> = run.head_list ( ) .into_iter ( ) .filter_map ( ChangeHash:: from_hex) .collect ( ) ;
5849
let state = hash_object ( ObjectType:: State, & StateObject:: new( heads_run) .encode ( ) ) ;
5850
let repo_id = h.app.repos.open ( & RepoName:: parse( " andy/demo" ) .unwrap ( ) ) .unwrap ( ) .id ( ) .unwrap ( ) ;
5851
5852
let sk = SecretKey:: generate( ) ;
5853
let key = sk.identity ( " test-1 job" , " " , 1 ) ;
5854
let proof = sk.sign_challenge ( & key.hash ( ) , & certify_challenge ( & token) ) .unwrap ( ) ;
5855
let ask = serde_json:: json! ( {
5856
" identity" : oxy_core:: dsse:: base64( & key.encode ( ) ) ,
5857
" proof" : oxy_core:: dsse:: base64( & oxy_core:: cbor:: encode( & proof.to_value ( ) ) ) ,
5858
" scope" : " seal" ,
5859
} ) ;
5860
let ( st, answer) = call ( & h.app, " POST" , " /api/v1/keyless/certify" , Some ( & token) , Some ( ask.to_string ( ) ) ) .await;
5861
assert_eq! ( st, StatusCode:: OK , " {answer}" ) ;
5862
let v: serde_json:: Value = serde_json:: from_str( & answer) .unwrap ( ) ;
5863
let cert = SignedCertificate:: decode( & oxy_core:: dsse:: base64_decode( v[ " certificate" ] .as_str ( ) .unwrap ( ) ) .unwrap ( ) ) .unwrap ( ) ;
5864
let fr: flow_core:: Receipt = serde_json:: from_value( r.clone ( ) ) .unwrap ( ) ;
5865
let signed = hex:: encode(
5866
SignedReceipt:: sign( fr.to_oxy ( ) , repo_id, state, & format! ( " {url} /flow/{} " , run.number) , & key, & sk)
5867
.with_certificate ( cert, h.app.hub_key.identity.clone ( ) )
5868
.encode ( ) ,
5869
) ;
5870
let done = serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 900 , " receipt" : r, " signed_receipt" : signed } ) ;
5871
assert_eq! ( finish ( & h.app, id, done) .await, StatusCode:: NO_CONTENT ) ;
5872
5873
let row = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
5874
let rel = db:: releases:: create( & h.app.db, row.id, " v1" , " One" , " " , false , None ) .await.unwrap ( ) ;
5875
oxygenhub:: web:: releases:: store_asset( & h.app, rel.id, " tool" , " " , & body) .await.unwrap ( ) ;
5876
let ( st, page) = call ( & h.app, " GET" , " /andy/demo/releases/trust/v1" , None , None ) .await;
5877
assert_eq! ( st, StatusCode:: OK ) ;
5878
for want in [
5879
" tool: signed receipt from run #1 build" ,
5880
" a short-lived key this hub certified for repo:andy/demo:bookmark:main" ,
5881
" 1 of 1 assets built and attested by OxygenFlow" ,
5882
] {
5883
assert! ( page.contains ( want) , " missing {want:?}" ) ;
5884
}
5885
assert! ( ! page.contains ( " not one this hub" ) , " not an unknown key" ) ;
5886
}
5887
965b950f Andy McCall 18 days ago ↶
5888
5889
5890
5891
5892
# [ tokio ::test ( flavor = " multi_thread" ) ]
5893
async fn a_push_is_logged_and_the_log_checks_out ( ) {
5894
use oxy_core:: tlog:: { self , Checkpoint, Entry, Hash, Note, Tile} ;
5895
let h = hub ( ) .await;
5896
create ( & h.app, " andy" , " demo" ) .await;
5897
let url = format! ( " {} /andy/demo" , h.base) ;
5898
let work = tempfile:: tempdir( ) .unwrap ( ) ;
5899
let mut c = client ( & url, work.path ( ) ) ;
5900
c.create_identity ( " Andy" , " [email protected] " , oxy_core:: repo:: IdentityScope:: Repo) .unwrap ( ) ;
5901
write ( work.path ( ) , " a.txt" , " a\n " ) ;
5902
c.commit ( " a" , & [ ] ) .unwrap ( ) ;
5903
c.policy_init ( ) .unwrap ( ) ;
5904
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
5905
let host = h.app.repos.open ( & RepoName:: parse( " andy/demo" ) .unwrap ( ) ) .unwrap ( ) ;
5906
let ( attest, _ ) = host.latest_ref_attest ( " main" ) .unwrap ( ) .unwrap ( ) ;
5907
5908
5909
let ( _ , sk) = oxygenhub:: keys:: load_secret( & h.app.config.keys_dir ( ) ) .unwrap ( ) ;
5910
h.app.tlog.refresh_checkpoint ( & h.app.hub_key.identity, & sk) .unwrap ( ) ;
5911
let ( st, note) = call ( & h.app, " GET" , " /log/checkpoint" , None , None ) .await;
5912
assert_eq! ( st, StatusCode:: OK , " {note}" ) ;
5913
let origin = oxygenhub:: tlog:: origin_of( h.app.config.base_url ( ) ) ;
5914
let cp = Checkpoint:: verify( & Note:: parse( & note) .unwrap ( ) , & origin, & h.app.hub_key.identity) .unwrap ( ) ;
5915
assert! ( cp.size > = 3 , " ref, seal and policy: {}" , cp.size) ;
5916
5917
5918
let bundle_path = Tile { level: 0 , n: 0 , width: cp.size as u32 } .entries_path ( ) ;
5919
let ( st, _ ) = call ( & h.app, " GET" , & format! ( " /log/{bundle_path} " ) , None , None ) .await;
5920
assert_eq! ( st, StatusCode:: OK ) ;
5921
let res =
5922
oxygenhub:: router( h.app.clone ( ) ) .oneshot ( Request:: get( format! ( " /log/{bundle_path} " ) ) .body ( Body:: empty( ) ) .unwrap ( ) ) .await.unwrap ( ) ;
5923
assert_eq! ( res.headers ( ) [ header:: CACHE_CONTROL ] , " public, max-age=5" , " a partial tile changes" ) ;
5924
let bytes = res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ;
5925
let entries: Vec< Entry> = tlog:: decode_bundle( & bytes) .unwrap ( ) .iter ( ) .map ( | b | Entry:: decode( b) .unwrap ( ) ) .collect ( ) ;
5926
let kinds: Vec< & str > = entries.iter ( ) .map ( | e | e.kind.as_str ( ) ) .collect ( ) ;
5927
assert! ( kinds.contains ( & " ref" ) & & kinds.contains ( & " seal" ) & & kinds.contains ( & " policy" ) , " {kinds:?}" ) ;
5928
let r = entries.iter ( ) .find ( | e | e.kind = = " ref" ) .unwrap ( ) ;
5929
assert_eq! ( r.object, attest) ;
5930
assert_eq! ( r.repo, c.id ( ) .unwrap ( ) ) ;
5931
5932
5933
5934
let mut stored: Vec< Hash> = Vec :: new( ) ;
5935
for ( n, e) in entries.iter ( ) .enumerate ( ) {
5936
stored.extend ( tlog:: stored_hashes_for_leaf( n as u64 , e.leaf_hash ( ) , & stored) .unwrap ( ) ) ;
5937
}
5938
assert_eq! ( tlog:: tree_hash( cp.size, & stored) .unwrap ( ) , cp.root) ;
5939
let t = Tile { level: 0 , n: 0 , width: cp.size as u32 } ;
5940
let res =
5941
oxygenhub:: router( h.app.clone ( ) ) .oneshot ( Request:: get( format! ( " /log/{} " , t.path ( ) ) ) .body ( Body:: empty( ) ) .unwrap ( ) ) .await.unwrap ( ) ;
5942
assert_eq! ( res.status ( ) , StatusCode:: OK ) ;
5943
let data = res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ;
5944
assert_eq! ( data.to_vec ( ) , t.read_data ( & stored) .unwrap ( ) ) ;
5945
let ( st, _ ) = call ( & h.app, " GET" , " /log/tile/0/000.p/200" , None , None ) .await;
5946
assert_eq! ( st, StatusCode:: NOT_FOUND , " not held yet" ) ;
5947
5948
5949
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
5950
assert_eq! ( h.app.tlog.size ( ) , cp.size) ;
5951
let ( st, page) = call ( & h.app, " GET" , " /log" , None , None ) .await;
5952
assert_eq! ( st, StatusCode:: OK ) ;
5953
assert! ( page.contains ( " Transparency log" ) & & page.contains ( " andy/demo" ) & & page.contains ( & origin) , " {page}" ) ;
5954
}
5955
e5b97ac3 Andy McCall 18 days ago ↶
5956
5957
5958
5959
5960
5961
# [ tokio ::test ( flavor = " multi_thread" ) ]
5962
async fn a_fetch_checks_the_hub_s_log_and_refuses_a_fork ( ) {
5963
use oxy_core:: tlog:: { Checkpoint, Note} ;
5964
let h = hub ( ) .await;
5965
create ( & h.app, " andy" , " demo" ) .await;
5966
let url = format! ( " {} /andy/demo" , h.base) ;
5967
let work = tempfile:: tempdir( ) .unwrap ( ) ;
5968
let mut a = client ( & url, & work.path ( ) .join ( " a" ) ) ;
5969
a.create_identity ( " Andy" , " [email protected] " , oxy_core:: repo:: IdentityScope:: Repo) .unwrap ( ) ;
5970
let ( _ , sk) = oxygenhub:: keys:: load_secret( & h.app.config.keys_dir ( ) ) .unwrap ( ) ;
5971
let hub_id = h.app.hub_key.identity.clone ( ) ;
5972
let checkpoint = | | h.app.tlog.refresh_checkpoint ( & hub_id, & sk) .unwrap ( ) ;
5973
let mut n = 0 ;
5974
let mut push = | a : & Repo| {
5975
n + = 1 ;
5976
write ( & work.path ( ) .join ( " a" ) , " a.txt" , & format! ( " {n} \n " ) ) ;
5977
a.commit ( & format! ( " change {n} " ) , & [ ] ) .unwrap ( ) ;
5978
assert! ( a.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
5979
} ;
5980
push ( & a) ;
5981
checkpoint ( ) ;
5982
5983
let home = work.path ( ) .join ( " home-b" ) ;
5984
let b = Repo:: clone( & url, & work.path ( ) .join ( " b" ) ) .unwrap ( ) .with_home ( home.clone ( ) ) ;
5985
let report = b.fetch ( " origin" ) .unwrap ( ) ;
5986
assert! ( report.log_notes.is_empty ( ) , " {:?}" , report.log_notes) ;
5987
let pin = std:: fs:: read_dir( home.join ( " tlog" ) ) .unwrap ( ) .next ( ) .unwrap ( ) .unwrap ( ) .path ( ) ;
5988
let seen = | p : & std:: path:: Path| {
5989
let t = std:: fs:: read_to_string( p) .unwrap ( ) ;
5990
Checkpoint:: parse( & Note:: parse( t.split_once ( ' \n ' ) .unwrap ( ) .1 ) .unwrap ( ) .text) .unwrap ( )
5991
} ;
5992
let first = seen ( & pin) ;
5993
5994
5995
push ( & a) ;
5996
checkpoint ( ) ;
5997
let report = b.fetch ( " origin" ) .unwrap ( ) ;
5998
assert! ( report.log_notes.is_empty ( ) , " {:?}" , report.log_notes) ;
5999
assert! ( seen ( & pin) .size > first.size) ;
6000
6001
6002
push ( & a) ;
6003
let report = b.fetch ( " origin" ) .unwrap ( ) ;
6004
assert! ( report.log_notes.iter ( ) .any ( | n | n.contains ( " not in its transparency log yet" ) ) , " {:?}" , report.log_notes) ;
6005
7dbb41b0 Andy McCall 18 days ago ↶
6006
6007
6008
let text = std:: fs:: read_to_string( & pin) .unwrap ( ) ;
6009
let other_sk = oxy_core:: identity:: SecretKey:: generate( ) ;
6010
let other = other_sk.identity ( " another hub" , " [email protected] " , 1 ) ;
6011
let note = text.split_once ( ' \n ' ) .unwrap ( ) .1 ;
6012
std:: fs:: write( & pin, format! ( " {} \n {note} " , oxy_core:: dsse:: base64( & other.encode ( ) ) ) ) .unwrap ( ) ;
6013
let report = b.fetch ( " origin" ) .unwrap ( ) ;
6014
assert! ( report.log_notes.iter ( ) .any ( | n | n.contains ( " signed by another key" ) ) , " {:?}" , report.log_notes) ;
6015
let report = b.fetch ( " origin" ) .unwrap ( ) ;
6016
assert! ( ! report.log_notes.iter ( ) .any ( | n | n.contains ( " another key" ) ) , " pinned now: {:?}" , report.log_notes) ;
6017
e5b97ac3 Andy McCall 18 days ago ↶
6018
6019
6020
let size = seen ( & pin) .size;
6021
let forged = Checkpoint { origin: oxygenhub:: tlog:: origin_of( h.app.config.base_url ( ) ) , size, root: [ 7 ; 32 ] , extensions: vec! [ ] } ;
6022
let text = std:: fs:: read_to_string( & pin) .unwrap ( ) ;
6023
let key = text.split_once ( ' \n ' ) .unwrap ( ) .0 ;
6024
std:: fs:: write( & pin, format! ( " {key} \n {} " , forged.sign ( & hub_id, & sk) .unwrap ( ) .encode ( ) ) ) .unwrap ( ) ;
6025
checkpoint ( ) ;
6026
let err = b.fetch ( " origin" ) .unwrap_err ( ) .to_string ( ) ;
6027
assert! ( err.contains ( " showed two histories" ) & & err.contains ( " does not extend" ) , " {err}" ) ;
6028
}
6029
4f40eed0 Andy McCall 18 days ago ↶
6030
6031
6032
6033
6034
6035
# [ tokio ::test ( flavor = " multi_thread" ) ]
6036
async fn a_hub_witnesses_another_hub_s_log ( ) {
6037
use oxy_core:: tlog:: { AddCheckpoint, Checkpoint, Note} ;
6038
let log = hub ( ) .await;
6039
let witness = hub ( ) .await;
6040
let origin = oxygenhub:: tlog:: origin_of( log.app.config.base_url ( ) ) ;
6041
witness.app.tlog.add_witness_for ( & origin, log.app.hub_key.identity.clone ( ) ) ;
6042
let name = witness.app.tlog.witness_name ( ) .to_string ( ) ;
6043
log.app.tlog.set_witnesses ( vec! [ oxygenhub:: tlog:: Peer {
6044
name: name.clone ( ) ,
6045
url: format! ( " {} /log/witness/add-checkpoint" , witness.base) ,
6046
identity: witness.app.hub_key.identity.clone ( ) ,
6047
} ] ) ;
6048
create ( & log.app, " andy" , " demo" ) .await;
6049
let work = tempfile:: tempdir( ) .unwrap ( ) ;
6050
let mut c = client ( & format! ( " {} /andy/demo" , log.base) , work.path ( ) ) ;
6051
c.create_identity ( " Andy" , " [email protected] " , oxy_core:: repo:: IdentityScope:: Repo) .unwrap ( ) ;
6052
let ( _ , sk) = oxygenhub:: keys:: load_secret( & log.app.config.keys_dir ( ) ) .unwrap ( ) ;
6053
let log_id = log.app.hub_key.identity.clone ( ) ;
6054
let wid = witness.app.hub_key.identity.clone ( ) ;
6055
for n in 0 .. 2 {
6056
write ( work.path ( ) , " a.txt" , & format! ( " {n} \n " ) ) ;
6057
c.commit ( & format! ( " change {n} " ) , & [ ] ) .unwrap ( ) ;
6058
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
6059
log.app.tlog.refresh_checkpoint ( & log_id, & sk) .unwrap ( ) ;
6060
oxygenhub:: tlog:: collect( & log.app) .await;
6061
let note = Note:: parse( & log.app.tlog.checkpoint ( ) .unwrap ( ) ) .unwrap ( ) ;
6062
assert! ( note.cosigned_by ( & name, & wid) .is_ok ( ) , " round {n}: {}" , note.encode ( ) ) ;
6063
6064
let ( _ , served) = call ( & log.app, " GET" , " /log/checkpoint" , None , None ) .await;
6065
assert! ( Note:: parse( & served) .unwrap ( ) .cosigned_by ( & name, & wid) .is_ok ( ) ) ;
6066
}
6067
let cp = Checkpoint:: parse( & Note:: parse( & log.app.tlog.checkpoint ( ) .unwrap ( ) ) .unwrap ( ) .text) .unwrap ( ) ;
13b27cb2 Andy McCall 18 days ago ↶
6068
6069
6070
6071
6072
let ( _ , listed) = call ( & log.app, " GET" , " /log/witnesses" , None , None ) .await;
6073
assert! ( listed.starts_with ( & format! ( " {name} " ) ) , " {listed}" ) ;
6074
let url = format! ( " {} /andy/demo" , log.base) ;
6075
let refs = tokio:: task:: spawn_blocking( move | | {
6076
let opts = RemoteOptions { token: Some ( " site" .into ( ) ) , .. Default :: default( ) } ;
6077
open_remote ( & url, & opts) .unwrap ( ) .refs ( ) .unwrap ( )
6078
} )
6079
.await
6080
.unwrap ( ) ;
6081
assert_eq! ( refs.witnesses.len ( ) , 1 ) ;
6082
assert_eq! ( refs.witnesses[ 0 ] .0 , name) ;
6083
let main = refs.refs.iter ( ) .find ( | r | r.name = = " main" ) .expect ( " main" ) ;
6084
let logged = main.logged.as_ref ( ) .expect ( " logged" ) ;
6085
assert! ( ! main.integration.is_empty ( ) , " a witnessed checkpoint holds it" ) ;
6086
let cosigned: Vec< ( & str , & [[u8 ; 32]]) > = main.integration.iter ( ) .map ( | i | ( i.note.as_str ( ) , i.proof.as_slice ( ) ) ) .collect ( ) ;
6087
let at = oxy_core:: tlog:: integrated_time( & logged.entry, logged.index, & cosigned, & origin, & log_id, & [ ( name.clone ( ) , wid.clone ( ) ) ] )
6088
.expect ( " an integrated time" ) ;
6089
assert_eq! ( at.witness, name) ;
6090
assert! ( at.size > logged.index) ;
4f40eed0 Andy McCall 18 days ago ↶
6091
6092
let ask = | req : AddCheckpoint| {
6093
let app = witness.app.clone ( ) ;
6094
async move { call ( & app, " POST" , " /log/witness/add-checkpoint" , None , Some ( req.encode ( ) ) ) .await }
6095
} ;
6096
6097
let forked = Checkpoint { root: [ 7 ; 32 ] , .. cp.clone ( ) } .sign ( & log_id, & sk) .unwrap ( ) .encode ( ) ;
6098
let ( st, body) = ask ( AddCheckpoint { old_size: cp.size, proof: vec! [ ] , note: forked.clone ( ) } ) .await;
6099
assert_eq! ( st, StatusCode:: UNPROCESSABLE_ENTITY , " {body}" ) ;
6100
6101
let ( st, body) = ask ( AddCheckpoint { old_size: 1 , proof: vec! [ ] , note: forked } ) .await;
6102
assert_eq! ( ( st, body.trim ( ) ) , ( StatusCode:: CONFLICT , cp.size.to_string ( ) .as_str ( ) ) ) ;
6103
6104
let stranger = oxy_core:: identity:: SecretKey:: generate( ) ;
6105
let sid = stranger.identity ( " elsewhere" , " [email protected] " , 1 ) ;
6106
let other = Checkpoint { origin: " elsewhere.example/log" .into ( ) , .. cp.clone ( ) } .sign ( & sid, & stranger) .unwrap ( ) .encode ( ) ;
6107
let ( st, _ ) = ask ( AddCheckpoint { old_size: 0 , proof: vec! [ ] , note: other } ) .await;
6108
assert_eq! ( st, StatusCode:: NOT_FOUND ) ;
6109
}
6110
8dcaeeb8 Andy McCall 18 days ago ↶
6111
6112
6113
6114
6115
# [ tokio ::test ( flavor = " multi_thread" ) ]
6116
async fn a_job_keeps_its_terraform_state_on_the_hub ( ) {
6117
let h = hub ( ) .await;
6118
create ( & h.app, " andy" , " infra" ) .await;
6119
create ( & h.app, " andy" , " other" ) .await;
6120
let work = tempfile:: tempdir( ) .unwrap ( ) ;
6121
let c = client ( & format! ( " {} /andy/infra" , h.base) , work.path ( ) ) ;
6122
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
6123
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
6124
c.snapshot ( " test" ) .unwrap ( ) ;
6125
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
6126
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
6127
assert_eq! ( runs_of ( & h.app, " andy" , " infra" ) .await.len ( ) , 1 ) ;
6128
let job = claim ( & h.app) .await.expect ( " a job" ) ;
6129
let token = job[ " token" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
6130
let basic = format! ( " Basic {} " , oxy_core:: dsse:: base64( format! ( " flow:{token} " ) .as_bytes ( ) ) ) ;
6131
let path = " /api/v1/repos/andy/infra/terraform/state/prod" ;
6132
let send = | method : & str , uri : String, auth : Option< String> , body : & str | {
6133
let mut req = Request:: builder( ) .method ( method) .uri ( uri) ;
6134
if let Some ( a) = auth {
6135
req = req.header ( header:: AUTHORIZATION , a) ;
6136
}
6137
let req = req.body ( Body:: from( body.to_string ( ) ) ) .unwrap ( ) ;
6138
let app = h.app.clone ( ) ;
6139
async move {
6140
let res = oxygenhub:: router( app) .oneshot ( req) .await.unwrap ( ) ;
6141
let st = res.status ( ) ;
6142
let b = res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ;
6143
( st, String :: from_utf8_lossy( & b) .into_owned ( ) )
6144
}
6145
} ;
6146
let lock = r # "{"ID":"lock-1","Operation":"OperationTypeApply","Who":"flow@runner","Version":"1.8.0"}"# ;
6147
6148
assert_eq! ( send ( " GET" , path.into ( ) , None , " " ) .await.0 , StatusCode:: UNAUTHORIZED ) ;
6149
assert_eq! ( send ( " GET" , path.into ( ) , Some ( basic.clone ( ) ) , " " ) .await.0 , StatusCode:: NO_CONTENT ) ;
6150
6151
let ( st, body) = send ( " LOCK" , path.into ( ) , Some ( basic.clone ( ) ) , lock) .await;
6152
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
6153
let other_lock = r # "{"ID":"lock-2","Operation":"OperationTypePlan"}"# ;
6154
let ( st, held) = send ( " LOCK" , path.into ( ) , Some ( basic.clone ( ) ) , other_lock) .await;
6155
assert_eq! ( st, StatusCode:: LOCKED ) ;
6156
assert! ( held.contains ( " lock-1" ) , " the holder's lock info: {held}" ) ;
6157
let state1 = r # "{"version":4,"serial":1,"lineage":"l","resources":[{"secret":"hunter2"}]}"# ;
6158
assert_eq! ( send ( " POST" , path.into ( ) , Some ( basic.clone ( ) ) , state1) .await.0 , StatusCode:: LOCKED , " not under its lock" ) ;
6159
assert_eq! ( send ( " POST" , format! ( " {path} ?ID=lock-1" ) , Some ( basic.clone ( ) ) , state1) .await.0 , StatusCode:: OK ) ;
6160
assert_eq! ( send ( " UNLOCK" , path.into ( ) , Some ( basic.clone ( ) ) , other_lock) .await.0 , StatusCode:: CONFLICT , " not its lock" ) ;
6161
assert_eq! ( send ( " UNLOCK" , path.into ( ) , Some ( basic.clone ( ) ) , lock) .await.0 , StatusCode:: OK ) ;
6162
let ( st, got) = send ( " GET" , path.into ( ) , Some ( basic.clone ( ) ) , " " ) .await;
6163
assert_eq! ( ( st, got.as_str ( ) ) , ( StatusCode:: OK , state1) ) ;
6164
6165
let state2 = state1.replace ( " \" serial\" :1" , " \" serial\" :2" ) ;
6166
assert_eq! ( send ( " POST" , path.into ( ) , Some ( basic.clone ( ) ) , & state2) .await.0 , StatusCode:: OK ) ;
6167
let row = db:: repos:: get( & h.app.db, " andy" , " infra" ) .await.unwrap ( ) .unwrap ( ) ;
6168
assert_eq! ( db:: terraform:: versions( & h.app.db, row.id, " prod" ) .await.unwrap ( ) , 1 ) ;
6169
let kept = db:: terraform:: get( & h.app.db, row.id, " prod" ) .await.unwrap ( ) .unwrap ( ) ;
6170
assert_eq! ( kept.serial, 2 ) ;
6171
assert! ( ! String :: from_utf8_lossy( & kept.body) .contains ( " hunter2" ) , " sealed at rest" ) ;
6172
6173
let ( st, _ ) = send ( " GET" , " /api/v1/repos/andy/other/terraform/state/prod" .into ( ) , Some ( basic.clone ( ) ) , " " ) .await;
6174
assert_eq! ( st, StatusCode:: FORBIDDEN ) ;
6175
assert_eq! (
6176
send ( " GET" , " /api/v1/repos/andy/infra/terraform/state/..x" .into ( ) , Some ( basic.clone ( ) ) , " " ) .await.0 ,
6177
StatusCode:: BAD_REQUEST
6178
) ;
bb018142 Andy McCall 18 days ago ↶
6179
6180
6181
let mallory = db:: accounts:: create_account( & h.app.db, " mallory" , " [email protected] " , " x" , " salt" , false ) .await.unwrap ( ) ;
6182
let ( secret, hash) = oxygenhub:: auth:: new_token( ) ;
6183
db:: accounts:: create_token( & h.app.db, mallory.id, " cli" , & hash, " repo:write" , " " , None ) .await.unwrap ( ) ;
6184
let ( st, got) = send ( " GET" , path.into ( ) , Some ( format! ( " Bearer {secret} " ) ) , " " ) .await;
6185
assert_eq! ( st, StatusCode:: FORBIDDEN , " {got}" ) ;
6186
assert! ( ! got.contains ( " hunter2" ) ) ;
6187
let ( _ , page) = call ( & h.app, " GET" , " /andy/infra/terraform" , Some ( & secret) , None ) .await;
6188
assert! ( ! page.contains ( " <strong>prod</strong>" ) & & page.contains ( " shown to people who can write" ) , " {page}" ) ;
8dcaeeb8 Andy McCall 18 days ago ↶
6189
6190
let ( st, _ ) = send ( " GET" , path.into ( ) , Some ( " Bearer site" .into ( ) ) , " " ) .await;
6191
assert_eq! ( st, StatusCode:: UNAUTHORIZED , " the site token is not a person's token" ) ;
6192
6193
6194
let issuer = h.app.oidc ( ) .await.unwrap ( ) ;
6195
let now = db:: now_ms( ) ;
6196
let aud = oxygenhub:: terraform:: audience( h.app.config.base_url ( ) ) ;
6197
let own = issuer.mint_for ( h.app.config.base_url ( ) , " repo:andy/infra:bookmark:main" , & aud, now, 600 ) ;
6198
let ( st, got) = send ( " GET" , path.into ( ) , Some ( format! ( " Bearer {own} " ) ) , " " ) .await;
6199
assert_eq! ( ( st, got.contains ( " \" serial\" :2" ) ) , ( StatusCode:: OK , true ) ) ;
6200
let other = issuer.mint_for ( h.app.config.base_url ( ) , " repo:andy/other:bookmark:main" , & aud, now, 600 ) ;
6201
assert_eq! ( send ( " GET" , path.into ( ) , Some ( format! ( " Bearer {other} " ) ) , " " ) .await.0 , StatusCode:: FORBIDDEN ) ;
6202
let elsewhere = issuer.mint_for ( h.app.config.base_url ( ) , " repo:andy/infra:bookmark:main" , " sts.amazonaws.com" , now, 600 ) ;
6203
assert_eq! ( send ( " GET" , path.into ( ) , Some ( format! ( " Bearer {elsewhere} " ) ) , " " ) .await.0 , StatusCode:: UNAUTHORIZED ) ;
16ee5d19 Andy McCall 18 days ago ↶
6204
bb018142 Andy McCall 18 days ago ↶
6205
6206
let res = oxygenhub:: router( h.app.clone ( ) )
6207
.oneshot (
6208
Request:: post( " /register" )
6209
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
6210
.body ( Body:: from( " username=andy&email=a%40example.com&password=correct-horse-battery&setup=test-setup" ) )
6211
.unwrap ( ) ,
6212
)
6213
.await
6214
.unwrap ( ) ;
6215
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
6216
let res = oxygenhub:: router( h.app.clone ( ) )
67523e06 Andy McCall 18 days ago ↶
6217
.oneshot ( Request:: get( " /andy/infra/terraform" ) .header ( header:: COOKIE , cookie.clone ( ) ) .body ( Body:: empty( ) ) .unwrap ( ) )
bb018142 Andy McCall 18 days ago ↶
6218
.await
6219
.unwrap ( ) ;
6220
let st = res.status ( ) ;
6221
let page = String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) ;
16ee5d19 Andy McCall 18 days ago ↶
6222
assert_eq! ( st, StatusCode:: OK , " {page}" ) ;
6223
assert! ( page.contains ( " <strong>prod</strong>" ) & & page.contains ( " /api/v1/repos/andy/infra/terraform/state/NAME" ) , " {page}" ) ;
6224
assert! ( page.contains ( " allow "" ) | | page.contains ( " allow \" " ) , " the egress line names the hub" ) ;
67523e06 Andy McCall 18 days ago ↶
6225
assert! ( page.contains ( " 1 kept version" ) & & page.contains ( " >restore</button>" ) , " its versions, restorable by its owner: {page}" ) ;
6226
6227
6228
6229
6230
let andy = db:: accounts:: account_by_name( & h.app.db, " andy" ) .await.unwrap ( ) .unwrap ( ) ;
6231
let ( pat, hash) = oxygenhub:: auth:: new_token( ) ;
6232
db:: accounts:: create_token( & h.app.db, andy.id, " cli" , & hash, " repo:write" , " " , None ) .await.unwrap ( ) ;
6233
let bearer = format! ( " Bearer {pat} " ) ;
6234
let ( st, list) = send ( " GET" , format! ( " {path} /versions" ) , Some ( bearer.clone ( ) ) , " " ) .await;
6235
assert_eq! ( st, StatusCode:: OK , " {list}" ) ;
6236
let list: serde_json:: Value = serde_json:: from_str( & list) .unwrap ( ) ;
6237
assert_eq! ( ( list.as_array ( ) .unwrap ( ) .len ( ) , list[ 0 ] [ " serial" ] .as_i64 ( ) ) , ( 1 , Some ( 1 ) ) , " {list}" ) ;
6238
let first = list[ 0 ] [ " id" ] .as_i64 ( ) .unwrap ( ) ;
6239
let ( st, got) = send ( " GET" , format! ( " {path} /versions/{first} " ) , Some ( bearer.clone ( ) ) , " " ) .await;
6240
assert_eq! ( ( st, got.as_str ( ) ) , ( StatusCode:: OK , state1) ) ;
6241
assert_eq! ( send ( " GET" , format! ( " {path} /versions" ) , Some ( format! ( " Bearer {secret} " ) ) , " " ) .await.0 , StatusCode:: FORBIDDEN , " a stranger" ) ;
6242
let restore = | id : i64 , auth : String| send ( " POST" , format! ( " {path} /versions/{id} /restore" ) , Some ( auth) , " " ) ;
6243
assert_eq! ( restore ( first, basic.clone ( ) ) .await.0 , StatusCode:: FORBIDDEN , " a job does not restore" ) ;
6244
assert_eq! ( restore ( first, format! ( " Bearer {secret} " ) ) .await.0 , StatusCode:: FORBIDDEN , " nor a stranger" ) ;
6245
assert_eq! ( restore ( 9999 , bearer.clone ( ) ) .await.0 , StatusCode:: NOT_FOUND ) ;
6246
assert_eq! ( send ( " LOCK" , path.into ( ) , Some ( basic.clone ( ) ) , lock) .await.0 , StatusCode:: OK ) ;
6247
assert_eq! ( restore ( first, bearer.clone ( ) ) .await.0 , StatusCode:: LOCKED , " not under someone's lock" ) ;
6248
assert_eq! ( send ( " UNLOCK" , path.into ( ) , Some ( basic.clone ( ) ) , lock) .await.0 , StatusCode:: OK ) ;
6249
let ( st, got) = restore ( first, bearer.clone ( ) ) .await;
6250
assert_eq! ( ( st, got.contains ( " \" serial\" :3" ) ) , ( StatusCode:: OK , true ) , " {got}" ) ;
6251
let ( _ , now) = send ( " GET" , path.into ( ) , Some ( bearer.clone ( ) ) , " " ) .await;
6252
let now: serde_json:: Value = serde_json:: from_str( & now) .unwrap ( ) ;
6253
assert_eq! (
6254
( now[ " serial" ] .as_i64 ( ) , now[ " lineage" ] .as_str ( ) , now[ " resources" ] [ 0 ] [ " secret" ] .as_str ( ) ) ,
6255
( Some ( 3 ) , Some ( " l" ) , Some ( " hunter2" ) )
6256
) ;
6257
6258
let ( _ , list) = send ( " GET" , format! ( " {path} /versions" ) , Some ( bearer.clone ( ) ) , " " ) .await;
6259
let list: serde_json:: Value = serde_json:: from_str( & list) .unwrap ( ) ;
6260
let serials: Vec< i64 > = list.as_array ( ) .unwrap ( ) .iter ( ) .map ( | v | v[ " serial" ] .as_i64 ( ) .unwrap ( ) ) .collect ( ) ;
6261
assert_eq! ( serials, vec! [ 2 , 1 ] , " {list}" ) ;
6262
let ( st, got) = restore ( list[ 0 ] [ " id" ] .as_i64 ( ) .unwrap ( ) , bearer.clone ( ) ) .await;
6263
assert_eq! ( ( st, got.contains ( " \" serial\" :4" ) ) , ( StatusCode:: OK , true ) , " {got}" ) ;
6264
6265
6266
let page_with = | req : axum:: http:: request:: Builder, body : Body| {
6267
let ( app, cookie) = ( h.app.clone ( ) , cookie.clone ( ) ) ;
6268
async move {
6269
let res = oxygenhub:: router( app) .oneshot ( req.header ( header:: COOKIE , cookie) .body ( body) .unwrap ( ) ) .await.unwrap ( ) ;
6270
let st = res.status ( ) ;
6271
( st, String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) )
6272
}
6273
} ;
6274
let ( st, file) = page_with ( Request:: get( format! ( " /andy/infra/terraform/prod/download?version={first} " ) ) , Body:: empty( ) ) .await;
6275
assert_eq! ( ( st, file.as_str ( ) ) , ( StatusCode:: OK , state1) ) ;
6276
let ( _ , page) = page_with ( Request:: get( " /andy/infra/terraform" ) , Body:: empty( ) ) .await;
6277
let csrf = page.split ( " name=\" _csrf\" value=\" " ) .nth ( 1 ) .unwrap ( ) .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( ) ;
6278
let ( st, page) = page_with (
6279
Request:: post( " /andy/infra/terraform/prod/restore" ) .header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) ,
6280
Body:: from( format! ( " _csrf={csrf} &version={first} " ) ) ,
6281
)
6282
.await;
6283
assert_eq! ( st, StatusCode:: OK , " {page}" ) ;
6284
assert! ( page.contains ( " Restored prod: it is now serial 5." ) , " {page}" ) ;
6285
let audit = db:: accounts:: audit_rows( & h.app.db, 50 ) .await.unwrap ( ) ;
6286
assert_eq! ( audit.iter ( ) .filter ( | r | r.action = = " terraform.restored" ) .count ( ) , 3 , " every restore audited" ) ;
8dcaeeb8 Andy McCall 18 days ago ↶
6287
}
6288
6289
6290
6291
6292
6293
# [ tokio ::test ( flavor = " multi_thread" ) ]
6294
# [ ignore ]
6295
async fn tofu_keeps_its_state_on_the_hub ( ) {
6296
let h = hub ( ) .await;
6297
create ( & h.app, " andy" , " infra" ) .await;
6298
let work = tempfile:: tempdir( ) .unwrap ( ) ;
6299
let c = client ( & format! ( " {} /andy/infra" , h.base) , work.path ( ) ) ;
6300
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
6301
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
6302
c.snapshot ( " test" ) .unwrap ( ) ;
6303
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
6304
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
6305
assert_eq! ( runs_of ( & h.app, " andy" , " infra" ) .await.len ( ) , 1 ) ;
6306
let job = claim ( & h.app) .await.expect ( " a job" ) ;
6307
let token = job[ " token" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
6308
let tf = tempfile:: tempdir( ) .unwrap ( ) ;
6309
let address = format! ( " {} /api/v1/repos/andy/infra/terraform/state/prod" , h.base) ;
6310
std:: fs:: write(
6311
tf.path ( ) .join ( " main.tf" ) ,
6312
format! (
6313
" terraform {{ \n backend \" http\" {{ \n address = \" {address} \" \n lock_address = \" {address} \" \n unlock_address = \" {address} \" \n username = \" flow\" \n }} \n }} \n resource \" terraform_data\" \" x\" {{ \n input = \" hello\" \n }} \n "
6314
) ,
6315
)
6316
.unwrap ( ) ;
6317
let tofu = | args : & [& str ]| {
6318
let out = std:: process:: Command:: new( " docker" )
6319
.args ( [ " run" , " --rm" , " --network" , " host" , " -v" ] )
6320
.arg ( format! ( " {} :/w" , tf.path ( ) .display ( ) ) )
6321
.args ( [ " -w" , " /w" , " -e" ] )
6322
.arg ( format! ( " TF_HTTP_PASSWORD={token} " ) )
6323
.args ( [ " --entrypoint" , " tofu" , " oxygenflow-tofu:latest" ] )
6324
.args ( args)
6325
.output ( )
6326
.expect ( " docker" ) ;
6327
let text = format! ( " {} {} " , String :: from_utf8_lossy( & out.stdout) , String :: from_utf8_lossy( & out.stderr) ) ;
6328
assert! ( out.status.success ( ) , " tofu {args:?}: {text}" ) ;
6329
text
6330
} ;
6331
tofu ( & [ " init" , " -input=false" , " -no-color" ] ) ;
6332
let applied = tofu ( & [ " apply" , " -auto-approve" , " -input=false" , " -no-color" ] ) ;
6333
assert! ( applied.contains ( " 1 added" ) , " {applied}" ) ;
6334
let again = tofu ( & [ " apply" , " -auto-approve" , " -input=false" , " -no-color" ] ) ;
6335
assert! ( again.contains ( " No changes" ) , " the state was kept: {again}" ) ;
6336
let row = db:: repos:: get( & h.app.db, " andy" , " infra" ) .await.unwrap ( ) .unwrap ( ) ;
6337
let kept = db:: terraform:: get( & h.app.db, row.id, " prod" ) .await.unwrap ( ) .unwrap ( ) ;
6338
assert! ( kept.lock_id.is_empty ( ) , " unlocked after each run" ) ;
6339
assert! ( kept.serial > = 1 ) ;
6340
}
6341
8a48dd83 Andy McCall 18 days ago ↶
6342
6343
6344
# [ tokio ::test ( flavor = " multi_thread" ) ]
6345
async fn metrics_are_private_and_count_flow_jobs ( ) {
6346
let h = hub ( ) .await;
6347
let ( st, _ ) = call ( & h.app, " GET" , " /metrics" , None , None ) .await;
6348
assert_eq! ( st, StatusCode:: NOT_FOUND , " off by default" ) ;
6349
let text = oxygenhub:: metrics:: render( & h.app) .await;
6350
for want in [
6351
" # TYPE oxyhub_uptime_seconds gauge" ,
6352
" oxyhub_flow_jobs{state=\" queued\" } 0" ,
6353
" oxyhub_bucket_requests_total{kind=\" list\" }" ,
6354
" oxyhub_tlog_entries" ,
a6d824c5 Andy McCall 18 days ago ↶
6355
" oxyhub_db_connections{idle=\" true\" }" ,
6356
" oxyhub_pushes_last_day{result=\" refused\" } 0" ,
6357
" # TYPE oxyhub_backup_last_run_ok gauge" ,
2c47eecf Andy McCall 18 days ago ↶
6358
" # TYPE oxyhub_flow_runner_last_seen_seconds gauge" ,
6359
" oxyhub_flow_runs_last_day{result=\" failed\" } 0" ,
6360
" oxyhub_mail_total{result=\" sent\" }" ,
8a48dd83 Andy McCall 18 days ago ↶
6361
] {
6362
assert! ( text.contains ( want) , " missing {want}:\n {text}" ) ;
6363
}
6364
create ( & h.app, " andy" , " demo" ) .await;
6365
let work = tempfile:: tempdir( ) .unwrap ( ) ;
6366
let c = client ( & format! ( " {} /andy/demo" , h.base) , work.path ( ) ) ;
6367
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
6368
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
6369
c.snapshot ( " test" ) .unwrap ( ) ;
6370
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
6371
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
6372
assert_eq! ( runs_of ( & h.app, " andy" , " demo" ) .await.len ( ) , 1 ) ;
6373
let text = oxygenhub:: metrics:: render( & h.app) .await;
6374
assert! ( ! text.contains ( " oxyhub_flow_jobs{state=\" queued\" } 0" ) , " the run's jobs are queued:\n {text}" ) ;
a6d824c5 Andy McCall 18 days ago ↶
6375
assert! ( text.contains ( " oxyhub_pushes_last_day{result=\" accepted\" }" ) , " {text}" ) ;
85ec74fd Andy McCall 18 days ago ↶
6376
6377
6378
let pushed = text
6379
.lines ( )
6380
.find_map ( | l | l.strip_prefix ( " oxyhub_protocol_bytes_total{endpoint=\" push\" ,direction=\" received\" } " ) )
6381
.and_then ( | v | v.parse:: < f64 > ( ) .ok ( ) )
6382
.unwrap_or ( 0. 0 ) ;
6383
assert! ( pushed > 0. 0 , " {text}" ) ;
b9b74d99 Andy McCall 18 days ago ↶
6384
6385
call ( & h.app, " GET" , " /andy/demo" , None , None ) .await;
6386
let text = oxygenhub:: metrics:: render( & h.app) .await;
6387
assert! ( text.contains ( " oxyhub_http_requests_total{route=\" page:repository\" ,method=\" GET\" ,status=\" 2xx\" }" ) , " {text}" ) ;
6388
assert! ( text.contains ( " oxyhub_http_request_duration_seconds_bucket{route=\" page:repository\" ,method=\" GET\" ,le=\" +Inf\" }" ) ) ;
8a48dd83 Andy McCall 18 days ago ↶
6389
}
6390
e97500b7 Andy McCall 18 days ago ↶
6391
6392
6393
6394
6395
# [ tokio ::test ( flavor = " multi_thread" ) ]
6396
async fn insights_chart_pickup_from_when_a_job_could_run ( ) {
6397
let h = hub ( ) .await;
6398
create ( & h.app, " andy" , " demo" ) .await;
6399
let work = tempfile:: tempdir( ) .unwrap ( ) ;
6400
let c = client ( & format! ( " {} /andy/demo" , h.base) , work.path ( ) ) ;
6401
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
6402
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
6403
c.snapshot ( " test" ) .unwrap ( ) ;
6404
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
6405
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
6406
assert_eq! ( runs_of ( & h.app, " andy" , " demo" ) .await.len ( ) , 1 ) ;
6407
let mut finished = std:: collections:: HashMap:: new( ) ;
6408
for want in [ " build" , " lint" , " test" ] {
6409
let j = claim ( & h.app) .await.expect ( " a job" ) ;
6410
assert_eq! ( j[ " job" ] [ " name" ] .as_str ( ) , Some ( want) ) ;
6411
let id = j[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
6412
let row = db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) ;
6413
let ( queued, ready, started) = ( row.queued_ms.unwrap ( ) , row.ready_ms.unwrap ( ) , row.started_ms.unwrap ( ) ) ;
6414
assert! ( queued < = ready & & ready < = started, " {want}: {queued} {ready} {started}" ) ;
6415
if want = = " test" {
6416
assert_eq! ( Some ( ready) , finished[ " build" ] , " ready when build passed, not when it was queued" ) ;
6417
}
6418
let ph = j[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) ;
6419
let st = finish ( & h.app, id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 , " receipt" : receipt ( want, ph, 0 ) } ) ) .await;
6420
assert_eq! ( st, StatusCode:: NO_CONTENT ) ;
6421
finished.insert ( want, db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) .finished_ms) ;
6422
}
6423
let ( st, page) = call ( & h.app, " GET" , " /andy/demo/insights" , None , None ) .await;
6424
assert_eq! ( st, StatusCode:: OK , " {page}" ) ;
6425
assert! ( page.contains ( " <span class=\" kpi-v\" >3</span><span class=\" kpi-l\" >jobs finished" ) , " {page}" ) ;
6426
assert! ( page.contains ( " 100% passed" ) , " {page}" ) ;
6427
assert_eq! ( page.matches ( " <svg class=\" chart\" " ) .count ( ) , 5 , " five charts" ) ;
6428
assert! ( page.contains ( " class=\" bar s-ok\" " ) , " the passed jobs drawn" ) ;
6429
assert! ( page.contains ( " data-refresh=\" 30\" data-fragment=\" insights\" " ) , " it refreshes itself" ) ;
6430
assert! ( page.contains ( " <summary class=\" on\" >More</summary>" ) , " the bar's More menu holds it" ) ;
6431
let ( _ , wall) = call ( & h.app, " GET" , " /andy/demo/insights?range=7d&wall=1" , None , None ) .await;
6432
assert! ( wall.contains ( " <body class=\" wall\" " ) & & wall.contains ( " the last 7 days" ) & & wall.contains ( " leave the wall view" ) , " {wall}" ) ;
6433
6434
assert_eq! ( call ( & h.app, " GET" , " /insights/andy" , None , None ) .await.0 , StatusCode:: NOT_FOUND ) ;
6435
assert_eq! ( call ( & h.app, " GET" , " /admin/insights" , None , None ) .await.0 , StatusCode:: NOT_FOUND ) ;
6436
let res = oxygenhub:: router( h.app.clone ( ) )
6437
.oneshot (
6438
Request:: post( " /register" )
6439
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
6440
.body ( Body:: from( " username=andy&email=a%40example.com&password=correct-horse-battery&setup=test-setup" ) )
6441
.unwrap ( ) ,
6442
)
6443
.await
6444
.unwrap ( ) ;
6445
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
6446
let andy = db:: accounts:: account_by_name( & h.app.db, " andy" ) .await.unwrap ( ) .unwrap ( ) ;
6447
db:: accounts:: set_admin( & h.app.db, andy.id, true ) .await.unwrap ( ) ;
6448
for path in [ " /insights/andy" , " /admin/insights?runner=test-1&range=1h" ] {
6449
let res = oxygenhub:: router( h.app.clone ( ) )
6450
.oneshot ( Request:: get( path) .header ( header:: COOKIE , cookie.clone ( ) ) .body ( Body:: empty( ) ) .unwrap ( ) )
6451
.await
6452
.unwrap ( ) ;
6453
let st = res.status ( ) ;
6454
let body = String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) ;
6455
assert_eq! ( st, StatusCode:: OK , " {path}: {body}" ) ;
6456
assert! ( body.contains ( " <svg class=\" chart\" " ) , " {path}" ) ;
6457
}
6458
}
6459
7c06f5b2 Andy McCall 18 days ago ↶
6460
6461
6462
6463
# [ tokio ::test ( flavor = " multi_thread" ) ]
6464
async fn reads_do_not_wait_for_a_writer ( ) {
6465
use std:: time:: Duration;
6466
let h = hub ( ) .await;
6467
create ( & h.app, " andy" , " demo" ) .await;
6468
let work = tempfile:: tempdir( ) .unwrap ( ) ;
6469
let c = client ( & format! ( " {} /andy/demo" , h.base) , work.path ( ) ) ;
6470
write ( work.path ( ) , " a.txt" , " a\n " ) ;
6471
c.commit ( " A change" , & [ ] ) .unwrap ( ) ;
6472
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
6473
let name = RepoName:: parse( " andy/demo" ) .unwrap ( ) ;
6474
let writer = h.app.repos.write_guard ( & name) .await;
6475
for path in [ " /andy/demo" , " /andy/demo/changes/main" , " /api/v1/repos/andy/demo/bookmarks" ] {
6476
let ( st, body) = tokio:: time:: timeout( Duration:: from_secs( 10 ) , call ( & h.app, " GET" , path, None , None ) )
6477
.await
6478
.unwrap_or_else ( | _| panic! ( " {path} waited for the writer" ) ) ;
6479
assert_eq! ( st, StatusCode:: OK , " {path}: {body}" ) ;
6480
}
6481
let refs = tokio:: time:: timeout( Duration:: from_secs( 10 ) , call ( & h.app, " POST" , " /andy/demo/oxy/v1/refs" , Some ( " site" ) , None ) ) .await;
6482
assert! ( refs.is_ok ( ) , " a fetch's refs waited for the writer" ) ;
6483
assert! ( tokio:: time:: timeout( Duration:: from_millis( 300 ) , h.app.repos.write_guard ( & name) ) .await.is_err ( ) , " writers take turns" ) ;
6484
assert! ( tokio:: time:: timeout( Duration:: from_millis( 300 ) , h.app.repos.exclusive_guard ( & name) ) .await.is_err ( ) , " a delete waits" ) ;
6485
drop ( writer) ;
6486
let all = tokio:: time:: timeout( Duration:: from_secs( 5 ) , h.app.repos.exclusive_guard ( & name) ) .await.expect ( " then it runs" ) ;
6487
assert! (
6488
tokio:: time:: timeout( Duration:: from_millis( 300 ) , h.app.repos.read_guard ( & name) ) .await.is_err ( ) ,
6489
" nothing reads a repository being deleted"
6490
) ;
6491
drop ( all) ;
6492
}
6493
a10446c6 Andy McCall 18 days ago ↶
6494
6495
6496
6497
6498
# [ tokio ::test ( flavor = " multi_thread" ) ]
6499
async fn the_issue_overview_and_diffusion_pages_listen_on_their_topics ( ) {
6500
let h = hub ( ) .await;
6501
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
6502
create ( & h.app, " andy" , " demo" ) .await;
6503
let row = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
6504
let work = tempfile:: tempdir( ) .unwrap ( ) ;
6505
let c = client ( & format! ( " {} /andy/demo" , h.base) , work.path ( ) ) ;
6506
write ( work.path ( ) , " a.txt" , " a\n " ) ;
6507
c.snapshot ( " t" ) .unwrap ( ) ;
6508
c.commit ( " a" , & [ ] ) .unwrap ( ) ;
6509
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
6510
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/issues" , Some ( " site" ) , Some ( r # "{"title":"Broken"}"# .into ( ) ) ) .await;
6511
assert! ( st.is_success ( ) , " {st} {body}" ) ;
6512
let repo = format! ( " repo:{} " , row.id) ;
6513
let ( _ , issue) = call ( & h.app, " GET" , " /andy/demo/issue/1" , None , None ) .await;
6514
assert! ( issue.contains ( & format! ( " data-live=\" {repo} /issue:1 {repo} \" data-fragment=\" timeline\" " ) ) , " the issue's timeline" ) ;
6515
let ( _ , overview) = call ( & h.app, " GET" , " /andy/demo" , None , None ) .await;
6516
assert! ( overview.contains ( & format! ( " data-live=\" {repo} \" data-fragment=\" latest\" " ) ) , " the latest change" ) ;
6517
assert! ( overview.contains ( & format! ( " data-live=\" {repo} \" data-fragment=\" bookmarks\" " ) ) , " the bookmarks" ) ;
6518
let ( _ , diffusion) = call ( & h.app, " GET" , " /andy/demo/diffusion" , None , None ) .await;
6519
assert! ( diffusion.contains ( & format! ( " data-live=\" diffusion:{} {repo} \" data-fragment=\" diffusion\" " , row.id) ) , " the Diffusion tab" ) ;
6520
}
6521
3eb491e9 Andy McCall 19 days ago ↶
6522
6523
6524
6525
6526
# [ tokio ::test ( flavor = " multi_thread" ) ]
6527
async fn the_hub_countersigns_an_update_by_a_key_it_certified ( ) {
6528
use oxy_core:: attest:: AttestBody;
6529
use oxy_core:: cert:: { Certificate, SignedCertificate} ;
6530
use oxy_core:: identity:: Scope;
6531
use oxy_core:: repo:: IdentityScope;
6532
let h = hub ( ) .await;
6533
create ( & h.app, " andy" , " demo" ) .await;
6534
let url = format! ( " {} /andy/demo" , h.base) ;
6535
let work = tempfile:: tempdir( ) .unwrap ( ) ;
6536
let mut a = client ( & url, & work.path ( ) .join ( " a" ) ) .with_home ( work.path ( ) .join ( " home-a" ) ) ;
6537
a.create_identity ( " Andy" , " [email protected] " , IdentityScope:: User) .unwrap ( ) ;
6538
write ( & work.path ( ) .join ( " a" ) , " a.txt" , " a\n " ) ;
6539
a.commit ( " a" , & [ ] ) .unwrap ( ) ;
6540
a.policy_init ( ) .unwrap ( ) ;
6541
assert! ( a.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
6542
let hub_id = h.app.hub_key.identity.clone ( ) ;
6543
a.policy_grant_subject ( " agent:claude@andy" , hub_id.clone ( ) , Scope:: CHANGES | Scope:: REFS , None ) .unwrap ( ) ;
6544
assert! ( a.push ( " origin" , false ) .unwrap ( ) .iter ( ) .any ( | r | r.name = = " policy" & & r.ok ) ) ;
6545
6546
6547
let mut b = Repo:: clone( & url, & work.path ( ) .join ( " b" ) ) .unwrap ( ) .with_home ( work.path ( ) .join ( " home-b" ) ) ;
6548
b.config.set ( " remote.origin.token" , " site" ) .unwrap ( ) ;
6549
b.save_config ( ) .unwrap ( ) ;
6550
let ( _ , key) = b.create_identity ( " claude" , " [email protected] " , IdentityScope:: User) .unwrap ( ) ;
6551
let ( _ , secret) = oxygenhub:: keys:: load_secret( & h.app.config.keys_dir ( ) ) .unwrap ( ) ;
6552
let now = db:: now_ms( ) as u64 ;
6553
let cert = SignedCertificate:: sign(
6554
Certificate {
6555
key,
6556
issuer: hub_id.hash ( ) ,
6557
subject: " agent:claude@andy" .into ( ) ,
6558
claims: vec! [ ] ,
6559
scope: Scope:: CHANGES | Scope:: REFS ,
6560
not_before: now - 60_000 ,
6561
expires: now + 600_000 ,
8aa5ef38 Andy McCall 19 days ago ↶
6562
grant: None ,
3eb491e9 Andy McCall 19 days ago ↶
6563
} ,
6564
& hub_id,
6565
& secret,
6566
) ;
6567
b.store ( ) .put_identity ( & hub_id) .unwrap ( ) ;
6568
b.store ( ) .put_certificate ( & cert) .unwrap ( ) ;
6569
write ( & work.path ( ) .join ( " b" ) , " b.txt" , " agent work\n " ) ;
6570
b.commit ( " work" , & [ ] ) .unwrap ( ) ;
6571
let r = b.push ( " origin" , false ) .unwrap ( ) ;
6572
assert! ( r.iter ( ) .all ( | x | x.ok ) , " {r:?}" ) ;
6573
6574
6575
let refs = open_remote ( & url, & RemoteOptions:: default( ) ) .unwrap ( ) .refs ( ) .unwrap ( ) ;
6576
let main = refs.refs.iter ( ) .find ( | r | r.name = = " main" ) .unwrap ( ) ;
6577
let ( attest, acc) = ( main.attestation.clone ( ) .unwrap ( ) , main.accepted.clone ( ) .expect ( " an acceptance" ) ) ;
6578
assert! ( matches! ( acc.attest.body, AttestBody:: Accepted { attest: r } if r = = attest.attest.hash ( ) ) ) ;
6579
assert_eq! ( acc.sigs[ 0 ] .key, hub_id.hash ( ) ) ;
6580
assert! ( acc.attest.hlc.wall_ms > = now) ;
6581
6582
a.pull ( " origin" ) .unwrap ( ) ;
6583
write ( & work.path ( ) .join ( " a" ) , " a.txt" , " again\n " ) ;
6584
a.commit ( " again" , & [ ] ) .unwrap ( ) ;
6585
assert! ( a.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
6586
let refs = open_remote ( & url, & RemoteOptions:: default( ) ) .unwrap ( ) .refs ( ) .unwrap ( ) ;
6587
assert! ( refs.refs.iter ( ) .find ( | r | r.name = = " main" ) .unwrap ( ) .accepted.is_none ( ) ) ;
6588
}
8aa5ef38 Andy McCall 19 days ago ↶
6589
b5c199bd Andy McCall 16 days ago ↶
6590
6591
6592
6593
6594
# [ tokio ::test ( flavor = " multi_thread" ) ]
6595
async fn a_tag_by_a_key_the_hub_certified_is_accepted_and_fetched ( ) {
6596
use oxy_core:: attest:: AttestBody;
6597
use oxy_core:: cert:: { Certificate, SignedCertificate} ;
6598
use oxy_core:: identity:: Scope;
6599
use oxy_core:: repo:: IdentityScope;
6600
use oxy_core:: tlog:: Entry;
6601
let h = hub ( ) .await;
6602
create ( & h.app, " andy" , " demo" ) .await;
6603
let url = format! ( " {} /andy/demo" , h.base) ;
6604
let work = tempfile:: tempdir( ) .unwrap ( ) ;
6605
let mut a = client ( & url, & work.path ( ) .join ( " a" ) ) .with_home ( work.path ( ) .join ( " home-a" ) ) ;
6606
a.create_identity ( " Andy" , " [email protected] " , IdentityScope:: User) .unwrap ( ) ;
6607
write ( & work.path ( ) .join ( " a" ) , " a.txt" , " a\n " ) ;
6608
a.commit ( " a" , & [ ] ) .unwrap ( ) ;
6609
a.policy_init ( ) .unwrap ( ) ;
6610
assert! ( a.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
6611
let hub_id = h.app.hub_key.identity.clone ( ) ;
6612
a.policy_grant_subject ( " agent:claude@andy" , hub_id.clone ( ) , Scope:: CHANGES | Scope:: REFS , None ) .unwrap ( ) ;
6613
assert! ( a.push ( " origin" , false ) .unwrap ( ) .iter ( ) .any ( | r | r.name = = " policy" & & r.ok ) ) ;
6614
6615
6616
let mut b = Repo:: clone( & url, & work.path ( ) .join ( " b" ) ) .unwrap ( ) .with_home ( work.path ( ) .join ( " home-b" ) ) ;
6617
b.config.set ( " remote.origin.token" , " site" ) .unwrap ( ) ;
6618
b.save_config ( ) .unwrap ( ) ;
6619
let ( _ , key) = b.create_identity ( " claude" , " [email protected] " , IdentityScope:: User) .unwrap ( ) ;
6620
let ( _ , secret) = oxygenhub:: keys:: load_secret( & h.app.config.keys_dir ( ) ) .unwrap ( ) ;
6621
let now = db:: now_ms( ) as u64 ;
6622
let cert = SignedCertificate:: sign(
6623
Certificate {
6624
key,
6625
issuer: hub_id.hash ( ) ,
6626
subject: " agent:claude@andy" .into ( ) ,
6627
claims: vec! [ ] ,
6628
scope: Scope:: CHANGES | Scope:: REFS ,
6629
not_before: now - 60_000 ,
6630
expires: now + 600_000 ,
6631
grant: None ,
6632
} ,
6633
& hub_id,
6634
& secret,
6635
) ;
6636
b.store ( ) .put_identity ( & hub_id) .unwrap ( ) ;
6637
b.store ( ) .put_certificate ( & cert) .unwrap ( ) ;
6638
let heads = b.bookmark_heads ( ) .unwrap ( ) ;
6639
b.tag_create ( " v0.2.0" , & heads, " OxygenFlow 0.2.0" ) .unwrap ( ) ;
6640
let r = b.push ( " origin" , false ) .unwrap ( ) ;
6641
assert! ( r.iter ( ) .any ( | x | x.name = = " tags/v0.2.0" & & x.ok ) , " {r:?}" ) ;
6642
6643
6644
6645
h.app.tlog.refresh_checkpoint ( & hub_id, & secret) .unwrap ( ) ;
6646
let refs = open_remote ( & url, & RemoteOptions:: default( ) ) .unwrap ( ) .refs ( ) .unwrap ( ) ;
6647
let tag = refs.tags.iter ( ) .find ( | t | t.name = = " v0.2.0" ) .expect ( " the tag" ) ;
6648
let ( attest, acc) = ( tag.attestation.clone ( ) .unwrap ( ) , tag.accepted.clone ( ) .expect ( " an acceptance" ) ) ;
6649
assert! ( matches! ( acc.attest.body, AttestBody:: Accepted { attest: r } if r = = attest.attest.hash ( ) ) ) ;
6650
assert_eq! ( acc.sigs[ 0 ] .key, hub_id.hash ( ) ) ;
6651
assert! ( acc.attest.hlc.wall_ms > = now) ;
6652
let logged = tag.logged.as_ref ( ) .expect ( " the tag's place in the log" ) ;
6653
let entry = Entry:: decode( & logged.entry) .unwrap ( ) ;
6654
assert_eq! ( ( entry.kind.as_str ( ) , entry.object) , ( " tag" , attest.attest.hash ( ) ) ) ;
6655
let logged_acceptance = db:: tlog:: entry_of( & h.app.db, & acc.attest.hash ( ) .to_hex ( ) , " accepted" ) .await.unwrap ( ) ;
6656
assert! ( logged_acceptance.is_some ( ) , " the acceptance is logged too" ) ;
6657
6658
6659
let c = Repo:: clone( & url, & work.path ( ) .join ( " c" ) ) .unwrap ( ) .with_home ( work.path ( ) .join ( " home-c" ) ) ;
6660
assert_eq! ( c.tag_heads ( " v0.2.0" ) .unwrap ( ) , Some ( heads) ) ;
6661
let report = c.fetch ( " origin" ) .unwrap ( ) ;
6662
assert! ( report.skipped.is_empty ( ) , " {:?}" , report.skipped) ;
6663
6664
a.pull ( " origin" ) .unwrap ( ) ;
6665
let t = a.tags ( ) .unwrap ( ) .into_iter ( ) .find ( | t | t.name = = " v0.2.0" ) .expect ( " pulled" ) ;
6666
assert_eq! ( t.signer.as_deref ( ) , Some ( " claude" ) ) ;
a0a1601e Andy McCall 16 days ago ↶
6667
6668
6669
6670
6671
a.tag_create ( " v0.1.0" , & a.bookmark_heads ( ) .unwrap ( ) , " by Andy" ) .unwrap ( ) ;
6672
assert! ( a.push ( " origin" , false ) .unwrap ( ) .iter ( ) .any ( | r | r.name = = " tags/v0.1.0" & & r.ok ) ) ;
6673
let bare = oxy_core:: protocol:: http_get( & format! ( " {url} /oxy/v1/refs" ) , & RemoteOptions:: default( ) ) .unwrap ( ) ;
6674
let ( _ , old) = oxy_core:: protocol:: wire:: decode_refs( & bare) .unwrap ( ) ;
6675
let names: Vec< String> = old.tags.into_iter ( ) .map ( | t | t.name ) .collect ( ) ;
6676
assert_eq! ( names, [ " v0.1.0" ] , " no Oxy-Features header" ) ;
6677
let refs = open_remote ( & url, & RemoteOptions:: default( ) ) .unwrap ( ) .refs ( ) .unwrap ( ) ;
6678
let mut names: Vec< String> = refs.tags.into_iter ( ) .map ( | t | t.name ) .collect ( ) ;
6679
names.sort ( ) ;
6680
assert_eq! ( names, [ " v0.1.0" , " v0.2.0" ] , " a current client declares certified-tags" ) ;
b5c199bd Andy McCall 16 days ago ↶
6681
}
6682
8aa5ef38 Andy McCall 19 days ago ↶
6683
6684
6685
6686
6687
6688
# [ tokio ::test ( flavor = " multi_thread" ) ]
6689
async fn an_agent_with_a_grant_pushes_in_its_owner_s_name ( ) {
6690
use oxy_core:: cert:: { Grant, SignedGrant} ;
6691
use oxy_core:: identity:: { Identity, Scope, SecretKey} ;
6692
use oxy_core:: repo:: { IdentityScope, SignatureStatus} ;
6693
let h = hub ( ) .await;
6694
let andy = db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
6695
let ( pat, pat_hash) = oxygenhub:: auth:: new_token( ) ;
6696
db:: accounts:: create_token( & h.app.db, andy.id, " cli" , & pat_hash, " repo:write" , " " , None ) .await.unwrap ( ) ;
6697
create ( & h.app, " andy" , " demo" ) .await;
6698
let url = format! ( " {} /andy/demo" , h.base) ;
6699
let work = tempfile:: tempdir( ) .unwrap ( ) ;
6700
6701
let mut a = client ( & url, & work.path ( ) .join ( " a" ) ) .with_home ( work.path ( ) .join ( " home-a" ) ) ;
6702
let ( andy_key, andy_id) = a.create_identity ( " Andy" , " [email protected] " , IdentityScope:: User) .unwrap ( ) ;
6703
db:: accounts:: link_identity( & h.app.db, andy.id, & andy_key.to_hex ( ) , " Andy" , " [email protected] " , & andy_id.encode ( ) ) .await.unwrap ( ) ;
6704
write ( & work.path ( ) .join ( " a" ) , " a.txt" , " a\n " ) ;
6705
a.commit ( " a" , & [ ] ) .unwrap ( ) ;
6706
a.policy_init ( ) .unwrap ( ) ;
6707
assert! ( a.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
6708
let hub_id = h.app.hub_key.identity.clone ( ) ;
6709
a.policy_grant_subject ( " agent:claude@andy" , hub_id.clone ( ) , Scope:: CHANGES | Scope:: REFS , None ) .unwrap ( ) ;
6710
assert! ( a.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
6711
6712
6713
let ( _ , andy_sk) = a.secret_key ( ) .unwrap ( ) .unwrap ( ) ;
6714
let now = db:: now_ms( ) as u64 ;
6715
let grant = | by : & Identity, sk : & SecretKey, subject : & str | {
6716
SignedGrant:: sign(
6717
Grant {
6718
grantor: by.hash ( ) ,
6719
subject: subject.into ( ) ,
6720
issuer: hub_id.hash ( ) ,
6721
scope: Scope:: CHANGES | Scope:: SEAL | Scope:: REFS ,
6722
not_before: now - 60_000 ,
6723
expires: now + 86_400_000 ,
6724
} ,
6725
by,
6726
sk,
6727
)
6728
} ;
6729
let post = | g : SignedGrant| {
6730
let ( app, pat) = ( h.app.clone ( ) , pat.clone ( ) ) ;
6731
let body = serde_json:: json! ( { " grant" : oxy_core:: dsse:: base64( & g.encode ( ) ) } ) .to_string ( ) ;
6732
async move { call ( & app, " POST" , " /api/v1/agents/grants" , Some ( & pat) , Some ( body) ) .await }
6733
} ;
6734
let stray_sk = SecretKey:: generate( ) ;
6735
let stray = stray_sk.identity ( " stray" , " [email protected] " , 1 ) ;
6736
let ( st, body) = post ( grant ( & stray, & stray_sk, " agent:claude@andy" ) ) .await;
6737
assert_eq! ( st, StatusCode:: FORBIDDEN , " a key not linked to the account: {body}" ) ;
6738
let ( st, body) = post ( grant ( & andy_id, & andy_sk, " agent:claude@mallory" ) ) .await;
6739
assert_eq! ( st, StatusCode:: BAD_REQUEST , " another account's agent: {body}" ) ;
6740
let ( st, body) = post ( grant ( & andy_id, & andy_sk, " agent:claude@andy" ) ) .await;
6741
assert_eq! ( st, StatusCode:: CREATED , " {body}" ) ;
6742
let v: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
6743
let token = v[ " token" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
6744
assert! ( token.starts_with ( " oxyagent_" ) ) ;
6745
let grant_id = v[ " id" ] .as_i64 ( ) .unwrap ( ) ;
205d8f86 Andy McCall 18 days ago ↶
6746
6747
6748
let ( st, me) = call ( & h.app, " GET" , " /api/v1/user" , Some ( & token) , None ) .await;
6749
assert! ( st = = StatusCode:: OK & & me.contains ( " \" andy\" " ) , " {st} {me}" ) ;
6750
let ( st, _ ) = call ( & h.app, " GET" , " /api/v1/agents/grants" , Some ( & token) , None ) .await;
6751
assert_eq! ( st, StatusCode:: FORBIDDEN , " an agent does not manage grants" ) ;
d4a0043f Andy McCall 18 days ago ↶
6752
6753
6754
let repo = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
6755
db:: notify:: set_watch( & h.app.db, andy.id, repo.id, true ) .await.unwrap ( ) ;
6756
let body = serde_json:: json! ( { " title" : " Filed by the agent" , " body" : " " } ) .to_string ( ) ;
6757
let ( st, out) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/issues" , Some ( & token) , Some ( body) ) .await;
6758
assert_eq! ( st, StatusCode:: CREATED , " {out}" ) ;
6759
let actor: String = sqlx:: query_scalar( " SELECT actor FROM notifications WHERE account_id = $1 AND title LIKE '%Filed by the agent%'" )
6760
.bind ( andy.id)
6761
.fetch_one ( & h.app.db)
6762
.await
6763
.expect ( " told of the agent's issue" ) ;
6764
assert! ( actor.starts_with ( " andy (agent token " ) , " {actor}" ) ;
8aa5ef38 Andy McCall 19 days ago ↶
6765
6766
6767
let agent_home = work.path ( ) .join ( " home-claude" ) ;
6768
oxy_core:: credentials:: Credentials:: at( & agent_home) .save_grant ( & h.base, & token) .unwrap ( ) ;
6769
let agent = | | {
6770
let n = std:: time:: SystemTime:: now( ) .duration_since ( std:: time:: UNIX_EPOCH ) .unwrap ( ) .as_nanos ( ) ;
6771
let dir = work.path ( ) .join ( format! ( " claude-{n} " ) ) ;
6772
let mut r = Repo:: clone( & url, & dir) .unwrap ( ) .with_home ( agent_home.clone ( ) ) ;
559c3920 Andy McCall 19 days ago ↶
6773
6774
r.config.set ( " remote.origin.token" , & token) .unwrap ( ) ;
8aa5ef38 Andy McCall 19 days ago ↶
6775
r.save_config ( ) .unwrap ( ) ;
6776
r.set_user_config ( " author.key" , & andy_key.to_hex ( ) ) .unwrap ( ) ;
6777
r.set_user_config ( " author.signer" , " keyless" ) .unwrap ( ) ;
6778
( r, dir)
6779
} ;
6780
let ( claude, dir) = agent ( ) ;
6781
assert! ( ! agent_home.join ( " keys" ) .exists ( ) , " no key on disk" ) ;
6782
write ( & dir, " work.txt" , " agent work\n " ) ;
6783
let change = claude.commit ( " Agent work" , & [ ] ) .unwrap ( ) ;
6784
let r = claude.push ( " origin" , false ) .unwrap ( ) ;
6785
assert! ( r.iter ( ) .all ( | x | x.ok ) , " {r:?}" ) ;
6786
let signer = claude.signing_key ( ) .expect ( " a certified key" ) ;
6787
assert_ne! ( signer, andy_key) ;
6788
assert! ( ! agent_home.join ( " keys" ) .exists ( ) , " still no key on disk" ) ;
6789
6790
6791
6792
let host = h.app.repos.open ( & RepoName:: parse( " andy/demo" ) .unwrap ( ) ) .unwrap ( ) ;
6793
assert_eq! ( host.store ( ) .get_change_meta ( & change) .unwrap ( ) .author, andy_key) ;
6794
assert! ( matches! (
6795
host.authenticate_change ( & change) .unwrap ( ) ,
6796
SignatureStatus:: Valid { key, sealed: true , .. } if key = = signer
6797
) ) ;
6798
let refs = open_remote ( & url, & RemoteOptions:: default( ) ) .unwrap ( ) .refs ( ) .unwrap ( ) ;
6799
assert! ( refs.refs.iter ( ) .find ( | r | r.name = = " main" ) .unwrap ( ) .accepted.is_some ( ) ) ;
6800
a.pull ( " origin" ) .unwrap ( ) ;
6801
assert! ( matches! ( a.authenticate_change ( & change) .unwrap ( ) , SignatureStatus:: Valid { .. } ) ) ;
6802
a66c05a6 Andy McCall 11 days ago ↶
6803
6804
6805
6806
create ( & h.app, " andy" , " fresh" ) .await;
6807
let fresh_url = format! ( " {} /andy/fresh" , h.base) ;
6808
let fresh_dir = work.path ( ) .join ( " claude-fresh" ) ;
6809
let mut fresh = client ( & fresh_url, & fresh_dir) .with_home ( agent_home.clone ( ) ) ;
6810
fresh.config.set ( " remote.origin.token" , & token) .unwrap ( ) ;
6811
fresh.save_config ( ) .unwrap ( ) ;
6812
fresh.set_user_config ( " author.key" , & andy_key.to_hex ( ) ) .unwrap ( ) ;
6813
fresh.set_user_config ( " author.signer" , " keyless" ) .unwrap ( ) ;
6814
assert! ( fresh.store ( ) .get_identity ( & andy_key) .is_err ( ) , " the fresh repository has never seen Andy" ) ;
6815
write ( & fresh_dir, " first.txt" , " first\n " ) ;
6816
let first = fresh.commit ( " First, by the agent" , & [ ] ) .unwrap ( ) ;
6817
let r = fresh.push ( " origin" , false ) .unwrap ( ) ;
6818
assert! ( r.iter ( ) .all ( | x | x.ok ) , " {r:?}" ) ;
6819
assert! ( matches! ( fresh.authenticate_change ( & first) .unwrap ( ) , SignatureStatus:: Valid { .. } ) , " locally" ) ;
6820
let fresh_host = h.app.repos.open ( & RepoName:: parse( " andy/fresh" ) .unwrap ( ) ) .unwrap ( ) ;
6821
assert! (
6822
matches! ( fresh_host.authenticate_change ( & first) .unwrap ( ) , SignatureStatus:: Valid { sealed: true , .. } ) ,
6823
" on the hub: {:?}" ,
6824
fresh_host.authenticate_change ( & first) .unwrap ( )
6825
) ;
6826
27babfa1 Andy McCall 10 days ago ↶
6827
6828
6829
6830
let rot = a.rotate_identity ( ) .unwrap ( ) ;
6831
db:: accounts:: link_identity( & h.app.db, andy.id, & rot.new.to_hex ( ) , " Andy" , " [email protected] " , & rot.identity.encode ( ) ) .await.unwrap ( ) ;
6832
assert! ( a.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
6833
let ( mut warm, dir) = agent ( ) ;
6834
warm.set_user_config ( " author.for" , " andy" ) .unwrap ( ) ;
6835
write ( & dir, " warm.txt" , " before the hub has said\n " ) ;
6836
warm.commit ( " Warm" , & [ ] ) .unwrap ( ) ;
6837
assert! ( warm.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
6838
let followed = std:: fs:: read_to_string( agent_home.join ( " keys" ) .join ( " author.for" ) ) .unwrap ( ) ;
6839
assert_eq! ( followed.trim ( ) , rot.new.to_hex ( ) , " the hub named Andy's newest identity" ) ;
6840
let ( follower, dir) = agent ( ) ;
6841
assert_eq! ( follower.config.author_key, Some ( rot.new) ) ;
6842
write ( & dir, " follow.txt" , " as Andy's newest identity\n " ) ;
6843
let followed = follower.commit ( " Followed" , & [ ] ) .unwrap ( ) ;
6844
assert! ( follower.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
6845
assert_eq! ( host.store ( ) .get_change_meta ( & followed) .unwrap ( ) .author, rot.new) ;
6846
assert! (
6847
matches! ( host.authenticate_change ( & followed) .unwrap ( ) , SignatureStatus:: Valid { sealed: true , .. } ) ,
6848
" {:?}" ,
6849
host.authenticate_change ( & followed) .unwrap ( )
6850
) ;
6851
8aa5ef38 Andy McCall 19 days ago ↶
6852
6853
let ( st, _ ) = call ( & h.app, " DELETE" , & format! ( " /api/v1/agents/grants/{grant_id} " ) , Some ( & pat) , None ) .await;
6854
assert_eq! ( st, StatusCode:: NO_CONTENT ) ;
6855
let ( late, dir) = agent ( ) ;
6856
write ( & dir, " late.txt" , " after the revocation\n " ) ;
6857
late.commit ( " Late" , & [ ] ) .unwrap ( ) ;
6858
let err = late.push ( " origin" , false ) .unwrap_err ( ) .to_string ( ) ;
559c3920 Andy McCall 19 days ago ↶
6859
assert! ( err.contains ( " not valid" ) | | err.contains ( " would not certify" ) , " {err}" ) ;
6860
6861
let by_agent: i64 = sqlx:: query_scalar( " SELECT COUNT(*) FROM pushes WHERE via = 'claude'" ) .fetch_one ( & h.app.db) .await.unwrap ( ) ;
6862
assert! ( by_agent > = 1 , " recorded as via the agent" ) ;
8aa5ef38 Andy McCall 19 days ago ↶
6863
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/agents/grants" , Some ( & pat) , None ) .await;
6864
let list: Vec< serde_json:: Value> = serde_json:: from_str( & body) .unwrap ( ) ;
6865
assert! ( list[ 0 ] [ " revoked_ms" ] .is_i64 ( ) & & list[ 0 ] .get ( " token" ) .is_none ( ) , " {body}" ) ;
6866
}
a9f1c0bf Andy McCall 17 days ago ↶
6867
6868
6869
6870
6871
6872
# [ tokio ::test ( flavor = " multi_thread" ) ]
6873
async fn newest_run_wins ( ) {
6874
use oxy_core:: hash:: ChangeHash;
6875
let h = hub ( ) .await;
6876
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
6877
create ( & h.app, " andy" , " newest" ) .await;
6878
let url = format! ( " {} /andy/newest" , h.base) ;
6879
let work = tempfile:: tempdir( ) .unwrap ( ) ;
6880
let c = client ( & url, work.path ( ) ) ;
6881
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
6882
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
6883
c.commit ( " Start" , & [ ] ) .unwrap ( ) ;
6884
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
6885
let pushed = runs_of ( & h.app, " andy" , " newest" ) .await[ 0 ] .clone ( ) ;
6886
assert_eq! ( ( pushed.number, pushed.trigger.as_str ( ) ) , ( 1 , " push" ) ) ;
6887
let heads: Vec< ChangeHash> = pushed.head_list ( ) .into_iter ( ) .filter_map ( ChangeHash:: from_hex) .collect ( ) ;
6888
let start = | row : db:: repos:: RepoRow, bookmark : & 'static str , trigger : & 'static str | {
6889
let ( app, state, heads) = ( h.app.clone ( ) , pushed.state.clone ( ) , heads.clone ( ) ) ;
6890
async move {
6891
oxygenhub:: flow:: start_run_scoped( & app, & row, bookmark, & state, & heads, " andy" , trigger, None ) .await.unwrap ( ) .expect ( " a run" )
6892
}
6893
} ;
6894
let row = db:: repos:: get( & h.app.db, " andy" , " newest" ) .await.unwrap ( ) .unwrap ( ) ;
6895
let run = | n : i64 | {
6896
let ( app, id) = ( h.app.clone ( ) , row.id) ;
6897
async move { db:: flow:: run( & app.db, id, n) .await.unwrap ( ) .unwrap ( ) }
6898
} ;
6899
let superseded_by = | n : i64 , by : i64 | {
6900
let app = h.app.clone ( ) ;
6901
let run = run ( n) ;
6902
async move {
6903
let r = run.await;
6904
assert_eq! ( r.status, " cancelled" , " run #{n} is superseded" ) ;
6905
let jobs = db:: flow:: jobs( & app.db, r.id) .await.unwrap ( ) ;
6906
let why = format! ( " superseded by run #{by} " ) ;
6907
assert! ( jobs.iter ( ) .all ( | j | j.status = = " cancelled" & & j.error = = why ) , " {jobs:?}" ) ;
6908
}
6909
} ;
6910
6911
6912
6913
assert_eq! ( start ( row.clone ( ) , " proposal/7" , " proposal" ) .await, 2 ) ;
6914
assert_eq! ( start ( row.clone ( ) , " proposal/7" , " proposal" ) .await, 3 ) ;
6915
superseded_by ( 2 , 3 ) .await;
6916
assert_eq! ( start ( row.clone ( ) , " proposal/8" , " proposal" ) .await, 4 ) ;
6917
assert! ( run ( 3 ) .await.live ( ) , " another proposal's run is not superseded" ) ;
6918
assert! ( run ( 1 ) .await.live ( ) , " a proposal's run leaves push runs alone" ) ;
6919
6920
6921
6922
assert_eq! ( start ( row.clone ( ) , " main" , " manual" ) .await, 5 ) ;
6923
assert_eq! ( start ( row.clone ( ) , " main" , " rerun:1:push" ) .await, 6 ) ;
6924
assert! ( run ( 1 ) .await.live ( ) , " a manual run or a rerun supersedes nothing" ) ;
6925
assert_eq! ( start ( row.clone ( ) , " main" , " push" ) .await, 7 ) ;
6926
superseded_by ( 1 , 7 ) .await;
6927
assert! ( run ( 5 ) .await.live ( ) & & run ( 6 ) .await.live ( ) , " a push leaves manual runs and reruns alone" ) ;
6928
6929
6930
6931
db:: repos:: set_flow_keep_running( & h.app.db, row.id, true ) .await.unwrap ( ) ;
6932
let kept = db:: repos:: get( & h.app.db, " andy" , " newest" ) .await.unwrap ( ) .unwrap ( ) ;
6933
assert! ( kept.flow_keep_running) ;
6934
let three = run ( 3 ) .await;
6935
sqlx:: query( " UPDATE flow_runs SET status = 'running' WHERE id = $1" ) .bind ( three.id) .execute ( & h.app.db) .await.unwrap ( ) ;
6936
sqlx:: query( " UPDATE flow_jobs SET status = 'running' WHERE run_id = $1 AND name = 'build'" )
6937
.bind ( three.id)
6938
.execute ( & h.app.db)
6939
.await
6940
.unwrap ( ) ;
6941
assert_eq! ( start ( kept.clone ( ) , " proposal/7" , " proposal" ) .await, 8 ) ;
6942
assert_eq! ( start ( kept.clone ( ) , " proposal/7" , " proposal" ) .await, 9 ) ;
6943
assert_eq! ( run ( 3 ) .await.status, " running" , " a running run is kept" ) ;
6944
superseded_by ( 8 , 9 ) .await;
6945
6946
db:: repos:: set_flow_keep_running( & h.app.db, row.id, false ) .await.unwrap ( ) ;
6947
let row = db:: repos:: get( & h.app.db, " andy" , " newest" ) .await.unwrap ( ) .unwrap ( ) ;
6948
assert_eq! ( start ( row.clone ( ) , " proposal/7" , " proposal" ) .await, 10 ) ;
6949
superseded_by ( 3 , 10 ) .await;
6950
superseded_by ( 9 , 10 ) .await;
6951
}
8501fd58 Andy McCall 10 days ago ↶
6952
6953
6954
6955
6956
# [ tokio ::test ( flavor = " multi_thread" ) ]
6957
async fn runners_are_offered_updates_by_their_nearest_setting ( ) {
6958
use oxy_core:: identity:: SecretKey;
6959
let h = hub ( ) .await;
6960
create ( & h.app, " andy" , " oxygenflow" ) .await;
6961
let res = oxygenhub:: router( h.app.clone ( ) )
6962
.oneshot (
6963
Request:: post( " /register" )
6964
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
6965
.body ( Body:: from( " username=carol&email=carol%40example.com&password=correct-horse-battery&setup=test-setup" ) )
6966
.unwrap ( ) ,
6967
)
6968
.await
6969
.unwrap ( ) ;
6970
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
6971
let admin = db:: accounts:: account_by_name( & h.app.db, " carol" ) .await.unwrap ( ) .unwrap ( ) ;
6972
db:: accounts:: set_admin( & h.app.db, admin.id, true ) .await.unwrap ( ) ;
6973
let web = | method : & 'static str , path : & str , body : Option< String> | {
6974
let ( app, cookie, path) = ( h.app.clone ( ) , cookie.clone ( ) , path.to_string ( ) ) ;
6975
async move {
6976
let mut req = Request:: builder( ) .method ( method) .uri ( path) .header ( header:: COOKIE , cookie) ;
6977
if body.is_some ( ) {
6978
req = req.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) ;
6979
}
6980
let res = oxygenhub:: router( app) .oneshot ( req.body ( body.map ( Body:: from) .unwrap_or_else ( Body:: empty) ) .unwrap ( ) ) .await.unwrap ( ) ;
6981
let st = res.status ( ) ;
6982
( st, String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) )
6983
}
6984
} ;
6985
let ( st, page) = web ( " GET" , " /admin/runners/updates" , None ) .await;
6986
assert_eq! ( st, StatusCode:: OK , " {page}" ) ;
6987
let csrf = {
6988
let i = page.find ( " name=\" _csrf\" value=\" " ) .unwrap ( ) + " name=\" _csrf\" value=\" " .len ( ) ;
6989
page[ i.. ] .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( )
6990
} ;
6991
assert_eq! ( call ( & h.app, " GET" , " /admin/runners/updates" , None , None ) .await.0 , StatusCode:: NOT_FOUND , " administrators only" ) ;
6992
let mut creds = vec! [ ] ;
6993
for name in [ " plex" , " narvi" ] {
6994
let ( _ , page) = web ( " POST" , " /admin/runners/token" , Some ( format! ( " _csrf={csrf} " ) ) ) .await;
6995
let i = page.find ( " --token " ) .unwrap ( ) + " --token " .len ( ) ;
6996
let token: String = page[ i.. ] .chars ( ) .take_while ( | c | ! c.is_whitespace ( ) ) .collect ( ) ;
6997
let identity = SecretKey:: generate( ) .identity ( & format! ( " {name} (OxygenFlow runner)" ) , " " , 1 ) ;
6998
let info = serde_json:: json! ( { " arch" : " x86_64" , " os" : " linux" , " version" : " 0.1.0 (build)" , " max_jobs" : 1 } ) ;
6999
let reg = serde_json:: json! ( { " token" : token, " name" : name, " identity" : hex:: encode( identity.encode ( ) ) , " info" : info } ) ;
7000
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await;
7001
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
7002
creds.push ( serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " credential" ] .as_str ( ) .unwrap ( ) .to_string ( ) ) ;
7003
}
7004
let ask = | who : usize | {
7005
let ( app, cred) = ( h.app.clone ( ) , creds[ who] .clone ( ) ) ;
7006
async move {
7007
let ( st, body) = call ( & app, " GET" , " /api/v1/flow/runner/update" , Some ( & cred) , None ) .await;
7008
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
7009
serde_json:: from_str:: < flow_core:: api:: UpdateAnswer> ( & body) .unwrap ( )
7010
}
7011
} ;
7012
let a = ask ( 0 ) .await;
7013
assert! ( ! a.policy.updates & & a.offer.is_none ( ) , " off unless chosen: {a:?}" ) ;
7014
assert_eq! ( a.note, " updates are off (default)" ) ;
7015
assert_eq! ( call ( & h.app, " GET" , " /api/v1/flow/runner/update" , Some ( " flowrunner_nope" ) , None ) .await.0 , StatusCode:: UNAUTHORIZED ) ;
7016
7017
7018
let repo = db:: repos:: get( & h.app.db, " andy" , " oxygenflow" ) .await.unwrap ( ) .unwrap ( ) ;
7019
let rel = db:: releases:: create( & h.app.db, repo.id, " v0.2.0" , " 0.2.0" , " " , false , None ) .await.unwrap ( ) ;
c26267b3 Andy McCall 7 days ago ↶
7020
let build: Vec< u8 > = ( 0 .. 300_000 u32 ) .map ( | i | ( i.wrapping_mul ( 2654435761 ) > > 13 ) as u8 ) .collect ( ) ;
7021
let build_hash = blake3:: hash( & build) .to_hex ( ) .to_string ( ) ;
7022
let at = h.app.config.data.join ( " assets" ) .join ( & build_hash[ .. 2 ] ) ;
7023
std:: fs:: create_dir_all( & at) .unwrap ( ) ;
7024
std:: fs:: write( at.join ( & build_hash) , & build) .unwrap ( ) ;
7025
db:: releases:: add_asset( & h.app.db, rel.id, " flow-linux-x86_64" , build.len ( ) as i64 , & build_hash, " application/octet-stream" )
7026
.await
7027
.unwrap ( ) ;
8501fd58 Andy McCall 10 days ago ↶
7028
let ( _ , page) = web ( " POST" , " /admin/runners/updates/source" , Some ( format! ( " _csrf={csrf} &repo=andy%2Fnope" ) ) ) .await;
7029
assert! ( page.contains ( " There is no repository" ) , " {page}" ) ;
7030
let ( _ , page) = web ( " POST" , " /admin/runners/updates/source" , Some ( format! ( " _csrf={csrf} &repo=andy%2Foxygenflow" ) ) ) .await;
7031
assert! ( page.contains ( " runner builds come from andy/oxygenflow" ) , " {page}" ) ;
7032
7033
7034
let plex = db:: runners:: list( & h.app.db) .await.unwrap ( ) .into_iter ( ) .find ( | r | r.name = = " plex" ) .unwrap ( ) ;
7035
let ( _ , page) =
7036
web ( " POST" , " /admin/runners/updates/groups" , Some ( format! ( " _csrf={csrf} &name=canary&mode=chosen&runner={} " , plex.id) ) ) .await;
7037
assert! ( page.contains ( " Test group canary made: 1 chosen runners" ) , " {page}" ) ;
7038
let group = db:: runner_updates:: groups( & h.app.db) .await.unwrap ( ) .remove ( 0 ) ;
7039
let ( _ , page) =
7040
web ( " POST" , " /admin/runners/updates/policy" , Some ( format! ( " _csrf={csrf} &scope=group&key={} &updates=on" , group.id) ) ) .await;
7041
assert! ( page.contains ( " Saved for group canary: updates on." ) , " {page}" ) ;
7042
let a = ask ( 0 ) .await;
7043
let offer = a.offer.clone ( ) .expect ( " plex is offered the build" ) ;
7044
assert_eq! ( ( offer.version.as_str ( ) , offer.asset.as_str ( ) , offer.blake3.len ( ) ) , ( " 0.2.0" , " flow-linux-x86_64" , 64 ) ) ;
7045
assert! ( offer.url.starts_with ( " /andy/oxygenflow/asset/" ) & & ! offer.older, " {offer:?}" ) ;
7046
assert_eq! ( a.policy.from[ 0 ] , " group canary" ) ;
7047
assert! ( ask ( 1 ) .await.offer.is_none ( ) , " narvi is not in the group" ) ;
c26267b3 Andy McCall 7 days ago ↶
7048
7049
7050
let mut shared = None ;
7051
for _ in 0 .. 100 {
7052
shared = h.app.images ( ) .manifest ( & build_hash) ;
7053
if shared.is_some ( ) {
7054
break ;
7055
}
7056
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
7057
}
7058
let m = shared.expect ( " the build is shared with the swarm" ) ;
7059
assert_eq! ( ( m.size, m.arch.as_deref ( ) ) , ( build.len ( ) as u64 , Some ( " x86_64" ) ) ) ;
7060
let mut whole = vec! [ ] ;
7061
for c in & m.chunks {
7062
whole.extend ( h.app.images ( ) .chunk ( c.hash ( ) ) .unwrap ( ) ) ;
7063
}
7064
assert_eq! ( whole, build) ;
7065
let kept = oxygenhub:: images:: kept( & h.app.db) .await.unwrap ( ) ;
7066
assert! ( kept.iter ( ) .any ( | k | k.hash = = build_hash & & k.uploader = = " runner build flow 0.2.0" ) ) ;
8501fd58 Andy McCall 10 days ago ↶
7067
7068
7069
let ( _ , page) = web ( " POST" , " /admin/runners/updates/policy" , Some ( format! ( " _csrf={csrf} &scope=site&updates=off&dry_run=on" ) ) ) .await;
7070
assert! ( page.contains ( " Saved for the site" ) , " {page}" ) ;
7071
let a = ask ( 0 ) .await;
7072
assert! ( a.policy.dry_run & & a.policy.from[ 3 ] = = " site" & & a.note.contains ( " dry run" ) , " {a:?}" ) ;
7073
let narvi = db:: runners:: list( & h.app.db) .await.unwrap ( ) .into_iter ( ) .find ( | r | r.name = = " narvi" ) .unwrap ( ) ;
7074
web ( " POST" , " /admin/runners/updates/policy" , Some ( format! ( " _csrf={csrf} &scope=runner&key={} &updates=on&channel=0.1.0" , narvi.id) ) )
7075
.await;
7076
let b = ask ( 1 ) .await;
7077
assert! ( b.policy.updates & & b.offer.is_none ( ) & & b.note.contains ( " has no release" ) , " {b:?}" ) ;
7078
7079
7080
let now = time:: OffsetDateTime:: now_utc( ) .hour ( ) ;
7081
let ( from, to) = ( ( now + 2 ) % 24 , ( now + 4 ) % 24 ) ;
7082
web (
7083
" POST" ,
7084
" /admin/runners/updates/policy" ,
7085
Some ( format! ( " _csrf={csrf} &scope=group&key={} &updates=on&window={from:02} -{to:02} " , group.id) ) ,
7086
)
7087
.await;
7088
let a = ask ( 0 ) .await;
7089
assert! ( a.offer.is_none ( ) & & a.note.contains ( " waits for its window" ) , " {a:?}" ) ;
7090
7091
7092
let ( _ , page) = web ( " POST" , " /admin/runners/updates/policy" , Some ( format! ( " _csrf={csrf} &scope=site&channel=latest" ) ) ) .await;
7093
assert! ( page.contains ( " A channel is stable, canary or a version" ) , " {page}" ) ;
7094
let ( _ , page) = web ( " POST" , " /admin/runners/updates/policy" , Some ( format! ( " _csrf={csrf} &scope=site&window=9" ) ) ) .await;
7095
assert! ( page.contains ( " a window is two hours" ) , " {page}" ) ;
7096
let ( _ , page) = web ( " POST" , " /admin/runners/updates/groups" , Some ( format! ( " _csrf={csrf} &name=farm&mode=percent&percent=0" ) ) ) .await;
7097
assert! ( page.contains ( " A percentage is a whole number from 1 to 100." ) , " {page}" ) ;
7098
7099
7100
let report = serde_json:: json! ( { " note" : " dry run: flow 0.2.0 checks out" } ) .to_string ( ) ;
7101
assert_eq! ( call ( & h.app, " POST" , " /api/v1/flow/runner/update" , Some ( & creds[ 0 ] ) , Some ( report) ) .await.0 , StatusCode:: NO_CONTENT ) ;
7102
let ( _ , page) = web ( " GET" , " /admin/runners/updates" , None ) .await;
7103
assert! ( page.contains ( " dry run: flow 0.2.0 checks out" ) & & page.contains ( " >canary</span>" ) , " {page}" ) ;
7104
assert! ( page.contains ( " updates on (group canary)" ) & & page.contains ( " updates on (runner) · channel 0.1.0 (runner)" ) , " {page}" ) ;
7105
9bbca80f Andy McCall 10 days ago ↶
7106
7107
let ( _ , tokens) = web ( " GET" , " /settings/tokens" , None ) .await;
7108
let tcsrf = {
7109
let i = tokens.find ( " name=\" _csrf\" value=\" " ) .unwrap ( ) + " name=\" _csrf\" value=\" " .len ( ) ;
7110
tokens[ i.. ] .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( )
7111
} ;
7112
let ( _ , made) = web ( " POST" , " /settings/tokens" , Some ( format! ( " _csrf={tcsrf} &name=admin&scopes=user%3Aread" ) ) ) .await;
7113
let i = made.find ( " oxyhub_" ) .unwrap ( ) ;
7114
let admin_token: String = made[ i.. ] .chars ( ) .take_while ( | c | c.is_ascii_alphanumeric ( ) | | * c = = ' _' ) .collect ( ) ;
7115
let ( st, body) = call ( & h.app, " GET" , " /api/v1/admin/runner-updates" , Some ( & admin_token) , None ) .await;
7116
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
7117
let all: hub_api:: RunnerUpdates = serde_json:: from_str( & body) .unwrap ( ) ;
7118
assert_eq! ( all.repo, " andy/oxygenflow" ) ;
7119
assert! ( all.settings.iter ( ) .any ( | s | s.scope = = " group" & & s.updates = = Some ( true ) & & s.window.is_some ( ) ) , " {all:?}" ) ;
7120
let plex_row = all.runners.iter ( ) .find ( | r | r.name = = " plex" ) .unwrap ( ) ;
7121
assert! ( plex_row.updates & & plex_row.from[ 0 ] = = " group canary" & & plex_row.groups = = [ " canary" ] , " {plex_row:?}" ) ;
7122
let put = | body : serde_json:: Value| {
7123
let ( app, t) = ( h.app.clone ( ) , admin_token.clone ( ) ) ;
7124
async move { call ( & app, " PUT" , " /api/v1/admin/runner-updates/settings" , Some ( & t) , Some ( body.to_string ( ) ) ) .await }
7125
} ;
7126
let ( st, body) =
7127
put ( serde_json:: json! ( { " scope" : " runner" , " key" : narvi.id.to_string ( ) , " updates" : true , " channel" : " 0.1.0" , " dry_run" : true } ) )
7128
.await;
7129
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
7130
assert! ( body.contains ( " Saved for runner narvi: updates on, channel 0.1.0, dry run on." ) , " {body}" ) ;
7131
let ( st, body) = put ( serde_json:: json! ( { " scope" : " site" , " channel" : " latest" } ) ) .await;
7132
assert_eq! ( st, StatusCode:: UNPROCESSABLE_ENTITY , " {body}" ) ;
7133
assert_eq! (
7134
call ( & h.app, " GET" , " /api/v1/admin/runner-updates" , Some ( & creds[ 0 ] ) , None ) .await.0 ,
7135
StatusCode:: UNAUTHORIZED ,
7136
" not a runner's"
7137
) ;
7138
faf5246c Andy McCall 10 days ago ↶
7139
7140
let report = serde_json:: json! ( { " note" : " rolled back from flow 0.2.0 to 0.1.0: 3 starts without reaching the hub" } ) .to_string ( ) ;
7141
assert_eq! ( call ( & h.app, " POST" , " /api/v1/flow/runner/update" , Some ( & creds[ 0 ] ) , Some ( report) ) .await.0 , StatusCode:: NO_CONTENT ) ;
7142
let a = ask ( 0 ) .await;
7143
assert! ( a.offer.is_none ( ) & & a.note.contains ( " flow 0.2.0 is held: it rolled back on plex (3 starts" ) , " {a:?}" ) ;
7144
let ( _ , page) = web ( " GET" , " /admin/runners/updates" , None ) .await;
7145
assert! ( page.contains ( " Held builds" ) & & page.contains ( " <strong>flow 0.2.0</strong>" ) , " {page}" ) ;
9bbca80f Andy McCall 10 days ago ↶
7146
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/admin/runner-updates" , Some ( & admin_token) , None ) .await;
7147
assert! ( body.contains ( " \" version\" :\" 0.2.0\" " ) & & body.contains ( " \" runner\" :\" plex\" " ) , " the hold through the API: {body}" ) ;
faf5246c Andy McCall 10 days ago ↶
7148
let ( _ , page) = web ( " POST" , " /admin/runners/updates/holds/lift" , Some ( format! ( " _csrf={csrf} &version=0.2.0" ) ) ) .await;
7149
assert! ( page.contains ( " flow 0.2.0 is offered again" ) , " {page}" ) ;
9bbca80f Andy McCall 10 days ago ↶
7150
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/admin/runner-updates/holds/0.2.0/lift" , Some ( & admin_token) , Some ( " {}" .into ( ) ) ) .await;
7151
assert_eq! ( st, StatusCode:: NOT_FOUND , " no longer held" ) ;
faf5246c Andy McCall 10 days ago ↶
7152
assert! ( ask ( 0 ) .await.note.contains ( " waits for its window" ) , " lifted: back to what its settings say" ) ;
7153
let ( _ , page) = web ( " POST" , " /admin/runners/updates/holds/lift" , Some ( format! ( " _csrf={csrf} &version=0.2.0" ) ) ) .await;
7154
assert! ( page.contains ( " flow 0.2.0 is not held." ) , " {page}" ) ;
7155
8501fd58 Andy McCall 10 days ago ↶
7156
7157
let ( _ , page) =
7158
web ( " POST" , & format! ( " /admin/runners/updates/groups/{} " , group.id) , Some ( format! ( " _csrf={csrf} &mode=percent&percent=100" ) ) ) .await;
7159
assert! ( page.contains ( " Test group canary is now 100% of its runners." ) , " {page}" ) ;
7160
assert! ( ask ( 1 ) .await.policy.from[ 0 ] = = " runner" , " narvi's own setting is nearer than any group" ) ;
7161
let ( _ , page) = web ( " POST" , & format! ( " /admin/runners/updates/groups/{} /delete" , group.id) , Some ( format! ( " _csrf={csrf} " ) ) ) .await;
7162
assert! ( page.contains ( " Test group canary removed" ) , " {page}" ) ;
7163
assert! ( ! ask ( 0 ) .await.policy.updates, " back to the site's off" ) ;
7164
}
d7050703 Andy McCall 10 days ago ↶
7165
1ec29691 Andy McCall 8 days ago ↶
7166
7167
7168
7169
# [ tokio ::test ( flavor = " multi_thread" ) ]
7170
async fn a_version_reaches_wave_two_once_wave_one_proves_it ( ) {
7171
use oxy_core:: identity:: SecretKey;
7172
let h = hub ( ) .await;
7173
create ( & h.app, " andy" , " oxygenflow" ) .await;
7174
let res = oxygenhub:: router( h.app.clone ( ) )
7175
.oneshot (
7176
Request:: post( " /register" )
7177
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
7178
.body ( Body:: from( " username=carol&email=carol%40example.com&password=correct-horse-battery&setup=test-setup" ) )
7179
.unwrap ( ) ,
7180
)
7181
.await
7182
.unwrap ( ) ;
7183
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
7184
let admin = db:: accounts:: account_by_name( & h.app.db, " carol" ) .await.unwrap ( ) .unwrap ( ) ;
7185
db:: accounts:: set_admin( & h.app.db, admin.id, true ) .await.unwrap ( ) ;
7186
let web = | method : & 'static str , path : & str , body : Option< String> | {
7187
let ( app, cookie, path) = ( h.app.clone ( ) , cookie.clone ( ) , path.to_string ( ) ) ;
7188
async move {
7189
let mut req = Request:: builder( ) .method ( method) .uri ( path) .header ( header:: COOKIE , cookie) ;
7190
if body.is_some ( ) {
7191
req = req.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) ;
7192
}
7193
let res = oxygenhub:: router( app) .oneshot ( req.body ( body.map ( Body:: from) .unwrap_or_else ( Body:: empty) ) .unwrap ( ) ) .await.unwrap ( ) ;
7194
( res.status ( ) , String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) )
7195
}
7196
} ;
7197
let ( _ , page) = web ( " GET" , " /admin/runners/updates" , None ) .await;
7198
let csrf = {
7199
let i = page.find ( " name=\" _csrf\" value=\" " ) .unwrap ( ) + " name=\" _csrf\" value=\" " .len ( ) ;
7200
page[ i.. ] .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( )
7201
} ;
7202
let mut creds = vec! [ ] ;
7203
for name in [ " plex" , " narvi" ] {
7204
let ( _ , page) = web ( " POST" , " /admin/runners/token" , Some ( format! ( " _csrf={csrf} " ) ) ) .await;
7205
let i = page.find ( " --token " ) .unwrap ( ) + " --token " .len ( ) ;
7206
let token: String = page[ i.. ] .chars ( ) .take_while ( | c | ! c.is_whitespace ( ) ) .collect ( ) ;
7207
let identity = SecretKey:: generate( ) .identity ( & format! ( " {name} (OxygenFlow runner)" ) , " " , 1 ) ;
7208
let info = serde_json:: json! ( { " arch" : " x86_64" , " os" : " linux" , " version" : " 0.1.0 (build)" , " max_jobs" : 1 } ) ;
7209
let reg = serde_json:: json! ( { " token" : token, " name" : name, " identity" : hex:: encode( identity.encode ( ) ) , " info" : info } ) ;
7210
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await;
7211
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
7212
creds.push ( serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " credential" ] .as_str ( ) .unwrap ( ) .to_string ( ) ) ;
7213
}
7214
let ask = | who : usize | {
7215
let ( app, cred) = ( h.app.clone ( ) , creds[ who] .clone ( ) ) ;
7216
async move {
7217
let ( st, body) = call ( & app, " GET" , " /api/v1/flow/runner/update" , Some ( & cred) , None ) .await;
7218
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
7219
serde_json:: from_str:: < flow_core:: api:: UpdateAnswer> ( & body) .unwrap ( )
7220
}
7221
} ;
7222
let repo = db:: repos:: get( & h.app.db, " andy" , " oxygenflow" ) .await.unwrap ( ) .unwrap ( ) ;
7223
let rel = db:: releases:: create( & h.app.db, repo.id, " v0.2.0" , " 0.2.0" , " " , false , None ) .await.unwrap ( ) ;
7224
db:: releases:: add_asset( & h.app.db, rel.id, " flow-linux-x86_64" , 42 , & " ab" .repeat ( 32 ) , " application/octet-stream" ) .await.unwrap ( ) ;
7225
let ( _ , page) =
7226
web ( " POST" , " /admin/runners/updates/source" , Some ( format! ( " _csrf={csrf} &repo=andy%2Foxygenflow&soak_jobs=2&soak_hours=48" ) ) ) .await;
7227
assert! ( page.contains ( " runner builds come from andy/oxygenflow" ) & & page.contains ( " value=\" 2\" " ) , " {page}" ) ;
7228
let ( _ , page) = web ( " POST" , " /admin/runners/updates/source" , Some ( format! ( " _csrf={csrf} &repo=andy%2Foxygenflow&soak_jobs=0" ) ) ) .await;
7229
assert! ( page.contains ( " is a whole number from 1 to 1000" ) , " {page}" ) ;
7230
web ( " POST" , " /admin/runners/updates/policy" , Some ( format! ( " _csrf={csrf} &scope=site&updates=on" ) ) ) .await;
7231
7232
7233
let runners = db:: runners:: list( & h.app.db) .await.unwrap ( ) ;
7234
let id = | name : & str | runners.iter ( ) .find ( | r | r.name = = name ) .unwrap ( ) .id ;
7235
let ( _ , page) =
7236
web ( " POST" , " /admin/runners/updates/groups" , Some ( format! ( " _csrf={csrf} &name=first&mode=chosen&runner={} &wave=1" , id ( " plex" ) ) ) )
7237
.await;
7238
assert! ( page.contains ( " Test group first made: 1 chosen runners, wave 1" ) & & page.contains ( " · wave 1</small>" ) , " {page}" ) ;
7239
let ( _ , page) =
7240
web ( " POST" , " /admin/runners/updates/groups" , Some ( format! ( " _csrf={csrf} &name=rest&mode=chosen&runner={} &wave=0" , id ( " narvi" ) ) ) )
7241
.await;
7242
assert! ( page.contains ( " A wave is a whole number from 1 to 9" ) , " {page}" ) ;
7243
web ( " POST" , " /admin/runners/updates/groups" , Some ( format! ( " _csrf={csrf} &name=rest&mode=chosen&runner={} &wave=2" , id ( " narvi" ) ) ) ) .await;
7244
assert_eq! ( ask ( 0 ) .await.offer.map ( | o | o.version ) .as_deref ( ) , Some ( " 0.2.0" ) , " wave 1 goes first" ) ;
7245
let b = ask ( 1 ) .await;
7246
assert! ( b.offer.is_none ( ) & & b.note.contains ( " flow 0.2.0 waits for wave 1 to prove it: 0 of 1 runner on it, 0 jobs" ) , " {b:?}" ) ;
7247
7248
7249
let run: i64 = sqlx:: query_scalar(
7250
" INSERT INTO flow_runs (repo_id, bookmark, state, heads, status, created_ms, number) VALUES ($1, 'main', 's', 'h', 'passed', 1, 1) RETURNING id" ,
7251
)
7252
.bind ( repo.id)
7253
.fetch_one ( & h.app.db)
7254
.await
7255
.unwrap ( ) ;
7256
let job = | name : & 'static str , status : & 'static str | {
7257
let ( app, plex) = ( h.app.clone ( ) , id ( " plex" ) ) ;
7258
async move {
7259
sqlx:: query(
7260
" INSERT INTO flow_jobs (run_id, name, depth, depends, spec, sandbox, runtime, status, runner, runner_id, started_ms, finished_ms, error, runner_version)
7261
VALUES ($1, $2, 0, '', '{}', 'microvm', 'microvm', $3, 'plex', $4, $5, $5, $6, '0.2.0')" ,
7262
)
7263
.bind ( run)
7264
.bind ( name)
7265
.bind ( status)
7266
.bind ( plex)
7267
.bind ( db:: now_ms( ) )
7268
.bind ( if status = = " error" { " the runner lost its disk" } else { " " } )
7269
.execute ( & app.db)
7270
.await
7271
.unwrap ( ) ;
7272
}
7273
} ;
7274
job ( " build" , " passed" ) .await;
7275
assert! ( ask ( 1 ) .await.note.contains ( " 1 of 1 runner on it, 1 job (each needs 2" ) , " still soaking" ) ;
7276
job ( " test" , " failed" ) .await;
7277
assert_eq! ( ask ( 1 ) .await.offer.map ( | o | o.version ) .as_deref ( ) , Some ( " 0.2.0" ) , " a failed test is not the runner's fault: proved" ) ;
7278
let ( _ , page) = web ( " GET" , " /admin/runners/updates" , None ) .await;
7279
assert! ( ! page.contains ( " waits for wave" ) , " {page}" ) ;
7280
7281
7282
job ( " lint" , " error" ) .await;
7283
let ( _ , page) = web ( " GET" , " /admin/runners/updates" , None ) .await;
7284
assert! ( page.contains ( " wave 1 stopped it on plex: job lint of andy/oxygenflow run #1: the runner lost its disk" ) , " {page}" ) ;
7285
assert! ( db:: runner_updates:: held( & h.app.db, " 0.2.0" ) .await.unwrap ( ) .is_none ( ) , " a page view holds nothing" ) ;
7286
let b = ask ( 1 ) .await;
7287
assert! ( b.offer.is_none ( ) & & b.note.contains ( " flow 0.2.0 is held: wave 1 stopped it on plex" ) , " {b:?}" ) ;
7288
assert! ( ask ( 0 ) .await.note.contains ( " is held" ) , " wave 1 too" ) ;
7289
let ( _ , page) = web ( " GET" , " /admin/runners/updates" , None ) .await;
7290
assert! ( page.contains ( " Held builds" ) & & page.contains ( " wave 1 stopped it on plex" ) , " {page}" ) ;
7291
}
7292
d7050703 Andy McCall 10 days ago ↶
7293
7294
# [ tokio ::test ( flavor = " multi_thread" ) ]
7295
async fn a_wasm_jobs_placement_names_no_image ( ) {
7296
let h = hub ( ) .await;
7297
create ( & h.app, " andy" , " wasm" ) .await;
7298
let work = tempfile:: tempdir( ) .unwrap ( ) ;
7299
let c = client ( & format! ( " {} /andy/wasm" , h.base) , work.path ( ) ) ;
7300
write (
7301
work.path ( ) ,
7302
" .flow/pipeline.kdl" ,
7303
" schema-version 1\n sandbox \" w\" {\n runtime \" wasm\" \n }\n job \" count\" {\n sandbox \" w\" \n exec \" count.wasm\" \n }\n " ,
7304
) ;
7305
write ( work.path ( ) , " count.wasm" , " \0 asm" ) ;
7306
c.commit ( " A WASM job" , & [ ] ) .unwrap ( ) ;
7307
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
7308
runs_of ( & h.app, " andy" , " wasm" ) .await;
7309
let ( st, body) =
7310
call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( r # "{"runner":"test-1","runtimes":["wasm"]}"# .into ( ) ) ) .await;
7311
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
7312
let id = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
7313
let placed = db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) .placed;
7314
assert! ( placed.starts_with ( " idle" ) & & ! placed.contains ( " image" ) , " {placed}" ) ;
c26267b3 Andy McCall 7 days ago ↶
7315
7316
7317
7318
let ( st, page) = call ( & h.app, " GET" , " /andy/wasm/flow/tab" , None , None ) .await;
7319
assert_eq! ( st, StatusCode:: OK , " {page}" ) ;
7320
assert! ( page.contains ( " Run all 1" ) & & page.contains ( " "exec":["count.wasm"]" ) , " {page}" ) ;
7321
assert! ( page.contains ( " data-archive=\" /andy/wasm/archive?ref=main&format=zip\" " ) , " {page}" ) ;
7322
let ( st, zip) = get_bytes ( & h.app, " /andy/wasm/archive?ref=main&format=zip" ) .await;
7323
assert! ( st = = StatusCode:: OK & & zip.starts_with ( b " PK" ) , " {st}" ) ;
7324
assert_eq! ( get_bytes ( & h.app, " /static/wasi-worker.js" ) .await.0 , StatusCode:: OK ) ;
7325
let ( _ , runs) = call ( & h.app, " GET" , " /andy/wasm/flow" , None , None ) .await;
7326
assert! ( runs.contains ( " /andy/wasm/flow/tab" ) , " linked from Flow" ) ;
7327
let ( _ , none) = call ( & h.app, " GET" , " /andy/wasm/flow/tab?bookmark=nope" , None , None ) .await;
7328
assert! ( none.contains ( " Run all 1" ) , " an unknown bookmark shows the default's" ) ;
d7050703 Andy McCall 10 days ago ↶
7329
}
dc5c6429 Andy McCall 10 days ago ↶
7330
7331
7332
7333
# [ tokio ::test ( flavor = " multi_thread" ) ]
7334
async fn an_organisation_sets_its_own_runners_updates ( ) {
7335
use oxy_core:: identity:: SecretKey;
7336
let h = hub ( ) .await;
7337
let res = oxygenhub:: router( h.app.clone ( ) )
7338
.oneshot (
7339
Request:: post( " /register" )
7340
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
7341
.body ( Body:: from( " username=andy&email=andy%40example.com&password=correct-horse-battery&setup=test-setup" ) )
7342
.unwrap ( ) ,
7343
)
7344
.await
7345
.unwrap ( ) ;
7346
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
7347
let web = | method : & 'static str , path : String, body : Option< String> | {
7348
let ( app, cookie) = ( h.app.clone ( ) , cookie.clone ( ) ) ;
7349
async move {
7350
let mut req = Request:: builder( ) .method ( method) .uri ( path) .header ( header:: COOKIE , cookie) ;
7351
if body.is_some ( ) {
7352
req = req.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) ;
7353
}
7354
let res = oxygenhub:: router( app) .oneshot ( req.body ( body.map ( Body:: from) .unwrap_or_else ( Body:: empty) ) .unwrap ( ) ) .await.unwrap ( ) ;
7355
let st = res.status ( ) ;
7356
( st, String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) )
7357
}
7358
} ;
7359
let csrf_of = | page : & str | {
7360
let i = page.find ( " name=\" _csrf\" value=\" " ) .unwrap ( ) + " name=\" _csrf\" value=\" " .len ( ) ;
7361
page[ i.. ] .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( )
7362
} ;
7363
let ( _ , page) = web ( " GET" , " /orgs/new" .into ( ) , None ) .await;
7364
let csrf = csrf_of ( & page) ;
7365
web ( " POST" , " /orgs/new" .into ( ) , Some ( format! ( " _csrf={csrf} &name=acme&display_name=Acme" ) ) ) .await;
7366
7367
let mut creds = vec! [ ] ;
7368
for name in [ " a1" , " a2" ] {
7369
let ( _ , page) = web ( " POST" , " /orgs/acme/settings/runners/token" .into ( ) , Some ( format! ( " _csrf={csrf} " ) ) ) .await;
7370
let i = page.find ( " --token " ) .unwrap ( ) + " --token " .len ( ) ;
7371
let token: String = page[ i.. ] .chars ( ) .take_while ( | c | ! c.is_whitespace ( ) ) .collect ( ) ;
7372
let identity = SecretKey:: generate( ) .identity ( name, " " , 1 ) ;
7373
let info = serde_json:: json! ( { " arch" : " x86_64" , " os" : " linux" , " version" : " 0.1.0" , " max_jobs" : 1 } ) ;
7374
let reg = serde_json:: json! ( { " token" : token, " name" : name, " identity" : hex:: encode( identity.encode ( ) ) , " info" : info } ) ;
7375
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await;
7376
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
7377
creds.push ( serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " credential" ] .as_str ( ) .unwrap ( ) .to_string ( ) ) ;
7378
}
7379
claim ( & h.app) .await;
7380
let all = db:: runners:: list( & h.app.db) .await.unwrap ( ) ;
7381
let id = | n : & str | all.iter ( ) .find ( | r | r.name = = n ) .unwrap ( ) .id ;
7382
let set = | form : String| web ( " POST" , " /orgs/acme/settings/runners/updates/policy" .into ( ) , Some ( format! ( " _csrf={csrf} &{form} " ) ) ) ;
7383
7384
let ( _ , page) = set ( " scope=owner&key=acme&updates=on" .into ( ) ) .await;
7385
assert! ( page.contains ( " Saved for owner acme: updates on." ) , " {page}" ) ;
7386
let ( _ , page) = set ( " scope=site&updates=on" .into ( ) ) .await;
7387
assert! ( page.contains ( " Choose the site, an owner, a test group or a runner." ) , " not the site's: {page}" ) ;
7388
let ( _ , page) = set ( " scope=owner&key=andy&updates=on" .into ( ) ) .await;
7389
assert! ( page.contains ( " An organisation sets its own runners" ) , " {page}" ) ;
7390
let ( _ , page) = set ( format! ( " scope=runner&key={} &updates=on" , id ( " test-1" ) ) ) .await;
7391
assert! ( page.contains ( " That runner is gone." ) , " the site's runner is not acme's: {page}" ) ;
7392
7393
7394
let ( _ , page) = web (
7395
" POST" ,
7396
" /orgs/acme/settings/runners/updates/groups" .into ( ) ,
7397
Some ( format! ( " _csrf={csrf} &name=canary&owner=&mode=chosen&runner={} &runner={} " , id ( " a1" ) , id ( " test-1" ) ) ) ,
7398
)
7399
.await;
7400
assert! ( page.contains ( " Test group canary made: 1 chosen runners" ) , " {page}" ) ;
7401
let canary = db:: runner_updates:: groups( & h.app.db) .await.unwrap ( ) .remove ( 0 ) ;
7402
assert_eq! ( canary.owner, " acme" , " an organisation's groups are its own" ) ;
7403
let ( _ , page) = set ( format! ( " scope=group&key={} &channel=canary" , canary.id) ) .await;
7404
assert! ( page.contains ( " Saved for group canary: channel canary." ) , " {page}" ) ;
7405
let ( _ , page) = set ( format! ( " scope=runner&key={} &updates=off" , id ( " a2" ) ) ) .await;
7406
assert! ( page.contains ( " Saved for runner a2: updates off." ) , " {page}" ) ;
7407
7408
7409
let site = db:: runner_updates:: create_group( & h.app.db, " " , " everyone" , Some ( 100 ) ) .await.unwrap ( ) ;
7410
let ( _ , page) =
7411
web ( " POST" , format! ( " /orgs/acme/settings/runners/updates/groups/{} /delete" , site.id) , Some ( format! ( " _csrf={csrf} " ) ) ) .await;
7412
assert! ( page.contains ( " That test group is gone." ) , " {page}" ) ;
7413
assert! ( db:: runner_updates:: group( & h.app.db, site.id) .await.unwrap ( ) .is_some ( ) ) ;
7414
7415
7416
let ( _ , page) = web ( " GET" , " /orgs/acme/settings" .into ( ) , None ) .await;
7417
assert! ( page.contains ( " id=\" runner-updates\" " ) & & page.contains ( " >canary <small" ) , " {page}" ) ;
7418
assert! ( page.contains ( " updates on (owner acme)" ) & & page.contains ( " updates off (runner)" ) , " {page}" ) ;
7419
let ask = | cred : String| {
7420
let app = h.app.clone ( ) ;
7421
async move {
7422
let ( _ , body) = call ( & app, " GET" , " /api/v1/flow/runner/update" , Some ( & cred) , None ) .await;
7423
serde_json:: from_str:: < flow_core:: api:: UpdateAnswer> ( & body) .unwrap ( ) .policy
7424
}
7425
} ;
7426
let a1 = ask ( creds[ 0 ] .clone ( ) ) .await;
7427
assert! ( a1.updates & & a1.channel = = " canary" & & a1.from[ .. 2 ] = = [ " owner acme" .to_string ( ) , " group canary" .to_string ( ) ] , " {a1:?}" ) ;
7428
assert! ( ! ask ( creds[ 1 ] .clone ( ) ) .await.updates, " a2's own setting wins" ) ;
7429
}