Merge pull request #5401 from systeminit/jkeiser/management-input-tests

test(sdf): Add regression tests for incomingConnections
systeminit · Jan 31, 2025 · cac5b0d · cac5b0d
2 parents 49160f6 + 09adebd
commit cac5b0d
Show file tree

Hide file tree

Showing 4 changed files with 483 additions and 40 deletions.
diff --git a/lib/dal-test/src/expected.rs b/lib/dal-test/src/expected.rs
@@ -3,6 +3,7 @@
 #![allow(clippy::expect_used)]
 
 use crate::helpers::ChangeSetTestHelpers;
+use dal::component::socket::ComponentInputSocket;
 use dal::diagram::geometry::RawGeometry;
 use dal::diagram::view::{View, ViewId};
 use dal::{
@@ -377,6 +378,30 @@ impl ExpectComponent {
             .await
     }
 
+    pub async fn list(ctx: &mut DalContext) -> Vec<ExpectComponent> {
+        Component::list_ids(ctx)
+            .await
+            .expect("list components")
+            .into_iter()
+            .map(Into::into)
+            .collect()
+    }
+
+    pub async fn find_opt(ctx: &mut DalContext, name: impl AsRef<str>) -> Option<ExpectComponent> {
+        for component in ExpectComponent::list(ctx).await {
+            if name.as_ref() == component.name(ctx).await {
+                return Some(component);
+            }
+        }
+        None
+    }
+
+    pub async fn find(ctx: &mut DalContext, name: impl AsRef<str>) -> ExpectComponent {
+        Self::find_opt(ctx, name)
+            .await
+            .expect("component not found")
+    }
+
     pub fn id(self) -> ComponentId {
         self.0
     }
@@ -387,6 +412,12 @@ impl ExpectComponent {
             .expect("get component by id")
     }
 
+    pub async fn name(&self, ctx: &mut DalContext) -> String {
+        Component::name_by_id(ctx, self.0)
+            .await
+            .expect("get component name")
+    }
+
     pub async fn geometry_for_default(self, ctx: &DalContext) -> RawGeometry {
         let view_id = ExpectView::get_id_for_default(ctx).await;
 
@@ -456,6 +487,17 @@ impl ExpectComponent {
         ExpectComponentInputSocket(self.0, input_socket_id)
     }
 
+    pub async fn input_connections(
+        self,
+        ctx: &DalContext,
+        input_socket_id: impl InputSocketKey,
+    ) -> Vec<ExpectComponentOutputSocket> {
+        self.input_socket(ctx, input_socket_id)
+            .await
+            .connections(ctx)
+            .await
+    }
+
     pub async fn output_socket(
         self,
         ctx: &DalContext,
@@ -475,7 +517,7 @@ impl ExpectComponent {
     }
 }
 
-#[derive(Debug, Copy, Clone)]
+#[derive(Debug, Copy, Clone, PartialEq, Eq)]
 pub struct ExpectComponentProp(ComponentId, PropId);
 
 impl ExpectComponentProp {
@@ -555,17 +597,38 @@ impl ExpectComponentProp {
     }
 }
 
-#[derive(Debug, Copy, Clone)]
+#[derive(Debug, Copy, Clone, PartialEq, Eq)]
 pub struct ExpectComponentInputSocket(ComponentId, InputSocketId);
 
 impl ExpectComponentInputSocket {
     pub fn component(self) -> ExpectComponent {
         ExpectComponent(self.0)
     }
-    pub fn prop(self) -> ExpectInputSocket {
+    pub fn input_socket(self) -> ExpectInputSocket {
         ExpectInputSocket(self.1)
     }
 
+    pub async fn connect(self, ctx: &DalContext, dest: ExpectComponentOutputSocket) {
+        dest.connect(ctx, self).await
+    }
+
+    pub async fn connections(self, ctx: &DalContext) -> Vec<ExpectComponentOutputSocket> {
+        let component_input_socket = ComponentInputSocket {
+            component_id: self.0,
+            input_socket_id: self.1,
+            attribute_value_id: self.attribute_value(ctx).await.id(),
+        };
+        component_input_socket
+            .connections(ctx)
+            .await
+            .expect("get connections")
+            .into_iter()
+            .map(|(component_id, output_socket_id, _)| {
+                ExpectComponentOutputSocket(component_id, output_socket_id)
+            })
+            .collect()
+    }
+
     pub async fn attribute_value(self, ctx: &DalContext) -> ExpectAttributeValue {
         InputSocket::component_attribute_value_for_input_socket_id(ctx, self.1, self.0)
             .await
@@ -616,14 +679,14 @@ impl ExpectComponentInputSocket {
     }
 }
 
-#[derive(Debug, Copy, Clone)]
+#[derive(Debug, Copy, Clone, PartialEq, Eq)]
 pub struct ExpectComponentOutputSocket(ComponentId, OutputSocketId);
 
 impl ExpectComponentOutputSocket {
     pub fn component(self) -> ExpectComponent {
         ExpectComponent(self.0)
     }
-    pub fn prop(self) -> ExpectOutputSocket {
+    pub fn output_socket(self) -> ExpectOutputSocket {
         ExpectOutputSocket(self.1)
     }