1 use crate::{newtypes::DbUrl, source::activity::*, traits::Crud};
2 use diesel::{dsl::*, result::Error, *};
6 io::{Error as IoError, ErrorKind},
9 impl Crud for Activity {
10 type Form = ActivityForm;
12 fn read(conn: &PgConnection, activity_id: i32) -> Result<Self, Error> {
13 use crate::schema::activity::dsl::*;
14 activity.find(activity_id).first::<Self>(conn)
17 fn create(conn: &PgConnection, new_activity: &ActivityForm) -> Result<Self, Error> {
18 use crate::schema::activity::dsl::*;
21 .get_result::<Self>(conn)
27 new_activity: &ActivityForm,
28 ) -> Result<Self, Error> {
29 use crate::schema::activity::dsl::*;
30 diesel::update(activity.find(activity_id))
32 .get_result::<Self>(conn)
34 fn delete(conn: &PgConnection, activity_id: i32) -> Result<usize, Error> {
35 use crate::schema::activity::dsl::*;
36 diesel::delete(activity.find(activity_id)).execute(conn)
47 ) -> Result<Activity, IoError>
51 let activity_form = ActivityForm {
53 data: serde_json::to_value(&data)?,
58 let result = Activity::create(conn, &activity_form);
61 Err(e) => Err(IoError::new(
63 format!("Failed to insert activity into database: {}", e),
68 pub fn read_from_apub_id(conn: &PgConnection, object_id: &DbUrl) -> Result<Activity, Error> {
69 use crate::schema::activity::dsl::*;
70 activity.filter(ap_id.eq(object_id)).first::<Self>(conn)
73 pub fn delete_olds(conn: &PgConnection) -> Result<usize, Error> {
74 use crate::schema::activity::dsl::*;
75 diesel::delete(activity.filter(published.lt(now - 6.months()))).execute(conn)
83 establish_unpooled_connection,
86 activity::{Activity, ActivityForm},
87 person::{Person, PersonForm},
90 use serde_json::Value;
91 use serial_test::serial;
97 let conn = establish_unpooled_connection();
99 let creator_form = PersonForm {
100 name: "activity_creator_pm".into(),
101 ..PersonForm::default()
104 let inserted_creator = Person::create(&conn, &creator_form).unwrap();
106 let ap_id: DbUrl = Url::parse(
107 "https://enterprise.lemmy.ml/activities/delete/f1b5d57c-80f8-4e03-a615-688d552e946c",
111 let test_json: Value = serde_json::from_str(
113 "@context": "https://www.w3.org/ns/activitystreams",
114 "id": "https://enterprise.lemmy.ml/activities/delete/f1b5d57c-80f8-4e03-a615-688d552e946c",
116 "actor": "https://enterprise.lemmy.ml/u/riker",
117 "to": "https://www.w3.org/ns/activitystreams#Public",
119 "https://enterprise.lemmy.ml/c/main/"
121 "object": "https://enterprise.lemmy.ml/post/32"
125 let activity_form = ActivityForm {
126 ap_id: ap_id.clone(),
127 data: test_json.to_owned(),
133 let inserted_activity = Activity::create(&conn, &activity_form).unwrap();
135 let expected_activity = Activity {
136 ap_id: Some(ap_id.clone()),
137 id: inserted_activity.id,
140 sensitive: Some(false),
141 published: inserted_activity.published,
145 let read_activity = Activity::read(&conn, inserted_activity.id).unwrap();
146 let read_activity_by_apub_id = Activity::read_from_apub_id(&conn, &ap_id).unwrap();
147 Person::delete(&conn, inserted_creator.id).unwrap();
148 Activity::delete(&conn, inserted_activity.id).unwrap();
150 assert_eq!(expected_activity, read_activity);
151 assert_eq!(expected_activity, read_activity_by_apub_id);
152 assert_eq!(expected_activity, inserted_activity);