1 files changed, 63 insertions, 2 deletions

diff --git a/tests/client_ui.rs b/tests/client_ui.rs
index 78faed0..a22fbe7 100644
--- a/tests/client_ui.rs
+++ b/tests/client_ui.rs
@@ -1,7 +1,8 @@
 use zagan_bet::client::ui::{
-    command_names, completion_lines, help_lines, input_cursor_column, lines_to_move_up,
-    parse_bet_arg, record_history,
+    command_names, completion_lines, display_width, first_contact_lines, format_pong,
+    help_lines, host_lines, input_cursor_column, lines_to_move_up, parse_bet_arg, record_history,
 };
+use std::time::Duration;
 
 #[test]
 fn help_has_no_source_indentation() {
@@ -129,3 +130,63 @@ fn completions_wrap_within_the_requested_width() {
         );
     }
 }
+
+#[test]
+fn display_width_counts_terminal_columns_not_bytes() {
+    assert_eq!(display_width("abc"), 3);
+    assert_eq!(display_width("ąćęłńóśźż"), 9);
+    assert_eq!(display_width("中"), 2);
+    assert_eq!(display_width("🎉"), 2);
+    assert_eq!(display_width("e\u{0301}"), 1);
+}
+
+#[test]
+fn cursor_column_uses_display_width() {
+    assert_eq!(input_cursor_column(2, display_width("ąć")), 4);
+    assert_eq!(input_cursor_column(2, display_width("中")), 4);
+}
+
+#[test]
+fn first_contact_message_stays_within_80_columns() {
+    let fingerprint = "a4d6f80784154c2cb739a1df61ab3e9038ed731ee2e05c7611457ac741b3834c";
+    let lines = first_contact_lines("127.0.0.1:11525", fingerprint);
+
+    assert!(lines.iter().any(|line| line.contains("127.0.0.1:11525")));
+    assert!(lines.iter().any(|line| line.contains(fingerprint)));
+    for line in lines {
+        assert!(
+            line.len() <= 80,
+            "first-contact line is too wide ({}): {line:?}",
+            line.len()
+        );
+    }
+}
+
+#[test]
+fn pong_is_formatted_to_one_decimal() {
+    assert_eq!(format_pong(Duration::from_micros(144)), "Pong (0.1ms)");
+    assert_eq!(format_pong(Duration::from_millis(41)), "Pong (41.0ms)");
+}
+
+#[test]
+fn host_lines_fit_an_80_column_terminal() {
+    let fingerprint = "a4d6f80784154c2cb739a1df61ab3e9038ed731ee2e05c7611457ac741b3834c";
+    let lines = host_lines(&[
+        (
+            "127.0.0.1:11525".to_string(),
+            Some(fingerprint.to_string()),
+        ),
+        ("unknown.example:443".to_string(), None),
+    ]);
+
+    assert!(lines.iter().any(|line| line.contains("127.0.0.1:11525")));
+    assert!(lines.iter().any(|line| line.contains(fingerprint)));
+    assert!(lines.iter().any(|line| line.contains("no fingerprint")));
+    for line in lines {
+        assert!(
+            line.len() <= 80,
+            "host line is too wide ({}): {line:?}",
+            line.len()
+        );
+    }
+}