1 use crate::protocol::{ImageObject, SourceCompat};
2 use html2md::parse_html;
3 use lemmy_apub_lib::verify::verify_domains_match;
4 use lemmy_utils::LemmyError;
12 pub mod private_message;
14 pub(crate) fn read_from_string_or_source(raw: &str, source: &Option<SourceCompat>) -> String {
15 if let Some(SourceCompat::Lemmy(s)) = source {
22 pub(crate) fn read_from_string_or_source_opt(
24 source: &Option<SourceCompat>,
26 if let Some(SourceCompat::Lemmy(s2)) = source {
27 Some(s2.content.clone())
29 raw.as_ref().map(|s| parse_html(s))
33 pub fn verify_image_domain_matches(a: &Url, b: &Option<ImageObject>) -> Result<(), LemmyError> {
35 verify_domains_match(a, &b.url)
42 pub(crate) mod tests {
45 r2d2::{ConnectionManager, Pool},
48 use lemmy_apub_lib::activity_queue::create_activity_queue;
49 use lemmy_db_schema::{
50 establish_unpooled_connection,
51 get_database_url_from_env,
52 source::secret::Secret,
55 rate_limit::{rate_limiter::RateLimiter, RateLimit},
56 request::build_user_agent,
57 settings::structs::Settings,
60 use lemmy_websocket::{chat_server::ChatServer, LemmyContext};
62 use reqwest_middleware::ClientBuilder;
64 use tokio::sync::Mutex;
66 // TODO: would be nice if we didnt have to use a full context for tests.
67 // or at least write a helper function so this code is shared with main.rs
68 pub(crate) fn init_context() -> LemmyContext {
69 let client = reqwest::Client::new().into();
70 // activity queue isnt used in tests, so worker count makes no difference
71 let queue_manager = create_activity_queue(client, 4);
72 let activity_queue = queue_manager.queue_handle().clone();
73 // call this to run migrations
74 establish_unpooled_connection();
75 let settings = Settings::init().unwrap();
76 let rate_limiter = RateLimit {
77 rate_limiter: Arc::new(Mutex::new(RateLimiter::default())),
78 rate_limit_config: settings.rate_limit.to_owned().unwrap_or_default(),
80 let client = Client::builder()
81 .user_agent(build_user_agent(&settings))
85 let client = ClientBuilder::new(client).build();
88 jwt_secret: "".to_string(),
90 let db_url = match get_database_url_from_env() {
92 Err(_) => settings.get_database_url(),
94 let manager = ConnectionManager::<PgConnection>::new(&db_url);
95 let pool = Pool::builder()
96 .max_size(settings.database.pool_size)
98 .unwrap_or_else(|_| panic!("Error connecting to {}", db_url));
99 async fn x() -> Result<String, LemmyError> {
102 let chat_server = ChatServer::startup(
105 |_, _, _, _| Box::pin(x()),
106 |_, _, _, _| Box::pin(x()),
108 activity_queue.clone(),
113 LemmyContext::create(pool, chat_server, client, activity_queue, settings, secret)