1 files changed, 35 insertions, 0 deletions

diff --git a/gui/tests/layout.rs b/gui/tests/layout.rs
index ab5a62e..5678665 100644
--- a/gui/tests/layout.rs
+++ b/gui/tests/layout.rs
@@ -5,8 +5,10 @@
 use std::f32::consts::TAU;
 
 use eframe::egui::{Pos2, Rect, pos2, vec2};
+use zagan_bet::client::view::{SlotsPhase, SlotsView};
 use zagan_bet::roulette::{BetTarget, Outside};
 use zagan_bet_gui::roulette::{Felt, number_under, size_for, target_name};
+use zagan_bet_gui::slots::winning_columns;
 use zagan_bet_gui::theme::{group_digits, money};
 
 #[test]
@@ -154,3 +156,36 @@ fn targets_read_as_bet_slips() {
         "Black"
     );
 }
+
+fn slots_view(win_cells: Vec<usize>, win_lines: Vec<usize>) -> SlotsView {
+    SlotsView {
+        bet: 10,
+        balance: Some(1000),
+        reels: "OOOOODDDDDOOOOOD".chars().collect(),
+        win_cells,
+        win_lines,
+        win_type: "win".to_string(),
+        winnings: Some(30),
+        phase: SlotsPhase::Result,
+        error: None,
+    }
+}
+
+#[test]
+fn a_payline_stops_at_the_last_matching_reel() {
+    // Middle line (line 2), four Os from reel 0: cells 1, 4, 7, 10.
+    let view = slots_view(vec![1, 4, 7, 10], vec![2]);
+    assert_eq!(winning_columns(&view, 2), vec![0, 1, 2, 3]);
+
+    // Three of a kind stops after reel 2.
+    let view = slots_view(vec![1, 4, 7], vec![2]);
+    assert_eq!(winning_columns(&view, 2), vec![0, 1, 2]);
+
+    // Cells on another line do not extend this one.
+    let view = slots_view(vec![0, 1, 4, 7], vec![2]);
+    assert_eq!(winning_columns(&view, 2), vec![0, 1, 2]);
+
+    // A line number that does not exist draws nothing.
+    let view = slots_view(vec![1, 4, 7], vec![9]);
+    assert!(winning_columns(&view, 9).is_empty());
+}