1 use crate::protocol::Source;
2 use html2md::parse_html;
9 pub mod private_message;
11 pub(crate) fn get_summary_from_string_or_source(
13 source: &Option<Source>,
15 if let Some(source) = &source {
16 Some(source.content.clone())
18 raw.as_ref().map(|s| parse_html(s))
23 pub(crate) mod tests {
25 use background_jobs::QueueHandle;
27 r2d2::{ConnectionManager, Pool},
30 use lemmy_db_schema::{
31 establish_unpooled_connection,
32 get_database_url_from_env,
33 source::secret::Secret,
36 rate_limit::{rate_limiter::RateLimiter, RateLimit},
37 request::build_user_agent,
38 settings::structs::Settings,
41 use lemmy_websocket::{chat_server::ChatServer, LemmyContext};
43 use reqwest_middleware::ClientBuilder;
44 use serde::de::DeserializeOwned;
45 use std::{fs::File, io::BufReader, sync::Arc};
46 use tokio::sync::Mutex;
48 // TODO: would be nice if we didnt have to use a full context for tests.
49 // or at least write a helper function so this code is shared with main.rs
50 pub(crate) fn init_context(activity_queue: QueueHandle) -> LemmyContext {
51 // call this to run migrations
52 establish_unpooled_connection();
53 let settings = Settings::init().unwrap();
54 let rate_limiter = RateLimit {
55 rate_limiter: Arc::new(Mutex::new(RateLimiter::default())),
56 rate_limit_config: settings.rate_limit.to_owned().unwrap_or_default(),
58 let client = Client::builder()
59 .user_agent(build_user_agent(&settings))
63 let client = ClientBuilder::new(client).build();
66 jwt_secret: "".to_string(),
68 let db_url = match get_database_url_from_env() {
70 Err(_) => settings.get_database_url(),
72 let manager = ConnectionManager::<PgConnection>::new(&db_url);
73 let pool = Pool::builder()
74 .max_size(settings.database.pool_size)
76 .unwrap_or_else(|_| panic!("Error connecting to {}", db_url));
77 async fn x() -> Result<String, LemmyError> {
80 let chat_server = ChatServer::startup(
83 |_, _, _, _| Box::pin(x()),
84 |_, _, _, _| Box::pin(x()),
86 activity_queue.clone(),
91 LemmyContext::create(pool, chat_server, client, activity_queue, settings, secret)
94 pub(crate) fn file_to_json_object<T: DeserializeOwned>(path: &str) -> Result<T, LemmyError> {
95 let file = File::open(path)?;
96 let reader = BufReader::new(file);
97 Ok(serde_json::from_reader(reader)?)