From 1f52bb93465f27ab4fbc991c5aee76e97023c115 Mon Sep 17 00:00:00 2001
From: Bruno BELANYI <bruno@belanyi.fr>
Date: Mon, 1 Apr 2024 21:56:01 +0100
Subject: [PATCH] Add tests for en-passant validation

---
 src/board/chess_board/mod.rs | 50 ++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)

diff --git a/src/board/chess_board/mod.rs b/src/board/chess_board/mod.rs
index e4cae1f..2a98537 100644
--- a/src/board/chess_board/mod.rs
+++ b/src/board/chess_board/mod.rs
@@ -588,6 +588,56 @@ mod test {
         assert_eq!(res.err().unwrap(), InvalidError::InvalidCastlingRights);
     }
 
+    #[test]
+    fn valid_en_passant() {
+        let mut builder = ChessBoardBuilder::new();
+        builder[Square::E1] = Some((Piece::King, Color::White));
+        builder[Square::E8] = Some((Piece::King, Color::Black));
+        builder[Square::A5] = Some((Piece::Pawn, Color::Black));
+        builder.with_en_passant(Square::A6);
+        TryInto::<ChessBoard>::try_into(builder).unwrap();
+    }
+
+    #[test]
+    fn invalid_en_passant_not_empty() {
+        let res = {
+            let mut builder = ChessBoardBuilder::new();
+            builder[Square::E1] = Some((Piece::King, Color::White));
+            builder[Square::E8] = Some((Piece::King, Color::Black));
+            builder[Square::A6] = Some((Piece::Rook, Color::Black));
+            builder[Square::A5] = Some((Piece::Pawn, Color::Black));
+            builder.with_en_passant(Square::A6);
+            TryInto::<ChessBoard>::try_into(builder)
+        };
+        assert_eq!(res.err().unwrap(), InvalidError::InvalidEnPassant);
+    }
+
+    #[test]
+    fn invalid_en_passant_not_behind_pawn() {
+        let res = {
+            let mut builder = ChessBoardBuilder::new();
+            builder[Square::E1] = Some((Piece::King, Color::White));
+            builder[Square::E8] = Some((Piece::King, Color::Black));
+            builder[Square::A5] = Some((Piece::Rook, Color::Black));
+            builder.with_en_passant(Square::A6);
+            TryInto::<ChessBoard>::try_into(builder)
+        };
+        assert_eq!(res.err().unwrap(), InvalidError::InvalidEnPassant);
+    }
+
+    #[test]
+    fn invalid_en_passant_incorrect_rank() {
+        let res = {
+            let mut builder = ChessBoardBuilder::new();
+            builder[Square::E1] = Some((Piece::King, Color::White));
+            builder[Square::E8] = Some((Piece::King, Color::Black));
+            builder[Square::A4] = Some((Piece::Pawn, Color::Black));
+            builder.with_en_passant(Square::A5);
+            TryInto::<ChessBoard>::try_into(builder)
+        };
+        assert_eq!(res.err().unwrap(), InvalidError::InvalidEnPassant);
+    }
+
     #[test]
     fn invalid_kings_next_to_each_other() {
         let res = {