diff --git a/src/db/spanner/models.rs b/src/db/spanner/models.rs index 849b6006..f5dd6b9b 100644 --- a/src/db/spanner/models.rs +++ b/src/db/spanner/models.rs @@ -1858,16 +1858,30 @@ impl Db for SpannerDb { Box::pin(async move { db.get_bso_timestamp_async(param).map_err(Into::into).await }) } + #[cfg(not(any(test, feature = "db_test")))] fn put_bso(&self, param: params::PutBso) -> DbFuture { let db = self.clone(); Box::pin(async move { db.put_bso_async(param).map_err(Into::into).await }) } + #[cfg(any(test, feature = "db_test"))] + fn put_bso(&self, param: params::PutBso) -> DbFuture { + let db = self.clone(); + Box::pin(block(move || db.put_bso_sync(param).map_err(Into::into)).map_err(Into::into)) + } + + #[cfg(not(any(test, feature = "db_test")))] fn post_bsos(&self, param: params::PostBsos) -> DbFuture { let db = self.clone(); Box::pin(async move { db.post_bsos_async(param).map_err(Into::into).await }) } + #[cfg(any(test, feature = "db_test"))] + fn post_bsos(&self, param: params::PostBsos) -> DbFuture { + let db = self.clone(); + Box::pin(block(move || db.post_bsos_sync(param).map_err(Into::into)).map_err(Into::into)) + } + fn validate_batch_id(&self, id: String) -> Result<()> { batch::validate_batch_id(&id) }