From e4d998a400f290365dc61dbd0965439eefde0a47 Mon Sep 17 00:00:00 2001
From: Sidharth-Singh10 <70999945+Sidharth-Singh10@users.noreply.github.com>
Date: Fri, 11 Apr 2025 19:56:07 +0530
Subject: [PATCH] Add test for double clicking to insert a gradient stop
 (#2527)

gradient tool test: Double click to insert a gradient stop

Co-authored-by: James Lindsay <78500760+0HyperCube@users.noreply.github.com>
---
 .../tool/tool_messages/gradient_tool.rs       | 55 +++++++++++++++++++
 1 file changed, 55 insertions(+)

diff --git a/editor/src/messages/tool/tool_messages/gradient_tool.rs b/editor/src/messages/tool/tool_messages/gradient_tool.rs
index 39b07a4e..9f9af721 100644
--- a/editor/src/messages/tool/tool_messages/gradient_tool.rs
+++ b/editor/src/messages/tool/tool_messages/gradient_tool.rs
@@ -522,6 +522,8 @@ impl Fsm for GradientToolFsmState {
 
 #[cfg(test)]
 mod test_gradient {
+	use crate::messages::input_mapper::utility_types::input_mouse::EditorMouseState;
+	use crate::messages::input_mapper::utility_types::input_mouse::ScrollDelta;
 	use crate::messages::portfolio::document::{graph_operation::utility_types::TransformIn, utility_types::misc::GroupFolderType};
 	pub use crate::test_utils::test_prelude::*;
 	use glam::DAffine2;
@@ -645,4 +647,57 @@ mod test_gradient {
 		assert!(transform.transform_point2(gradient.start).abs_diff_eq(DVec2::new(2., 3.), 1e-10));
 		assert!(transform.transform_point2(gradient.end).abs_diff_eq(DVec2::new(24., 4.), 1e-10));
 	}
+
+	#[tokio::test]
+	async fn double_click_insert_stop() {
+		let mut editor = EditorTestUtils::create();
+		editor.new_document().await;
+
+		editor.drag_tool(ToolType::Rectangle, -5., -3., 100., 100., ModifierKeys::empty()).await;
+
+		editor.select_primary_color(Color::GREEN).await;
+		editor.select_secondary_color(Color::BLUE).await;
+
+		editor.drag_tool(ToolType::Gradient, 0., 0., 100., 0., ModifierKeys::empty()).await;
+
+		// Get initial gradient state (should have 2 stops)
+		let initial_fills = get_fills(&mut editor).await;
+		assert_eq!(initial_fills.len(), 1);
+		let (initial_fill, _) = initial_fills.first().unwrap();
+		let initial_gradient = initial_fill.as_gradient().unwrap();
+		assert_eq!(initial_gradient.stops.len(), 2);
+
+		editor.select_tool(ToolType::Gradient).await;
+
+		// Simulate a double click
+		let click_position = DVec2::new(50., 0.);
+		let modifier_keys = ModifierKeys::empty();
+
+		// Send the DoubleClick event
+		editor
+			.handle_message(InputPreprocessorMessage::DoubleClick {
+				editor_mouse_state: EditorMouseState {
+					editor_position: click_position,
+					mouse_keys: MouseKeys::LEFT,
+					scroll_delta: ScrollDelta::default(),
+				},
+				modifier_keys,
+			})
+			.await;
+
+		// Check that a new stop has been added
+		let updated_fills = get_fills(&mut editor).await;
+		assert_eq!(updated_fills.len(), 1);
+		let (updated_fill, _) = updated_fills.first().unwrap();
+		let updated_gradient = updated_fill.as_gradient().unwrap();
+
+		assert_eq!(updated_gradient.stops.len(), 3);
+
+		let positions: Vec<f64> = updated_gradient.stops.iter().map(|(pos, _)| *pos).collect();
+		assert!(
+			positions.iter().any(|pos| (pos - 0.5).abs() < 0.1),
+			"Expected to find a stop near position 0.5, but found: {:?}",
+			positions
+		);
+	}
 }