cloudflare · kflansburg · Mar 11, 2024 · Mar 8, 2024 · Mar 8, 2024 · Mar 8, 2024
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/package.json b/package.json
@@ -22,6 +22,6 @@
     "vitest": "^0.32.0"
   },
   "scripts": {
-    "test": ""
+    "test": "cd worker-sandbox && worker-build --dev && vitest"
   }
 }
diff --git a/worker-sandbox/Cargo.toml b/worker-sandbox/Cargo.toml
@@ -14,6 +14,7 @@ default = ["console_error_panic_hook"]
 
 [dependencies]
 blake2 = "0.10.6"
+bytes = "1.4.0"
 chrono = { version = "0.4.26", default-features = false, features = [
     "wasmbind",
     "clock",
@@ -26,19 +27,21 @@ http = "0.2.9"
 regex = "1.8.4"
 serde = { version = "1.0.164", features = ["derive"] }
 serde_json = "1.0.96"
-worker = { path = "../worker", version = "0.0.17", features= ["queue", "d1"] }
+worker = { path = "../worker", version = "0.0.17", features = ["queue", "d1"] }
 futures-channel = "0.3.28"
 futures-util = { version = "0.3.28", default-features = false }
 rand = "0.8.5"
-uuid = {version = "1.3.3", features = ["v4", "serde"]}
+uuid = { version = "1.3.3", features = ["v4", "serde"] }
 serde-wasm-bindgen = "0.5.0"
 md5 = "0.7.0"
 router-service = { git = "https://github.com/zebp/router-service", version = "0.1.0" }
 tower = "0.4.13"
 
 [dev-dependencies]
 futures-channel = { version = "0.3.28", features = ["sink"] }
-futures-util = { version = "0.3.28", default-features = false, features = ["sink"] }
+futures-util = { version = "0.3.28", default-features = false, features = [
+    "sink",
+] }
 reqwest = { version = "0.11.18", features = [
     "blocking",
     "json",

diff --git a/worker-sandbox/src/d1.rs b/worker-sandbox/src/d1.rs
@@ -8,10 +8,7 @@ struct Person {
     age: u32,
 }
 
-pub async fn prepared_statement(
-    _req: http::Request<Body>,
-    env: Env,
-) -> Result<http::Response<Body>> {
+pub async fn prepared_statement(env: &Env) -> Result<http::Response<Body>> {
     let db = env.d1("DB")?;
     let stmt = worker::query!(&db, "SELECT * FROM people WHERE name = ?", "Ryan Upton")?;
 
@@ -47,7 +44,7 @@ pub async fn prepared_statement(
     Ok(http::Response::new("ok".into()))
 }
 
-pub async fn batch(_req: http::Request<Body>, env: Env) -> Result<http::Response<Body>> {
+pub async fn batch(env: &Env) -> Result<http::Response<Body>> {
     let db = env.d1("DB")?;
     let mut results = db
         .batch(vec![
@@ -71,7 +68,7 @@ pub async fn batch(_req: http::Request<Body>, env: Env) -> Result<http::Response
     Ok(http::Response::new("ok".into()))
 }
 
-pub async fn exec(req: http::Request<Body>, env: Env) -> Result<http::Response<Body>> {
+pub async fn exec(req: http::Request<Body>, env: &Env) -> Result<http::Response<Body>> {
     let db = env.d1("DB")?;
 
     let result = db
@@ -84,21 +81,24 @@ pub async fn exec(req: http::Request<Body>, env: Env) -> Result<http::Response<B
     ))
 }
 
-pub async fn dump(_req: http::Request<Body>, env: Env) -> Result<http::Response<Body>> {
+pub async fn dump(env: &Env) -> Result<http::Response<Body>> {
     let db = env.d1("DB")?;
     let bytes = db.dump().await?;
     Ok(http::Response::new(bytes.into()))
 }
 
-pub async fn error(_req: http::Request<Body>, env: Env) -> Result<http::Response<Body>> {
+pub async fn error(env: &Env) -> Result<http::Response<Body>> {
     let db = env.d1("DB")?;
     let error = db
         .exec("THIS IS NOT VALID SQL")
         .await
         .expect_err("did not get error");
 
     if let Error::D1(error) = error {
-        assert_eq!(error.cause(), "Error in line 1: THIS IS NOT VALID SQL: ERROR 9009: SQL prepare error: near \"THIS\": syntax error in THIS IS NOT VALID SQL at offset 0")
+        assert_eq!(
+            error.cause(),
+            "Error in line 1: THIS IS NOT VALID SQL: SqliteError: near \"THIS\": syntax error"
+        )
     } else {
         panic!("expected D1 error");
     }