9ab074170b
Some checks failed
ci/woodpecker/pr/server-audit Pipeline was successful
ci/woodpecker/pr/server-lint Pipeline failed
ci/woodpecker/pr/server-vet Pipeline failed
ci/woodpecker/pr/server-test Pipeline was successful
ci/woodpecker/push/server-audit Pipeline was successful
ci/woodpecker/push/server-lint Pipeline failed
ci/woodpecker/push/server-vet Pipeline failed
ci/woodpecker/push/server-test Pipeline was successful
ci/woodpecker/push/useragent-analyze Pipeline failed
91 lines
2.2 KiB
Rust
91 lines
2.2 KiB
Rust
#![allow(
|
|
dead_code,
|
|
reason = "Common test utilities that may not be used in every test"
|
|
)]
|
|
use arbiter_crypto::safecell::{SafeCell, SafeCellHandle as _};
|
|
use arbiter_proto::transport::{Bi, Error, Receiver, Sender};
|
|
use arbiter_server::{
|
|
actors::{GlobalActors, vault::Vault},
|
|
db::{self, schema},
|
|
};
|
|
|
|
use async_trait::async_trait;
|
|
use diesel::QueryDsl;
|
|
use diesel_async::RunQueryDsl;
|
|
use tokio::sync::mpsc;
|
|
|
|
pub(crate) async fn bootstrapped_vault(db: &db::DatabasePool) -> Vault {
|
|
let mut actor = Vault::new(db.clone(), GlobalActors::spawn_message_bus())
|
|
.await
|
|
.unwrap();
|
|
actor
|
|
.bootstrap(SafeCell::new(b"test-seal-key".to_vec()))
|
|
.await
|
|
.unwrap();
|
|
actor
|
|
}
|
|
|
|
pub(crate) async fn root_key_history_id(db: &db::DatabasePool) -> i32 {
|
|
let mut conn = db.get().await.unwrap();
|
|
let id = schema::arbiter_settings::table
|
|
.select(schema::arbiter_settings::root_key_id)
|
|
.first::<Option<i32>>(&mut conn)
|
|
.await
|
|
.unwrap();
|
|
id.expect("root_key_id should be set after bootstrap")
|
|
}
|
|
|
|
pub(crate) struct ChannelTransport<T, Y> {
|
|
receiver: mpsc::Receiver<T>,
|
|
sender: mpsc::Sender<Y>,
|
|
}
|
|
|
|
impl<T, Y> ChannelTransport<T, Y> {
|
|
pub(crate) fn new() -> (Self, ChannelTransport<Y, T>) {
|
|
let (tx1, rx1) = mpsc::channel(10);
|
|
let (tx2, rx2) = mpsc::channel(10);
|
|
(
|
|
Self {
|
|
receiver: rx1,
|
|
sender: tx2,
|
|
},
|
|
ChannelTransport {
|
|
receiver: rx2,
|
|
sender: tx1,
|
|
},
|
|
)
|
|
}
|
|
}
|
|
|
|
#[async_trait]
|
|
impl<T, Y> Sender<Y> for ChannelTransport<T, Y>
|
|
where
|
|
T: Send + Sync + 'static,
|
|
Y: Send + Sync + 'static,
|
|
{
|
|
async fn send(&mut self, item: Y) -> Result<(), Error> {
|
|
self.sender
|
|
.send(item)
|
|
.await
|
|
.map_err(|_| Error::ChannelClosed)
|
|
}
|
|
}
|
|
|
|
#[async_trait]
|
|
impl<T, Y> Receiver<T> for ChannelTransport<T, Y>
|
|
where
|
|
T: Send + Sync + 'static,
|
|
Y: Send + Sync + 'static,
|
|
{
|
|
async fn recv(&mut self) -> Option<T> {
|
|
self.receiver.recv().await
|
|
}
|
|
}
|
|
|
|
impl<T, Y> Bi<T, Y> for ChannelTransport<T, Y>
|
|
where
|
|
T: Send + Sync + 'static,
|
|
Y: Send + Sync + 'static,
|
|
{
|
|
}
|