From 82828589417c3f30ebb6839e797ff7404259adfd Mon Sep 17 00:00:00 2001
From: Mikayla Maki <mikayla.c.maki@icloud.com>
Date: Fri, 15 Jul 2022 11:30:14 -0700
Subject: [PATCH 1/2] Updated copy integration test to match retries on
 terminal integration test

---
 crates/terminal/src/terminal.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/crates/terminal/src/terminal.rs b/crates/terminal/src/terminal.rs
index c2d8fc1bab..9f5ab4ce7f 100644
--- a/crates/terminal/src/terminal.rs
+++ b/crates/terminal/src/terminal.rs
@@ -496,11 +496,11 @@ mod tests {
     }
 
     /// Integration test for selections, clipboard, and terminal execution
-    #[gpui::test]
+    #[gpui::test(retries = 5)]
     async fn test_copy(cx: &mut TestAppContext) {
         let mut result_line: i32 = 0;
         let terminal = cx.add_view(Default::default(), |cx| Terminal::new(None, false, cx));
-        cx.set_condition_duration(Some(Duration::from_secs(2)));
+        cx.set_condition_duration(Some(Duration::from_secs(5)));
 
         terminal.update(cx, |terminal, cx| {
             terminal.connection.update(cx, |connection, _| {

From e156675640302d2ba8a86ea2179703162dcf15bd Mon Sep 17 00:00:00 2001
From: Mikayla Maki <mikayla.c.maki@icloud.com>
Date: Fri, 15 Jul 2022 12:15:55 -0700
Subject: [PATCH 2/2] Revamped testing in terminals

---
 crates/terminal/src/terminal.rs               | 85 ++++++-------------
 .../src/tests/terminal_test_context.rs        | 82 ++++++++++++++++++
 2 files changed, 107 insertions(+), 60 deletions(-)
 create mode 100644 crates/terminal/src/tests/terminal_test_context.rs

diff --git a/crates/terminal/src/terminal.rs b/crates/terminal/src/terminal.rs
index 9f5ab4ce7f..2ae3faf103 100644
--- a/crates/terminal/src/terminal.rs
+++ b/crates/terminal/src/terminal.rs
@@ -458,85 +458,59 @@ fn get_wd_for_workspace(workspace: &Workspace, cx: &AppContext) -> Option<PathBu
 #[cfg(test)]
 mod tests {
 
+    use crate::tests::terminal_test_context::TerminalTestContext;
+
     use super::*;
     use alacritty_terminal::{
-        grid::GridIterator,
         index::{Column, Line, Point, Side},
         selection::{Selection, SelectionType},
-        term::cell::Cell,
     };
     use gpui::TestAppContext;
-    use itertools::Itertools;
 
-    use std::{path::Path, time::Duration};
+    use std::path::Path;
     use workspace::AppState;
 
+    mod terminal_test_context;
+
     ///Basic integration test, can we get the terminal to show up, execute a command,
     //and produce noticable output?
     #[gpui::test(retries = 5)]
     async fn test_terminal(cx: &mut TestAppContext) {
-        let terminal = cx.add_view(Default::default(), |cx| Terminal::new(None, false, cx));
+        let mut cx = TerminalTestContext::new(cx);
 
-        terminal.update(cx, |terminal, cx| {
-            terminal.connection.update(cx, |connection, _| {
-                connection.write_to_pty("expr 3 + 4".to_string());
-            });
-            terminal.carriage_return(&Return, cx);
-        });
-
-        cx.set_condition_duration(Some(Duration::from_secs(5)));
-        terminal
-            .condition(cx, |terminal, cx| {
-                let term = terminal.connection.read(cx).term.clone();
-                let content = grid_as_str(term.lock().renderable_content().display_iter);
-                content.contains("7")
-            })
+        cx.execute_and_wait("expr 3 + 4", |content, _cx| content.contains("7"))
             .await;
-        cx.set_condition_duration(None);
     }
 
-    /// Integration test for selections, clipboard, and terminal execution
+    /// TODO: I don't think this is actually testing anything anymore.
+    ///Integration test for selections, clipboard, and terminal execution
     #[gpui::test(retries = 5)]
     async fn test_copy(cx: &mut TestAppContext) {
-        let mut result_line: i32 = 0;
-        let terminal = cx.add_view(Default::default(), |cx| Terminal::new(None, false, cx));
-        cx.set_condition_duration(Some(Duration::from_secs(5)));
+        let mut cx = TerminalTestContext::new(cx);
 
-        terminal.update(cx, |terminal, cx| {
-            terminal.connection.update(cx, |connection, _| {
-                connection.write_to_pty("expr 3 + 4".to_string());
-            });
-            terminal.carriage_return(&Return, cx);
-        });
-
-        terminal
-            .condition(cx, |terminal, cx| {
-                let term = terminal.connection.read(cx).term.clone();
-                let content = grid_as_str(term.lock().renderable_content().display_iter);
-
-                if content.contains("7") {
-                    let idx = content.chars().position(|c| c == '7').unwrap();
-                    result_line = content.chars().take(idx).filter(|c| *c == '\n').count() as i32;
-                    true
-                } else {
-                    false
-                }
-            })
+        let grid_content = cx
+            .execute_and_wait("expr 3 + 4", |content, _cx| content.contains("7"))
             .await;
 
-        terminal.update(cx, |terminal, cx| {
-            let mut term = terminal.connection.read(cx).term.lock();
+        //Get the position of the result
+        let idx = grid_content.chars().position(|c| c == '7').unwrap();
+        let result_line = grid_content
+            .chars()
+            .take(idx)
+            .filter(|c| *c == '\n')
+            .count() as i32;
+
+        let copy_res = cx.update_connection(|connection, _cx| {
+            let mut term = connection.term.lock();
             term.selection = Some(Selection::new(
                 SelectionType::Semantic,
-                Point::new(Line(2), Column(0)),
+                Point::new(Line(result_line), Column(0)),
                 Side::Right,
             ));
-            drop(term);
-            terminal.copy(&Copy, cx)
+            term.selection_to_string()
         });
 
-        cx.assert_clipboard_content(Some(&"7"));
-        cx.set_condition_duration(None);
+        assert_eq!(copy_res.unwrap(), "7");
     }
 
     ///Working directory calculation tests
@@ -726,13 +700,4 @@ mod tests {
             assert_eq!(res, Some((Path::new("/root/")).to_path_buf()));
         });
     }
-
-    pub(crate) fn grid_as_str(grid_iterator: GridIterator<Cell>) -> String {
-        let lines = grid_iterator.group_by(|i| i.point.line.0);
-        lines
-            .into_iter()
-            .map(|(_, line)| line.map(|i| i.c).collect::<String>())
-            .collect::<Vec<String>>()
-            .join("\n")
-    }
 }
diff --git a/crates/terminal/src/tests/terminal_test_context.rs b/crates/terminal/src/tests/terminal_test_context.rs
new file mode 100644
index 0000000000..ab0809c3a1
--- /dev/null
+++ b/crates/terminal/src/tests/terminal_test_context.rs
@@ -0,0 +1,82 @@
+use std::time::Duration;
+
+use alacritty_terminal::term::SizeInfo;
+use gpui::{AppContext, ModelContext, ModelHandle, ReadModelWith, TestAppContext};
+use itertools::Itertools;
+
+use crate::{
+    connection::TerminalConnection, DEBUG_CELL_WIDTH, DEBUG_LINE_HEIGHT, DEBUG_TERMINAL_HEIGHT,
+    DEBUG_TERMINAL_WIDTH,
+};
+
+pub struct TerminalTestContext<'a> {
+    pub cx: &'a mut TestAppContext,
+    pub connection: ModelHandle<TerminalConnection>,
+}
+
+impl<'a> TerminalTestContext<'a> {
+    pub fn new(cx: &'a mut TestAppContext) -> Self {
+        cx.set_condition_duration(Some(Duration::from_secs(5)));
+
+        let size_info = SizeInfo::new(
+            DEBUG_TERMINAL_WIDTH,
+            DEBUG_TERMINAL_HEIGHT,
+            DEBUG_CELL_WIDTH,
+            DEBUG_LINE_HEIGHT,
+            0.,
+            0.,
+            false,
+        );
+
+        let connection = cx.add_model(|cx| TerminalConnection::new(None, size_info, cx));
+
+        TerminalTestContext { cx, connection }
+    }
+
+    pub async fn execute_and_wait<F>(&mut self, command: &str, f: F) -> String
+    where
+        F: Fn(String, &AppContext) -> bool,
+    {
+        let command = command.to_string();
+        self.connection.update(self.cx, |connection, _| {
+            connection.write_to_pty(command);
+            connection.write_to_pty("\r".to_string());
+        });
+
+        self.connection
+            .condition(self.cx, |conn, cx| {
+                let content = Self::grid_as_str(conn);
+                f(content, cx)
+            })
+            .await;
+
+        self.cx
+            .read_model_with(&self.connection, &mut |conn, _: &AppContext| {
+                Self::grid_as_str(conn)
+            })
+    }
+
+    pub fn update_connection<F, S>(&mut self, f: F) -> S
+    where
+        F: FnOnce(&mut TerminalConnection, &mut ModelContext<TerminalConnection>) -> S,
+    {
+        self.connection.update(self.cx, |conn, cx| f(conn, cx))
+    }
+
+    fn grid_as_str(connection: &TerminalConnection) -> String {
+        let term = connection.term.lock();
+        let grid_iterator = term.renderable_content().display_iter;
+        let lines = grid_iterator.group_by(|i| i.point.line.0);
+        lines
+            .into_iter()
+            .map(|(_, line)| line.map(|i| i.c).collect::<String>())
+            .collect::<Vec<String>>()
+            .join("\n")
+    }
+}
+
+impl<'a> Drop for TerminalTestContext<'a> {
+    fn drop(&mut self) {
+        self.cx.set_condition_duration(None);
+    }
+}