refactor(tests): Allow converting EventBuilder to state

Author: robintown

testing/matrix-sdk-test/src/event_factory.rs

@@ -51,8 +51,9 @@ use ruma::{
             topic::RoomTopicEventContent,
         },
         typing::TypingEventContent,
-        AnySyncTimelineEvent, AnyTimelineEvent, BundledMessageLikeRelations, EventContent,
-        RedactedMessageLikeEventContent, RedactedStateEventContent,
+        AnyStateEvent, AnySyncStateEvent, AnySyncTimelineEvent, AnyTimelineEvent,
+        BundledMessageLikeRelations, EventContent, RedactedMessageLikeEventContent,
+        RedactedStateEventContent, StateEventContent,
     },
     serde::Raw,
     server_name, EventId, MilliSecondsSinceUnixEpoch, MxcUri, OwnedEventId, OwnedMxcUri,
@@ -263,6 +264,14 @@ where
         Raw::new(&self.construct_json(false)).unwrap().cast()
     }
 
+    pub fn into_raw_state(self) -> Raw<AnyStateEvent> {
+        Raw::new(&self.construct_json(true)).unwrap().cast()
+    }
+
+    pub fn into_raw_sync_state(self) -> Raw<AnySyncStateEvent> {
+        Raw::new(&self.construct_json(false)).unwrap().cast()
+    }
+
     pub fn into_event(self) -> TimelineEvent {
         TimelineEvent::new(self.into_raw_sync())
     }
@@ -372,6 +381,24 @@ where
     }
 }
 
+impl<E: StateEventContent> From<EventBuilder<E>> for Raw<AnySyncStateEvent>
+where
+    E::EventType: Serialize,
+{
+    fn from(val: EventBuilder<E>) -> Self {
+        val.into_raw_sync_state()
+    }
+}
+
+impl<E: StateEventContent> From<EventBuilder<E>> for Raw<AnyStateEvent>
+where
+    E::EventType: Serialize,
+{
+    fn from(val: EventBuilder<E>) -> Self {
+        val.into_raw_state()
+    }
+}
+
 #[derive(Debug, Default)]
 pub struct EventFactory {
     next_ts: AtomicU64,