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 ( ) ;
44
config.auth.push_tokens = vec! [ " site" .into ( ) ] ;
4d2e44d7 Andy McCall 1 month ago ↶
45
config.flow.runner_tokens = vec! [ " runner-secret" .into ( ) , " runner-two" .into ( ) , " runner-three" .into ( ) ] ;
42fa9554 Andy McCall 1 month ago ↶
46
config.flow.repos = vec! [ " andy/*" .into ( ) ] ;
10708ab8 Andy McCall 1 month ago ↶
47
config.registration.mode = " open" .into ( ) ;
42fa9554 Andy McCall 1 month ago ↶
48
let listener = tokio:: net:: TcpListener:: bind( " 127.0.0.1:0" ) .await.unwrap ( ) ;
49
let port = listener.local_addr ( ) .unwrap ( ) .port ( ) ;
50
config.server.base_url = format! ( " http://127.0.0.1:{port} " ) ;
51
let app = oxygenhub:: build( config) .await.unwrap ( ) ;
52
let router = oxygenhub:: router( app.clone ( ) ) ;
53
tokio:: spawn( async move { axum:: serve( listener, router) .await.unwrap ( ) } ) ;
54
Hub { app, base: format! ( " http://127.0.0.1:{port} " ) , _dir: dir }
55
}
56
57
async fn call ( app : & Shared, method : & str , path : & str , bearer : Option< & str > , json : Option< String> ) -> ( StatusCode, String) {
58
let mut req = Request:: builder( ) .method ( method) .uri ( path) ;
59
if let Some ( b) = bearer {
60
req = req.header ( header:: AUTHORIZATION , format! ( " Bearer {b} " ) ) ;
61
}
62
let body = match json {
63
Some ( j) => {
64
req = req.header ( header:: CONTENT_TYPE , " application/json" ) ;
65
Body:: from( j)
66
}
67
None => Body:: empty( ) ,
68
} ;
69
let res = oxygenhub:: router( app.clone ( ) ) .oneshot ( req.body ( body) .unwrap ( ) ) .await.unwrap ( ) ;
70
let status = res.status ( ) ;
71
let bytes = res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ;
72
( status, String :: from_utf8_lossy( & bytes) .into_owned ( ) )
73
}
74
736f6813 Andy McCall 1 month ago ↶
75
76
async fn head ( app : & Shared, path : & str ) -> ( StatusCode, String, String) {
77
let res = oxygenhub:: router( app.clone ( ) ) .oneshot ( Request:: get( path) .body ( Body:: empty( ) ) .unwrap ( ) ) .await.unwrap ( ) ;
78
let h = | k : header:: HeaderName| res.headers ( ) .get ( k) .and_then ( | v | v.to_str ( ) .ok ( ) ) .unwrap_or ( " " ) .to_string ( ) ;
79
( res.status ( ) , h ( header:: LOCATION ) , h ( header:: CONTENT_TYPE ) )
80
}
81
42fa9554 Andy McCall 1 month ago ↶
82
async fn claim ( app : & Shared) -> Option< serde_json:: Value> {
83
let ( st, body) =
84
call ( app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( r # "{"runner":"test-1","runtimes":["microvm"]}"# .into ( ) ) ) .await;
85
match st {
86
StatusCode:: NO_CONTENT => None ,
87
StatusCode:: OK => Some ( serde_json:: from_str( & body) .unwrap ( ) ) ,
88
other => panic! ( " claim: {other} {body}" ) ,
89
}
90
}
91
92
fn receipt ( job : & str , pipeline_hash : & str , exit : u32 ) -> serde_json:: Value {
93
serde_json:: json! ( {
94
" job_name" : job, " pipeline_hash" : pipeline_hash, " scope_tree_hash" : " 11" .repeat ( 32 ) ,
95
" inputs" : [ ] , " environment" : " 22" .repeat ( 32 ) , " outputs" : [ ] , " exit_code" : exit, " duration_ms" : 640
96
} )
97
}
98
99
async fn finish ( app : & Shared, id : i64 , body : serde_json:: Value) -> StatusCode {
100
call ( app, " POST" , & format! ( " /api/v1/flow/jobs/{id} /finish" ) , Some ( " runner-secret" ) , Some ( body.to_string ( ) ) ) .await.0
101
}
102
103
async fn create ( app : & Shared, owner : & str , name : & str ) {
104
app.repos.create ( & RepoName:: parse( & format! ( " {owner} /{name} " ) ) .unwrap ( ) , None ) .unwrap ( ) ;
105
db:: repos:: create( & app.db, owner, name, " " , hub_api:: Visibility:: Public) .await.unwrap ( ) ;
106
}
107
108
fn client ( url : & str , work : & Path) -> Repo {
109
let mut c = Repo:: init( work) .unwrap ( ) ;
110
c.config.set ( " remote.origin.url" , url) .unwrap ( ) ;
111
c.config.set ( " remote.origin.token" , " site" ) .unwrap ( ) ;
112
c.save_config ( ) .unwrap ( ) ;
113
c
114
}
115
116
fn pipeline ( extra : & str ) -> String {
117
format! (
118
" 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} "
119
)
120
}
121
122
async fn runs_of ( app : & Shared, owner : & str , name : & str ) -> Vec< db:: flow:: Run> {
123
let row = db:: repos:: get( & app.db, owner, name) .await.unwrap ( ) .unwrap ( ) ;
124
for _ in 0 .. 100 {
125
let r = db:: flow:: runs( & app.db, row.id, 10 , None ) .await.unwrap ( ) ;
126
if ! r.is_empty ( ) {
127
return r;
128
}
129
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
130
}
131
vec! [ ]
132
}
133
90326c96 Andy McCall 1 month ago ↶
134
135
async fn checks_when ( app : & Shared, repo_id : i64 , state : & str , n : usize ) -> Vec< db:: hooks:: Check> {
136
for _ in 0 .. 100 {
137
let c = db:: hooks:: checks_for_state( & app.db, repo_id, state) .await.unwrap ( ) ;
138
if c.len ( ) > = n {
139
return c;
140
}
141
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
142
}
143
db:: hooks:: checks_for_state( & app.db, repo_id, state) .await.unwrap ( )
144
}
145
42fa9554 Andy McCall 1 month ago ↶
146
# [ tokio ::test ( flavor = " multi_thread" ) ]
147
async fn a_push_runs_the_pipeline_through_a_runner ( ) {
148
let h = hub ( ) .await;
149
create ( & h.app, " andy" , " demo" ) .await;
150
create ( & h.app, " andy" , " secret" ) .await;
151
create ( & h.app, " other" , " plain" ) .await;
152
let url = format! ( " {} /andy/demo" , h.base) ;
153
let work = tempfile:: tempdir( ) .unwrap ( ) ;
154
let mut c = client ( & url, work.path ( ) ) ;
155
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
156
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
157
c.snapshot ( " test" ) .unwrap ( ) ;
158
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
159
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
160
161
162
let runs = runs_of ( & h.app, " andy" , " demo" ) .await;
163
assert_eq! ( runs.len ( ) , 1 ) ;
164
let run = & runs[ 0 ] ;
165
assert_eq! ( ( run.bookmark.as_str ( ) , run.status.as_str ( ) , run.actor.as_str ( ) ) , ( " main" , " queued" , " site token" ) ) ;
166
assert_eq! ( run.pipeline_hash.len ( ) , 64 ) ;
167
let jobs = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) ;
168
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 ↶
169
170
let checks = checks_when ( & h.app, run.repo_id, & run.state, 3 ) .await;
42fa9554 Andy McCall 1 month ago ↶
171
assert_eq! (
172
checks.iter ( ) .map ( | c | ( c.name.as_str ( ) , c.status.as_str ( ) ) ) .collect:: < Vec< _ > > ( ) ,
173
[ ( " flow/build" , " queued" ) , ( " flow/lint" , " queued" ) , ( " flow/test" , " queued" ) ]
174
) ;
175
176
177
assert_eq! (
178
call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " site" ) , Some ( r # "{"runner":"x","runtimes":["microvm"]}"# .into ( ) ) ) .await.0 ,
179
StatusCode:: UNAUTHORIZED
180
) ;
181
assert_eq! (
182
call ( & h.app, " POST" , " /api/v1/flow/claim" , None , Some ( r # "{"runner":"x","runtimes":["microvm"]}"# .into ( ) ) ) .await.0 ,
183
StatusCode:: UNAUTHORIZED
184
) ;
185
186
let ( st, _ ) =
187
call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( r # "{"runner":"x","runtimes":["wasm"]}"# .into ( ) ) ) .await;
188
assert_eq! ( st, StatusCode:: NO_CONTENT ) ;
189
190
191
let a = claim ( & h.app) .await.unwrap ( ) ;
192
let b = claim ( & h.app) .await.unwrap ( ) ;
193
assert_eq! ( ( a[ " job" ] [ " name" ] .as_str ( ) , b[ " job" ] [ " name" ] .as_str ( ) ) , ( Some ( " build" ) , Some ( " lint" ) ) ) ;
194
assert! ( claim ( & h.app) .await.is_none ( ) , " test waits for build" ) ;
195
assert_eq! ( a[ " repo" ] , " andy/demo" ) ;
196
assert_eq! ( a[ " clone_url" ] , url.as_str ( ) ) ;
197
assert_eq! ( a[ " sandbox" ] [ " rootfs" ] , ROOTFS ) ;
198
assert_eq! ( a[ " job" ] [ " exec" ] , serde_json:: json! ( [ " make" ] ) ) ;
199
assert_eq! ( a[ " timeout_s" ] , 3600 ) ;
9a065bc0 Andy McCall 1 month ago ↶
200
201
assert_eq! (
202
( a[ " state" ] .as_str ( ) , a[ " bookmark" ] .as_str ( ) , a[ " run_number" ] .as_i64 ( ) ) ,
203
( Some ( run.state.as_str ( ) ) , Some ( " main" ) , Some ( run.number) )
204
) ;
42fa9554 Andy McCall 1 month ago ↶
205
let heads: Vec< String> = a[ " heads" ] .as_array ( ) .unwrap ( ) .iter ( ) .map ( | v | v.as_str ( ) .unwrap ( ) .to_string ( ) ) .collect ( ) ;
206
assert_eq! ( heads.join ( " +" ) , run.heads) ;
207
let checks = db:: hooks:: checks_for_state( & h.app.db, run.repo_id, & run.state) .await.unwrap ( ) ;
208
assert_eq! ( checks.iter ( ) .find ( | c | c.name = = " flow/build" ) .unwrap ( ) .status, " running" ) ;
209
210
211
let token = a[ " token" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
212
assert! ( token.starts_with ( " flowjob_" ) ) ;
8a3334b7 Andy McCall 1 month ago ↶
213
let opts = RemoteOptions { token: Some ( token.clone ( ) ) , ca: None , repo_id: None , cache_dir: None } ;
42fa9554 Andy McCall 1 month ago ↶
214
open_remote ( & url, & opts) .unwrap ( ) .refs ( ) .unwrap ( ) ;
215
assert! ( open_remote ( & format! ( " {} /andy/secret" , h.base) , & opts) .unwrap ( ) .refs ( ) .is_err ( ) , " another repository" ) ;
8a3334b7 Andy McCall 1 month ago ↶
216
let bad = RemoteOptions { token: Some ( " flowjob_00" .into ( ) ) , ca: None , repo_id: None , cache_dir: None } ;
42fa9554 Andy McCall 1 month ago ↶
217
assert! ( open_remote ( & url, & bad) .unwrap ( ) .refs ( ) .is_err ( ) , " an unknown job token" ) ;
218
219
c.config.set ( " remote.origin.token" , & token) .unwrap ( ) ;
220
c.save_config ( ) .unwrap ( ) ;
221
write ( work.path ( ) , " more.txt" , " x\n " ) ;
222
c.snapshot ( " test" ) .unwrap ( ) ;
223
c.commit ( " More" , & [ ] ) .unwrap ( ) ;
224
let pushed = c.push ( " origin" , false ) ;
225
assert! ( pushed.is_err ( ) | | pushed.unwrap ( ) .iter ( ) .all ( | r | ! r.ok) , " a job token cannot push" ) ;
226
c.config.set ( " remote.origin.token" , " site" ) .unwrap ( ) ;
227
c.save_config ( ) .unwrap ( ) ;
228
229
230
let a_id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
231
let b_id = b[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
232
let log = serde_json:: json! ( { " records" : [
233
{ " t" : 5 , " stream" : " flow" , " text" : " booting" } ,
234
{ " t" : 120 , " stream" : " out" , " text" : " cc -o demo demo.c" } ,
235
{ " t" : 130 , " stream" : " weird" , " text" : " x" .repeat ( 20_000 ) } ,
236
] } ) ;
237
assert_eq! (
238
call ( & h.app, " POST" , & format! ( " /api/v1/flow/jobs/{a_id} /log" ) , Some ( " runner-secret" ) , Some ( log.to_string ( ) ) ) .await.0 ,
239
StatusCode:: NO_CONTENT
240
) ;
241
let lines = db:: flow:: log( & h.app.db, a_id, 0 , 100 ) .await.unwrap ( ) ;
242
assert_eq! ( lines.len ( ) , 3 ) ;
243
assert_eq! ( ( lines[ 2 ] .stream.as_str ( ) , lines[ 2 ] .text.len ( ) ) , ( " console" , 16 * 1024 ) , " streams are known ones, text is capped" ) ;
244
let ph = a[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) ;
245
assert_eq! (
246
finish ( & h.app, a_id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 , " receipt" : receipt ( " lint" , ph, 0 ) } ) ) .await,
247
StatusCode:: BAD_REQUEST ,
248
" a receipt for another job"
249
) ;
250
assert_eq! (
251
finish ( & h.app, a_id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 , " receipt" : receipt ( " build" , ph, 0 ) } ) ) .await,
252
StatusCode:: NO_CONTENT
253
) ;
254
assert_eq! ( finish ( & h.app, a_id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 } ) ) .await, StatusCode:: CONFLICT , " only once" ) ;
255
assert! ( open_remote ( & url, & opts) .unwrap ( ) .refs ( ) .is_err ( ) , " the token ends with its job" ) ;
256
assert_eq! (
257
finish ( & h.app, b_id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 300 , " receipt" : receipt ( " lint" , ph, 0 ) } ) ) .await,
258
StatusCode:: NO_CONTENT
259
) ;
260
let t = claim ( & h.app) .await.unwrap ( ) ;
261
assert_eq! ( t[ " job" ] [ " name" ] , " test" ) ;
262
let t_id = t[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
263
assert_eq! (
264
finish ( & h.app, t_id, serde_json:: json! ( { " exit_code" : 3 , " duration_ms" : 500 , " receipt" : receipt ( " test" , ph, 3 ) } ) ) .await,
265
StatusCode:: NO_CONTENT
266
) ;
267
268
let run = db:: flow:: run_by_id( & h.app.db, run.id) .await.unwrap ( ) .unwrap ( ) ;
269
assert_eq! ( run.status, " failed" ) ;
270
let jobs = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) ;
271
let build = jobs.iter ( ) .find ( | j | j.name = = " build" ) .unwrap ( ) ;
272
assert_eq! ( ( build.status.as_str ( ) , build.exit_code, build.receipt_hash.len ( ) ) , ( " passed" , Some ( 0 ) , 64 ) ) ;
273
assert_eq! ( jobs.iter ( ) .find ( | j | j.name = = " test" ) .unwrap ( ) .status, " failed" ) ;
274
let checks = db:: hooks:: checks_for_state( & h.app.db, run.repo_id, & run.state) .await.unwrap ( ) ;
275
assert_eq! (
276
checks.iter ( ) .map ( | c | ( c.name.as_str ( ) , c.status.as_str ( ) ) ) .collect:: < Vec< _ > > ( ) ,
277
[ ( " flow/build" , " passed" ) , ( " flow/lint" , " passed" ) , ( " flow/test" , " failed" ) ]
278
) ;
a0d4bbfb Andy McCall 1 month ago ↶
279
assert! ( checks[ 2 ] .summary.contains ( " exit 3" ) & & checks[ 2 ] .url.ends_with ( & format! ( " /andy/demo/flow/{} " , run.number) ) ) ;
42fa9554 Andy McCall 1 month ago ↶
280
281
282
let ( st, list) = call ( & h.app, " GET" , " /andy/demo/flow" , None , None ) .await;
283
assert_eq! ( st, StatusCode:: OK ) ;
a0d4bbfb Andy McCall 1 month ago ↶
284
assert! ( list.contains ( & format! ( " /andy/demo/flow/{} " , run.number) ) & & list.contains ( " st-failed" ) , " {list}" ) ;
285
let ( st, page) = call ( & h.app, " GET" , & format! ( " /andy/demo/flow/{} " , run.number) , None , None ) .await;
42fa9554 Andy McCall 1 month ago ↶
286
assert_eq! ( st, StatusCode:: OK ) ;
287
assert! ( page.contains ( " <svg class=\" flow-graph\" " ) , " graph" ) ;
288
assert_eq! ( page.matches ( " data-job-node=" ) .count ( ) , 3 ) ;
289
assert_eq! ( page.matches ( " <path class=\" edge" ) .count ( ) , 1 , " one dependency edge" ) ;
290
assert! ( page.contains ( " cc -o demo demo.c" ) , " log shown" ) ;
291
assert! ( ! page.contains ( " data-flow-live" ) , " a finished run does not poll" ) ;
a0d4bbfb Andy McCall 1 month ago ↶
292
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 ↶
293
assert_eq! ( st, StatusCode:: OK ) ;
294
let live: serde_json:: Value = serde_json:: from_str( & live) .unwrap ( ) ;
295
assert_eq! ( live[ " status" ] , " failed" ) ;
296
let build_live = live[ " jobs" ] .as_array ( ) .unwrap ( ) .iter ( ) .find ( | j | j[ " name" ] = = " build" ) .unwrap ( ) .clone ( ) ;
297
assert_eq! ( build_live[ " lines" ] .as_array ( ) .unwrap ( ) .len ( ) , 2 , " lines after seq 1" ) ;
298
assert_eq! ( call ( & h.app, " GET" , " /andy/demo/flow/999" , None , None ) .await.0 , StatusCode:: NOT_FOUND ) ;
299
300
301
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
302
write ( work.path ( ) , " test.sh" , " echo changed\n " ) ;
303
c.snapshot ( " test" ) .unwrap ( ) ;
304
c.commit ( " Change the test" , & [ ] ) .unwrap ( ) ;
305
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
306
let mut second = None ;
307
for _ in 0 .. 100 {
308
let r = runs_of ( & h.app, " andy" , " demo" ) .await;
309
if r.len ( ) = = 2 {
310
second = Some ( r[ 0 ] .clone ( ) ) ;
311
break ;
312
}
313
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
314
}
315
let second = second.expect ( " a second run" ) ;
316
let x = claim ( & h.app) .await.unwrap ( ) ;
317
let y = claim ( & h.app) .await.unwrap ( ) ;
318
let ( build_id, lint_id) = if x[ " job" ] [ " name" ] = = " build" {
319
( x[ " job_id" ] .as_i64 ( ) .unwrap ( ) , y[ " job_id" ] .as_i64 ( ) .unwrap ( ) )
320
} else {
321
( y[ " job_id" ] .as_i64 ( ) .unwrap ( ) , x[ " job_id" ] .as_i64 ( ) .unwrap ( ) )
322
} ;
323
assert_eq! (
324
finish ( & h.app, build_id, serde_json:: json! ( { " exit_code" : null, " duration_ms" : 10 , " error" : " rootfs is not on this runner" } ) )
325
.await,
326
StatusCode:: NO_CONTENT
327
) ;
328
assert_eq! ( finish ( & h.app, lint_id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 10 } ) ) .await, StatusCode:: NO_CONTENT ) ;
329
assert! ( claim ( & h.app) .await.is_none ( ) , " test was skipped" ) ;
330
let jobs = db:: flow:: jobs( & h.app.db, second.id) .await.unwrap ( ) ;
331
assert_eq! (
332
jobs.iter ( ) .map ( | j | ( j.name.as_str ( ) , j.status.as_str ( ) ) ) .collect:: < Vec< _ > > ( ) ,
333
[ ( " build" , " error" ) , ( " lint" , " passed" ) , ( " test" , " skipped" ) ]
334
) ;
335
assert_eq! ( db:: flow:: run_by_id( & h.app.db, second.id) .await.unwrap ( ) .unwrap ( ) .status, " failed" ) ;
336
337
338
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " job \" broken\" {\n sandbox \" missing\" \n exec \" x\" \n }\n " ) ) ;
339
c.snapshot ( " test" ) .unwrap ( ) ;
340
c.commit ( " Break the pipeline" , & [ ] ) .unwrap ( ) ;
341
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
342
let mut third = None ;
343
for _ in 0 .. 100 {
344
let r = runs_of ( & h.app, " andy" , " demo" ) .await;
345
if r.len ( ) = = 3 {
346
third = Some ( r[ 0 ] .clone ( ) ) ;
347
break ;
348
}
349
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
350
}
351
let third = third.expect ( " a third run" ) ;
352
assert_eq! ( third.status, " error" ) ;
353
assert! ( third.error.contains ( " line 18" ) & & third.error.contains ( " \" missing\" " ) , " {}" , third.error) ;
a0d4bbfb Andy McCall 1 month ago ↶
354
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/demo/flow/{} " , third.number) , None , None ) .await;
42fa9554 Andy McCall 1 month ago ↶
355
assert! ( page.contains ( " The pipeline is not valid" ) ) ;
356
736f6813 Andy McCall 1 month ago ↶
357
358
write ( work.path ( ) , " README.md" , " # Demo\n \n [](.flow/run)\n \n \n " ) ;
359
std:: fs:: create_dir_all( work.path ( ) .join ( " img" ) ) .unwrap ( ) ;
360
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 " )
361
.unwrap ( ) ;
362
c.snapshot ( " test" ) .unwrap ( ) ;
363
c.commit ( " A README with a stamp" , & [ ] ) .unwrap ( ) ;
364
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
365
let mut fourth = None ;
366
for _ in 0 .. 100 {
367
let r = runs_of ( & h.app, " andy" , " demo" ) .await;
368
if r.len ( ) = = 4 {
369
fourth = Some ( r[ 0 ] .clone ( ) ) ;
370
break ;
371
}
372
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
373
}
374
let fourth = fourth.expect ( " a fourth run" ) ;
375
let ( _ , overview) = call ( & h.app, " GET" , " /andy/demo" , None , None ) .await;
376
assert! ( overview.contains ( " src=\" /andy/demo/raw/main/.flow/stamp\" " ) , " {overview}" ) ;
377
assert! ( overview.contains ( " href=\" /andy/demo/file/main/.flow/run\" " ) ) ;
378
assert! ( overview.contains ( " src=\" /andy/demo/raw/main/img/logo.png\" " ) , " images use the raw route" ) ;
379
let ( st, _ , ct) = head ( & h.app, " /andy/demo/raw/main/img/logo.png" ) .await;
380
assert_eq! ( ( st, ct.as_str ( ) ) , ( StatusCode:: OK , " image/png" ) ) ;
381
let ( st, loc, _ ) = head ( & h.app, " /andy/demo/raw/main/.flow/stamp" ) .await;
382
assert! ( st.is_redirection ( ) & & loc = = " /andy/demo/flow/stamp/main" , " {st} {loc}" ) ;
383
let ( st, loc, _ ) = head ( & h.app, " /andy/demo/flow/stamp/main" ) .await;
384
assert! ( st.is_redirection ( ) & & loc = = format! ( " /andy/demo/flow/stamp/{} " , fourth.heads) , " a bookmark's stamp names its state: {loc}" ) ;
385
let ( st, _ , ct) = head ( & h.app, & loc) .await;
386
assert_eq! ( ( st, ct.as_str ( ) ) , ( StatusCode:: OK , " image/svg+xml; charset=utf-8" ) ) ;
387
let ( _ , svg) = call ( & h.app, " GET" , & loc, None , None ) .await;
388
assert! ( svg.starts_with ( " <svg" ) & & svg.contains ( " error ·" ) & & svg.contains ( & fourth.heads[ .. 12 ] ) , " {svg}" ) ;
389
let ( _ , svg1) = call ( & h.app, " GET" , & format! ( " /andy/demo/flow/stamp/{} " , run.heads) , None , None ) .await;
390
assert! ( svg1.contains ( " failed ·" ) & & svg1.contains ( & format! ( " run #{} failed" , run.id) ) , " an older state keeps its own result: {svg1}" ) ;
391
assert! ( svg1.contains ( & build.receipt_hash) , " receipt hashes in the metadata" ) ;
392
assert_eq! ( svg1.matches ( " text-anchor=\" middle\" " ) .count ( ) , 3 , " one glyph per job" ) ;
393
let ( st, loc, _ ) = head ( & h.app, " /andy/demo/file/main/.flow/run" ) .await;
394
assert! ( st.is_redirection ( ) & & loc = = " /andy/demo/flow/at/main" , " {loc}" ) ;
395
let ( st, loc, _ ) = head ( & h.app, " /andy/demo/flow/at/main" ) .await;
396
assert! ( st.is_redirection ( ) & & loc = = format! ( " /andy/demo/flow/{} " , fourth.id) , " {loc}" ) ;
397
42fa9554 Andy McCall 1 month ago ↶
398
399
let work2 = tempfile:: tempdir( ) .unwrap ( ) ;
400
let c2 = client ( & format! ( " {} /other/plain" , h.base) , work2.path ( ) ) ;
401
write ( work2.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
402
c2.snapshot ( " test" ) .unwrap ( ) ;
403
c2.commit ( " Pipeline" , & [ ] ) .unwrap ( ) ;
404
assert! ( c2.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
405
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 500 ) ) .await;
406
let row = db:: repos:: get( & h.app.db, " other" , " plain" ) .await.unwrap ( ) .unwrap ( ) ;
407
assert! ( db:: flow:: runs( & h.app.db, row.id, 10 , None ) .await.unwrap ( ) .is_empty ( ) ) ;
408
let ( _ , page) = call ( & h.app, " GET" , " /other/plain/flow" , None , None ) .await;
409
assert! ( page.contains ( " not enabled for this repository" ) ) ;
736f6813 Andy McCall 1 month ago ↶
410
let ( _ , loc, _ ) = head ( & h.app, " /other/plain/flow/stamp/main" ) .await;
411
let ( _ , svg) = call ( & h.app, " GET" , & loc, None , None ) .await;
412
assert! ( svg.contains ( " no run ·" ) , " {svg}" ) ;
413
let ( _ , loc, _ ) = head ( & h.app, " /other/plain/flow/at/main" ) .await;
414
assert_eq! ( loc, " /other/plain/flow" ) ;
42fa9554 Andy McCall 1 month ago ↶
415
}
fd3db644 Andy McCall 1 month ago ↶
416
417
# [ tokio ::test ( flavor = " multi_thread" ) ]
418
async fn claims_carry_pinned_subrepos_and_the_job_token_reads_only_those ( ) {
419
let h = hub ( ) .await;
25091e5c Andy McCall 1 month ago ↶
420
for n in [ " base" , " lib" , " vault" , " other" , " app" ] {
fd3db644 Andy McCall 1 month ago ↶
421
create ( & h.app, " andy" , n) .await;
422
}
423
let push = | name : & str , file : & str | {
424
let work = tempfile:: tempdir( ) .unwrap ( ) ;
425
let c = client ( & format! ( " {} /andy/{name} " , h.base) , work.path ( ) ) ;
426
write ( work.path ( ) , file, " x\n " ) ;
427
c.snapshot ( " t" ) .unwrap ( ) ;
428
c.commit ( " first" , & [ ] ) .unwrap ( ) ;
429
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
430
let heads = c.bookmark_heads ( ) .unwrap ( ) ;
431
( work, heads)
432
} ;
25091e5c Andy McCall 1 month ago ↶
433
let ( _b, base_heads) = push ( " base" , " base.rs" ) ;
434
435
let lib_work = tempfile:: tempdir( ) .unwrap ( ) ;
436
let lib = client ( & format! ( " {} /andy/lib" , h.base) , lib_work.path ( ) ) ;
437
lib.submodule_add ( & format! ( " {} /andy/base" , h.base) , " deps/base" , Some ( " main" ) ) .unwrap ( ) ;
438
write ( lib_work.path ( ) , " lib.rs" , " x\n " ) ;
439
lib.snapshot ( " t" ) .unwrap ( ) ;
440
lib.commit ( " lib on base" , & [ ] ) .unwrap ( ) ;
441
assert! ( lib.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
442
let lib_heads = lib.bookmark_heads ( ) .unwrap ( ) ;
fd3db644 Andy McCall 1 month ago ↶
443
let ( _v, _ ) = push ( " vault" , " secret.rs" ) ;
444
let ( _o, _ ) = push ( " other" , " o.rs" ) ;
445
446
447
let work = tempfile:: tempdir( ) .unwrap ( ) ;
448
let app = client ( & format! ( " {} /andy/app" , h.base) , work.path ( ) ) ;
449
app.submodule_add ( & format! ( " {} /andy/lib" , h.base) , " deps/lib" , Some ( " main" ) ) .unwrap ( ) ;
a4f01d51 Andy McCall 1 month ago ↶
450
2bbe88a8 Andy McCall 1 month ago ↶
451
452
fd3db644 Andy McCall 1 month ago ↶
453
sqlx:: query( " UPDATE repositories SET visibility = 'private' WHERE name = 'vault'" ) .execute ( & h.app.db) .await.unwrap ( ) ;
a4f01d51 Andy McCall 1 month ago ↶
454
let bare_work = tempfile:: tempdir( ) .unwrap ( ) ;
455
let mut bare = Repo:: init( bare_work.path ( ) ) .unwrap ( ) ;
456
bare.config.set ( " remote.origin.url" , & format! ( " {} /andy/bare" , h.base) ) .unwrap ( ) ;
457
bare.save_config ( ) .unwrap ( ) ;
458
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 ↶
459
app.submodule_add ( & format! ( " {} /andy/vault" , h.base) , " deps/vault" , Some ( " main" ) ) .unwrap ( ) ;
a4f01d51 Andy McCall 1 month ago ↶
460
let nested = Repo:: open( & work.path ( ) .join ( " deps/vault" ) ) .unwrap ( ) ;
2bbe88a8 Andy McCall 1 month ago ↶
461
assert! ( nested.config.remote_tokens.is_empty ( ) , " no token is copied into a subrepo" ) ;
462
nested.fetch ( " origin" ) .unwrap ( ) ;
370035a4 Andy McCall 1 month ago ↶
463
fd3db644 Andy McCall 1 month ago ↶
464
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
465
write ( work.path ( ) , " Makefile" , " all:\n " ) ;
466
app.snapshot ( " t" ) .unwrap ( ) ;
467
app.commit ( " app with deps" , & [ ] ) .unwrap ( ) ;
468
assert! ( app.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
469
assert! ( ! runs_of ( & h.app, " andy" , " app" ) .await.is_empty ( ) ) ;
470
471
let job = claim ( & h.app) .await.unwrap ( ) ;
472
let subs = job[ " subrepos" ] .as_array ( ) .cloned ( ) .unwrap_or_default ( ) ;
25091e5c Andy McCall 1 month ago ↶
473
assert_eq! ( subs.len ( ) , 2 , " lib and what lib pins; vault is private and the pusher has no account: {subs:?}" ) ;
474
assert_eq! ( subs[ 1 ] [ " path" ] , " deps/lib/deps/base" , " nested under its parent, after it" ) ;
475
assert_eq! ( subs[ 1 ] [ " heads" ] , serde_json:: json! ( base_heads.iter ( ) .map ( | h | h.to_hex ( ) ) .collect:: < Vec< _ > > ( ) ) ) ;
370035a4 Andy McCall 1 month ago ↶
476
let _ = std:: fs:: remove_dir_all( work.path ( ) .join ( " deps/vault2" ) ) ;
fd3db644 Andy McCall 1 month ago ↶
477
assert_eq! ( subs[ 0 ] [ " path" ] , " deps/lib" ) ;
478
assert_eq! ( subs[ 0 ] [ " repo" ] , " andy/lib" ) ;
479
assert_eq! ( subs[ 0 ] [ " heads" ] , serde_json:: json! ( lib_heads.iter ( ) .map ( | h | h.to_hex ( ) ) .collect:: < Vec< _ > > ( ) ) ) ;
480
assert! ( subs[ 0 ] [ " clone_url" ] .as_str ( ) .unwrap ( ) .ends_with ( " /andy/lib" ) ) ;
afd0be38 Andy McCall 1 month ago ↶
481
482
let job_id = job[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
483
let lines = db:: flow:: log( & h.app.db, job_id, 0 , 100 ) .await.unwrap ( ) ;
484
assert! (
485
lines.iter ( ) .any ( | l | l.stream = = " flow" & & l.text.contains ( " subrepo deps/vault (andy/vault) is left out" ) ) ,
486
" {:?}" ,
487
lines.iter ( ) .map ( | l | & l.text) .collect:: < Vec< _ > > ( )
488
) ;
fd3db644 Andy McCall 1 month ago ↶
489
490
491
let token = job[ " token" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
492
let read = | name : & str | {
493
let opts = RemoteOptions { token: Some ( token.clone ( ) ) , .. Default :: default( ) } ;
494
open_remote ( & format! ( " {} /andy/{name} " , h.base) , & opts) .and_then ( | r | r.refs ( ) ) .is_ok ( )
495
} ;
25091e5c Andy McCall 1 month ago ↶
496
assert! ( read ( " app" ) & & read ( " lib" ) & & read ( " base" ) ) ;
fd3db644 Andy McCall 1 month ago ↶
497
assert! ( ! read ( " vault" ) , " not pinned for this pusher" ) ;
498
assert! ( ! read ( " other" ) , " public, but not the job's to read" ) ;
499
}
941abdd6 Andy McCall 1 month ago ↶
500
501
# [ tokio ::test ( flavor = " multi_thread" , worker_threads = 8 ) ]
502
async fn concurrent_log_appends_never_fail_on_a_locked_database ( ) {
503
let h = hub ( ) .await;
504
create ( & h.app, " andy" , " demo" ) .await;
505
let work = tempfile:: tempdir( ) .unwrap ( ) ;
506
let c = client ( & format! ( " {} /andy/demo" , h.base) , work.path ( ) ) ;
507
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
508
write ( work.path ( ) , " Makefile" , " all:\n " ) ;
509
c.snapshot ( " t" ) .unwrap ( ) ;
510
c.commit ( " pipeline" , & [ ] ) .unwrap ( ) ;
511
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
512
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
513
let job = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) .remove ( 0 ) ;
514
515
516
let mut tasks = Vec :: new( ) ;
517
for w in 0 .. 16 {
518
let pool = h.app.db.clone ( ) ;
519
tasks.push ( tokio:: spawn( async move {
520
let records: Vec< flow_core:: api:: Record> =
521
( 0 .. 20 ) .map ( | i | flow_core:: api:: Record { t: i, stream: " out" .into ( ) , text: format! ( " writer {w} line {i} " ) } ) .collect ( ) ;
522
db:: flow:: append_log( & pool, job.id, & records, 100_000 ) .await
523
} ) ) ;
524
}
525
for t in tasks {
526
assert_eq! ( t.await.unwrap ( ) .unwrap ( ) , 20 , " every append succeeds" ) ;
527
}
528
let lines = db:: flow:: log( & h.app.db, job.id, 0 , 1_000 ) .await.unwrap ( ) ;
529
assert_eq! ( lines.len ( ) , 320 ) ;
530
let seqs: std:: collections:: BTreeSet< i64 > = lines.iter ( ) .map ( | l | l.seq ) .collect ( ) ;
531
assert_eq! ( seqs.len ( ) , 320 , " sequence numbers are unique" ) ;
532
}
9008bf6c Andy McCall 1 month ago ↶
533
534
535
# [ tokio ::test ( flavor = " multi_thread" ) ]
536
async fn a_repository_setting_turns_flow_on_or_off ( ) {
537
let h = hub ( ) .await;
538
create ( & h.app, " other" , " plain" ) .await;
539
create ( & h.app, " andy" , " quiet" ) .await;
a0d4bbfb Andy McCall 1 month ago ↶
540
541
create ( & h.app, " andy" , " first" ) .await;
542
{
543
let work = tempfile:: tempdir( ) .unwrap ( ) ;
544
let c = client ( & format! ( " {} /andy/first" , h.base) , work.path ( ) ) ;
545
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
546
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
547
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
548
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
549
assert_eq! ( runs_of ( & h.app, " andy" , " first" ) .await.len ( ) , 1 ) ;
550
}
9008bf6c Andy McCall 1 month ago ↶
551
let row = db:: repos:: get( & h.app.db, " other" , " plain" ) .await.unwrap ( ) .unwrap ( ) ;
552
db:: repos:: set_flow( & h.app.db, row.id, " on" ) .await.unwrap ( ) ;
553
let quiet = db:: repos:: get( & h.app.db, " andy" , " quiet" ) .await.unwrap ( ) .unwrap ( ) ;
554
db:: repos:: set_flow( & h.app.db, quiet.id, " off" ) .await.unwrap ( ) ;
555
for ( owner, name) in [ ( " other" , " plain" ) , ( " andy" , " quiet" ) ] {
556
let work = tempfile:: tempdir( ) .unwrap ( ) ;
557
let c = client ( & format! ( " {} /{owner} /{name} " , h.base) , work.path ( ) ) ;
558
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
559
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
560
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
561
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
562
}
a0d4bbfb Andy McCall 1 month ago ↶
563
let plain = runs_of ( & h.app, " other" , " plain" ) .await;
564
assert_eq! ( plain.len ( ) , 1 , " turned on" ) ;
bf3d0754 Andy McCall 1 month ago ↶
565
566
let a = claim ( & h.app) .await.unwrap ( ) ;
567
assert_eq! ( a[ " repo" ] , " andy/first" ) ;
568
let ( _ , page) = call ( & h.app, " GET" , " /other/plain/flow/1" , None , None ) .await;
569
assert! ( page.contains ( " Waiting for a runner: 3rd in line" ) & & page.contains ( " Running now: andy/first #1" ) , " {page}" ) ;
570
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/repos/other/plain/flow/runs/1" , None , None ) .await;
571
assert_eq! ( serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " queue_position" ] , 3 ) ;
572
let ( _ , list) = call ( & h.app, " GET" , " /other/plain/flow" , None , None ) .await;
573
assert! ( list.contains ( " queued, 3rd in line" ) , " {list}" ) ;
a0d4bbfb Andy McCall 1 month ago ↶
574
575
576
assert_eq! ( plain[ 0 ] .number, 1 ) ;
90326c96 Andy McCall 1 month ago ↶
577
let checks = checks_when ( & h.app, row.id, & plain[ 0 ] .state, 3 ) .await;
a0d4bbfb Andy McCall 1 month ago ↶
578
assert! ( checks.iter ( ) .all ( | c | c.url.ends_with ( " /other/plain/flow/1" ) ) , " {checks:?}" ) ;
579
let ( st, page) = call ( & h.app, " GET" , " /other/plain/flow/1" , None , None ) .await;
580
assert! ( st = = StatusCode:: OK & & page.contains ( " Run #1" ) , " {st}" ) ;
9008bf6c Andy McCall 1 month ago ↶
581
assert! ( runs_of ( & h.app, " andy" , " quiet" ) .await.is_empty ( ) , " turned off despite andy/*" ) ;
582
}
1edac4b6 Andy McCall 1 month ago ↶
583
584
585
586
# [ tokio ::test ( flavor = " multi_thread" ) ]
587
async fn runs_start_without_a_push_and_finished_runs_re_run ( ) {
588
let h = hub ( ) .await;
589
590
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
591
create ( & h.app, " andy" , " demo" ) .await;
592
let url = format! ( " {} /andy/demo" , h.base) ;
593
let work = tempfile:: tempdir( ) .unwrap ( ) ;
594
let c = client ( & url, work.path ( ) ) ;
595
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
596
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
597
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
598
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
599
let first = runs_of ( & h.app, " andy" , " demo" ) .await;
600
assert_eq! ( ( first.len ( ) , first[ 0 ] .trigger.as_str ( ) ) , ( 1 , " push" ) ) ;
601
602
603
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( " {}" .into ( ) ) ) .await;
604
assert_eq! ( st, StatusCode:: CREATED , " {body}" ) ;
605
let v: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
606
assert_eq! ( v[ " number" ] , 2 ) ;
607
assert! ( v[ " url" ] .as_str ( ) .unwrap ( ) .ends_with ( " /andy/demo/flow/2" ) ) ;
608
let second = db:: flow:: run( & h.app.db, first[ 0 ] .repo_id, 2 ) .await.unwrap ( ) .unwrap ( ) ;
609
assert_eq! ( ( second.trigger.as_str ( ) , second.actor.as_str ( ) , second.state.as_str ( ) ) , ( " manual" , " andy" , first[ 0 ] .state.as_str ( ) ) ) ;
610
assert_eq! ( db:: flow:: jobs( & h.app.db, second.id) .await.unwrap ( ) .len ( ) , 3 ) ;
611
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/flow/2" , None , None ) .await;
612
assert! ( page.contains ( " started by andy" ) , " the run page says what started it" ) ;
613
4f692533 Andy McCall 1 month ago ↶
614
615
let head = first[ 0 ] .head_list ( ) [ 0 ] .to_string ( ) ;
616
let ( st, body) =
617
call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( format! ( r # "{{ "change":"{head} "}} "# ) ) ) .await;
618
assert_eq! ( st, StatusCode:: CREATED , " {body}" ) ;
619
let on_change = db:: flow:: run( & h.app.db, first[ 0 ] .repo_id, 3 ) .await.unwrap ( ) .unwrap ( ) ;
620
assert_eq! (
621
( on_change.bookmark.as_str ( ) , on_change.heads.as_str ( ) , on_change.state.as_str ( ) ) ,
622
( " " , head.as_str ( ) , first[ 0 ] .state.as_str ( ) )
623
) ;
624
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/flow/3" , None , None ) .await;
625
assert! ( page.contains ( " title=\" Run on one change, not a bookmark\" >change</span>" ) , " run page" ) ;
626
let ( st, body) =
627
call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( format! ( r # "{{ "change":"{} "}} "# , " ab" .repeat ( 32 ) ) ) )
628
.await;
629
assert! ( st = = StatusCode:: CONFLICT & & body.contains ( " no change" ) , " {st} {body}" ) ;
630
631
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/flow" , None , None ) .await;
632
assert! ( page.contains ( " Sign in with write access to run the pipeline." ) & & ! page.contains ( " primary-btn" ) ) ;
633
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/bookmarks" , None , None ) .await;
634
assert! ( ! page.contains ( " /flow/start" ) , " no run buttons for anonymous viewers" ) ;
635
1edac4b6 Andy McCall 1 month ago ↶
636
637
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs/1/rerun" , Some ( " site" ) , None ) .await;
638
assert! ( st = = StatusCode:: CONFLICT & & body.contains ( " still queued" ) , " {st} {body}" ) ;
639
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( r # "{"bookmark":"nope"}"# .into ( ) ) ) .await;
640
assert_eq! ( st, StatusCode:: CONFLICT ) ;
641
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs/99/rerun" , Some ( " site" ) , None ) .await;
642
assert_eq! ( st, StatusCode:: NOT_FOUND ) ;
643
644
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , None , Some ( " {}" .into ( ) ) ) .await;
645
assert_ne! ( st, StatusCode:: CREATED ) ;
646
647
648
let row = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
649
for r in [ & first[ 0 ] , & second] {
650
for j in db:: flow:: jobs( & h.app.db, r.id) .await.unwrap ( ) {
651
sqlx:: query( " UPDATE flow_jobs SET status = 'passed' WHERE id = ?" ) .bind ( j.id) .execute ( & h.app.db) .await.unwrap ( ) ;
652
}
653
sqlx:: query( " UPDATE flow_runs SET status = 'passed', finished_ms = 1 WHERE id = ?" ) .bind ( r.id) .execute ( & h.app.db) .await.unwrap ( ) ;
654
}
655
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs/1/rerun" , Some ( " site" ) , None ) .await;
656
assert_eq! ( st, StatusCode:: CREATED , " {body}" ) ;
4f692533 Andy McCall 1 month ago ↶
657
let third = db:: flow:: run( & h.app.db, row.id, 4 ) .await.unwrap ( ) .unwrap ( ) ;
348f8c4b Andy McCall 1 month ago ↶
658
assert_eq! (
659
( third.trigger_label ( ) .as_str ( ) , third.status.as_str ( ) , third.state.as_str ( ) ) ,
660
( " rerun:1" , " queued" , first[ 0 ] .state.as_str ( ) )
661
) ;
c889911b Andy McCall 1 month ago ↶
662
assert_eq! ( third.trigger, " rerun:1:push" , " a rerun keeps the event it repeats" ) ;
1edac4b6 Andy McCall 1 month ago ↶
663
let checks = db:: hooks:: checks_for_state( & h.app.db, row.id, & third.state) .await.unwrap ( ) ;
664
assert! (
4f692533 Andy McCall 1 month ago ↶
665
checks.iter ( ) .all ( | c | c.status = = " queued" & & c.url.ends_with ( " /flow/4" ) ) ,
1edac4b6 Andy McCall 1 month ago ↶
666
" the re-run's checks replace the old ones: {checks:?}"
667
) ;
4f692533 Andy McCall 1 month ago ↶
668
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/flow/4" , None , None ) .await;
1edac4b6 Andy McCall 1 month ago ↶
669
assert! ( page.contains ( " re-run of #1 by andy" ) ) ;
670
assert! ( ! page.contains ( " RE-RUN" ) , " anonymous viewers get no button" ) ;
671
0650d94e Andy McCall 26 days ago ↶
672
673
674
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( r # "{"jobs":["test"]}"# .into ( ) ) ) .await;
675
assert_eq! ( st, StatusCode:: CREATED , " {body}" ) ;
676
let n = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " number" ] .as_i64 ( ) .unwrap ( ) ;
677
let only = db:: flow:: run( & h.app.db, first[ 0 ] .repo_id, n) .await.unwrap ( ) .unwrap ( ) ;
678
let mut names: Vec< String> = db:: flow:: jobs( & h.app.db, only.id) .await.unwrap ( ) .into_iter ( ) .map ( | j | j.name ) .collect ( ) ;
679
names.sort ( ) ;
680
assert_eq! ( names, [ " build" , " test" ] , " test and the job it depends on, not lint" ) ;
681
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( r # "{"jobs":["deploy"]}"# .into ( ) ) ) .await;
682
assert! ( st = = StatusCode:: CONFLICT & & body.contains ( " not in the pipeline" ) & & body.contains ( " build, lint, test" ) , " {st} {body}" ) ;
683
1edac4b6 Andy McCall 1 month ago ↶
684
685
db:: repos:: set_flow( & h.app.db, row.id, " off" ) .await.unwrap ( ) ;
686
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( " {}" .into ( ) ) ) .await;
687
assert! ( st = = StatusCode:: CONFLICT & & body.contains ( " not enabled" ) , " {body}" ) ;
4f692533 Andy McCall 1 month ago ↶
688
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/flow" , None , None ) .await;
689
assert! ( page.contains ( " Flow is off for this repository" ) , " and the Flow tab says so" ) ;
90326c96 Andy McCall 1 month ago ↶
690
}
691
692
693
694
# [ tokio ::test ( flavor = " multi_thread" ) ]
695
async fn runs_jobs_and_logs_are_in_the_api ( ) {
696
let h = hub ( ) .await;
697
create ( & h.app, " andy" , " demo" ) .await;
698
let url = format! ( " {} /andy/demo" , h.base) ;
699
let work = tempfile:: tempdir( ) .unwrap ( ) ;
700
let c = client ( & url, work.path ( ) ) ;
701
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
702
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
703
let head = c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
704
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
705
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
706
707
708
let a = claim ( & h.app) .await.unwrap ( ) ;
709
let id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
710
let recs =
711
serde_json:: json! ( { " records" : [ { " t" : 1 , " stream" : " stdout" , " text" : " one" } , { " t" : 3 , " stream" : " stderr" , " text" : " three" } ] } ) ;
712
let ( st, _ ) = call ( & h.app, " POST" , & format! ( " /api/v1/flow/jobs/{id} /log" ) , Some ( " runner-secret" ) , Some ( recs.to_string ( ) ) ) .await;
713
assert! ( st.is_success ( ) , " {st}" ) ;
714
715
let ( st, body) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/flow/runs" , None , None ) .await;
716
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
717
let runs: Vec< hub_api:: FlowRun> = serde_json:: from_str( & body) .unwrap ( ) ;
718
assert_eq! ( runs.len ( ) , 1 ) ;
719
assert_eq! ( ( runs[ 0 ] .number, runs[ 0 ] .state.as_str ( ) , runs[ 0 ] .trigger.as_str ( ) ) , ( 1 , run.state.as_str ( ) , " push" ) ) ;
720
assert! ( runs[ 0 ] .url.ends_with ( " /andy/demo/flow/1" ) & & runs[ 0 ] .jobs.is_empty ( ) ) ;
721
722
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/flow/runs/1" , None , None ) .await;
723
let one: hub_api:: FlowRun = serde_json:: from_str( & body) .unwrap ( ) ;
724
let names: Vec< ( & str , & str ) > = one.jobs.iter ( ) .map ( | j | ( j.name.as_str ( ) , j.status.as_str ( ) ) ) .collect ( ) ;
725
assert_eq! ( names.len ( ) , 3 ) ;
726
let running = one.jobs.iter ( ) .find ( | j | j.status = = " running" ) .unwrap ( ) ;
727
assert_eq! ( running.name, a[ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) ) ;
728
assert! ( one.jobs.iter ( ) .any ( | j | j.name = = " test" & & j.depends_on = = [ " build" ] ) ) ;
729
730
let job = running.name.clone ( ) ;
731
let ( st, body) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/demo/flow/runs/1/jobs/{job} /log" ) , None , None ) .await;
732
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
733
let lines: Vec< hub_api:: FlowLogLine> = serde_json:: from_str( & body) .unwrap ( ) ;
734
assert! ( lines.iter ( ) .any ( | l | l.text = = " three" ) , " {body}" ) ;
735
let last = lines.last ( ) .unwrap ( ) .seq;
736
let ( _ , body) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/demo/flow/runs/1/jobs/{job} /log?after={last} " ) , None , None ) .await;
737
assert_eq! ( body, " []" , " nothing after the last line" ) ;
738
assert_eq! ( call ( & h.app, " GET" , " /api/v1/repos/andy/demo/flow/runs/9" , None , None ) .await.0 , StatusCode:: NOT_FOUND ) ;
739
assert_eq! ( call ( & h.app, " GET" , " /api/v1/repos/andy/demo/flow/runs/1/jobs/nope/log" , None , None ) .await.0 , StatusCode:: NOT_FOUND ) ;
740
741
742
let by = | key : String| {
743
let app = h.app.clone ( ) ;
744
async move {
745
let ( st, body) = call ( & app, " GET" , & format! ( " /api/v1/repos/andy/demo/checks/{key} " ) , None , None ) .await;
746
assert_eq! ( st, StatusCode:: OK , " {key}: {body}" ) ;
747
serde_json:: from_str:: < Vec< hub_api:: Check> > ( & body) .unwrap ( ) .len ( )
748
}
749
} ;
750
checks_when ( & h.app, run.repo_id, & run.state, 3 ) .await;
751
assert_eq! ( by ( run.state.clone ( ) ) .await, 3 ) ;
752
assert_eq! ( by ( head.to_hex ( ) ) .await, 3 , " a single head names its state" ) ;
753
assert_eq! ( by ( " main" .into ( ) ) .await, 3 ) ;
754
let ( st, _ ) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/checks/no-such-bookmark" , None , None ) .await;
755
assert_eq! ( st, StatusCode:: NOT_FOUND ) ;
756
757
758
let ( st, body) = call ( & h.app, " GET" , " /api/v1/users/andy/repos" , None , None ) .await;
759
assert_eq! ( st, StatusCode:: NOT_FOUND ) ;
760
let v: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
761
assert_eq! ( v[ " status" ] , 404 , " {body}" ) ;
1edac4b6 Andy McCall 1 month ago ↶
762
}
cc19697f Andy McCall 1 month ago ↶
763
764
async fn get_bytes ( app : & Shared, path : & str ) -> ( StatusCode, Vec< u8 > ) {
765
let res = oxygenhub:: router( app.clone ( ) ) .oneshot ( Request:: get( path) .body ( Body:: empty( ) ) .unwrap ( ) ) .await.unwrap ( ) ;
766
let status = res.status ( ) ;
767
( status, res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) .to_vec ( ) )
768
}
769
770
# [ tokio ::test ( flavor = " multi_thread" ) ]
771
async fn signed_receipts_are_checked_kept_and_shown ( ) {
772
use oxy_core:: hash:: { hash_object, ChangeHash, ObjectRef, ObjectType} ;
773
use oxy_core:: identity:: SecretKey;
774
use oxy_core:: objects:: StateObject;
775
use oxy_core:: receipt:: SignedReceipt;
776
let h = hub ( ) .await;
777
create ( & h.app, " andy" , " demo" ) .await;
778
let url = format! ( " {} /andy/demo" , h.base) ;
779
let work = tempfile:: tempdir( ) .unwrap ( ) ;
780
let c = client ( & url, work.path ( ) ) ;
781
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
782
c.snapshot ( " test" ) .unwrap ( ) ;
783
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
784
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
785
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
786
let heads: Vec< ChangeHash> = run.head_list ( ) .into_iter ( ) .filter_map ( ChangeHash:: from_hex) .collect ( ) ;
787
let state = hash_object ( ObjectType:: State, & StateObject:: new( heads.clone ( ) ) .encode ( ) ) ;
788
let repo_id = h.app.repos.open ( & RepoName:: parse( " andy/demo" ) .unwrap ( ) ) .unwrap ( ) .id ( ) .unwrap ( ) ;
789
let sk = SecretKey:: generate( ) ;
790
let runner = sk.identity ( " test-1 (OxygenFlow runner)" , " " , 1 ) ;
791
let sign = | r : & serde_json:: Value, state : ObjectRef, sk : & SecretKey, id : & oxy_core:: identity:: Identity| {
792
let r: flow_core:: Receipt = serde_json:: from_value( r.clone ( ) ) .unwrap ( ) ;
793
let run_url = format! ( " {url} /flow/{} " , run.number) ;
794
hex:: encode( SignedReceipt:: sign( r.to_oxy ( ) , repo_id, state, & run_url, id, sk) .encode ( ) )
795
} ;
796
797
let a = claim ( & h.app) .await.unwrap ( ) ;
798
let b = claim ( & h.app) .await.unwrap ( ) ;
799
let ph = a[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
800
let ( a_id, b_id) = ( a[ " job_id" ] .as_i64 ( ) .unwrap ( ) , b[ " job_id" ] .as_i64 ( ) .unwrap ( ) ) ;
801
802
let rb = receipt ( " build" , & ph, 0 ) ;
803
let body = serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 , " receipt" : rb, " signed_receipt" : sign ( & rb, state, & sk, & runner) } ) ;
804
assert_eq! ( finish ( & h.app, a_id, body) .await, StatusCode:: NO_CONTENT ) ;
805
806
let other = SecretKey:: generate( ) ;
807
let rl = receipt ( " lint" , & ph, 0 ) ;
808
let body = serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 300 , " receipt" : rl,
809
" signed_receipt" : sign ( & rl, state, & other, & other.identity ( " mallory" , " " , 1 ) ) } ) ;
810
assert_eq! ( finish ( & h.app, b_id, body) .await, StatusCode:: NO_CONTENT ) ;
811
812
let t = claim ( & h.app) .await.unwrap ( ) ;
813
let rt = receipt ( " test" , & ph, 0 ) ;
814
let body = serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 500 , " receipt" : rt,
815
" signed_receipt" : sign ( & rt, ObjectRef( [ 9 ; 32 ] ) , & sk, & runner) } ) ;
816
assert_eq! ( finish ( & h.app, t[ " job_id" ] .as_i64 ( ) .unwrap ( ) , body) .await, StatusCode:: NO_CONTENT ) ;
817
818
let jobs = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) ;
819
let job = | n : & str | jobs.iter ( ) .find ( | j | j.name = = n ) .unwrap ( ) .clone ( ) ;
820
assert_eq! ( job ( " build" ) .status, " passed" ) ;
821
assert_eq! ( ( job ( " build" ) .receipt_attest.len ( ) , job ( " build" ) .receipt_signer.as_str ( ) ) , ( 64 , " test-1 (OxygenFlow runner)" ) ) ;
822
assert_eq! ( job ( " lint" ) .status, " passed" , " the job's result stands" ) ;
823
assert! ( job ( " lint" ) .receipt_attest.is_empty ( ) ) ;
824
assert! ( job ( " lint" ) .receipt_note.contains ( " not the key this runner signs with" ) , " {}" , job ( " lint" ) .receipt_note) ;
825
assert! ( job ( " test" ) .receipt_note.contains ( " another state" ) , " {}" , job ( " test" ) .receipt_note) ;
826
827
828
829
let kept = h.app.repos.open ( & RepoName:: parse( " andy/demo" ) .unwrap ( ) ) .unwrap ( ) .receipts ( & heads) .unwrap ( ) ;
830
assert_eq! ( kept.len ( ) , 1 ) ;
831
let attest = job ( " build" ) .receipt_attest;
832
let ( st, bytes) = get_bytes ( & h.app, & format! ( " /andy/demo/receipts/{attest} " ) ) .await;
833
assert_eq! ( st, StatusCode:: OK ) ;
834
let sr = SignedReceipt:: decode( & bytes) .unwrap ( ) ;
835
sr.verify ( ) .unwrap ( ) ;
836
assert_eq! ( ( sr.state ( ) , sr.receipt.job_name.as_str ( ) ) , ( state, " build" ) ) ;
837
assert_eq! ( get_bytes ( & h.app, & format! ( " /andy/demo/receipts/{} " , " 0" .repeat ( 64 ) ) ) .await.0 , StatusCode:: NOT_FOUND ) ;
838
839
840
841
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/demo/flow/{} " , run.number) , None , None ) .await;
842
assert! ( page.contains ( " signed receipt</a> by test-1" ) , " run page" ) ;
843
assert! ( page.contains ( " receipt not kept" ) , " run page" ) ;
844
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/demo/change/{} " , heads[ 0 ] .to_hex ( ) ) , None , None ) .await;
845
assert! ( page.contains ( " <dt>Receipts</dt>" ) & & page.contains ( & format! ( " /andy/demo/receipts/{attest} " ) ) , " change page" ) ;
846
}
a3c8288b Andy McCall 1 month ago ↶
847
848
async fn upload ( app : & Shared, job : i64 , name : & str , bytes : & [u8 ]) -> ( StatusCode, String) {
849
let req = Request:: post( format! ( " /api/v1/flow/jobs/{job} /artifacts/{name} " ) )
850
.header ( header:: AUTHORIZATION , " Bearer runner-secret" )
851
.header ( header:: CONTENT_TYPE , " application/octet-stream" )
852
.body ( Body:: from( bytes.to_vec ( ) ) )
853
.unwrap ( ) ;
854
let res = oxygenhub:: router( app.clone ( ) ) .oneshot ( req) .await.unwrap ( ) ;
855
let status = res.status ( ) ;
856
( status, String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) )
857
}
858
859
# [ tokio ::test ( flavor = " multi_thread" ) ]
860
async fn artifacts_are_kept_named_by_the_receipt_and_served ( ) {
861
let h = hub ( ) .await;
862
create ( & h.app, " andy" , " demo" ) .await;
863
let url = format! ( " {} /andy/demo" , h.base) ;
864
let work = tempfile:: tempdir( ) .unwrap ( ) ;
865
let c = client ( & url, work.path ( ) ) ;
866
write (
867
work.path ( ) ,
868
" .flow/pipeline.kdl" ,
869
& format! (
870
" 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 "
871
) ,
872
) ;
873
c.snapshot ( " test" ) .unwrap ( ) ;
874
c.commit ( " Build a tool" , & [ ] ) .unwrap ( ) ;
875
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
876
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
877
let a = claim ( & h.app) .await.unwrap ( ) ;
878
assert_eq! ( a[ " job" ] [ " outputs" ] , serde_json:: json! ( [ { " name" : " tool" , " path" : " /src/out/tool" } ] ) , " the runner is told" ) ;
879
let id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
880
881
let body: Vec< u8 > = ( 0 .. 300_000 u32 ) .map ( | i | ( i % 253 ) as u8 ) .collect ( ) ;
882
let ( st, text) = upload ( & h.app, id, " other" , & body) .await;
883
assert_eq! ( st, StatusCode:: BAD_REQUEST , " an undeclared artifact: {text}" ) ;
884
let ( st, text) = upload ( & h.app, id, " tool" , & body) .await;
885
assert_eq! ( st, StatusCode:: OK , " {text}" ) ;
886
let blob = serde_json:: from_str:: < serde_json:: Value> ( & text) .unwrap ( ) [ " blob" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
887
assert_eq! ( blob, oxy_core:: repo:: Repo:: blob_ref( & body) .to_hex ( ) , " the same blob the runner computes" ) ;
888
889
890
let mut r = receipt ( " build" , a[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) , 0 ) ;
891
assert_eq! (
892
finish ( & h.app, id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 , " receipt" : r.clone ( ) } ) ) .await,
893
StatusCode:: BAD_REQUEST
894
) ;
895
r[ " outputs" ] = serde_json:: json! ( [ blob] ) ;
896
assert_eq! ( finish ( & h.app, id, serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 700 , " receipt" : r } ) ) .await, StatusCode:: NO_CONTENT ) ;
897
898
899
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/demo/flow/{} " , run.number) , None , None ) .await;
900
let link = format! ( " /andy/demo/flow/{} /artifacts/build/tool" , run.number) ;
901
assert! ( page.contains ( & link) , " run page lists the artifact" ) ;
902
let ( st, got) = get_bytes ( & h.app, & link) .await;
903
assert_eq! ( ( st, got = = body) , ( StatusCode:: OK , true ) ) ;
904
assert_eq! ( get_bytes ( & h.app, & format! ( " /andy/demo/flow/{} /artifacts/build/none" , run.number) ) .await.0 , StatusCode:: NOT_FOUND ) ;
10708ab8 Andy McCall 1 month ago ↶
905
45977501 Andy McCall 27 days ago ↶
906
907
908
{
909
let name = RepoName:: parse( " andy/demo" ) .unwrap ( ) ;
910
let mut hosted = oxy_core:: repo:: Repo:: open( & h.app.repos.dir ( & name) ) .unwrap ( ) ;
911
hosted.config.set ( " core.gc-grace" , " 0" ) .unwrap ( ) ;
912
hosted.save_config ( ) .unwrap ( ) ;
913
}
914
let done = oxygenhub:: maintenance:: gc_all( & h.app) .await;
915
assert! ( done.iter ( ) .all ( | d | ! d.failed) , " {done:?}" ) ;
916
let ( st, got) = get_bytes ( & h.app, & link) .await;
917
assert_eq! ( ( st, got = = body) , ( StatusCode:: OK , true ) , " the artifact survives gc" ) ;
918
10708ab8 Andy McCall 1 month ago ↶
919
920
921
let form = | f : String| {
922
Request:: post( " /register" ) .header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) .body ( Body:: from( f) ) .unwrap ( )
923
} ;
924
let res = oxygenhub:: router( h.app.clone ( ) )
925
.oneshot ( form ( " username=andy&email=a%40example.com&password=correct-horse-battery" .into ( ) ) )
926
.await
927
.unwrap ( ) ;
928
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
929
let with_cookie = | method : & str , path : & str , body : Option< String> | {
930
let mut req = Request:: builder( ) .method ( method) .uri ( path) .header ( header:: COOKIE , cookie.clone ( ) ) ;
931
if body.is_some ( ) {
932
req = req.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) ;
933
}
934
req.body ( body.map ( Body:: from) .unwrap_or_else ( Body:: empty) ) .unwrap ( )
935
} ;
936
let row = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
937
db:: releases:: create( & h.app.db, row.id, " v1" , " One" , " " , false , None ) .await.unwrap ( ) ;
938
let res = oxygenhub:: router( h.app.clone ( ) ) .oneshot ( with_cookie ( " GET" , & format! ( " /andy/demo/flow/{} " , run.number) , None ) ) .await.unwrap ( ) ;
939
let page = String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) ;
940
let attach = format! ( " /andy/demo/flow/{} /artifacts/build/tool/attach" , run.number) ;
941
assert! ( page.contains ( & attach) , " a writer is offered the release" ) ;
942
let i = page.find ( " name=\" _csrf\" value=\" " ) .unwrap ( ) + " name=\" _csrf\" value=\" " .len ( ) ;
943
let csrf = page[ i.. ] .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( ) ;
944
let release_id = db:: releases:: list( & h.app.db, row.id) .await.unwrap ( ) [ 0 ] .id;
945
let res = oxygenhub:: router( h.app.clone ( ) )
946
.oneshot ( with_cookie ( " POST" , & attach, Some ( format! ( " _csrf={csrf} &release={release_id} " ) ) ) )
947
.await
948
.unwrap ( ) ;
949
assert_eq! ( res.status ( ) , StatusCode:: SEE_OTHER ) ;
950
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/releases/tag/v1" , None , None ) .await;
951
assert! (
952
page.contains ( " >tool</a>" ) & & page.contains ( & format! ( " built by <a href=\" /andy/demo/flow/{0} \" >run #{0} </a> build" , run.number) ) ,
953
" {page}"
954
) ;
955
let ( _ , api) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/releases" , None , None ) .await;
956
let api: serde_json:: Value = serde_json:: from_str( & api) .unwrap ( ) ;
957
let asset = & api[ 0 ] [ " assets" ] [ 0 ] ;
958
assert_eq! (
959
( asset[ " name" ] .as_str ( ) , asset[ " built_by" ] [ " run" ] .as_i64 ( ) , asset[ " built_by" ] [ " job" ] .as_str ( ) ) ,
960
( Some ( " tool" ) , Some ( run.number) , Some ( " build" ) )
961
) ;
962
assert! ( asset[ " built_by" ] [ " receipt" ] .is_null ( ) , " that receipt was not signed" ) ;
a3c8288b Andy McCall 1 month ago ↶
963
}
df940d75 Andy McCall 1 month ago ↶
964
d29070fa Andy McCall 1 month ago ↶
965
966
a36bab6b Andy McCall 27 days ago ↶
967
async fn upload_report_file ( app : & Shared, job : i64 , name : & str , path : & str , bytes : & [u8 ]) -> ( StatusCode, String) {
968
let req = Request:: post( format! ( " /api/v1/flow/jobs/{job} /reports/{name} /files" ) )
969
.header ( header:: AUTHORIZATION , " Bearer runner-secret" )
970
.header ( header:: CONTENT_TYPE , " application/octet-stream" )
971
.header ( " X-Flow-Report-Path" , path)
972
.body ( Body:: from( bytes.to_vec ( ) ) )
973
.unwrap ( ) ;
974
let res = oxygenhub:: router( app.clone ( ) ) .oneshot ( req) .await.unwrap ( ) ;
975
let status = res.status ( ) ;
976
( status, String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) )
977
}
978
979
980
981
982
# [ tokio ::test ( flavor = " multi_thread" ) ]
983
async fn reports_are_kept_file_by_file_and_indexed ( ) {
984
let h = hub ( ) .await;
985
create ( & h.app, " andy" , " demo" ) .await;
986
let url = format! ( " {} /andy/demo" , h.base) ;
987
let work = tempfile:: tempdir( ) .unwrap ( ) ;
988
let c = client ( & url, work.path ( ) ) ;
989
write (
990
work.path ( ) ,
991
" .flow/pipeline.kdl" ,
992
& format! (
993
" 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 "
994
) ,
995
) ;
996
c.snapshot ( " test" ) .unwrap ( ) ;
997
c.commit ( " Measure frames" , & [ ] ) .unwrap ( ) ;
998
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
999
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
1000
let a = claim ( & h.app) .await.unwrap ( ) ;
1001
let id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
1002
1003
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 " ;
1004
let page = b " <p>frames</p>" .to_vec ( ) ;
1005
let data: Vec< u8 > = ( 0 .. 100_000 u32 ) .map ( | i | ( i % 241 ) as u8 ) .collect ( ) ;
c8df33a6 Andy McCall 26 days ago ↶
1006
let ( page_bytes, data_bytes) = ( page.clone ( ) , data.clone ( ) ) ;
a36bab6b Andy McCall 27 days ago ↶
1007
let files: [ ( & str , & [ u8 ] ) ; 3 ] = [ ( " data/frames.csv" , & data) , ( " index.html" , & page) , ( " report.toml" , toml) ] ;
1008
1009
assert_eq! ( upload_report_file ( & h.app, id, " tool" , " report.toml" , toml) .await.0 , StatusCode:: BAD_REQUEST , " not a report" ) ;
1010
assert_eq! ( upload_report_file ( & h.app, id, " frame-timing" , " ../x" , toml) .await.0 , StatusCode:: BAD_REQUEST , " not a clean path" ) ;
1011
let mut listed = vec! [ ] ;
1012
for ( path, bytes) in files {
1013
let ( st, text) = upload_report_file ( & h.app, id, " frame-timing" , path, bytes) .await;
1014
assert_eq! ( st, StatusCode:: OK , " {text}" ) ;
1015
let blob = serde_json:: from_str:: < serde_json:: Value> ( & text) .unwrap ( ) [ " blob" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
1016
assert_eq! ( blob, oxy_core:: repo:: Repo:: blob_ref( bytes) .to_hex ( ) ) ;
1017
listed.push ( serde_json:: json! ( { " path" : path, " blob" : blob, " size" : bytes.len ( ) } ) ) ;
1018
}
1019
let mut missing = listed.clone ( ) ;
1020
missing.push ( serde_json:: json! ( { " path" : " assets/x.js" , " blob" : " ab" .repeat ( 32 ) , " size" : 1 } ) ) ;
1021
let ( st, text) =
1022
upload ( & h.app, id, " frame-timing" , & serde_json:: to_vec( & serde_json:: json! ( { " version" : 1 , " files" : missing } ) ) .unwrap ( ) ) .await;
1023
assert_eq! ( st, StatusCode:: BAD_REQUEST , " a file never uploaded: {text}" ) ;
1024
assert! ( text.contains ( " assets/x.js" ) , " {text}" ) ;
1025
let manifest = serde_json:: to_vec( & serde_json:: json! ( { " version" : 1 , " files" : listed } ) ) .unwrap ( ) ;
1026
let ( st, text) = upload ( & h.app, id, " frame-timing" , & manifest) .await;
1027
assert_eq! ( st, StatusCode:: OK , " {text}" ) ;
1028
1029
let ( kind, title, summary, files_n, bytes) : ( String , String , String , i64 , i64 ) =
1030
sqlx:: query_as( " SELECT kind, title, summary, files, bytes FROM reports WHERE job_id = ? AND name = 'frame-timing'" )
1031
.bind ( id)
1032
.fetch_one ( & h.app.db)
1033
.await
1034
.unwrap ( ) ;
1035
assert_eq! ( ( kind.as_str ( ) , title.as_str ( ) , summary.as_str ( ) ) , ( " performance" , " Frame timing" , " Median 16.4 ms" ) ) ;
1036
assert_eq! ( ( files_n, bytes) , ( 3 , ( data.len ( ) + page.len ( ) + toml.len ( ) ) as i64 ) ) ;
1037
let rid: i64 = sqlx:: query_scalar( " SELECT id FROM reports WHERE job_id = ?" ) .bind ( id) .fetch_one ( & h.app.db) .await.unwrap ( ) ;
1038
let m = oxygenhub:: reports:: metrics( & h.app.db, rid) .await.unwrap ( ) ;
1039
assert_eq! ( ( m[ 0 ] .key.as_str ( ) , m[ 0 ] .value, m[ 0 ] .fail_above) , ( " median_ms" , 16. 4 , Some ( 20. 0 ) ) ) ;
1040
let log = db:: flow:: log( & h.app.db, id, 0 , 1000 ) .await.unwrap ( ) ;
1041
assert! ( log.iter ( ) .any ( | r | r.text.contains ( " report frame-timing: performance \" Frame timing\" , 1 metric" ) ) , " {log:?}" ) ;
c8df33a6 Andy McCall 26 days ago ↶
1042
1043
1044
let checks = db:: hooks:: checks_for_state( & h.app.db, run.repo_id, & run.state) .await.unwrap ( ) ;
1045
let c = checks.iter ( ) .find ( | c | c.name = = " report/frame-timing/median_ms" ) .expect ( " a check per threshold" ) ;
1046
assert_eq! ( c.status, " passed" , " {c:?}" ) ;
1047
assert! ( c.summary.contains ( " 16.4 ms" ) & & c.url.contains ( " /reports/frame-timing?id=" ) , " {c:?}" ) ;
1048
1049
1050
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo" , None , None ) .await;
1051
assert! ( page.contains ( " >Reports</a>" ) , " the tab" ) ;
1052
let ( st, page) = call ( & h.app, " GET" , " /andy/demo/reports" , None , None ) .await;
1053
assert! ( st = = StatusCode:: OK & & page.contains ( " Frame timing" ) & & page.contains ( " median_ms" ) , " {st}" ) ;
1054
let ( st, page) = call ( & h.app, " GET" , " /andy/demo/reports/frame-timing" , None , None ) .await;
1055
assert! ( st = = StatusCode:: OK & & page.contains ( " fails above 20" ) & & page.contains ( " sandbox=\" allow-scripts\" " ) , " {st}" ) ;
1056
let src = page.split ( " class=\" rep-frame\" src=\" " ) .nth ( 1 ) .unwrap ( ) .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( ) ;
1057
assert! ( src.ends_with ( " /index.html" ) , " {src}" ) ;
1058
let res = oxygenhub:: router( h.app.clone ( ) ) .oneshot ( Request:: get( & src) .body ( Body:: empty( ) ) .unwrap ( ) ) .await.unwrap ( ) ;
1059
assert_eq! ( res.status ( ) , StatusCode:: OK ) ;
1060
let csp = res.headers ( ) [ header:: CONTENT_SECURITY_POLICY ] .to_str ( ) .unwrap ( ) .to_string ( ) ;
1061
assert! ( csp.starts_with ( " sandbox allow-scripts;" ) & & csp.contains ( " default-src 'none'" ) , " {csp}" ) ;
1062
assert_eq! ( res.headers ( ) [ header:: ACCESS_CONTROL_ALLOW_ORIGIN ] , " *" ) ;
f3e5e2bc Andy McCall 26 days ago ↶
1063
1064
let served = res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) .to_vec ( ) ;
1065
assert! ( served.starts_with ( & page_bytes) , " the report's page" ) ;
1066
assert! ( String :: from_utf8_lossy( & served) .contains ( " /static/report-frame.js" ) , " with the frame script" ) ;
1067
let ( st, _ ) = get_bytes ( & h.app, " /static/report.css" ) .await;
1068
assert_eq! ( st, StatusCode:: OK , " the stylesheet a report may link" ) ;
c8df33a6 Andy McCall 26 days ago ↶
1069
let data = src.replace ( " /index.html" , " /data/frames.csv" ) ;
1070
assert_eq! ( get_bytes ( & h.app, & data) .await, ( StatusCode:: OK , data_bytes.clone ( ) ) ) ;
1071
95b525b0 Andy McCall 26 days ago ↶
1072
1073
1074
let first: i64 = sqlx:: query_scalar( " SELECT id FROM reports WHERE name = 'frame-timing'" ) .fetch_one ( & h.app.db) .await.unwrap ( ) ;
1075
let second: i64 = sqlx:: query_scalar(
1076
" INSERT INTO reports (repo_id, name, kind, title, summary, context, run_id, job_id, run_number, job, bookmark, state, manifest, files, bytes, created_ms) \
1077
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 = ? RETURNING id" ,
1078
)
1079
.bind ( first)
1080
.fetch_one ( & h.app.db)
1081
.await
1082
.unwrap ( ) ;
1083
sqlx:: query(
1084
" INSERT INTO report_metrics (report_id, key, value, unit, better, fail_above) VALUES (?, 'median_ms', 18.0, 'ms', 'lower', 20)" ,
1085
)
1086
.bind ( second)
1087
.execute ( & h.app.db)
1088
.await
1089
.unwrap ( ) ;
1090
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/reports/frame-timing" , None , None ) .await;
1091
assert_eq! ( page.matches ( " data-tl-stop" ) .count ( ) , 2 , " a stop per run" ) ;
1092
assert! ( page.contains ( " Median 18.0 ms" ) , " the newest run is shown" ) ;
1093
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/demo/reports/frame-timing?id={first} &compare={second} &play=3" ) , None , None ) .await;
1094
assert! ( page.contains ( " This run against run #2" ) & & page.contains ( " -1.6" ) , " every metric against the other run" ) ;
1095
let stop = page.split ( " data-tl-stop" ) .next ( ) .unwrap_or ( " " ) .rsplit ( " href=" ) .next ( ) .unwrap_or ( " " ) .to_string ( ) ;
1096
assert! ( page.contains ( " data-play=\" 3\" " ) , " playing" ) ;
1097
assert! ( stop.contains ( " play" ) & & stop.contains ( ' 3' ) , " playing carries on from stop to stop: {stop}" ) ;
1098
assert! ( page.contains ( " data-report-runs=" ) , " the frame interface knows the runs" ) ;
c8df33a6 Andy McCall 26 days ago ↶
1099
let forged = src.replacen ( " /files/" , " /files/9" , 1 ) ;
1100
assert_eq! ( get_bytes ( & h.app, & forged) .await.0 , StatusCode:: NOT_FOUND ) ;
1101
assert_eq! ( get_bytes ( & h.app, & src.replace ( " /index.html" , " /secret.txt" ) ) .await.0 , StatusCode:: NOT_FOUND ) ;
a36bab6b Andy McCall 27 days ago ↶
1102
let _ = run;
1103
1104
1105
{
1106
let name = RepoName:: parse( " andy/demo" ) .unwrap ( ) ;
1107
let mut hosted = oxy_core:: repo:: Repo:: open( & h.app.repos.dir ( & name) ) .unwrap ( ) ;
1108
hosted.config.set ( " core.gc-grace" , " 0" ) .unwrap ( ) ;
1109
hosted.save_config ( ) .unwrap ( ) ;
1110
}
1111
let done = oxygenhub:: maintenance:: gc_all( & h.app) .await;
1112
assert! ( done.iter ( ) .all ( | d | ! d.failed) , " {done:?}" ) ;
1113
let hosted = h.app.repos.open ( & RepoName:: parse( " andy/demo" ) .unwrap ( ) ) .unwrap ( ) ;
1114
for ( path, bytes) in files {
1115
assert_eq! ( hosted.read_blob ( & oxy_core:: repo:: Repo:: blob_ref( bytes) ) .unwrap ( ) , bytes, " {path} survives gc" ) ;
1116
}
1117
}
1118
d29070fa Andy McCall 1 month ago ↶
1119
# [ tokio ::test ( flavor = " multi_thread" ) ]
1120
async fn a_failing_job_keeps_an_artifact_outside_its_receipt ( ) {
1121
let h = hub ( ) .await;
1122
create ( & h.app, " andy" , " demo" ) .await;
1123
let url = format! ( " {} /andy/demo" , h.base) ;
1124
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1125
let c = client ( & url, work.path ( ) ) ;
1126
write (
1127
work.path ( ) ,
1128
" .flow/pipeline.kdl" ,
1129
& format! (
336c953a Andy McCall 1 month ago ↶
1130
" 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 ↶
1131
) ,
1132
) ;
1133
c.snapshot ( " test" ) .unwrap ( ) ;
1134
c.commit ( " Test" , & [ ] ) .unwrap ( ) ;
1135
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1136
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
1137
let a = claim ( & h.app) .await.unwrap ( ) ;
336c953a Andy McCall 1 month ago ↶
1138
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 ↶
1139
let id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
0b203bbf Andy McCall 1 month ago ↶
1140
1141
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>"# ;
1142
let ( st, text) = upload ( & h.app, id, " junit" , xml) .await;
d29070fa Andy McCall 1 month ago ↶
1143
assert_eq! ( st, StatusCode:: OK , " {text}" ) ;
1144
let blob = serde_json:: from_str:: < serde_json:: Value> ( & text) .unwrap ( ) [ " blob" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
1145
let mut r = receipt ( " test" , a[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) , 1 ) ;
1146
r[ " outputs" ] = serde_json:: json! ( [ blob] ) ;
1147
assert_eq! (
1148
finish ( & h.app, id, serde_json:: json! ( { " exit_code" : 1 , " duration_ms" : 700 , " receipt" : r.clone ( ) } ) ) .await,
1149
StatusCode:: BAD_REQUEST ,
1150
" a failing receipt names no outputs"
1151
) ;
1152
r[ " outputs" ] = serde_json:: json! ( [ ] ) ;
1153
assert_eq! ( finish ( & h.app, id, serde_json:: json! ( { " exit_code" : 1 , " duration_ms" : 700 , " receipt" : r } ) ) .await, StatusCode:: NO_CONTENT ) ;
1154
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/demo/flow/{} " , run.number) , None , None ) .await;
1155
assert! ( page.contains ( & format! ( " /andy/demo/flow/{} /artifacts/test/junit" , run.number) ) , " listed" ) ;
1156
assert! ( page.contains ( " not in its receipt" ) , " marked" ) ;
0b203bbf Andy McCall 1 month ago ↶
1157
assert! ( page.contains ( " 2 tests: 1 failed" ) , " counts beside the job" ) ;
1158
assert! ( page.contains ( " tests.test_io::test_write" ) & & page.contains ( " disk full" ) & & page.contains ( " OSError" ) , " the failure" ) ;
1159
assert! ( page.contains ( " tests: 2 tests: 1 failed in 2.0 s" ) , " said in the job's log" ) ;
d29070fa Andy McCall 1 month ago ↶
1160
}
1161
df940d75 Andy McCall 1 month ago ↶
1162
1163
1164
1165
# [ tokio ::test ( flavor = " multi_thread" ) ]
1166
async fn a_release_is_audited_from_authors_to_receipts ( ) {
1167
use oxy_core:: hash:: { hash_object, ChangeHash, ObjectType} ;
1168
use oxy_core:: identity:: SecretKey;
1169
use oxy_core:: objects:: StateObject;
1170
use oxy_core:: receipt:: SignedReceipt;
1171
let h = hub ( ) .await;
1172
create ( & h.app, " andy" , " demo" ) .await;
1173
let url = format! ( " {} /andy/demo" , h.base) ;
1174
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1175
let mut c = client ( & url, work.path ( ) ) ;
1176
c.create_identity ( " Andy" , " [email protected] " , oxy_core:: repo:: IdentityScope:: Repo) .unwrap ( ) ;
1177
write (
1178
work.path ( ) ,
1179
" .flow/pipeline.kdl" ,
1180
& format! (
1181
" 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 "
1182
) ,
1183
) ;
1184
c.snapshot ( " test" ) .unwrap ( ) ;
1185
c.commit ( " Build a tool" , & [ ] ) .unwrap ( ) ;
1186
let heads = c.bookmark_heads ( ) .unwrap ( ) ;
1187
c.tag_create ( " v1" , & heads, " first" ) .unwrap ( ) ;
1188
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1189
1190
1191
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
1192
let a = claim ( & h.app) .await.unwrap ( ) ;
1193
let id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
1194
let body: Vec< u8 > = b " #!/bin/sh\n echo tool\n " .to_vec ( ) ;
1195
let ( _ , text) = upload ( & h.app, id, " tool" , & body) .await;
1196
let blob = serde_json:: from_str:: < serde_json:: Value> ( & text) .unwrap ( ) [ " blob" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
1197
let mut r = receipt ( " build" , a[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) , 0 ) ;
1198
r[ " outputs" ] = serde_json:: json! ( [ blob] ) ;
1199
let heads_run: Vec< ChangeHash> = run.head_list ( ) .into_iter ( ) .filter_map ( ChangeHash:: from_hex) .collect ( ) ;
1200
let state = hash_object ( ObjectType:: State, & StateObject:: new( heads_run) .encode ( ) ) ;
1201
let repo_id = h.app.repos.open ( & RepoName:: parse( " andy/demo" ) .unwrap ( ) ) .unwrap ( ) .id ( ) .unwrap ( ) ;
1202
let sk = SecretKey:: generate( ) ;
1203
let runner = sk.identity ( " test-1 (OxygenFlow runner)" , " " , 1 ) ;
1204
let fr: flow_core:: Receipt = serde_json:: from_value( r.clone ( ) ) .unwrap ( ) ;
1205
let signed = hex:: encode( SignedReceipt:: sign( fr.to_oxy ( ) , repo_id, state, & format! ( " {url} /flow/1" ) , & runner, & sk) .encode ( ) ) ;
1206
let done = serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 900 , " receipt" : r, " signed_receipt" : signed } ) ;
1207
assert_eq! ( finish ( & h.app, id, done) .await, StatusCode:: NO_CONTENT ) ;
1208
1209
1210
let row = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
1211
let rel = db:: releases:: create( & h.app.db, row.id, " v1" , " One" , " " , false , None ) .await.unwrap ( ) ;
1212
oxygenhub:: web:: releases:: store_asset( & h.app, rel.id, " tool" , " " , & body) .await.unwrap ( ) ;
1213
oxygenhub:: web:: releases:: store_asset( & h.app, rel.id, " notes.txt" , " " , b " by hand\n " ) .await.unwrap ( ) ;
1214
1215
let ( st, page) = call ( & h.app, " GET" , " /andy/demo/releases/trust/v1" , None , None ) .await;
1216
assert_eq! ( st, StatusCode:: OK ) ;
1217
for want in [
1218
" Tag v1 is signed" ,
1219
" The one change in the release is authenticated" ,
1220
" tool: signed receipt from run #1 build" ,
1221
" a runner key of this hub" ,
1222
" tool: built from the tagged state" ,
1223
" tool: pinned environment" ,
1224
" notes.txt: no build found" ,
1225
" <svg class=\" trust-graph\" " ,
bf5101d3 Andy McCall 1 month ago ↶
1226
1227
1228
" Tag signed by" ,
1229
" 1 of 1 changes signed or sealed" ,
1230
" 1 of 2 assets built and attested by OxygenFlow" ,
1231
" attested build" ,
1232
" uploaded by hand" ,
1233
" receipt verified" ,
1234
" class=\" shackle\" " ,
1235
" <title>notes.txt:" ,
df940d75 Andy McCall 1 month ago ↶
1236
] {
1237
assert! ( page.contains ( want) , " missing {want:?}" ) ;
1238
}
bf5101d3 Andy McCall 1 month ago ↶
1239
let svg = & page[ page.find ( " <svg class=\" trust-graph\" " ) .unwrap ( ) .. ] ;
1240
assert! ( svg[ .. 200 ] .contains ( " width=\" " ) & & svg[ .. 200 ] .contains ( " height=\" " ) , " the graph has its own size: {}" , & svg[ .. 200 ] ) ;
df940d75 Andy McCall 1 month ago ↶
1241
let ( _ , releases) = call ( & h.app, " GET" , " /andy/demo/releases" , None , None ) .await;
1242
assert! ( releases.contains ( " /andy/demo/releases/trust/v1" ) , " the release links its audit" ) ;
1243
1244
1245
let ( st, json) = call ( & h.app, " GET" , " /andy/demo/releases/trust/v1?format=json" , None , None ) .await;
1246
assert_eq! ( st, StatusCode:: OK ) ;
1247
let t: serde_json:: Value = serde_json:: from_str( & json) .unwrap ( ) ;
1248
assert_eq! ( ( t[ " tag" ] .as_str ( ) , t[ " tag_signature" ] .as_str ( ) , t[ " changes" ] .as_i64 ( ) ) , ( Some ( " v1" ) , Some ( " valid" ) , Some ( 1 ) ) ) ;
1249
assert_eq! ( t[ " authors" ] [ 0 ] [ " who" ] [ " name" ] , " Andy" ) ;
1250
let tool = t[ " assets" ] .as_array ( ) .unwrap ( ) .iter ( ) .find ( | a | a[ " name" ] = = " tool" ) .unwrap ( ) ;
1251
let hex_receipt = tool[ " build" ] [ " signed_receipt" ] .as_str ( ) .unwrap ( ) ;
1252
let sr = SignedReceipt:: decode( & hex:: decode( hex_receipt) .unwrap ( ) ) .unwrap ( ) ;
1253
sr.verify ( ) .unwrap ( ) ;
1254
assert_eq! ( sr.receipt.outputs[ 0 ] .to_hex ( ) , blob) ;
1255
let audit = t[ " audit" ] .as_array ( ) .unwrap ( ) ;
1256
assert! ( audit.iter ( ) .any ( | c | c[ " verdict" ] = = " warn" & & c[ " title" ] .as_str ( ) .unwrap ( ) .starts_with ( " notes.txt" ) ) ) ;
1257
assert! ( ! audit.iter ( ) .any ( | c | c[ " verdict" ] = = " fail" ) , " {audit:?}" ) ;
1258
1259
let ( st, api) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/releases/trust/v1" , None , None ) .await;
1260
assert_eq! ( st, StatusCode:: OK ) ;
1261
assert_eq! ( serde_json:: from_str:: < serde_json:: Value> ( & api) .unwrap ( ) [ " assets" ] .as_array ( ) .unwrap ( ) .len ( ) , 2 ) ;
1262
assert_eq! ( call ( & h.app, " GET" , " /api/v1/repos/andy/demo/releases/trust/v9" , None , None ) .await.0 , StatusCode:: NOT_FOUND ) ;
1263
assert_eq! ( call ( & h.app, " GET" , " /andy/demo/releases/trust/v9" , None , None ) .await.0 , StatusCode:: NOT_FOUND ) ;
1264
}
b6b7e4cf Andy McCall 1 month ago ↶
1265
1266
1267
1268
1269
# [ tokio ::test ( flavor = " multi_thread" ) ]
1270
async fn runners_register_show_their_history_and_are_drained_and_removed ( ) {
1271
use oxy_core:: identity:: SecretKey;
1272
let h = hub ( ) .await;
1273
create ( & h.app, " andy" , " demo" ) .await;
1274
1275
let res = oxygenhub:: router( h.app.clone ( ) )
1276
.oneshot (
1277
Request:: post( " /register" )
1278
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
1279
.body ( Body:: from( " username=carol&email=carol%40example.com&password=correct-horse-battery" ) )
1280
.unwrap ( ) ,
1281
)
1282
.await
1283
.unwrap ( ) ;
1284
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
1285
let admin = db:: accounts:: account_by_name( & h.app.db, " carol" ) .await.unwrap ( ) .unwrap ( ) ;
1286
db:: accounts:: set_admin( & h.app.db, admin.id, true ) .await.unwrap ( ) ;
1287
let web = | method : & 'static str , path : String, body : Option< String> | {
1288
let ( app, cookie) = ( h.app.clone ( ) , cookie.clone ( ) ) ;
1289
async move {
1290
let mut req = Request:: builder( ) .method ( method) .uri ( path) .header ( header:: COOKIE , cookie) ;
1291
if body.is_some ( ) {
1292
req = req.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) ;
1293
}
1294
let res = oxygenhub:: router( app) .oneshot ( req.body ( body.map ( Body:: from) .unwrap_or_else ( Body:: empty) ) .unwrap ( ) ) .await.unwrap ( ) ;
1295
let st = res.status ( ) ;
1296
( st, String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) )
1297
}
1298
} ;
1299
let csrf_of = | page : & str | {
1300
let i = page.find ( " name=\" _csrf\" value=\" " ) .unwrap ( ) + " name=\" _csrf\" value=\" " .len ( ) ;
1301
page[ i.. ] .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( )
1302
} ;
1303
let ( st, page) = web ( " GET" , " /admin/runners" .into ( ) , None ) .await;
1304
assert_eq! ( st, StatusCode:: OK ) ;
1305
assert! ( page.contains ( " No runners yet" ) ) ;
1306
assert_eq! ( call ( & h.app, " GET" , " /admin/runners" , None , None ) .await.0 , StatusCode:: NOT_FOUND , " administrators only" ) ;
1307
1308
1309
let ( _ , page) = web ( " POST" , " /admin/runners/token" .into ( ) , Some ( format! ( " _csrf={} " , csrf_of ( & page) ) ) ) .await;
1310
let i = page.find ( " --token " ) .unwrap ( ) + " --token " .len ( ) ;
1311
let token: String = page[ i.. ] .chars ( ) .take_while ( | c | ! c.is_whitespace ( ) ) .collect ( ) ;
1312
assert! ( token.starts_with ( " flowreg_" ) , " {token}" ) ;
1313
1314
1315
let sk = SecretKey:: generate( ) ;
1316
let identity = sk.identity ( " plex (OxygenFlow runner)" , " " , 1 ) ;
1317
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 } ) ;
1318
let reg = serde_json:: json! ( { " token" : token, " name" : " plex" , " identity" : hex:: encode( identity.encode ( ) ) , " info" : info } ) ;
1319
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await;
1320
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1321
let credential = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " credential" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
1322
assert! ( credential.starts_with ( " flowrunner_" ) ) ;
1323
assert_eq! (
1324
call ( & h.app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await.0 ,
1325
StatusCode:: FORBIDDEN ,
1326
" used once"
1327
) ;
1328
1329
1330
let claim_as = | who : String| {
1331
let app = h.app.clone ( ) ;
1332
let info = info.clone ( ) ;
1333
async move {
1334
let body = serde_json:: json! ( { " runner" : " ignored" , " runtimes" : [ " microvm" ] , " info" : info } ) .to_string ( ) ;
1335
call ( & app, " POST" , " /api/v1/flow/claim" , Some ( & who) , Some ( body) ) .await
1336
}
1337
} ;
1338
assert_eq! ( claim_as ( credential.clone ( ) ) .await.0 , StatusCode:: NO_CONTENT , " nothing queued" ) ;
1339
assert_eq! ( claim ( & h.app) .await, None ) ;
1340
let ( _ , page) = web ( " GET" , " /admin/runners" .into ( ) , None ) .await;
1341
assert! (
1342
page.contains ( " >plex</strong>" ) & & page.contains ( " st-online" ) & & page.contains ( " x86_64 · linux" ) & & page.contains ( " big-disk" ) ,
1343
" {page}"
1344
) ;
1345
assert! ( page.contains ( " >test-1</strong>" ) & & page.contains ( " hub.toml" ) , " the configured runner, by its claimed name" ) ;
1346
let plex = db:: runners:: list( & h.app.db) .await.unwrap ( ) .into_iter ( ) .find ( | r | r.name = = " plex" ) .unwrap ( ) ;
1347
assert_eq! ( plex.key, identity.hash ( ) .to_hex ( ) , " bound at registration" ) ;
1348
1349
1350
let url = format! ( " {} /andy/demo" , h.base) ;
1351
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1352
let c = client ( & url, work.path ( ) ) ;
1353
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
1354
c.snapshot ( " test" ) .unwrap ( ) ;
1355
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
1356
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1357
runs_of ( & h.app, " andy" , " demo" ) .await;
1358
let ( st, body) = claim_as ( credential.clone ( ) ) .await;
1359
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1360
let job: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
1361
let ( _ , page) = web ( " GET" , format! ( " /admin/runners/{} " , plex.id) , None ) .await;
1362
assert! (
1363
page.contains ( " andy/demo" ) & & page.contains ( job[ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) ) & & page.contains ( " Running now</dt><dd>1 job" ) ,
1364
" {page}"
1365
) ;
b66e70b7 Andy McCall 1 month ago ↶
1366
1367
let ( _ , page) = web ( " GET" , " /admin/runners" .into ( ) , None ) .await;
e9b9619c Andy McCall 21 days ago ↶
1368
assert! ( page.contains ( " id=\" queue\" " ) & & page.contains ( " >plex</a></td>" ) , " {page}" ) ;
1369
1370
1371
assert! ( page.contains ( " running 1 of " ) & & page.contains ( " /admin/runners?runner=plex#queue" ) , " {page}" ) ;
1372
let ( _ , only) = web ( " GET" , " /admin/runners?runner=plex" .into ( ) , None ) .await;
1373
assert! ( only.contains ( " >plex</a></td>" ) & & only.contains ( " of " ) & & only.contains ( " >all</a>" ) , " {only}" ) ;
1374
let ( _ , none) = web ( " GET" , " /admin/runners?runner=nobody&status=running" .into ( ) , None ) .await;
1375
assert! ( none.contains ( " Nothing matches." ) , " {none}" ) ;
1376
1377
sqlx:: query( " UPDATE flow_runners SET last_seen_ms = 0 WHERE id = $1" ) .bind ( plex.id) .execute ( & h.app.db) .await.unwrap ( ) ;
1378
let ( _ , page) = web ( " GET" , " /admin/runners" .into ( ) , None ) .await;
1379
assert! ( page.contains ( " st-offline" ) , " quiet for long: offline" ) ;
1380
assert_eq! ( call ( & h.app, " POST" , " /api/v1/flow/runners/heartbeat" , Some ( & credential) , None ) .await.0 , StatusCode:: NO_CONTENT ) ;
1381
assert_eq! ( call ( & h.app, " POST" , " /api/v1/flow/runners/heartbeat" , Some ( " flowrunner_nope" ) , None ) .await.0 , StatusCode:: UNAUTHORIZED ) ;
1382
let plex_now = db:: runners:: list( & h.app.db) .await.unwrap ( ) .into_iter ( ) .find ( | r | r.name = = " plex" ) .unwrap ( ) ;
1383
assert_eq! ( plex_now.connectivity ( ) , " online" ) ;
b66e70b7 Andy McCall 1 month ago ↶
1384
let ( _ , tokens) = web ( " GET" , " /settings/tokens" .into ( ) , None ) .await;
1385
let ( _ , made) = web ( " POST" , " /settings/tokens" .into ( ) , Some ( format! ( " _csrf={} &name=admin&scopes=user%3Aread" , csrf_of ( & tokens) ) ) ) .await;
1386
let i = made.find ( " oxyhub_" ) .unwrap ( ) ;
1387
let admin_token: String = made[ i.. ] .chars ( ) .take_while ( | c | c.is_ascii_alphanumeric ( ) | | * c = = ' _' ) .collect ( ) ;
1388
let ( st, body) = call ( & h.app, " GET" , " /api/v1/runners" , Some ( & admin_token) , None ) .await;
1389
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1390
let listed: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
1391
let plex_row = listed.as_array ( ) .unwrap ( ) .iter ( ) .find ( | r | r[ " name" ] = = " plex" ) .unwrap ( ) ;
1392
assert_eq! ( ( plex_row[ " running" ] .as_i64 ( ) , plex_row[ " arch" ] .as_str ( ) ) , ( Some ( 1 ) , Some ( " x86_64" ) ) ) ;
7f5f6c05 Andy McCall 1 month ago ↶
1393
1394
let ( _ , page) = web ( " GET" , format! ( " /admin/runners/{} " , plex.id) , None ) .await;
1395
let ( st, _ ) = web (
1396
" POST" ,
1397
format! ( " /admin/runners/{} /limits" , plex.id) ,
1398
Some ( format! ( " _csrf={} &labels=gpu%2C+plex&timeout_minutes=5" , csrf_of ( & page) ) ) ,
1399
)
1400
.await;
1401
assert_eq! ( st, StatusCode:: SEE_OTHER ) ;
1402
let ( _ , page) = web ( " GET" , format! ( " /admin/runners/{} " , plex.id) , None ) .await;
1403
assert! ( page.contains ( " >gpu<" ) & & page.contains ( " 5 min for its jobs" ) , " {page}" ) ;
1404
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/runners" , Some ( & admin_token) , None ) .await;
1405
let listed: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
1406
let plex_row = listed.as_array ( ) .unwrap ( ) .iter ( ) .find ( | r | r[ " name" ] = = " plex" ) .unwrap ( ) ;
1407
assert! ( plex_row[ " labels" ] .as_array ( ) .unwrap ( ) .iter ( ) .any ( | l | l = = " gpu" ) , " {plex_row}" ) ;
1408
let ( st, _ ) = web (
1409
" POST" ,
1410
format! ( " /admin/runners/{} /limits" , plex.id) ,
1411
Some ( format! ( " _csrf={} &labels=bad%20label%21&timeout_minutes=" , csrf_of ( & page) ) ) ,
1412
)
1413
.await;
1414
assert_eq! ( st, StatusCode:: BAD_REQUEST , " a label with a space or punctuation is refused" ) ;
b66e70b7 Andy McCall 1 month ago ↶
1415
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/runners/queue" , Some ( & admin_token) , None ) .await;
1416
let queue: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
1417
assert! ( queue.as_array ( ) .unwrap ( ) .iter ( ) .any ( | q | q[ " status" ] = = " running" & & q[ " runner" ] = = " plex" ) , " {queue}" ) ;
1418
let ( _ , body) = call ( & h.app, " GET" , & format! ( " /api/v1/runners/{} /jobs" , plex.id) , Some ( & admin_token) , None ) .await;
1419
let jobs: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
1420
assert_eq! ( jobs[ 0 ] [ " repo" ] , " andy/demo" , " {jobs}" ) ;
1421
assert_eq! ( call ( & h.app, " GET" , " /api/v1/runners" , None , None ) .await.0 , StatusCode:: NOT_FOUND , " administrators only" ) ;
b6b7e4cf Andy McCall 1 month ago ↶
1422
1423
1424
let ( _ , page) = web ( " GET" , " /admin/runners" .into ( ) , None ) .await;
1425
let csrf = csrf_of ( & page) ;
1426
web ( " POST" , format! ( " /admin/runners/{} /state" , plex.id) , Some ( format! ( " _csrf={csrf} &state=draining" ) ) ) .await;
1427
assert_eq! ( claim_as ( credential.clone ( ) ) .await.0 , StatusCode:: NO_CONTENT , " a draining runner takes nothing" ) ;
1428
let ( _ , page) = web ( " POST" , format! ( " /admin/runners/{} /remove" , plex.id) , Some ( format! ( " _csrf={csrf} " ) ) ) .await;
1429
assert! ( page.contains ( " is running jobs: drain it first" ) , " {page}" ) ;
1430
1431
let id = job[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
1432
let done = call (
1433
& h.app,
1434
" POST" ,
1435
& format! ( " /api/v1/flow/jobs/{id} /finish" ) ,
1436
Some ( & credential) ,
1437
Some ( r # "{"exit_code":0,"duration_ms":5}"# .into ( ) ) ,
1438
)
1439
.await;
1440
assert_eq! ( done.0 , StatusCode:: NO_CONTENT ) ;
1441
web ( " POST" , format! ( " /admin/runners/{} /state" , plex.id) , Some ( format! ( " _csrf={csrf} &state=disabled" ) ) ) .await;
1442
assert_eq! ( claim_as ( credential.clone ( ) ) .await.0 , StatusCode:: FORBIDDEN ) ;
1443
let ( _ , page) = web ( " POST" , format! ( " /admin/runners/{} /remove" , plex.id) , Some ( format! ( " _csrf={csrf} " ) ) ) .await;
1444
assert! ( page.contains ( " Removed plex." ) , " {page}" ) ;
1445
assert_eq! ( claim_as ( credential) .await.0 , StatusCode:: UNAUTHORIZED ) ;
1446
}
74be9796 Andy McCall 1 month ago ↶
1447
1448
1449
1450
# [ tokio ::test ( flavor = " multi_thread" ) ]
1451
async fn scoped_runners_take_only_their_scope ( ) {
1452
use oxy_core:: identity:: SecretKey;
1453
let h = hub ( ) .await;
1454
create ( & h.app, " andy" , " demo" ) .await;
1455
create ( & h.app, " andy" , " other" ) .await;
1456
let register = | scope : db:: runners:: Scope, name : & 'static str | {
1457
let app = h.app.clone ( ) ;
1458
async move {
1459
let token = format! ( " flowreg_{name} " ) ;
1460
db:: runners:: add_registration( & app.db, & blake3:: hash( token.as_bytes ( ) ) .to_hex ( ) , " admin" , & scope) .await.unwrap ( ) ;
1461
let identity = SecretKey:: generate( ) .identity ( name, " " , 1 ) ;
1462
let info = serde_json:: json! ( { " arch" : " x86_64" , " os" : " linux" , " version" : " 0.1.0" , " labels" : [ ] , " max_jobs" : 1 } ) ;
1463
let reg = serde_json:: json! ( { " token" : token, " name" : name, " identity" : hex:: encode( identity.encode ( ) ) , " info" : info } ) ;
1464
let ( st, body) = call ( & app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await;
1465
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1466
serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " credential" ] .as_str ( ) .unwrap ( ) .to_string ( )
1467
}
1468
} ;
1469
let demo = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
1470
let only_demo = register ( db:: runners:: Scope { owner: " andy" .into ( ) , repo_id: Some ( demo.id) } , " demo-runner" ) .await;
1471
let carols = register ( db:: runners:: Scope { owner: " carol" .into ( ) , repo_id: None } , " carol-runner" ) .await;
1472
let andys = register ( db:: runners:: Scope { owner: " andy" .into ( ) , repo_id: None } , " andy-runner" ) .await;
1473
let claim_as = | who : String| {
1474
let app = h.app.clone ( ) ;
1475
async move {
1476
let body = serde_json:: json! ( { " runner" : " x" , " runtimes" : [ " microvm" ] } ) .to_string ( ) ;
1477
let ( st, body) = call ( & app, " POST" , " /api/v1/flow/claim" , Some ( & who) , Some ( body) ) .await;
1478
( st = = StatusCode:: OK ) .then ( | | serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " repo" ] .as_str ( ) .unwrap ( ) .to_string ( ) )
1479
}
1480
} ;
1481
1482
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1483
let c = client ( & format! ( " {} /andy/other" , h.base) , work.path ( ) ) ;
1484
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
1485
c.snapshot ( " t" ) .unwrap ( ) ;
1486
c.commit ( " pipeline" , & [ ] ) .unwrap ( ) ;
1487
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1488
assert! ( ! runs_of ( & h.app, " andy" , " other" ) .await.is_empty ( ) ) ;
1489
assert_eq! ( claim_as ( only_demo.clone ( ) ) .await, None , " nothing of andy/demo is queued" ) ;
1490
assert_eq! ( claim_as ( carols) .await, None , " nothing of carol's" ) ;
1491
assert_eq! ( claim_as ( andys) .await.as_deref ( ) , Some ( " andy/other" ) , " the owner's runner takes its repositories' jobs" ) ;
1492
1493
assert_eq! ( claim ( & h.app) .await.unwrap ( ) [ " repo" ] , " andy/other" ) ;
1494
assert_eq! ( claim_as ( only_demo) .await, None , " still nothing of its own" ) ;
1495
}
7974adb0 Andy McCall 1 month ago ↶
1496
1497
1498
1499
1500
# [ tokio ::test ( flavor = " multi_thread" ) ]
1501
async fn repository_admins_add_and_remove_their_own_runners ( ) {
1502
use oxy_core:: identity:: SecretKey;
1503
let h = hub ( ) .await;
1504
let res = oxygenhub:: router( h.app.clone ( ) )
1505
.oneshot (
1506
Request:: post( " /register" )
1507
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
1508
.body ( Body:: from( " username=andy&email=andy%40example.com&password=correct-horse-battery" ) )
1509
.unwrap ( ) ,
1510
)
1511
.await
1512
.unwrap ( ) ;
1513
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
1514
create ( & h.app, " andy" , " demo" ) .await;
1515
let demo = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
1516
let web = | method : & 'static str , path : String, body : Option< String> | {
1517
let ( app, cookie) = ( h.app.clone ( ) , cookie.clone ( ) ) ;
1518
async move {
1519
let mut req = Request:: builder( ) .method ( method) .uri ( path) .header ( header:: COOKIE , cookie) ;
1520
if body.is_some ( ) {
1521
req = req.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) ;
1522
}
1523
let res = oxygenhub:: router( app) .oneshot ( req.body ( body.map ( Body:: from) .unwrap_or_else ( Body:: empty) ) .unwrap ( ) ) .await.unwrap ( ) ;
1524
let st = res.status ( ) ;
1525
( st, String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) )
1526
}
1527
} ;
1528
let csrf_of = | page : & str | {
1529
let i = page.find ( " name=\" _csrf\" value=\" " ) .unwrap ( ) + " name=\" _csrf\" value=\" " .len ( ) ;
1530
page[ i.. ] .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( )
1531
} ;
1532
let ( st, page) = web ( " GET" , " /andy/demo/settings" .into ( ) , None ) .await;
1533
assert_eq! ( st, StatusCode:: OK ) ;
1534
assert! ( page.contains ( " <h2>Runners</h2>" ) , " the owner sees the section" ) ;
1535
let ( _ , page) = web ( " POST" , " /andy/demo/settings/runners/token" .into ( ) , Some ( format! ( " _csrf={} " , csrf_of ( & page) ) ) ) .await;
1536
let i = page.find ( " --token " ) .unwrap ( ) + " --token " .len ( ) ;
1537
let token: String = page[ i.. ] .chars ( ) .take_while ( | c | ! c.is_whitespace ( ) ) .collect ( ) ;
1538
let identity = SecretKey:: generate( ) .identity ( " mine" , " " , 1 ) ;
1539
let info = serde_json:: json! ( { " arch" : " x86_64" , " os" : " linux" , " version" : " 0.1.0" , " labels" : [ ] , " max_jobs" : 1 } ) ;
1540
let reg = serde_json:: json! ( { " token" : token, " name" : " mine" , " identity" : hex:: encode( identity.encode ( ) ) , " info" : info } ) ;
1541
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await;
1542
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1543
let mine = db:: runners:: list( & h.app.db) .await.unwrap ( ) .into_iter ( ) .find ( | r | r.name = = " mine" ) .unwrap ( ) ;
1544
assert_eq! ( mine.scope ( ) , db:: runners:: Scope { owner: " andy" .into ( ) , repo_id: Some ( demo.id) } ) ;
1545
1546
claim ( & h.app) .await;
1547
let site = db:: runners:: list( & h.app.db) .await.unwrap ( ) .into_iter ( ) .find ( | r | r.name = = " test-1" ) .unwrap ( ) ;
1548
let ( _ , page) = web ( " GET" , " /andy/demo/settings" .into ( ) , None ) .await;
1549
assert! ( page.contains ( " <strong>mine</strong>" ) & & page.contains ( " this repository" ) , " {page}" ) ;
1550
assert! ( page.contains ( " <strong>test-1</strong>" ) & & page.contains ( " the whole site" ) , " {page}" ) ;
1551
let csrf = csrf_of ( & page) ;
1552
let ( st, _ ) = web ( " POST" , format! ( " /andy/demo/settings/runners/{} /remove" , site.id) , Some ( format! ( " _csrf={csrf} " ) ) ) .await;
1553
assert_eq! ( st, StatusCode:: NOT_FOUND , " not the repository's own" ) ;
1554
let ( _ , page) = web ( " POST" , format! ( " /andy/demo/settings/runners/{} /remove" , mine.id) , Some ( format! ( " _csrf={csrf} " ) ) ) .await;
1555
assert! ( page.contains ( " mine removed" ) , " {page}" ) ;
1556
assert! ( db:: runners:: get( & h.app.db, mine.id) .await.unwrap ( ) .is_none ( ) ) ;
1557
1558
1559
let ( _ , page) = web ( " GET" , " /orgs/new" .into ( ) , None ) .await;
1560
web ( " POST" , " /orgs/new" .into ( ) , Some ( format! ( " _csrf={} &name=acme&display_name=Acme" , csrf_of ( & page) ) ) ) .await;
1561
let ( st, page) = web ( " GET" , " /orgs/acme/settings" .into ( ) , None ) .await;
1562
assert_eq! ( st, StatusCode:: OK ) ;
1563
let ( _ , page) = web ( " POST" , " /orgs/acme/settings/runners/token" .into ( ) , Some ( format! ( " _csrf={} " , csrf_of ( & page) ) ) ) .await;
1564
let i = page.find ( " --token " ) .unwrap ( ) + " --token " .len ( ) ;
1565
let token: String = page[ i.. ] .chars ( ) .take_while ( | c | ! c.is_whitespace ( ) ) .collect ( ) ;
1566
let identity = SecretKey:: generate( ) .identity ( " acme-runner" , " " , 1 ) ;
1567
let reg = serde_json:: json! ( { " token" : token, " name" : " acme-runner" , " identity" : hex:: encode( identity.encode ( ) ) , " info" : info } ) ;
1568
assert_eq! ( call ( & h.app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await.0 , StatusCode:: OK ) ;
1569
let acme = db:: runners:: list( & h.app.db) .await.unwrap ( ) .into_iter ( ) .find ( | r | r.name = = " acme-runner" ) .unwrap ( ) ;
1570
assert_eq! ( acme.scope ( ) , db:: runners:: Scope { owner: " acme" .into ( ) , repo_id: None } ) ;
1571
let ( _ , page) = web ( " GET" , " /orgs/acme/settings" .into ( ) , None ) .await;
1572
assert! ( page.contains ( " <strong>acme-runner</strong>" ) & & page.contains ( " the organisation's repositories" ) , " {page}" ) ;
1573
let ( _ , page) = web ( " POST" , format! ( " /orgs/acme/settings/runners/{} /remove" , acme.id) , Some ( format! ( " _csrf={} " , csrf_of ( & page) ) ) ) .await;
1574
assert! ( page.contains ( " acme-runner removed" ) , " {page}" ) ;
1575
}
c4c87f48 Andy McCall 1 month ago ↶
1576
1577
1578
1579
# [ tokio ::test ( flavor = " multi_thread" ) ]
1580
async fn jobs_that_ask_for_labels_wait_for_a_runner_with_them ( ) {
1581
use oxy_core:: identity:: SecretKey;
1582
let h = hub ( ) .await;
1583
create ( & h.app, " andy" , " demo" ) .await;
1584
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1585
let c = client ( & format! ( " {} /andy/demo" , h.base) , work.path ( ) ) ;
1586
let cross = " job \" cross\" {\n sandbox \" alpine\" \n runs-on \" riscv64\" \n exec \" true\" \n }\n " ;
1587
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( cross) ) ;
1588
c.snapshot ( " t" ) .unwrap ( ) ;
1589
c.commit ( " pipeline" , & [ ] ) .unwrap ( ) ;
1590
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1591
assert! ( ! runs_of ( & h.app, " andy" , " demo" ) .await.is_empty ( ) ) ;
1592
1593
let mut taken = Vec :: new( ) ;
1594
while let Some ( j) = claim ( & h.app) .await {
1595
taken.push ( j[ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) .to_string ( ) ) ;
1596
}
1597
assert! ( ! taken.contains ( & " cross" .to_string ( ) ) & & taken.contains ( & " lint" .to_string ( ) ) , " {taken:?}" ) ;
1598
let queue = db:: flow:: queue( & h.app.db) .await.unwrap ( ) ;
1599
let waiting = queue.iter ( ) .find ( | q | q.job = = " cross" ) .unwrap ( ) ;
1600
assert_eq! ( ( waiting.status.as_str ( ) , waiting.needs ( ) ) , ( " queued" , vec! [ " riscv64" .to_string ( ) ] ) ) ;
1601
1602
let token = " flowreg_riscv" ;
1603
db:: runners:: add_registration( & h.app.db, & blake3:: hash( token.as_bytes ( ) ) .to_hex ( ) , " admin" , & Default :: default( ) ) .await.unwrap ( ) ;
1604
let identity = SecretKey:: generate( ) .identity ( " visionfive" , " " , 1 ) ;
1605
let info = serde_json:: json! ( { " arch" : " riscv64" , " os" : " linux" , " version" : " 0.1.0" , " labels" : [ " riscv64" ] , " max_jobs" : 1 } ) ;
1606
let reg = serde_json:: json! ( { " token" : token, " name" : " visionfive" , " identity" : hex:: encode( identity.encode ( ) ) , " info" : info } ) ;
1607
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/runners/register" , None , Some ( reg.to_string ( ) ) ) .await;
1608
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1609
let credential = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " credential" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
1610
let claim_body = serde_json:: json! ( { " runner" : " x" , " runtimes" : [ " microvm" ] , " info" : info } ) .to_string ( ) ;
1611
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( & credential) , Some ( claim_body) ) .await;
1612
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1613
assert_eq! ( serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " job" ] [ " name" ] , " cross" ) ;
1614
}
070a7b64 Andy McCall 1 month ago ↶
1615
1616
1617
1618
1619
1620
# [ tokio ::test ( flavor = " multi_thread" ) ]
1621
async fn a_job_no_runner_reports_is_ended_after_the_time_limit ( ) {
1622
let h = hub ( ) .await;
1623
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
1624
create ( & h.app, " andy" , " lost" ) .await;
1625
let url = format! ( " {} /andy/lost" , h.base) ;
1626
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1627
let c = client ( & url, work.path ( ) ) ;
1628
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
1629
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
1630
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
1631
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1632
assert_eq! ( runs_of ( & h.app, " andy" , " lost" ) .await.len ( ) , 1 ) ;
1633
let job = claim ( & h.app) .await.expect ( " a job to claim" ) ;
1634
let id = job[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
1635
1636
1637
assert_eq! ( oxygenhub:: flow:: reap_lost_jobs( & h.app) .await.unwrap ( ) , 0 ) ;
1638
1639
let old = db:: now_ms( ) - 24 * 3600 * 1000 ;
1640
sqlx:: query( " UPDATE flow_jobs SET started_ms = ? WHERE id = ?" ) .bind ( old) .bind ( id) .execute ( & h.app.db) .await.unwrap ( ) ;
1641
assert_eq! ( oxygenhub:: flow:: reap_lost_jobs( & h.app) .await.unwrap ( ) , 1 ) ;
1642
let j = db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) ;
1643
assert_eq! ( j.status, " error" ) ;
1644
assert! ( j.error.starts_with ( " lost:" ) , " {}" , j.error) ;
1645
let run = & runs_of ( & h.app, " andy" , " lost" ) .await[ 0 ] ;
1646
let jobs = db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) ;
1647
assert! ( jobs.iter ( ) .all ( | j | j.status ! = " running" ) , " nothing left running" ) ;
1648
1649
assert_ne! ( finish ( & h.app, id, serde_json:: json! ( { " exit_code" : 0 } ) ) .await, StatusCode:: NO_CONTENT ) ;
1650
assert_eq! ( db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) .status, " error" ) ;
1651
}
b0612e7d Andy McCall 1 month ago ↶
1652
1653
1654
1655
# [ tokio ::test ( flavor = " multi_thread" ) ]
1656
async fn a_runner_is_offered_only_jobs_whose_image_it_holds ( ) {
1657
let h = hub ( ) .await;
1658
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
1659
create ( & h.app, " andy" , " images" ) .await;
1660
let url = format! ( " {} /andy/images" , h.base) ;
1661
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1662
let c = client ( & url, work.path ( ) ) ;
1663
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
1664
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
1665
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
1666
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1667
assert_eq! ( runs_of ( & h.app, " andy" , " images" ) .await.len ( ) , 1 ) ;
1668
let claim_with = | images : Vec< String> | {
1669
let body = serde_json:: json! ( { " runner" : " test-1" , " runtimes" : [ " microvm" ] , " images" : images } ) .to_string ( ) ;
1670
let app = h.app.clone ( ) ;
1671
async move {
1672
let ( st, body) = call ( & app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( body) ) .await;
1673
( st, body)
1674
}
1675
} ;
1676
let ( st, _ ) = claim_with ( vec! [ " ab" .repeat ( 32 ) ] ) .await;
1677
assert_eq! ( st, StatusCode:: NO_CONTENT , " no job for a runner without the image" ) ;
1678
let jobs = db:: flow:: jobs( & h.app.db, runs_of ( & h.app, " andy" , " images" ) .await[ 0 ] .id) .await.unwrap ( ) ;
1679
let rootfs: serde_json:: Value = serde_json:: from_str( & jobs[ 0 ] .sandbox) .unwrap ( ) ;
1680
let rootfs = rootfs[ " rootfs" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
1681
let ( st, body) = claim_with ( vec! [ " ab" .repeat ( 32 ) , rootfs] ) .await;
1682
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1683
}
eef677c4 Andy McCall 1 month ago ↶
1684
1685
1686
1687
1688
# [ tokio ::test ( flavor = " multi_thread" ) ]
1689
async fn a_job_its_runner_no_longer_holds_is_ended_at_its_next_claim ( ) {
1690
let h = hub ( ) .await;
1691
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
1692
create ( & h.app, " andy" , " held" ) .await;
1693
let url = format! ( " {} /andy/held" , h.base) ;
1694
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1695
let c = client ( & url, work.path ( ) ) ;
1696
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
1697
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
1698
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
1699
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1700
assert_eq! ( runs_of ( & h.app, " andy" , " held" ) .await.len ( ) , 1 ) ;
1701
let id = claim ( & h.app) .await.expect ( " a job to claim" ) [ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
1702
let claim_holding = | holding : Vec< i64 > | {
1703
let body = serde_json:: json! ( { " runner" : " test-1" , " runtimes" : [ " microvm" ] , " holding" : holding } ) .to_string ( ) ;
1704
let app = h.app.clone ( ) ;
1705
async move { call ( & app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( body) ) .await.0 }
1706
} ;
1707
claim_holding ( vec! [ id] ) .await;
1708
assert_eq! ( db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) .status, " running" , " a held job stays" ) ;
1709
claim_holding ( vec! [ ] ) .await;
1710
let j = db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) ;
1711
assert_eq! ( j.status, " error" ) ;
1712
assert! ( j.error.starts_with ( " lost:" ) & & j.error.contains ( " no longer holds" ) , " {}" , j.error) ;
1713
let run = & runs_of ( & h.app, " andy" , " held" ) .await[ 0 ] ;
1714
assert! ( db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) .iter ( ) .all ( | j | j.status ! = " running" ) , " the run settles" ) ;
1715
}
d7ce41ac Andy McCall 1 month ago ↶
1716
1717
1718
1719
1720
# [ tokio ::test ( flavor = " multi_thread" ) ]
1721
async fn a_scoped_job_runs_only_when_a_push_touches_its_paths ( ) {
1722
let h = hub ( ) .await;
1723
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
1724
create ( & h.app, " andy" , " scoped" ) .await;
1725
let url = format! ( " {} /andy/scoped" , h.base) ;
1726
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1727
let c = client ( & url, work.path ( ) ) ;
1728
let extra = " job \" dist\" {\n sandbox \" alpine\" \n scope \" Cargo.toml\" \" packaging/**\" \n exec \" true\" \n }\n " ;
1729
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( extra) ) ;
1730
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
1731
write ( work.path ( ) , " Cargo.toml" , " version = \" 0.1.0\" \n " ) ;
1732
c.commit ( " Start" , & [ ] ) .unwrap ( ) ;
1733
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1734
let names = | app : Shared, n : usize | async move {
1735
let mut runs = Vec :: new( ) ;
1736
for _ in 0 .. 100 {
1737
runs = runs_of ( & app, " andy" , " scoped" ) .await;
1738
if runs.len ( ) > = n {
1739
break ;
1740
}
1741
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
1742
}
1743
assert_eq! ( runs.len ( ) , n, " runs" ) ;
1744
let mut j: Vec< String> = db:: flow:: jobs( & app.db, runs[ 0 ] .id) .await.unwrap ( ) .into_iter ( ) .map ( | j | j.name ) .collect ( ) ;
1745
j.sort ( ) ;
745d225d Andy McCall 1 month ago ↶
1746
1747
sqlx:: query( " UPDATE flow_runs SET status = 'passed' WHERE id = ?" ) .bind ( runs[ 0 ] .id) .execute ( & app.db) .await.unwrap ( ) ;
d7ce41ac Andy McCall 1 month ago ↶
1748
j
1749
} ;
1750
assert_eq! ( names ( h.app.clone ( ) , 1 ) .await, [ " build" , " dist" , " lint" , " test" ] , " the first run has every job" ) ;
1751
write ( work.path ( ) , " test.sh" , " echo still ok\n " ) ;
1752
c.commit ( " Edit the test" , & [ ] ) .unwrap ( ) ;
1753
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1754
assert_eq! ( names ( h.app.clone ( ) , 2 ) .await, [ " build" , " lint" , " test" ] , " dist is out of scope" ) ;
1755
write ( work.path ( ) , " Cargo.toml" , " version = \" 0.1.1\" \n " ) ;
1756
c.commit ( " Release 0.1.1" , & [ ] ) .unwrap ( ) ;
1757
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1758
assert_eq! ( names ( h.app.clone ( ) , 3 ) .await, [ " build" , " dist" , " lint" , " test" ] , " a version bump runs dist" ) ;
1759
}
5dead1f4 Andy McCall 1 month ago ↶
1760
745d225d Andy McCall 1 month ago ↶
1761
1762
1763
1764
# [ tokio ::test ( flavor = " multi_thread" ) ]
1765
async fn runs_are_cancelled_by_hand_or_by_a_newer_push ( ) {
1766
let h = hub ( ) .await;
1767
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
1768
create ( & h.app, " andy" , " stop" ) .await;
1769
let url = format! ( " {} /andy/stop" , h.base) ;
1770
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1771
let c = client ( & url, work.path ( ) ) ;
1772
let extra = " job \" dist\" {\n sandbox \" alpine\" \n scope \" Cargo.toml\" \n exec \" true\" \n }\n " ;
1773
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( extra) ) ;
1774
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
1775
write ( work.path ( ) , " Cargo.toml" , " version = \" 0.1.0\" \n " ) ;
1776
c.commit ( " Start" , & [ ] ) .unwrap ( ) ;
1777
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1778
let wait_runs = | app : Shared, n : usize | async move {
1779
for _ in 0 .. 100 {
1780
let runs = runs_of ( & app, " andy" , " stop" ) .await;
1781
if runs.len ( ) > = n {
1782
return runs;
1783
}
1784
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
1785
}
1786
panic! ( " no run {n}" ) ;
1787
} ;
1788
let first = wait_runs ( h.app.clone ( ) , 1 ) .await;
1789
sqlx:: query( " UPDATE flow_runs SET status = 'passed' WHERE id = ?" ) .bind ( first[ 0 ] .id) .execute ( & h.app.db) .await.unwrap ( ) ;
1790
sqlx:: query( " UPDATE flow_jobs SET status = 'passed' WHERE run_id = ?" ) .bind ( first[ 0 ] .id) .execute ( & h.app.db) .await.unwrap ( ) ;
1791
1792
1793
write ( work.path ( ) , " Cargo.toml" , " version = \" 0.1.1\" \n " ) ;
1794
c.commit ( " Release 0.1.1" , & [ ] ) .unwrap ( ) ;
1795
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1796
wait_runs ( h.app.clone ( ) , 2 ) .await;
1797
let job = claim ( & h.app) .await.expect ( " a job of run 2" ) ;
1798
let id = job[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
1799
write ( work.path ( ) , " test.sh" , " echo still ok\n " ) ;
1800
c.commit ( " Edit the test" , & [ ] ) .unwrap ( ) ;
1801
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1802
let runs = wait_runs ( h.app.clone ( ) , 3 ) .await;
1803
let third = & runs[ 0 ] ;
1804
let mut second = runs[ 1 ] .clone ( ) ;
1805
for _ in 0 .. 100 {
1806
second = db:: flow:: run_by_id( & h.app.db, runs[ 1 ] .id) .await.unwrap ( ) .unwrap ( ) ;
1807
if ! second.live ( ) {
1808
break ;
1809
}
1810
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
1811
}
1812
assert_eq! ( second.status, " cancelled" , " a newer push supersedes a run still going" ) ;
1813
let jobs = db:: flow:: jobs( & h.app.db, second.id) .await.unwrap ( ) ;
1814
assert! ( jobs.iter ( ) .all ( | j | j.status = = " cancelled" & & j.error = = " superseded by run #3" ) , " {jobs:?}" ) ;
1815
let mut names: Vec< String> = db:: flow:: jobs( & h.app.db, third.id) .await.unwrap ( ) .into_iter ( ) .map ( | j | j.name ) .collect ( ) ;
1816
names.sort ( ) ;
1817
assert_eq! ( names, [ " build" , " dist" , " lint" , " test" ] , " run 3 also checks what run 2 would have: the version bump" ) ;
1818
let mut checks = vec! [ ] ;
1819
for _ in 0 .. 100 {
1820
checks = db:: hooks:: checks_for_state( & h.app.db, second.repo_id, & second.state) .await.unwrap ( ) ;
1821
if checks.iter ( ) .all ( | c | c.status = = " cancelled" ) {
1822
break ;
1823
}
1824
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
1825
}
1826
assert! ( checks.iter ( ) .all ( | c | c.status = = " cancelled" ) , " {checks:?}" ) ;
1827
1828
let ( st, _ ) =
1829
call ( & h.app, " POST" , & format! ( " /api/v1/flow/jobs/{id} /log" ) , Some ( " runner-secret" ) , Some ( r # "{"records":[]}"# .into ( ) ) ) .await;
1830
assert_eq! ( st, StatusCode:: CONFLICT ) ;
1831
1832
1833
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/stop/flow/runs/3/cancel" , Some ( " site" ) , None ) .await;
1834
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
1835
let third = db:: flow:: run_by_id( & h.app.db, third.id) .await.unwrap ( ) .unwrap ( ) ;
1836
assert_eq! ( third.status, " cancelled" ) ;
1837
assert! ( db:: flow:: jobs( & h.app.db, third.id) .await.unwrap ( ) .iter ( ) .all ( | j | j.error = = " cancelled by andy" ) ) ;
1838
let ( st, body) = call ( & h.app, " POST" , " /api/v1/repos/andy/stop/flow/runs/3/cancel" , Some ( " site" ) , None ) .await;
1839
assert! ( st = = StatusCode:: CONFLICT & & body.contains ( " already cancelled" ) , " {st} {body}" ) ;
1840
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/repos/andy/stop/flow/runs/3/cancel" , None , None ) .await;
1841
assert_ne! ( st, StatusCode:: OK , " anonymous callers cannot cancel" ) ;
1842
let ( _ , page) = call ( & h.app, " GET" , " /andy/stop/flow/3" , None , None ) .await;
1843
assert! ( page.contains ( " ⊘" ) & & ! page.contains ( " CANCEL</button>" ) , " a cancelled run shows so, with no button" ) ;
1844
}
1845
5dead1f4 Andy McCall 1 month ago ↶
1846
1847
1848
1849
# [ tokio ::test ( flavor = " multi_thread" ) ]
1850
async fn a_job_runs_under_its_own_time_limit_within_the_sites ( ) {
1851
let h = hub ( ) .await;
1852
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
1853
db:: accounts:: set_setting( & h.app.db, oxygenhub:: flow:: SETTING_DEFAULT , " 30" ) .await.unwrap ( ) ;
1854
db:: accounts:: set_setting( & h.app.db, oxygenhub:: flow:: SETTING_MAX , " 120" ) .await.unwrap ( ) ;
1855
create ( & h.app, " andy" , " limits" ) .await;
1856
let url = format! ( " {} /andy/limits" , h.base) ;
1857
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1858
let c = client ( & url, work.path ( ) ) ;
1859
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 " ;
1860
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( extra) ) ;
1861
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
1862
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
1863
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1864
assert_eq! ( runs_of ( & h.app, " andy" , " limits" ) .await.len ( ) , 1 ) ;
1865
let mut limits = std:: collections:: HashMap:: new( ) ;
1866
let mut ids = std:: collections:: HashMap:: new( ) ;
1867
while let Some ( j) = claim ( & h.app) .await {
1868
let name = j[ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
1869
limits.insert ( name.clone ( ) , j[ " timeout_s" ] .as_u64 ( ) .unwrap ( ) ) ;
1870
ids.insert ( name, j[ " job_id" ] .as_i64 ( ) .unwrap ( ) ) ;
1871
}
1872
assert_eq! ( limits[ " slow" ] , 5400 , " its own" ) ;
1873
assert_eq! ( limits[ " huge" ] , 7200 , " cut to the site's maximum" ) ;
1874
assert_eq! ( limits[ " build" ] , 1800 , " the site's default" ) ;
1875
1876
let started = db:: now_ms( ) - 40 * 60 * 1000 ;
1877
sqlx:: query( " UPDATE flow_jobs SET started_ms = ?" ) .bind ( started) .execute ( & h.app.db) .await.unwrap ( ) ;
1878
oxygenhub:: flow:: reap_lost_jobs( & h.app) .await.unwrap ( ) ;
1879
let status = | id : i64 | {
1880
let app = h.app.clone ( ) ;
1881
async move { db:: flow:: job( & app.db, id) .await.unwrap ( ) .unwrap ( ) .status }
1882
} ;
1883
assert_eq! ( status ( ids[ " build" ] ) .await, " error" , " past its 30 minutes" ) ;
1884
assert_eq! ( status ( ids[ " slow" ] ) .await, " running" , " within its 90" ) ;
1885
assert_eq! ( status ( ids[ " huge" ] ) .await, " running" , " within its 120" ) ;
1886
}
c889911b Andy McCall 1 month ago ↶
1887
1888
1889
1890
1891
# [ tokio ::test ( flavor = " multi_thread" ) ]
1892
async fn on_decides_which_events_start_a_job ( ) {
1893
let h = hub ( ) .await;
1894
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
1895
create ( & h.app, " andy" , " events" ) .await;
1896
let url = format! ( " {} /andy/events" , h.base) ;
1897
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1898
let c = client ( & url, work.path ( ) ) ;
1899
let text = format! (
1900
" on {{ \n push \" main\" \n manual\n }} \n {} job \" dist\" {{ \n sandbox \" alpine\" \n on {{ \n tag \" v*\" \n }} \n exec \" true\" \n }} \n " ,
1901
pipeline ( " " )
1902
) ;
1903
1904
let text = text.replacen (
1905
" on {\n push \" main\" \n manual\n }\n schema-version 1\n " ,
1906
" schema-version 1\n on {\n push \" main\" \n manual\n }\n " ,
1907
1 ,
1908
) ;
1909
write ( work.path ( ) , " .flow/pipeline.kdl" , & text) ;
1910
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
1911
c.commit ( " Start" , & [ ] ) .unwrap ( ) ;
1912
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1913
let jobs_of_latest = | app : Shared, n : usize | async move {
1914
let mut runs = Vec :: new( ) ;
1915
for _ in 0 .. 100 {
1916
runs = runs_of ( & app, " andy" , " events" ) .await;
1917
if runs.len ( ) > = n {
1918
break ;
1919
}
1920
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
1921
}
1922
assert_eq! ( runs.len ( ) , n, " runs" ) ;
1923
let mut j: Vec< String> = db:: flow:: jobs( & app.db, runs[ 0 ] .id) .await.unwrap ( ) .into_iter ( ) .map ( | j | j.name ) .collect ( ) ;
1924
j.sort ( ) ;
1925
( runs[ 0 ] .clone ( ) , j)
1926
} ;
1927
let ( _ , j) = jobs_of_latest ( h.app.clone ( ) , 1 ) .await;
1928
assert_eq! ( j, [ " build" , " lint" , " test" ] , " a push to main: no dist" ) ;
1929
1930
let heads = c.bookmark_heads ( ) .unwrap ( ) ;
1931
c.tag_create ( " v1.0.0" , & heads, " first" ) .unwrap ( ) ;
1932
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1933
let ( run, j) = jobs_of_latest ( h.app.clone ( ) , 2 ) .await;
1934
assert_eq! ( j, [ " dist" ] , " a tag runs only what names it" ) ;
1935
assert_eq! ( run.trigger, " tag:v1.0.0" ) ;
1936
1937
let row = db:: repos:: get( & h.app.db, " andy" , " events" ) .await.unwrap ( ) .unwrap ( ) ;
1938
db:: flow:: finish(
1939
& h.app.db,
1940
db:: flow:: jobs( & h.app.db, run.id) .await.unwrap ( ) [ 0 ] .id,
1941
db:: flow:: Outcome { status: " passed" , exit_code: Some ( 0 ) , error: " " , receipt: " " , receipt_hash: " " } ,
1942
)
1943
.await
1944
.unwrap ( ) ;
1945
let run = db:: flow:: run_by_id( & h.app.db, run.id) .await.unwrap ( ) .unwrap ( ) ;
1946
oxygenhub:: flow:: rerun( & h.app, & row, & run, " andy" ) .await.unwrap ( ) ;
1947
let ( _ , j) = jobs_of_latest ( h.app.clone ( ) , 3 ) .await;
1948
assert_eq! ( j, [ " dist" ] , " the rerun is the tag's run again" ) ;
1949
1950
c.bookmark_create ( " develop" , None ) .unwrap ( ) ;
1951
assert! ( c.push_bookmark ( " origin" , " develop" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1952
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 300 ) ) .await;
1953
assert_eq! ( runs_of ( & h.app, " andy" , " events" ) .await.len ( ) , 3 , " no run for develop" ) ;
348f8c4b Andy McCall 1 month ago ↶
1954
}
1955
1956
1957
1958
# [ tokio ::test ( flavor = " multi_thread" ) ]
1959
async fn scheduled_jobs_run_when_due ( ) {
1960
let h = hub ( ) .await;
1961
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
1962
create ( & h.app, " andy" , " nightly" ) .await;
1963
let url = format! ( " {} /andy/nightly" , h.base) ;
1964
let work = tempfile:: tempdir( ) .unwrap ( ) ;
1965
let c = client ( & url, work.path ( ) ) ;
1966
1967
let extra = " job \" nightly\" {\n sandbox \" alpine\" \n on {\n schedule \" 17 4 29 2 *\" \n }\n exec \" true\" \n }\n " ;
1968
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( extra) ) ;
1969
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
1970
c.commit ( " Start" , & [ ] ) .unwrap ( ) ;
1971
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
1972
assert_eq! ( runs_of ( & h.app, " andy" , " nightly" ) .await.len ( ) , 1 ) ;
1973
let first = runs_of ( & h.app, " andy" , " nightly" ) .await;
1974
let names: Vec< String> = db:: flow:: jobs( & h.app.db, first[ 0 ] .id) .await.unwrap ( ) .into_iter ( ) .map ( | j | j.name ) .collect ( ) ;
1975
assert! ( ! names.contains ( & " nightly" .to_string ( ) ) , " a push does not run a scheduled job" ) ;
1976
use flow_core:: pipeline:: When;
1977
let not_yet = When { minute: 16 , hour: 4 , day: 29 , month: 2 , weekday: 4 } ;
1978
assert_eq! ( oxygenhub:: flow:: run_due( & h.app, not_yet) .await, 0 ) ;
1979
let due = When { minute: 17 , .. not_yet } ;
1980
assert_eq! ( oxygenhub:: flow:: run_due( & h.app, due) .await, 1 ) ;
1981
let runs = runs_of ( & h.app, " andy" , " nightly" ) .await;
1982
assert_eq! ( runs.len ( ) , 2 ) ;
1983
assert_eq! ( runs[ 0 ] .trigger, " schedule" ) ;
1984
let names: Vec< String> = db:: flow:: jobs( & h.app.db, runs[ 0 ] .id) .await.unwrap ( ) .into_iter ( ) .map ( | j | j.name ) .collect ( ) ;
1985
assert_eq! ( names, [ " nightly" ] ) ;
1986
assert_eq! ( runs[ 0 ] .bookmark, " main" , " on the default bookmark" ) ;
c889911b Andy McCall 1 month ago ↶
1987
}
2a83f530 Andy McCall 1 month ago ↶
1988
1989
# [ tokio ::test ( flavor = " multi_thread" ) ]
1990
async fn variables_from_three_levels_reach_a_job_inner_overriding_outer ( ) {
1991
use oxygenhub:: levels:: Scope;
1992
use oxygenhub:: variables;
1993
let h = hub ( ) .await;
1994
let app = & h.app;
1995
1996
let res = oxygenhub:: router( app.clone ( ) )
1997
.oneshot (
1998
Request:: post( " /register" )
1999
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
2000
.body ( Body:: from( " username=andy&email=a%40example.com&password=correct-horse-battery" ) )
2001
.unwrap ( ) ,
2002
)
2003
.await
2004
.unwrap ( ) ;
2005
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
2006
let andy = db:: accounts:: account_by_name( & app.db, " andy" ) .await.unwrap ( ) .unwrap ( ) ;
2007
db:: accounts:: create_org( & app.db, " acme" , " Acme" , andy.id) .await.unwrap ( ) ;
2008
create ( app, " acme" , " app" ) .await;
2009
let page = | path : & str , form : Option< String> | {
2010
let mut req =
2011
Request:: builder( ) .method ( if form.is_some ( ) { " POST" } else { " GET" } ) .uri ( path) .header ( header:: COOKIE , cookie.clone ( ) ) ;
2012
if form.is_some ( ) {
2013
req = req.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) ;
2014
}
2015
let req = req.body ( Body:: from( form.unwrap_or_default ( ) ) ) .unwrap ( ) ;
2016
let app = app.clone ( ) ;
2017
async move {
2018
let res = oxygenhub:: router( app) .oneshot ( req) .await.unwrap ( ) ;
2019
let status = res.status ( ) ;
2020
let body = String :: from_utf8( res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) .to_vec ( ) ) .unwrap ( ) ;
2021
( status, body)
2022
}
2023
} ;
2024
let csrf_of = | page : & str | {
2025
let i = page.find ( " name=\" _csrf\" value=\" " ) .expect ( " csrf field" ) + " name=\" _csrf\" value=\" " .len ( ) ;
2026
page[ i.. ] .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( )
2027
} ;
2028
2029
2030
2031
variables:: set( & app.db, Scope:: Site, " CC" , " gcc" ) .await.unwrap ( ) ;
2032
variables:: set( & app.db, Scope:: Site, " LANG" , " C" ) .await.unwrap ( ) ;
2033
let org = db:: accounts:: org_by_name( & app.db, " acme" ) .await.unwrap ( ) .unwrap ( ) ;
2034
variables:: set( & app.db, Scope:: Org( org.id) , " CC" , " clang" ) .await.unwrap ( ) ;
2035
let ( st, settings) = page ( " /acme/app/settings" , None ) .await;
2036
assert_eq! ( st, StatusCode:: OK , " {settings}" ) ;
2037
assert! ( settings.contains ( " Inherited" ) & & settings.contains ( " the acme organisation" ) & & settings.contains ( " overridden" ) , " {settings}" ) ;
2038
let csrf = csrf_of ( & settings) ;
2039
let ( _ , body) = page ( " /acme/app/settings/variables" , Some ( format! ( " _csrf={csrf} &action=set&name=RUST_LOG&value=debug" ) ) ) .await;
2040
assert! ( body.contains ( " RUST_LOG set." ) & & body.contains ( " <code>debug</code>" ) , " {body}" ) ;
2041
let ( _ , body) = page ( " /acme/app/settings/variables" , Some ( format! ( " _csrf={csrf} &action=set&name=FLOW_REPO&value=x" ) ) ) .await;
2042
assert! ( body.contains ( " not FLOW_" ) , " {body}" ) ;
2043
let repo = db:: repos:: get( & app.db, " acme" , " app" ) .await.unwrap ( ) .unwrap ( ) ;
2044
db:: repos:: set_flow( & app.db, repo.id, " on" ) .await.unwrap ( ) ;
2045
let env = variables:: env_for_repo( & app.db, & repo) .await.unwrap ( ) ;
2046
assert_eq! (
2047
env.iter ( ) .map ( | ( k , v ) | ( k.as_str ( ) , v.as_str ( ) ) ) .collect:: < Vec< _ > > ( ) ,
2048
[ ( " CC" , " clang" ) , ( " LANG" , " C" ) , ( " RUST_LOG" , " debug" ) ] ,
2049
" the organisation's CC wins over the site's"
2050
) ;
2051
2052
2053
let url = format! ( " {} /acme/app" , h.base) ;
2054
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2055
let c = client ( & url, work.path ( ) ) ;
2056
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
2057
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2058
c.snapshot ( " test" ) .unwrap ( ) ;
2059
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
2060
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2061
let runs = runs_of ( app, " acme" , " app" ) .await;
2062
assert_eq! ( runs.len ( ) , 1 , " {runs:?}" ) ;
2063
let got = claim ( app) .await.unwrap_or_else ( | | panic! ( " a job: {:?}" , runs[ 0 ] ) ) ;
2064
assert_eq! ( got[ " env" ] , serde_json:: json! ( { " CC" : " clang" , " LANG" : " C" , " RUST_LOG" : " debug" } ) , " {got}" ) ;
2065
2066
2067
let ( _ , admin) = page ( " /admin" , None ) .await;
2068
assert! ( admin.contains ( " Variables" ) & & admin.contains ( " <code>gcc</code>" ) , " {admin}" ) ;
2069
let ( _ , body) = page ( " /admin/variables" , Some ( format! ( " _csrf={} &action=delete&name=LANG" , csrf_of ( & admin) ) ) ) .await;
2070
assert! ( body.contains ( " LANG removed." ) , " {body}" ) ;
2071
let got = claim ( app) .await.expect ( " another job" ) ;
2072
assert_eq! ( got[ " env" ] , serde_json:: json! ( { " CC" : " clang" , " RUST_LOG" : " debug" } ) , " {got}" ) ;
2073
}
4ac01a73 Andy McCall 1 month ago ↶
2074
2075
# [ tokio ::test ( flavor = " multi_thread" ) ]
2076
async fn secrets_a_job_declares_come_with_the_claim_and_never_reach_the_log ( ) {
2077
let h = hub ( ) .await;
2078
let app = & h.app;
2079
let res = oxygenhub:: router( app.clone ( ) )
2080
.oneshot (
2081
Request:: post( " /register" )
2082
.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" )
2083
.body ( Body:: from( " username=andy&email=a%40example.com&password=correct-horse-battery" ) )
2084
.unwrap ( ) ,
2085
)
2086
.await
2087
.unwrap ( ) ;
2088
let cookie = res.headers ( ) .get ( header:: SET_COOKIE ) .unwrap ( ) .to_str ( ) .unwrap ( ) .split ( ' ;' ) .next ( ) .unwrap ( ) .to_string ( ) ;
2089
let page = | path : & str , form : Option< String> | {
2090
let mut req =
2091
Request:: builder( ) .method ( if form.is_some ( ) { " POST" } else { " GET" } ) .uri ( path) .header ( header:: COOKIE , cookie.clone ( ) ) ;
2092
if form.is_some ( ) {
2093
req = req.header ( header:: CONTENT_TYPE , " application/x-www-form-urlencoded" ) ;
2094
}
2095
let req = req.body ( Body:: from( form.unwrap_or_default ( ) ) ) .unwrap ( ) ;
2096
let app = app.clone ( ) ;
2097
async move {
2098
let res = oxygenhub:: router( app) .oneshot ( req) .await.unwrap ( ) ;
2099
let status = res.status ( ) ;
2100
let body = String :: from_utf8( res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) .to_vec ( ) ) .unwrap ( ) ;
2101
( status, body)
2102
}
2103
} ;
2104
let csrf_of = | page : & str | {
2105
let i = page.find ( " name=\" _csrf\" value=\" " ) .expect ( " csrf field" ) + " name=\" _csrf\" value=\" " .len ( ) ;
2106
page[ i.. ] .split ( ' "' ) .next ( ) .unwrap ( ) .to_string ( )
2107
} ;
2108
create ( app, " andy" , " demo" ) .await;
2109
2110
2111
let ( _ , admin) = page ( " /admin" , None ) .await;
2112
let csrf = csrf_of ( & admin) ;
2113
let ( _ , body) = page ( " /admin/secrets" , Some ( format! ( " _csrf={csrf} &action=set&name=deploy-token&value=site-level-value" ) ) ) .await;
2114
assert! ( body.contains ( " deploy-token set." ) & & ! body.contains ( " site-level-value" ) , " {body}" ) ;
2115
let ( _ , body) = page ( " /admin/secrets" , Some ( format! ( " _csrf={csrf} &action=set&name=bad+name&value=x" ) ) ) .await;
2116
assert! ( body.contains ( " name is 1 to 64 of" ) , " {body}" ) ;
2117
let ( _ , settings) = page ( " /andy/demo/settings" , None ) .await;
2118
assert! (
2119
settings.contains ( " Secrets" ) & & settings.contains ( " <code>deploy-token</code>" ) & & settings.contains ( " the site</td>" ) ,
2120
" {settings}"
2121
) ;
2122
let csrf_r = csrf_of ( & settings) ;
2123
let ( _ , body) =
2124
page ( " /andy/demo/settings/secrets" , Some ( format! ( " _csrf={csrf_r} &action=set&name=deploy-token&value=hunter22-repo-value" ) ) ) .await;
2125
assert! ( body.contains ( " deploy-token set." ) & & body.contains ( " overridden" ) & & ! body.contains ( " hunter22" ) , " {body}" ) ;
2126
let repo = db:: repos:: get( & app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
2127
assert! (
2128
! String :: from_utf8_lossy( & oxygenhub:: secrets:: list( & app.db, oxygenhub:: levels:: Scope:: Repo( repo.id) ) .await.unwrap ( ) [ 0 ] .value)
2129
.contains ( " hunter22" ) ,
2130
" sealed at rest"
2131
) ;
2132
2133
2134
let url = format! ( " {} /andy/demo" , h.base) ;
2135
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2136
let c = client ( & url, work.path ( ) ) ;
2137
write (
2138
work.path ( ) ,
2139
" .flow/pipeline.kdl" ,
2140
& 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 " ) ,
2141
) ;
2142
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2143
c.snapshot ( " test" ) .unwrap ( ) ;
2144
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
2145
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2146
let runs = runs_of ( app, " andy" , " demo" ) .await;
2147
let jobs = match runs.first ( ) {
2148
Some ( r) => db:: flow:: jobs( & app.db, r.id) .await.unwrap ( ) .iter ( ) .map ( | j | format! ( " {} :{} " , j.name, j.status) ) .collect:: < Vec< _ > > ( ) ,
2149
None => vec! [ ] ,
2150
} ;
2151
let mut deploy = None ;
2152
for _ in 0 .. 50 {
2153
let Some ( got) = claim ( app) .await else {
2154
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 100 ) ) .await;
2155
continue ;
2156
} ;
2157
if got[ " job" ] [ " name" ] = = " deploy" {
2158
deploy = Some ( got) ;
2159
break ;
2160
}
2161
assert_eq! ( got[ " secrets" ] , serde_json:: Value:: Null, " {got}" ) ;
2162
}
2163
let got = deploy.unwrap_or_else ( | | panic! ( " the deploy job: runs {runs:?} jobs {jobs:?}" ) ) ;
2164
assert_eq! (
2165
got[ " secrets" ] ,
2166
serde_json:: json! ( [ { " name" : " deploy-token" , " path" : " /run/secrets/token" , " value" : " hunter22-repo-value" } ] ) ,
2167
" the repository's value, the missing one left out: {got}"
2168
) ;
2169
let id = got[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
2170
let log = db:: flow:: log( & app.db, id, 0 , 100 ) .await.unwrap ( ) ;
2171
assert! ( log.iter ( ) .any ( | l | l.stream = = " flow" & & l.text.contains ( " secret \" missing\" is not set" ) ) , " {log:?}" ) ;
2172
2173
2174
let records = serde_json:: json! ( { " records" : [
2175
{ " t" : 1 , " stream" : " out" , " text" : " token=hunter22-repo-value" } ,
2176
{ " t" : 2 , " stream" : " err" , " text" : " b64 aHVudGVyMjItcmVwby12YWx1ZQ== end" } ,
2177
] } ) ;
2178
let ( st, _ ) = call ( app, " POST" , & format! ( " /api/v1/flow/jobs/{id} /log" ) , Some ( " runner-secret" ) , Some ( records.to_string ( ) ) ) .await;
2179
assert_eq! ( st, StatusCode:: NO_CONTENT ) ;
2180
let log = db:: flow:: log( & app.db, id, 0 , 100 ) .await.unwrap ( ) ;
2181
let texts: Vec< & str > = log.iter ( ) .map ( | l | l.text.as_str ( ) ) .collect ( ) ;
2182
assert! ( texts.contains ( & " token=***" ) & & texts.contains ( & " b64 *** end" ) , " {texts:?}" ) ;
2183
assert! ( ! texts.iter ( ) .any ( | t | t.contains ( " hunter22" ) ) , " {texts:?}" ) ;
2184
}
d1727c57 Andy McCall 1 month ago ↶
2185
2186
# [ tokio ::test ( flavor = " multi_thread" ) ]
2187
async fn a_job_s_identity_is_a_signed_token_the_discovery_document_lets_anyone_verify ( ) {
2188
let h = hub ( ) .await;
2189
let app = & h.app;
2190
let ( st, disc) = call ( app, " GET" , " /.well-known/openid-configuration" , None , None ) .await;
2191
assert_eq! ( st, StatusCode:: OK , " {disc}" ) ;
2192
let disc: serde_json:: Value = serde_json:: from_str( & disc) .unwrap ( ) ;
2193
assert_eq! ( disc[ " issuer" ] , h.base.as_str ( ) ) ;
2194
assert_eq! ( disc[ " jwks_uri" ] , format! ( " {} /.well-known/jwks.json" , h.base) ) ;
2195
let ( st, jwks) = call ( app, " GET" , " /.well-known/jwks.json" , None , None ) .await;
2196
assert_eq! ( st, StatusCode:: OK , " {jwks}" ) ;
2197
let jwks: serde_json:: Value = serde_json:: from_str( & jwks) .unwrap ( ) ;
2198
assert_eq! ( jwks[ " keys" ] [ 0 ] [ " alg" ] , " RS256" ) ;
2199
let kid = jwks[ " keys" ] [ 0 ] [ " kid" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
2200
2201
create ( app, " andy" , " demo" ) .await;
2202
let url = format! ( " {} /andy/demo" , h.base) ;
2203
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2204
let c = client ( & url, work.path ( ) ) ;
2205
write (
2206
work.path ( ) ,
2207
" .flow/pipeline.kdl" ,
2208
& pipeline ( " job \" deploy\" {\n sandbox \" alpine\" \n identity \" sts.amazonaws.com\" {\n mount-path \" /run/secrets/aws-token\" \n }\n exec \" deploy\" \n }\n " ) ,
2209
) ;
2210
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2211
c.snapshot ( " test" ) .unwrap ( ) ;
2212
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
2213
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2214
let mut deploy = None ;
2215
for _ in 0 .. 50 {
2216
let Some ( got) = claim ( app) .await else {
2217
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 100 ) ) .await;
2218
continue ;
2219
} ;
2220
if got[ " job" ] [ " name" ] = = " deploy" {
2221
deploy = Some ( got) ;
2222
break ;
2223
}
2224
}
2225
let got = deploy.expect ( " the deploy job" ) ;
2226
let secrets = got[ " secrets" ] .as_array ( ) .expect ( " secrets" ) ;
2227
assert_eq! ( secrets.len ( ) , 1 ) ;
2228
assert_eq! (
2229
( secrets[ 0 ] [ " name" ] .as_str ( ) , secrets[ 0 ] [ " path" ] .as_str ( ) ) ,
2230
( Some ( " identity:sts.amazonaws.com" ) , Some ( " /run/secrets/aws-token" ) )
2231
) ;
2232
let token = secrets[ 0 ] [ " value" ] .as_str ( ) .unwrap ( ) .to_string ( ) ;
2233
let issuer = app.oidc ( ) .await.unwrap ( ) ;
2234
assert_eq! ( issuer.kid, kid, " the key the JWKS publishes signed it" ) ;
2235
let claims = issuer.verify ( & token, " sts.amazonaws.com" , db:: now_ms( ) ) .expect ( " verifies" ) ;
2236
assert_eq! ( claims[ " sub" ] , " repo:andy/demo:bookmark:main" ) ;
2237
assert_eq! ( claims[ " repository" ] , " andy/demo" ) ;
2238
assert_eq! ( claims[ " job" ] , " deploy" ) ;
2239
assert_eq! ( claims[ " run_number" ] , 1 ) ;
2240
assert_eq! ( claims[ " runner" ] , " test-1" ) ;
2241
assert_eq! ( claims[ " iss" ] , h.base.as_str ( ) ) ;
2242
assert! ( claims[ " state" ] .as_str ( ) .unwrap ( ) .len ( ) = = 64 ) ;
2243
2244
let id = got[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
2245
let records = serde_json:: json! ( { " records" : [ { " t" : 1 , " stream" : " out" , " text" : format! ( " TOKEN={token} end" ) } ] } ) ;
2246
let ( st, _ ) = call ( app, " POST" , & format! ( " /api/v1/flow/jobs/{id} /log" ) , Some ( " runner-secret" ) , Some ( records.to_string ( ) ) ) .await;
2247
assert_eq! ( st, StatusCode:: NO_CONTENT ) ;
2248
let log = db:: flow:: log( & app.db, id, 0 , 100 ) .await.unwrap ( ) ;
2249
assert! ( log.iter ( ) .any ( | l | l.text = = " TOKEN=*** end" ) , " {log:?}" ) ;
2250
}
efd29a00 Andy McCall 1 month ago ↶
2251
2252
# [ tokio ::test ( flavor = " multi_thread" ) ]
2253
async fn jobs_are_offered_only_to_runners_of_their_architecture ( ) {
2254
let h = hub ( ) .await;
2255
let app = & h.app;
2256
create ( app, " andy" , " demo" ) .await;
2257
let url = format! ( " {} /andy/demo" , h.base) ;
2258
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2259
let c = client ( & url, work.path ( ) ) ;
2260
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 ) ) ) ;
2261
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2262
c.snapshot ( " test" ) .unwrap ( ) ;
2263
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
2264
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2265
let claim_as = | arch : & 'static str , archs : serde_json:: Value| {
2266
let app = app.clone ( ) ;
2267
async move {
2268
let body = serde_json:: json! ( {
2269
" runner" : format! ( " {arch} -1" ) , " runtimes" : [ " microvm" ] ,
2270
" info" : { " arch" : arch, " os" : " linux" , " version" : " t" , " labels" : [ ] , " max_jobs" : 1 } ,
2271
" image_archs" : archs,
2272
} ) ;
2273
let ( st, body) = call ( & app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( body.to_string ( ) ) ) .await;
2274
match st {
2275
StatusCode:: NO_CONTENT => None ,
2276
StatusCode:: OK => {
2277
Some ( serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " job" ] [ " name" ] .as_str ( ) .unwrap ( ) .to_string ( ) )
2278
}
2279
other => panic! ( " claim: {other} {body}" ) ,
2280
}
2281
}
2282
} ;
2283
2284
2285
let rootfs = pipeline ( " " ) .split ( " cas://blob/" ) .nth ( 1 ) .unwrap ( ) [ .. 64 ] .to_string ( ) ;
2286
for _ in 0 .. 50 {
2287
if ! runs_of ( app, " andy" , " demo" ) .await.is_empty ( ) {
2288
break ;
2289
}
2290
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 100 ) ) .await;
2291
}
2292
let runs = runs_of ( app, " andy" , " demo" ) .await;
2293
assert_eq! ( runs.len ( ) , 1 , " {runs:?}" ) ;
2294
let mut taken = vec! [ ] ;
2295
while let Some ( j) = claim_as ( " x86_64" , serde_json:: json! ( { rootfs.clone ( ) : " x86_64" } ) ) .await {
2296
taken.push ( j) ;
2297
}
2298
assert_eq! ( taken, [ " build" , " lint" ] , " test waits for build; flash is for riscv64" ) ;
2299
2300
assert_eq! ( claim_as ( " riscv64" , serde_json:: json! ( { } ) ) .await.as_deref ( ) , Some ( " flash" ) ) ;
2301
assert_eq! ( claim_as ( " riscv64" , serde_json:: json! ( { } ) ) .await, None ) ;
2302
let ( _ , queue) = call ( app, " GET" , " /admin/runners" , None , None ) .await;
2303
let _ = queue;
2304
}
4d2e44d7 Andy McCall 1 month ago ↶
2305
2306
2307
2308
2309
2310
# [ tokio ::test ( flavor = " multi_thread" ) ]
2311
async fn a_declined_job_goes_to_another_runner_or_ends_unplaceable ( ) {
2312
let h = hub ( ) .await;
2313
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
2314
create ( & h.app, " andy" , " declines" ) .await;
2315
let url = format! ( " {} /andy/declines" , h.base) ;
2316
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2317
let c = client ( & url, work.path ( ) ) ;
2318
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
2319
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2320
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
2321
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2322
let run = runs_of ( & h.app, " andy" , " declines" ) .await[ 0 ] .clone ( ) ;
2323
let claim_as = | token : & 'static str , name : & 'static str | {
2324
let app = h.app.clone ( ) ;
2325
async move {
2326
let body = serde_json:: json! ( { " runner" : name, " runtimes" : [ " microvm" ] } ) .to_string ( ) ;
2327
let ( st, body) = call ( & app, " POST" , " /api/v1/flow/claim" , Some ( token) , Some ( body) ) .await;
2328
( st = = StatusCode:: OK ) .then ( | | serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " job_id" ] .as_i64 ( ) .unwrap ( ) )
2329
}
2330
} ;
2331
let decline = | token : & 'static str , id : i64 , reason : & 'static str | {
2332
let app = h.app.clone ( ) ;
2333
async move {
2334
let body = serde_json:: json! ( { " reason" : reason } ) .to_string ( ) ;
2335
call ( & app, " POST" , & format! ( " /api/v1/flow/jobs/{id} /decline" ) , Some ( token) , Some ( body) ) .await.0
2336
}
2337
} ;
2338
let job = claim_as ( " runner-secret" , " earth-1" ) .await.expect ( " a job" ) ;
2339
2340
assert_eq! ( decline ( " runner-two" , job, " not mine" ) .await, StatusCode:: CONFLICT , " only the runner it runs on may decline it" ) ;
2341
assert_eq! ( decline ( " runner-secret" , job, " the MicroVM tier needs /dev/kvm" ) .await, StatusCode:: NO_CONTENT ) ;
2342
let j = db:: flow:: job( & h.app.db, job) .await.unwrap ( ) .unwrap ( ) ;
2343
assert_eq! ( ( j.status.as_str ( ) , j.runner.as_str ( ) ) , ( " queued" , " " ) ) ;
2344
assert_eq! ( j.waiting, " declined by earth-1: the MicroVM tier needs /dev/kvm; waiting for another runner" ) ;
2345
2346
let next = claim_as ( " runner-secret" , " earth-1" ) .await.expect ( " the other ready job" ) ;
2347
assert_ne! ( next, job) ;
2348
let page = call ( & h.app, " GET" , & format! ( " /andy/declines/flow/{} " , run.number) , None , None ) .await.1 ;
2349
assert! ( page.contains ( " Waiting: declined by earth-1: the MicroVM tier needs /dev/kvm" ) , " {page}" ) ;
2350
assert_eq! ( claim_as ( " runner-two" , " plex-1" ) .await, Some ( job) , " another runner takes it" ) ;
2351
assert_eq! ( decline ( " runner-two" , job, " rootfs 36fc4e323804 is not on this runner" ) .await, StatusCode:: NO_CONTENT ) ;
2352
assert_eq! ( claim_as ( " runner-three" , " narvi" ) .await, Some ( job) ) ;
2353
assert_eq! ( decline ( " runner-three" , job, " rootfs 36fc4e323804 is for x86_64; this runner is riscv64" ) .await, StatusCode:: NO_CONTENT ) ;
2354
let j = db:: flow:: job( & h.app.db, job) .await.unwrap ( ) .unwrap ( ) ;
2355
assert_eq! ( j.status, " error" ) ;
2356
assert_eq! (
2357
j.error,
2358
" 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"
2359
) ;
2360
let r = db:: runners:: list( & h.app.db) .await.unwrap ( ) .into_iter ( ) .find ( | r | r.name = = " plex-1" ) .unwrap ( ) ;
2361
let declined = db:: flow:: declines_of_runner( & h.app.db, r.id, 10 ) .await.unwrap ( ) ;
2362
assert_eq! ( declined.len ( ) , 1 ) ;
2363
assert_eq! ( ( declined[ 0 ] .job_id, declined[ 0 ] .reason.as_str ( ) ) , ( job, " rootfs 36fc4e323804 is not on this runner" ) ) ;
2364
assert_eq! ( r.tier_list ( ) , vec! [ " microvm" ] ) ;
2365
}
554ccacf Andy McCall 1 month ago ↶
2366
2367
2368
2369
2370
# [ tokio ::test ( flavor = " multi_thread" ) ]
2371
async fn a_coverage_artifact_becomes_a_coverage_report ( ) {
2372
let h = hub ( ) .await;
2373
create ( & h.app, " andy" , " cov" ) .await;
2374
let url = format! ( " {} /andy/cov" , h.base) ;
2375
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2376
let c = client ( & url, work.path ( ) ) ;
2377
write (
2378
work.path ( ) ,
2379
" .flow/pipeline.kdl" ,
2380
& format! (
2381
" 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 "
2382
) ,
2383
) ;
2384
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 " ) ;
2385
c.snapshot ( " test" ) .unwrap ( ) ;
2386
c.commit ( " Code and coverage" , & [ ] ) .unwrap ( ) ;
2387
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2388
let run = runs_of ( & h.app, " andy" , " cov" ) .await.remove ( 0 ) ;
c8df33a6 Andy McCall 26 days ago ↶
2389
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 ↶
2390
let id = claim ( & h.app) .await.unwrap ( ) [ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
2391
2392
let ( st, text) = upload ( & h.app, id, " coverage" , b " this is not a report" ) .await;
2393
assert_eq! ( st, StatusCode:: OK , " a bad report never fails the upload: {text}" ) ;
2394
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 \
2395
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 " ;
2396
let ( st, text) = upload ( & h.app, id, " coverage" , lcov.as_bytes ( ) ) .await;
2397
assert_eq! ( st, StatusCode:: OK , " {text}" ) ;
2398
let log: Vec< String> = db:: flow:: log( & h.app.db, id, 0 , 50 ) .await.unwrap ( ) .into_iter ( ) .map ( | l | l.text ) .collect ( ) ;
2399
assert! ( log.iter ( ) .any ( | l | l.contains ( " coverage: this is not an LCOV report" ) ) , " {log:?}" ) ;
2400
assert! (
2401
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)" ) ,
2402
" {log:?}"
2403
) ;
2404
2405
let ( st, page) = call ( & h.app, " GET" , " /andy/cov/coverage" , None , None ) .await;
2406
assert_eq! ( st, StatusCode:: OK ) ;
c987c865 Andy McCall 1 month ago ↶
2407
assert! (
c8df33a6 Andy McCall 26 days ago ↶
2408
page.contains ( " >Reports</a>" )
c987c865 Andy McCall 1 month ago ↶
2409
& & page.contains ( " 50.0%" )
2410
& & page.contains ( " <strong>2</strong> of <strong>4</strong> relevant lines covered" ) ,
2411
" {page}"
2412
) ;
554ccacf Andy McCall 1 month ago ↶
2413
assert! ( page.contains ( " 1 of 2" ) & & page.contains ( " /andy/cov/coverage/" ) & & page.contains ( " /file/src/lib.rs" ) , " {page}" ) ;
2414
let report = crate_report ( & h.app, run.id) .await;
2415
let ( st, file) = call ( & h.app, " GET" , & format! ( " /andy/cov/coverage/{report} /file/src/lib.rs" ) , None , None ) .await;
2416
assert_eq! ( st, StatusCode:: OK , " {file}" ) ;
2417
assert_eq! ( file.matches ( " class=\" cov-hit\" " ) .count ( ) , 2 , " {file}" ) ;
2418
assert_eq! ( file.matches ( " class=\" cov-miss\" " ) .count ( ) , 2 , " {file}" ) ;
2419
assert! ( file.contains ( " 3×" ) , " {file}" ) ;
2420
assert_eq! ( call ( & h.app, " GET" , & format! ( " /andy/cov/coverage/{report} /file/src/gone.rs" ) , None , None ) .await.0 , StatusCode:: NOT_FOUND ) ;
2421
let ( _ , badge) = call ( & h.app, " GET" , " /andy/cov/coverage/badge.svg" , None , None ) .await;
2422
assert! ( badge.contains ( " coverage: 50.0%" ) , " {badge}" ) ;
2423
let ( _ , run_page) = call ( & h.app, " GET" , & format! ( " /andy/cov/flow/{} " , run.number) , None , None ) .await;
2424
assert! ( run_page.contains ( " coverage 50.0%" ) , " {run_page}" ) ;
617c2840 Andy McCall 1 month ago ↶
2425
2426
let ( _ , list) = call ( & h.app, " GET" , " /api/v1/repos/andy/cov/coverage" , None , None ) .await;
2427
let list: serde_json:: Value = serde_json:: from_str( & list) .unwrap ( ) ;
2428
assert_eq! ( ( list[ 0 ] [ " percent" ] .as_f64 ( ) , list[ 0 ] [ " format" ] .as_str ( ) ) , ( Some ( 50. 0 ) , Some ( " lcov" ) ) , " {list}" ) ;
2429
let ( _ , one) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/cov/coverage/{report} " ) , None , None ) .await;
2430
let one: serde_json:: Value = serde_json:: from_str( & one) .unwrap ( ) ;
2431
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 ↶
2432
2433
2434
2435
let row = db:: repos:: get( & h.app.db, " andy" , " cov" ) .await.unwrap ( ) .unwrap ( ) ;
2436
let check = | checks : Vec< db:: hooks:: Check> | checks.into_iter ( ) .find ( | c | c.name = = " coverage" ) .expect ( " a coverage check" ) ;
2437
let c = check ( db:: hooks:: checks_for_state( & h.app.db, row.id, & run.state) .await.unwrap ( ) ) ;
2438
assert_eq! ( ( c.status.as_str ( ) , c.summary.as_str ( ) ) , ( " passed" , " 50.0%" ) ) ;
2439
assert! ( c.url.ends_with ( & format! ( " /andy/cov/coverage?report={report} " ) ) , " {}" , c.url) ;
2440
oxygenhub:: coverage:: set_rules( & h.app.db, row.id, oxygenhub:: coverage:: Rules { min: Some ( 60. 0 ) , max_drop: None } ) .await.unwrap ( ) ;
2441
assert_eq! ( upload ( & h.app, id, " coverage" , lcov.as_bytes ( ) ) .await.0 , StatusCode:: OK ) ;
2442
let c = check ( db:: hooks:: checks_for_state( & h.app.db, row.id, & run.state) .await.unwrap ( ) ) ;
2443
assert_eq! ( ( c.status.as_str ( ) , c.summary.as_str ( ) ) , ( " failed" , " 50.0%: below the minimum of 60%" ) ) ;
554ccacf Andy McCall 1 month ago ↶
2444
}
2445
2446
async fn crate_report ( app : & Shared, run_id : i64 ) -> i64 {
2447
oxygenhub:: coverage:: for_run( & app.db, run_id) .await.unwrap ( ) [ 0 ] .id
2448
}
36e5a9c1 Andy McCall 1 month ago ↶
2449
2450
2451
2452
# [ tokio ::test ( flavor = " multi_thread" ) ]
2453
async fn junit_reports_make_a_tests_tab ( ) {
2454
let h = hub ( ) .await;
2455
create ( & h.app, " andy" , " t" ) .await;
2456
let url = format! ( " {} /andy/t" , h.base) ;
2457
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2458
let c = client ( & url, work.path ( ) ) ;
2459
write (
2460
work.path ( ) ,
2461
" .flow/pipeline.kdl" ,
2462
& format! (
2463
" 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 "
2464
) ,
2465
) ;
c8df33a6 Andy McCall 26 days ago ↶
2466
assert! ( ! call ( & h.app, " GET" , " /andy/t" , None , None ) .await.1. contains ( " >Reports</a>" ) ) ;
36e5a9c1 Andy McCall 1 month ago ↶
2467
let case = | name : & str , ok : bool , secs : & str | {
2468
if ok {
2469
format! ( r # "<testcase classname="t.suite" name="{name} " time="{secs} "/>"# )
2470
} else {
2471
format! (
2472
r # "<testcase classname="t.suite" name="{name} " time="{secs} "><failure message="{name} broke">AssertionError</failure></testcase>"#
2473
)
2474
}
2475
} ;
2476
let runs = [
2477
vec! [ case ( " a" , true , " 0.1" ) , case ( " b" , true , " 2.5" ) , case ( " c" , true , " 0.2" ) ] ,
2478
vec! [ case ( " a" , true , " 0.1" ) , case ( " b" , false , " 2.6" ) , case ( " c" , true , " 0.2" ) , case ( " d" , true , " 0.3" ) ] ,
2479
vec! [ case ( " a" , true , " 0.1" ) , case ( " b" , true , " 2.4" ) , case ( " c" , true , " 0.2" ) , case ( " d" , true , " 0.3" ) ] ,
2480
] ;
2481
for ( i, cases) in runs.iter ( ) .enumerate ( ) {
2482
write ( work.path ( ) , " n.txt" , & i.to_string ( ) ) ;
2483
c.snapshot ( " test" ) .unwrap ( ) ;
2484
c.commit ( & format! ( " Run {i} " ) , & [ ] ) .unwrap ( ) ;
2485
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2486
2487
let mut a = None ;
2488
for _ in 0 .. 100 {
2489
a = claim ( & h.app) .await;
2490
if a.is_some ( ) {
2491
break ;
2492
}
2493
tokio:: time:: sleep( std:: time:: Duration:: from_millis( 50 ) ) .await;
2494
}
2495
let a = a.expect ( " a job to claim" ) ;
2496
let id = a[ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
2497
let xml = format! ( " <testsuites><testsuite name=\" t\" >{} </testsuite></testsuites>" , cases.concat ( ) ) ;
2498
assert_eq! ( upload ( & h.app, id, " junit" , xml.as_bytes ( ) ) .await.0 , StatusCode:: OK ) ;
2499
let failed = cases.iter ( ) .any ( | c | c.contains ( " <failure" ) ) ;
2500
let mut r = receipt ( " test" , a[ " pipeline_hash" ] .as_str ( ) .unwrap ( ) , failed as u32 ) ;
2501
if ! failed {
2502
r[ " outputs" ] = serde_json:: json! ( [ oxy_core:: repo:: Repo:: blob_ref( xml.as_bytes ( ) ) .to_hex ( ) ] ) ;
2503
}
2504
let code = failed as i32 ;
2505
assert_eq! (
2506
finish ( & h.app, id, serde_json:: json! ( { " exit_code" : code, " duration_ms" : 700 , " receipt" : r } ) ) .await,
2507
StatusCode:: NO_CONTENT
2508
) ;
2509
}
2510
let ( st, page) = call ( & h.app, " GET" , " /andy/t/tests" , None , None ) .await;
2511
assert_eq! ( st, StatusCode:: OK , " {page}" ) ;
2512
if let Some ( out) = std:: env:: var_os( " OXYHUB_DUMP_TESTS_PAGE" ) {
2513
std:: fs:: write( out, & page) .unwrap ( ) ;
2514
}
c8df33a6 Andy McCall 26 days ago ↶
2515
assert! ( page.contains ( " >Reports</a>" ) , " the tab" ) ;
36e5a9c1 Andy McCall 1 month ago ↶
2516
assert! ( page.contains ( " 100%" ) & & page.contains ( " of 4 tests passed" ) , " the latest run passes" ) ;
2517
assert! ( page.contains ( " 1 fixed" ) & & page.contains ( " t.suite::b" ) , " b fixed against the run before" ) ;
2518
assert! ( page.contains ( " 1 ✕" ) & & page.contains ( " 2 ✓" ) , " b is flaky: {page}" ) ;
2519
assert! ( page.contains ( " <svg class=\" cov-chart\" " ) , " charts" ) ;
2520
2521
let slow = page.find ( " Slowest tests" ) .unwrap ( ) ;
2522
assert! ( page[ slow.. ] .find ( " t.suite::b" ) .unwrap ( ) < page[ slow.. ] .find ( " t.suite::d" ) .unwrap ( ) ) ;
2523
2524
let row = db:: repos:: get( & h.app.db, " andy" , " t" ) .await.unwrap ( ) .unwrap ( ) ;
2525
let second = oxygenhub:: junit:: on_bookmark( & h.app.db, row.id, " main" , 10 ) .await.unwrap ( ) [ 1 ] .id;
2526
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/t/tests?report={second} " ) , None , None ) .await;
2527
assert! ( page.contains ( " 1 failing" ) & & page.contains ( " Newly failing" ) & & page.contains ( " b broke" ) , " {page}" ) ;
2528
assert! ( page.contains ( " 1 new tests" ) , " d is new" ) ;
617c2840 Andy McCall 1 month ago ↶
2529
2530
2531
2532
let ( st, list) = call ( & h.app, " GET" , " /api/v1/repos/andy/t/tests" , None , None ) .await;
2533
assert_eq! ( st, StatusCode:: OK , " {list}" ) ;
2534
let list: serde_json:: Value = serde_json:: from_str( & list) .unwrap ( ) ;
2535
assert_eq! ( list.as_array ( ) .unwrap ( ) .len ( ) , 3 ) ;
2536
assert_eq! ( ( list[ 0 ] [ " tests" ] .as_i64 ( ) , list[ 1 ] [ " failed" ] .as_i64 ( ) ) , ( Some ( 4 ) , Some ( 1 ) ) ) ;
2537
let ( _ , one) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/t/tests/{second} ?status=failed" ) , None , None ) .await;
2538
let one: serde_json:: Value = serde_json:: from_str( & one) .unwrap ( ) ;
2539
assert_eq! ( one[ " cases" ] .as_array ( ) .unwrap ( ) .len ( ) , 1 ) ;
2540
assert_eq! ( ( one[ " cases" ] [ 0 ] [ " name" ] .as_str ( ) , one[ " cases" ] [ 0 ] [ " message" ] .as_str ( ) ) , ( Some ( " b" ) , Some ( " b broke" ) ) ) ;
2541
let ( _ , all) = call ( & h.app, " GET" , & format! ( " /api/v1/repos/andy/t/tests/{second} " ) , None , None ) .await;
2542
assert_eq! ( serde_json:: from_str:: < serde_json:: Value> ( & all) .unwrap ( ) [ " cases" ] .as_array ( ) .unwrap ( ) .len ( ) , 4 ) ;
2543
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 ↶
2544
2545
2546
let reports = oxygenhub:: junit:: on_bookmark( & h.app.db, row.id, " main" , 10 ) .await.unwrap ( ) ;
2547
let check = | state : String| {
2548
let app = h.app.clone ( ) ;
2549
async move { db:: hooks:: checks_for_state( & app.db, row.id, & state) .await.unwrap ( ) .into_iter ( ) .find ( | c | c.name = = " tests" ) .unwrap ( ) }
2550
} ;
2551
let c = check ( reports[ 1 ] .state.clone ( ) ) .await;
2552
assert_eq! ( ( c.status.as_str ( ) , c.summary.as_str ( ) ) , ( " failed" , " 1 failing; 1 new against main: t.suite::b" ) ) ;
2553
let c = check ( reports[ 0 ] .state.clone ( ) ) .await;
2554
assert_eq! ( ( c.status.as_str ( ) , c.summary.as_str ( ) ) , ( " passed" , " 4 passed" ) ) ;
2555
2556
2557
let ( _ , tab) = call ( & h.app, " GET" , " /andy/t/tests" , None , None ) .await;
2558
assert! (
2559
tab.contains ( " /andy/t/tests/case?classname=t.suite&name=b" ) | | tab.contains ( " /andy/t/tests/case?classname=t.suite&name=b" ) ,
2560
" linked"
2561
) ;
2562
let ( st, hist) = call ( & h.app, " GET" , " /andy/t/tests/case?classname=t.suite&name=b" , None , None ) .await;
2563
assert_eq! ( st, StatusCode:: OK , " {hist}" ) ;
2564
assert! ( hist.contains ( " 2 passed" ) & & hist.contains ( " 1 failed" ) & & hist.contains ( " b broke" ) , " {hist}" ) ;
2565
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 ↶
2566
}
67040b15 Andy McCall 26 days ago ↶
2567
2568
2569
2570
# [ tokio ::test ( flavor = " multi_thread" ) ]
2571
async fn a_job_no_runner_can_take_says_why ( ) {
2572
let h = hub ( ) .await;
2573
create ( & h.app, " andy" , " demo" ) .await;
2574
let url = format! ( " {} /andy/demo" , h.base) ;
2575
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2576
let c = client ( & url, work.path ( ) ) ;
2577
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
2578
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2579
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
2580
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2581
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
2582
2583
2584
let r = db:: runners:: configured( & h.app.db, & blake3:: hash( b " elsewhere" ) .to_hex ( ) , " earth-9" ) .await.unwrap ( ) ;
2585
db:: runners:: seen( & h.app.db, r.id, None , None ) .await.unwrap ( ) ;
2586
let other = " ab" .repeat ( 32 ) ;
92a797b2 Andy McCall 26 days ago ↶
2587
db:: runners:: set_offer( & h.app.db, r.id, & [ " microvm" .to_string ( ) ] , Some ( & [ other] ) , false , " p" ) .await.unwrap ( ) ;
67040b15 Andy McCall 26 days ago ↶
2588
oxygenhub:: flow:: explain_waits( & h.app) .await.unwrap ( ) ;
2589
2590
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/flow/runs/1" , Some ( " site" ) , None ) .await;
2591
let v: serde_json:: Value = serde_json:: from_str( & body) .unwrap ( ) ;
2592
let build = v[ " jobs" ] .as_array ( ) .unwrap ( ) .iter ( ) .find ( | j | j[ " name" ] = = " build" ) .unwrap ( ) .clone ( ) ;
2593
let why = build[ " waiting" ] .as_str ( ) .unwrap_or_default ( ) .to_string ( ) ;
2594
assert! ( why.contains ( " earth-9 does not hold image" ) & & why.contains ( & ROOTFS [ .. 12 ] ) , " {body}" ) ;
2595
let ( _ , page) = call ( & h.app, " GET" , " /andy/demo/flow/1" , None , None ) .await;
2596
assert! ( page.contains ( " data-waiting" ) & & page.contains ( " does not hold image" ) , " the run page says it" ) ;
2597
let checks = db:: hooks:: checks_for_state( & h.app.db, run.repo_id, & run.state) .await.unwrap ( ) ;
2598
let check = checks.iter ( ) .find ( | c | c.name = = " flow/build" ) .unwrap ( ) ;
2599
assert! ( check.summary.contains ( " does not hold image" ) , " {check:?}" ) ;
2600
}
92a797b2 Andy McCall 26 days ago ↶
2601
2602
async fn put_bytes ( app : & Shared, path : & str , bearer : & str , bytes : Vec< u8 > ) -> ( StatusCode, String) {
2603
let req = Request:: put( path)
2604
.header ( header:: AUTHORIZATION , format! ( " Bearer {bearer} " ) )
2605
.header ( header:: CONTENT_TYPE , " application/octet-stream" )
2606
.body ( Body:: from( bytes) )
2607
.unwrap ( ) ;
2608
let res = oxygenhub:: router( app.clone ( ) ) .oneshot ( req) .await.unwrap ( ) ;
2609
let st = res.status ( ) ;
2610
( st, String :: from_utf8_lossy( & res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) ) .into_owned ( ) )
2611
}
2612
2613
2614
2615
2616
2617
# [ tokio ::test ( flavor = " multi_thread" ) ]
2618
async fn images_are_sent_by_chunk_and_offered_to_runners_that_fetch ( ) {
2619
let h = hub ( ) .await;
2620
2621
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
2622
2623
let mut image: Vec< u8 > = ( 0 .. 400_000 u32 ) .map ( | i | ( i % 251 ) as u8 ) .collect ( ) ;
2624
image.extend ( std:: iter:: repeat_n( 0 u8 , 2 < < 20 ) ) ;
2625
image.extend ( ( 0 .. 100_000 u32 ) .map ( | i | ( i % 239 ) as u8 ) ) ;
2626
let mut chunks = std:: collections:: HashMap:: new( ) ;
2627
let m = jet:: images:: scan( & image[ .. ] , Some ( " x86_64" .into ( ) ) , | c , b | {
2628
chunks.insert ( c.hash ( ) .to_string ( ) , b.to_vec ( ) ) ;
2629
Ok ( ( ) )
2630
} )
2631
.unwrap ( ) ;
2632
let hash = m.hash.clone ( ) ;
2633
2634
2635
let ask = serde_json:: json! ( { " chunks" : m.needed ( ) } ) .to_string ( ) ;
2636
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/images/missing" , Some ( " site" ) , Some ( ask.clone ( ) ) ) .await;
2637
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
2638
let missing: Vec< String> = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " missing" ]
2639
.as_array ( )
2640
.unwrap ( )
2641
.iter ( )
2642
.map ( | v | v.as_str ( ) .unwrap ( ) .to_string ( ) )
2643
.collect ( ) ;
2644
assert_eq! ( missing.len ( ) , m.needed ( ) .len ( ) ) ;
2645
let manifest = serde_json:: to_string( & m) .unwrap ( ) ;
2646
let ( st, body) = call ( & h.app, " PUT" , & format! ( " /api/v1/flow/images/{hash} " ) , Some ( " site" ) , Some ( manifest.clone ( ) ) ) .await;
2647
assert! ( st = = StatusCode:: CONFLICT & & body.contains ( " was not sent" ) , " {st} {body}" ) ;
2648
2649
let first = & missing[ 0 ] ;
2650
assert_eq! ( put_bytes ( & h.app, & format! ( " /api/v1/flow/images/chunks/{first} " ) , " site" , vec! [ 1 , 2 , 3 ] ) .await.0 , StatusCode:: BAD_REQUEST ) ;
2651
assert_ne! (
2652
put_bytes ( & h.app, & format! ( " /api/v1/flow/images/chunks/{first} " ) , " runner-secret" , chunks[ first] .clone ( ) ) .await.0 ,
2653
StatusCode:: NO_CONTENT
2654
) ;
2655
for c in & missing {
2656
let ( st, body) = put_bytes ( & h.app, & format! ( " /api/v1/flow/images/chunks/{c} " ) , " site" , chunks[ c] .clone ( ) ) .await;
2657
assert_eq! ( st, StatusCode:: NO_CONTENT , " {body}" ) ;
2658
}
2659
let ( _ , body) = call ( & h.app, " POST" , " /api/v1/flow/images/missing" , Some ( " site" ) , Some ( ask) ) .await;
2660
assert! ( body.contains ( " \" missing\" :[]" ) , " {body}" ) ;
750579ec Andy McCall 25 days ago ↶
2661
2662
let many: Vec< String> = ( 0 .. 60_000 u32 ) .map ( | i | blake3:: hash( & i.to_le_bytes ( ) ) .to_hex ( ) .to_string ( ) ) .collect ( ) ;
2663
let ( st, body) =
2664
call ( & h.app, " POST" , " /api/v1/flow/images/missing" , Some ( " site" ) , Some ( serde_json:: json! ( { " chunks" : many } ) .to_string ( ) ) ) .await;
2665
assert_eq! ( st, StatusCode:: OK , " {}" , & body[ .. body.len ( ) .min ( 200 ) ] ) ;
2666
assert_eq! ( serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " missing" ] .as_array ( ) .unwrap ( ) .len ( ) , 60_000 ) ;
92a797b2 Andy McCall 26 days ago ↶
2667
2668
let mut wrong = m.clone ( ) ;
2669
wrong.hash = " ab" .repeat ( 32 ) ;
2670
let ( st, _ ) =
2671
call ( & h.app, " PUT" , & format! ( " /api/v1/flow/images/{} " , wrong.hash) , Some ( " site" ) , Some ( serde_json:: to_string( & wrong) .unwrap ( ) ) )
2672
.await;
2673
assert_eq! ( st, StatusCode:: CONFLICT , " chunks that make another image" ) ;
2674
let ( st, body) = call ( & h.app, " PUT" , & format! ( " /api/v1/flow/images/{hash} " ) , Some ( " site" ) , Some ( manifest) ) .await;
2675
assert_eq! ( st, StatusCode:: CREATED , " {body}" ) ;
2676
2677
2678
let ( st, body) = call ( & h.app, " GET" , & format! ( " /api/v1/flow/images/{hash} " ) , Some ( " runner-secret" ) , None ) .await;
2679
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
2680
let got: jet:: images:: Manifest = serde_json:: from_str( & body) .unwrap ( ) ;
2681
assert_eq! ( got, m) ;
2682
let dir = tempfile:: tempdir( ) .unwrap ( ) ;
2683
let out = dir.path ( ) .join ( " img.ext4" ) ;
2684
let mut fetched = Vec :: new( ) ;
2685
for c in got.needed ( ) {
2686
let req = Request:: get( format! ( " /api/v1/flow/images/chunks/{c} " ) )
2687
.header ( header:: AUTHORIZATION , " Bearer runner-secret" )
2688
.body ( Body:: empty( ) )
2689
.unwrap ( ) ;
2690
let res = oxygenhub:: router( h.app.clone ( ) ) .oneshot ( req) .await.unwrap ( ) ;
2691
assert_eq! ( res.status ( ) , StatusCode:: OK ) ;
2692
let bytes = res.into_body ( ) .collect ( ) .await.unwrap ( ) .to_bytes ( ) .to_vec ( ) ;
2693
fetched.push ( ( c.to_string ( ) , bytes) ) ;
2694
}
2695
jet:: images:: assemble( & got, & out, | c | Ok ( fetched.iter ( ) .find ( | ( h , _) | h = = c.hash ( ) ) .unwrap ( ) .1. clone ( ) ) ) .unwrap ( ) ;
2696
assert_eq! ( std:: fs:: read( & out) .unwrap ( ) , image) ;
2697
assert_eq! (
2698
call ( & h.app, " GET" , & format! ( " /api/v1/flow/images/{} " , " cd" .repeat ( 32 ) ) , Some ( " runner-secret" ) , None ) .await.0 ,
2699
StatusCode:: NOT_FOUND
2700
) ;
2701
assert_ne! ( call ( & h.app, " GET" , & format! ( " /api/v1/flow/images/{hash} " ) , None , None ) .await.0 , StatusCode:: OK , " not anonymous" ) ;
2702
2703
2704
2705
create ( & h.app, " andy" , " demo" ) .await;
2706
let url = format! ( " {} /andy/demo" , h.base) ;
2707
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2708
let c = client ( & url, work.path ( ) ) ;
2709
write (
2710
work.path ( ) ,
2711
" .flow/pipeline.kdl" ,
2712
& format! ( " schema-version 1\n sandbox \" img\" {{ \n rootfs \" cas://blob/{hash} \" \n }} \n job \" build\" {{ \n sandbox \" img\" \n exec \" make\" \n }} \n " ) ,
2713
) ;
2714
c.commit ( " Build on the image" , & [ ] ) .unwrap ( ) ;
2715
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2716
runs_of ( & h.app, " andy" , " demo" ) .await;
1c8276b9 Andy McCall 26 days ago ↶
2717
2718
let row = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
2719
let run1 = db:: flow:: run( & h.app.db, row.id, 1 ) .await.unwrap ( ) .unwrap ( ) ;
2720
let checks = checks_when ( & h.app, row.id, & run1.state, 1 ) .await;
2721
assert! ( checks.iter ( ) .all ( | c | c.name ! = " pipeline/images" ) , " {checks:?}" ) ;
92a797b2 Andy McCall 26 days ago ↶
2722
let ask = | fetches : bool | format! ( r # "{{ "runner":"test-1","runtimes":["microvm"],"images":[],"fetches_images":{fetches} }} "# ) ;
2723
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( ask ( false ) ) ) .await;
2724
assert_eq! ( st, StatusCode:: NO_CONTENT , " it lacks the image and does not fetch" ) ;
2725
oxygenhub:: flow:: explain_waits( & h.app) .await.unwrap ( ) ;
2726
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/flow/runs/1" , Some ( " site" ) , None ) .await;
2727
assert! ( body.contains ( " does not hold image" ) , " {body}" ) ;
2728
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( ask ( true ) ) ) .await;
2729
assert_eq! ( st, StatusCode:: OK , " it fetches, and the hub holds the image: {body}" ) ;
2730
}
40dd3f86 Andy McCall 26 days ago ↶
2731
2732
2733
2734
2735
# [ tokio ::test ( flavor = " multi_thread" ) ]
2736
async fn coverage_uploaded_from_a_workstation_says_so ( ) {
2737
let h = hub ( ) .await;
2738
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
2739
create ( & h.app, " andy" , " vk" ) .await;
2740
let url = format! ( " {} /andy/vk" , h.base) ;
2741
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2742
let c = client ( & url, work.path ( ) ) ;
2743
write ( work.path ( ) , " src/game.c" , " int a;\n int b;\n int c;\n int d;\n " ) ;
2744
c.commit ( " Game" , & [ ] ) .unwrap ( ) ;
2745
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2746
2747
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 " ;
2748
let body = serde_json:: json! ( { " bookmark" : " main" , " content" : lcov } ) .to_string ( ) ;
2749
let ( st, out) = call ( & h.app, " POST" , " /api/v1/repos/andy/vk/coverage" , Some ( " site" ) , Some ( body) ) .await;
2750
assert_eq! ( st, StatusCode:: CREATED , " {out}" ) ;
2751
let v: serde_json:: Value = serde_json:: from_str( & out) .unwrap ( ) ;
2752
assert_eq! ( ( v[ " uploaded_by" ] .as_str ( ) , v[ " lines_hit" ] .as_i64 ( ) , v[ " lines_found" ] .as_i64 ( ) ) , ( Some ( " andy" ) , Some ( 3 ) , Some ( 4 ) ) ) ;
2753
assert_eq! ( v[ " dropped" ] , 1 , " a file outside the state is left out" ) ;
2754
assert_eq! ( v[ " bookmark" ] , " main" ) ;
2755
2756
let ( _ , page) = call ( & h.app, " GET" , " /andy/vk/coverage" , None , None ) .await;
2757
assert! ( page.contains ( " uploaded by andy" ) & & page.contains ( " 75.0%" ) , " the Coverage page says where it came from" ) ;
2758
let ( _ , page) = call ( & h.app, " GET" , " /andy/vk/reports" , None , None ) .await;
2759
assert! ( page.contains ( " uploaded by andy" ) , " and the Reports card" ) ;
2760
let row = db:: repos:: get( & h.app.db, " andy" , " vk" ) .await.unwrap ( ) .unwrap ( ) ;
2761
let state = v[ " state" ] .as_str ( ) .unwrap ( ) ;
2762
let checks = db:: hooks:: checks_for_state( & h.app.db, row.id, state) .await.unwrap ( ) ;
2763
assert! ( checks.iter ( ) .all ( | c | c.name ! = " coverage" ) , " an upload sets no check: {checks:?}" ) ;
2764
2765
2766
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/repos/andy/vk/coverage" , None , Some ( r # "{"content":"x"}"# .into ( ) ) ) .await;
2767
assert_ne! ( st, StatusCode:: CREATED ) ;
2768
let ( st, _ ) = call (
2769
& h.app,
2770
" POST" ,
2771
" /api/v1/repos/andy/vk/coverage" ,
2772
Some ( " site" ) ,
2773
Some ( r # "{"bookmark":"nope","content":"SF:a\nend_of_record\n"}"# .into ( ) ) ,
2774
)
2775
.await;
2776
assert_eq! ( st, StatusCode:: BAD_REQUEST ) ;
2777
}
2fb70d01 Andy McCall 26 days ago ↶
2778
2779
2780
# [ tokio ::test ( flavor = " multi_thread" ) ]
2781
async fn a_runners_facts_are_kept_from_its_claim ( ) {
2782
let h = hub ( ) .await;
2783
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
2784
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"}}"# ;
2785
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( body.into ( ) ) ) .await;
2786
assert_eq! ( st, StatusCode:: NO_CONTENT , " nothing to run, but the facts are kept" ) ;
2787
let ( st, out) = call ( & h.app, " GET" , " /api/v1/runners" , Some ( " site" ) , None ) .await;
2788
assert_eq! ( st, StatusCode:: OK , " {out}" ) ;
2789
let v: serde_json:: Value = serde_json:: from_str( & out) .unwrap ( ) ;
2790
let r = & v[ 0 ] ;
2791
assert_eq! (
2792
( r[ " speed" ] .as_i64 ( ) , r[ " speed_class" ] .as_i64 ( ) , r[ " free_disk_mib" ] .as_i64 ( ) , r[ " load_x100" ] .as_i64 ( ) ) ,
2793
( Some ( 100 ) , Some ( 3 ) , Some ( 42000 ) , Some ( 112 ) )
2794
) ;
2795
let row = db:: runners:: list( & h.app.db) .await.unwrap ( ) .remove ( 0 ) ;
2796
assert_eq! ( row.facts_line ( ) , " speed 100 (class 3) · load 1.12 · 41 GiB free" ) ;
2797
}
44a3d2eb Andy McCall 26 days ago ↶
2798
2799
2800
2801
# [ tokio ::test ( flavor = " multi_thread" ) ]
2802
async fn a_jobs_next_run_needs_the_disk_its_last_took ( ) {
2803
let h = hub ( ) .await;
2804
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
2805
create ( & h.app, " andy" , " demo" ) .await;
2806
let url = format! ( " {} /andy/demo" , h.base) ;
2807
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2808
let c = client ( & url, work.path ( ) ) ;
2809
write (
2810
work.path ( ) ,
2811
" .flow/pipeline.kdl" ,
2812
& format! ( " schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" build\" {{ \n sandbox \" alpine\" \n exec \" make\" \n }} \n " ) ,
2813
) ;
2814
c.commit ( " Build" , & [ ] ) .unwrap ( ) ;
2815
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2816
runs_of ( & h.app, " andy" , " demo" ) .await;
2817
let claim_with = | free : u64 | format! ( r # "{{ "runner":"test-1","runtimes":["microvm"],"facts":{{ "free_disk_mib":{free} }} }} "# ) ;
2818
2819
2820
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( claim_with ( 500 ) ) ) .await;
2821
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
2822
let id = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
2823
let done = serde_json:: json! ( { " exit_code" : 0 , " duration_ms" : 10 , " peak_disk_mib" : 8000 , " peak_memory_mib" : 900 } ) ;
2824
assert_eq! ( finish ( & h.app, id, done) .await, StatusCode:: NO_CONTENT ) ;
2825
let job = db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) ;
2826
assert_eq! ( ( job.peak_disk_mib, job.peak_memory_mib) , ( Some ( 8000 ) , Some ( 900 ) ) ) ;
2827
2828
2829
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( " {}" .into ( ) ) ) .await;
2830
assert_eq! ( st, StatusCode:: CREATED ) ;
2831
let repo = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
2832
let run2 = db:: flow:: run( & h.app.db, repo.id, 2 ) .await.unwrap ( ) .unwrap ( ) ;
2833
let next = db:: flow:: jobs( & h.app.db, run2.id) .await.unwrap ( ) .remove ( 0 ) ;
2834
assert_eq! ( next.need_disk_mib, Some ( 8000 + 2000 + 64 ) ) ;
2835
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( claim_with ( 9000 ) ) ) .await;
2836
assert_eq! ( st, StatusCode:: NO_CONTENT , " 9000 MiB free is not enough" ) ;
2837
oxygenhub:: flow:: explain_waits( & h.app) .await.unwrap ( ) ;
2838
let waiting = db:: flow:: job( & h.app.db, next.id) .await.unwrap ( ) .unwrap ( ) .waiting;
2839
assert! ( waiting.contains ( " 8.8 GiB free; the job needs ~9.8 GiB" ) , " {waiting}" ) ;
2840
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( claim_with ( 20_000 ) ) ) .await;
2841
assert_eq! ( st, StatusCode:: OK , " a roomier runner takes it: {body}" ) ;
2842
}
501d8b26 Andy McCall 26 days ago ↶
2843
2844
2845
2846
2847
# [ tokio ::test ( flavor = " multi_thread" ) ]
2848
async fn a_job_waits_briefly_for_a_runner_that_holds_its_image ( ) {
2849
let h = hub ( ) .await;
2850
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
2851
2852
db:: runners:: configured( & h.app.db, & blake3:: hash( b " other-secret" ) .to_hex ( ) , " earth-9" ) .await.unwrap ( ) ;
2853
let ask = | images : & str | {
2854
format! (
2855
r # "{{ "runner":"r","runtimes":["microvm"],"images":[{images} ],"fetches_images":true,"facts":{{ "speed":100,"speed_class":3,"free_disk_mib":50000}} }} "#
2856
)
2857
} ;
2858
let holder = ask ( & format! ( " \" {ROOTFS} \" " ) ) ;
2859
2860
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " other-secret" ) , Some ( holder.clone ( ) ) ) .await;
2861
assert_eq! ( st, StatusCode:: NO_CONTENT ) ;
2862
2863
sqlx:: query( " INSERT INTO jet_images (hash, size, arch, chunks, uploader, created_ms) VALUES (?, 1, 'x86_64', 1, 'andy', 0)" )
2864
.bind ( ROOTFS )
2865
.execute ( & h.app.db)
2866
.await
2867
.unwrap ( ) ;
2868
2869
create ( & h.app, " andy" , " demo" ) .await;
2870
let url = format! ( " {} /andy/demo" , h.base) ;
2871
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2872
let c = client ( & url, work.path ( ) ) ;
2873
write (
2874
work.path ( ) ,
2875
" .flow/pipeline.kdl" ,
2876
& format! ( " schema-version 1\n sandbox \" alpine\" {{ \n rootfs \" cas://blob/{ROOTFS} \" \n }} \n job \" build\" {{ \n sandbox \" alpine\" \n exec \" make\" \n }} \n " ) ,
2877
) ;
2878
c.commit ( " Build" , & [ ] ) .unwrap ( ) ;
2879
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2880
runs_of ( & h.app, " andy" , " demo" ) .await;
2881
2882
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( ask ( " " ) ) ) .await;
2883
assert_eq! ( st, StatusCode:: NO_CONTENT , " held for the runner that holds the image" ) ;
2884
let ( st, body) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " other-secret" ) , Some ( holder) ) .await;
2885
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
2886
let id = serde_json:: from_str:: < serde_json:: Value> ( & body) .unwrap ( ) [ " job_id" ] .as_i64 ( ) .unwrap ( ) ;
2887
let placed = db:: flow:: job( & h.app.db, id) .await.unwrap ( ) .unwrap ( ) .placed;
2888
assert! ( placed.starts_with ( " holds the image, idle, speed 100 (class 3)" ) , " {placed}" ) ;
2889
assert! ( placed.contains ( " held for a better runner" ) , " {placed}" ) ;
2890
let ( _ , body) = call ( & h.app, " GET" , " /api/v1/repos/andy/demo/flow/runs/1" , Some ( " site" ) , None ) .await;
2891
assert! ( body.contains ( " \" placed\" :\" holds the image" ) , " {body}" ) ;
2892
}
20b8fe85 Andy McCall 26 days ago ↶
2893
2894
2895
2896
2897
# [ tokio ::test ( flavor = " multi_thread" ) ]
2898
async fn report_files_expire_under_retention ( ) {
2899
let h = hub ( ) .await;
2900
create ( & h.app, " andy" , " demo" ) .await;
2901
let repo = db:: repos:: get( & h.app.db, " andy" , " demo" ) .await.unwrap ( ) .unwrap ( ) ;
2902
let add = | bookmark : & 'static str , at : i64 | {
2903
let db = h.app.db.clone ( ) ;
2904
async move {
2905
let id: i64 = sqlx:: query_scalar(
2906
" INSERT INTO reports (repo_id, name, kind, title, summary, context, job, bookmark, state, files, bytes, created_ms) \
2907
VALUES (?, 'perf', 'performance', 'Perf', '', '{}', 'perf', ?, 's', 1, 1, ?) RETURNING id" ,
2908
)
2909
.bind ( repo.id)
2910
.bind ( bookmark)
2911
.bind ( at)
2912
.fetch_one ( & db)
2913
.await
2914
.unwrap ( ) ;
2915
sqlx:: query( " INSERT INTO report_files (report_id, path, blob, size) VALUES (?, 'report.toml', ?, 1)" )
2916
.bind ( id)
2917
.bind ( format! ( " {:064x}" , id) )
2918
.execute ( & db)
2919
.await
2920
.unwrap ( ) ;
2921
sqlx:: query( " INSERT INTO report_metrics (report_id, key, value, unit, better) VALUES (?, 'ms', 1, '', '')" )
2922
.bind ( id)
2923
.execute ( & db)
2924
.await
2925
.unwrap ( ) ;
2926
id
2927
}
2928
} ;
2929
let main_old = add ( " main" , 1 ) .await;
2930
add ( " main" , 2 ) .await;
2931
let mut feature = Vec :: new( ) ;
2932
for at in 0 .. 52 {
2933
feature.push ( add ( " feature" , 10 + at) .await) ;
2934
}
2935
assert_eq! ( oxygenhub:: reports:: expire( & h.app.db) .await.unwrap ( ) , 2 , " the two oldest feature runs" ) ;
2936
let files = | id : i64 | {
2937
let db = h.app.db.clone ( ) ;
2938
async move {
2939
sqlx:: query_scalar:: < _ , i64 > ( " SELECT COUNT(*) FROM report_files WHERE report_id = ?" ) .bind ( id) .fetch_one ( & db) .await.unwrap ( )
2940
}
2941
} ;
2942
assert_eq! ( ( files ( feature[ 0 ] ) .await, files ( feature[ 1 ] ) .await, files ( feature[ 2 ] ) .await) , ( 0 , 0 , 1 ) ) ;
2943
assert_eq! ( files ( main_old) .await, 1 , " the default bookmark keeps every run" ) ;
2944
assert_eq! ( oxygenhub:: reports:: metrics( & h.app.db, feature[ 0 ] ) .await.unwrap ( ) .len ( ) , 1 , " metrics are kept" ) ;
2945
assert_eq! ( oxygenhub:: reports:: expire( & h.app.db) .await.unwrap ( ) , 0 , " once" ) ;
2946
let ( _ , page) = call ( & h.app, " GET" , & format! ( " /andy/demo/reports/perf?id={} " , feature[ 0 ] ) , None , None ) .await;
2947
assert! ( page.contains ( " files have expired" ) , " the page says so" ) ;
1c8276b9 Andy McCall 26 days ago ↶
2948
}
2949
2950
2951
2952
2953
# [ tokio ::test ( flavor = " multi_thread" ) ]
2954
async fn a_pipeline_pinning_an_image_nobody_holds_fails_a_check ( ) {
2955
let h = hub ( ) .await;
2956
db:: accounts:: create_account( & h.app.db, " andy" , " [email protected] " , " x" , " salt" , true ) .await.unwrap ( ) ;
2957
2958
let none = r # "{"runner":"test-1","runtimes":["microvm"],"images":[]}"# ;
2959
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( none.into ( ) ) ) .await;
2960
assert_eq! ( st, StatusCode:: NO_CONTENT ) ;
2961
create ( & h.app, " andy" , " demo" ) .await;
2962
let url = format! ( " {} /andy/demo" , h.base) ;
2963
let work = tempfile:: tempdir( ) .unwrap ( ) ;
2964
let c = client ( & url, work.path ( ) ) ;
2965
write ( work.path ( ) , " .flow/pipeline.kdl" , & pipeline ( " " ) ) ;
2966
write ( work.path ( ) , " test.sh" , " echo ok\n " ) ;
2967
c.commit ( " Add a pipeline" , & [ ] ) .unwrap ( ) ;
2968
assert! ( c.push ( " origin" , false ) .unwrap ( ) .iter ( ) .all ( | r | r.ok ) ) ;
2969
let run = runs_of ( & h.app, " andy" , " demo" ) .await.remove ( 0 ) ;
2970
let checks = checks_when ( & h.app, run.repo_id, & run.state, 4 ) .await;
2971
let check = checks.iter ( ) .find ( | c | c.name = = " pipeline/images" ) .expect ( " an images check" ) ;
2972
assert! ( check.status = = " failed" & & check.summary.contains ( & ROOTFS [ .. 12 ] ) & & check.summary.contains ( " flow rootfs push" ) , " {check:?}" ) ;
2973
2974
let holds = format! ( r # "{{ "runner":"test-1","runtimes":["microvm"],"images":["{ROOTFS} "]}} "# ) ;
2975
call ( & h.app, " POST" , " /api/v1/flow/claim" , Some ( " runner-secret" ) , Some ( holds) ) .await;
2976
let ( st, _ ) = call ( & h.app, " POST" , " /api/v1/repos/andy/demo/flow/runs" , Some ( " site" ) , Some ( " {}" .into ( ) ) ) .await;
2977
assert_eq! ( st, StatusCode:: CREATED ) ;
2978
let checks = db:: hooks:: checks_for_state( & h.app.db, run.repo_id, & run.state) .await.unwrap ( ) ;
2979
let check = checks.iter ( ) .find ( | c | c.name = = " pipeline/images" ) .unwrap ( ) ;
2980
assert_eq! ( check.status, " passed" , " {check:?}" ) ;
20b8fe85 Andy McCall 26 days ago ↶
2981
}
a46d15e7 Andy McCall 25 days ago ↶
2982
2983
2984
2985
2986
# [ tokio ::test ( flavor = " multi_thread" ) ]
2987
async fn the_tracker_names_peers_in_the_same_site_and_signs_a_ticket ( ) {
2988
let h = hub ( ) .await;
2989
let app = & h.app;
2990
let image = " ab" .repeat ( 32 ) ;
2991
let claim_as = | token : & 'static str , name : & 'static str , body : serde_json:: Value| async move {
2992
let mut b = serde_json:: json! ( { " runner" : name, " runtimes" : [ " microvm" ] } ) ;
2993
b.as_object_mut ( ) .unwrap ( ) .extend ( body.as_object ( ) .unwrap ( ) .clone ( ) ) ;
2994
let ( st, out) = call ( app, " POST" , " /api/v1/flow/claim" , Some ( token) , Some ( b.to_string ( ) ) ) .await;
2995
assert_eq! ( st, StatusCode:: NO_CONTENT , " {out}" ) ;
2996
} ;
2997
let announce = | token : & 'static str , state : & 'static str | {
2998
let body = serde_json:: json! ( { " image" : image, " state" : state } ) .to_string ( ) ;
2999
async move { call ( app, " POST" , " /api/v1/flow/swarm/announce" , Some ( token) , Some ( body) ) .await }
3000
} ;
3001
3002
claim_as ( " runner-secret" , " a" , serde_json:: json! ( { " images" : [ image] , " swarm" : { " addr" : " 10.0.0.1:7401" , " site" : " lab" } } ) ) .await;
3003
claim_as ( " runner-two" , " b" , serde_json:: json! ( { " images" : [ ] , " swarm" : { " addr" : " 10.0.0.2:7401" , " site" : " lab" } } ) ) .await;
3004
claim_as ( " runner-three" , " c" , serde_json:: json! ( { " images" : [ ] , " swarm" : { " addr" : " 10.0.0.3:7401" , " site" : " lab" } } ) ) .await;
3005
3006
let ( st, body) = announce ( " runner-two" , " fetching" ) .await;
3007
assert_eq! ( st, StatusCode:: OK , " {body}" ) ;
3008
let a: flow_core:: api:: Announced = serde_json:: from_str( & body) .unwrap ( ) ;
3009
let names: Vec< ( & str , bool ) > = a.peers.iter ( ) .map ( | p | ( p.addr.as_str ( ) , p.complete) ) .collect ( ) ;
3010
assert_eq! ( names, vec! [ ( " 10.0.0.1:7401" , true ) ] , " only A holds it so far" ) ;
3011
let t = jet:: swarm:: Ticket:: open( & a.ticket, & app.hub_key.identity, oxygenhub:: db:: now_ms( ) as u64 ) .unwrap ( ) ;
3012
assert_eq! ( ( t.name.as_str ( ) , t.site.as_str ( ) ) , ( " b" , " lab" ) ) ;
3013
3014
3015
let ( _ , body) = announce ( " runner-three" , " fetching" ) .await;
3016
let a: flow_core:: api:: Announced = serde_json:: from_str( & body) .unwrap ( ) ;
3017
let mut peers: Vec< ( & str , bool ) > = a.peers.iter ( ) .map ( | p | ( p.addr.as_str ( ) , p.complete) ) .collect ( ) ;
3018
peers.sort ( ) ;
3019
assert_eq! ( peers, vec! [ ( " 10.0.0.1:7401" , true ) , ( " 10.0.0.2:7401" , false ) ] ) ;
3020
3021
3022
announce ( " runner-two" , " done" ) .await;
3023
claim_as ( " runner-three" , " c" , serde_json:: json! ( { " images" : [ ] , " swarm" : { " addr" : " 10.0.0.3:7401" , " site" : " elsewhere" } } ) ) .await;
3024
let ( _ , body) = announce ( " runner-three" , " fetching" ) .await;
3025
let a: flow_core:: api:: Announced = serde_json:: from_str( & body) .unwrap ( ) ;
3026
assert! ( a.peers.is_empty ( ) , " {:?}" , a.peers) ;
3027
3028
3029
claim_as ( " runner-two" , " b" , serde_json:: json! ( { " images" : [ ] } ) ) .await;
3030
let ( st, body) = announce ( " runner-two" , " fetching" ) .await;
3031
assert_eq! ( st, StatusCode:: CONFLICT ) ;
3032
assert! ( body.contains ( " [runner.swarm]" ) , " {body}" ) ;
3033
let ( st, _ ) =
3034
call ( app, " POST" , " /api/v1/flow/swarm/announce" , Some ( " runner-secret" ) , Some ( r # "{"image":"nope","state":"fetching"}"# .into ( ) ) )
3035
.await;
3036
assert_eq! ( st, StatusCode:: BAD_REQUEST ) ;
3037
let ( st, _ ) = call (
3038
app,
3039
" POST" ,
3040
" /api/v1/flow/swarm/announce" ,
3041
None ,
3042
Some ( serde_json:: json! ( { " image" : image, " state" : " fetching" } ) .to_string ( ) ) ,
3043
)
3044
.await;
3045
assert_eq! ( st, StatusCode:: UNAUTHORIZED ) ;
3046
}