Something went wrong. Try again.
A lexicon-driven AppView for ATProto.
Something went wrong. Try again.
Rust
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189mod common;
use axum::body::Body;use axum::http::{Request, StatusCode};use happyview::db::{adapt_sql, now_rfc3339};use http_body_util::BodyExt;use serde_json::Value;use serial_test::serial;use tower::ServiceExt;
use common::app::TestApp;use common::fixtures;
// ---------------------------------------------------------------------------// Helpers// ---------------------------------------------------------------------------
async fn json_body(resp: axum::response::Response) -> Value { let body = resp.into_body().collect().await.unwrap().to_bytes(); serde_json::from_slice(&body).unwrap()}
fn admin_get( uri: &str, cookie: (axum::http::HeaderName, axum::http::HeaderValue),) -> Request<Body> { Request::builder() .uri(uri) .header(cookie.0, cookie.1) .body(Body::empty()) .unwrap()}
fn admin_delete( uri: &str, cookie: (axum::http::HeaderName, axum::http::HeaderValue),) -> Request<Body> { Request::builder() .method("DELETE") .uri(uri) .header(cookie.0, cookie.1) .body(Body::empty()) .unwrap()}
async fn seed_network_lexicon(app: &TestApp, nsid: &str, authority_did: &str) { let lexicon_json = fixtures::game_record_lexicon(); let backend = app.state.db_backend; let sql = adapt_sql( r#" INSERT INTO lexicons (id, lexicon_json, backfill, source, authority_did, last_fetched_at, created_at) VALUES (?, ?, 0, 'network', ?, ?, ?) ON CONFLICT (id) DO NOTHING "#, backend, ); let now = now_rfc3339(); sqlx::query(&sql) .bind(nsid) .bind(serde_json::to_string(&lexicon_json).unwrap_or_default()) .bind(authority_did) .bind(&now) .bind(&now) .execute(&app.state.db) .await .expect("failed to seed network lexicon");}
// ---------------------------------------------------------------------------// Network lexicon CRUD// ---------------------------------------------------------------------------
#[tokio::test]#[serial]#[ignore]async fn network_lexicon_list_empty() { let app = TestApp::new().await;
let resp = app .router .clone() .oneshot(admin_get("/admin/network-lexicons", app.admin_cookie())) .await .unwrap();
assert_eq!(resp.status(), StatusCode::OK); let json = json_body(resp).await; assert!(json.as_array().unwrap().is_empty());}
#[tokio::test]#[serial]#[ignore]async fn network_lexicon_list_returns_seeded() { let app = TestApp::new().await;
seed_network_lexicon(&app, "games.gamesgamesgamesgames.game", "did:plc:authority").await;
let resp = app .router .clone() .oneshot(admin_get("/admin/network-lexicons", app.admin_cookie())) .await .unwrap();
assert_eq!(resp.status(), StatusCode::OK); let json = json_body(resp).await; let arr = json.as_array().unwrap(); assert_eq!(arr.len(), 1); assert_eq!(arr[0]["nsid"], "games.gamesgamesgamesgames.game"); assert_eq!(arr[0]["authority_did"], "did:plc:authority");}
#[tokio::test]#[serial]#[ignore]async fn network_lexicon_delete_removes_tracking_and_lexicon() { let app = TestApp::new().await; let backend = app.state.db_backend;
let nsid = "games.gamesgamesgamesgames.game"; seed_network_lexicon(&app, nsid, "did:plc:authority").await;
let resp = app .router .clone() .clone() .oneshot(admin_delete( &format!("/admin/network-lexicons/{nsid}"), app.admin_cookie(), )) .await .unwrap();
assert_eq!(resp.status(), StatusCode::NO_CONTENT);
// Verify lexicon is removed. let sql = adapt_sql( "SELECT COUNT(*) FROM lexicons WHERE id = ? AND source = 'network'", backend, ); let count: (i64,) = sqlx::query_as(&sql) .bind(nsid) .fetch_one(&app.state.db) .await .unwrap(); assert_eq!(count.0, 0);}
#[tokio::test]#[serial]#[ignore]async fn network_lexicon_delete_not_found() { let app = TestApp::new().await;
let resp = app .router .clone() .oneshot(admin_delete( "/admin/network-lexicons/nonexistent.lexicon", app.admin_cookie(), )) .await .unwrap();
assert_eq!(resp.status(), StatusCode::NOT_FOUND);}
#[tokio::test]#[serial]#[ignore]async fn network_lexicon_no_auth_returns_401() { let app = TestApp::new().await;
let resp = app .router .clone() .oneshot( Request::builder() .uri("/admin/network-lexicons") .body(Body::empty()) .unwrap(), ) .await .unwrap();
assert_eq!(resp.status(), StatusCode::UNAUTHORIZED);}