/**

  * The MIT License (MIT)

  *

  * Copyright (C) 2013 Jaroslav Tulach <jaroslav.tulach@apidesign.org>

  *

  * Permission is hereby granted, free of charge, to any person obtaining a copy

  * of this software and associated documentation files (the "Software"), to deal

  * in the Software without restriction, including without limitation the rights

  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

  * copies of the Software, and to permit persons to whom the Software is

  * furnished to do so, subject to the following conditions:

  *

  * The above copyright notice and this permission notice shall be included in

  * all copies or substantial portions of the Software.

  *

  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN

  * THE SOFTWARE.

  */

 package org.apidesign.demo.minesweeper;

 import org.apidesign.demo.minesweeper.MinesModel.SquareType;

 import static org.testng.Assert.*;

 import org.testng.annotations.Test;

 public class MinesModelTest {

     @Test public void tenTenTen() {

         Mines m = new Mines();

         m.init(10, 10, 10);

         assertEquals(m.getRows().size(), 10, "Ten rows");

         int cnt = 0;

         for (Row row : m.getRows()) {

             assertEquals(row.getColumns().size(), 10, "Ten columns in each row");

             for (Square square : row.getColumns()) {

                 if (square.isMine()) {

                     cnt++;

                 }

             }

         }

         assertEquals(cnt, 10, "Ten mines");

     }

     @Test public void clickRemovesMarkedSign() {

         Mines m = new Mines();

         m.init(10, 10, 10);

         final Square sq = m.getRows().get(5).getColumns().get(5);

         MinesModel.markMine(m);

         MinesModel.click(m, sq);

         assertEquals(sq.getState(), SquareType.MARKED, "Changed to marked");

         MinesModel.click(m, sq);

         assertEquals(sq.getState(), SquareType.UNKNOWN, "Changed back to unknown");

     }

     @Test public void gameWonWhenAllMarked() {

         Mines m = new Mines();

         m.init(10, 10, 10);

         for (Row row : m.getRows()) {

             for (Square sq : row.getColumns()) {

                 if (sq.isMine()) {

                     MinesModel.markMine(m);

                     MinesModel.click(m, sq);

                 }

             }

         }

         assertEquals(m.getState(), MinesModel.GameState.WON, "All mines found. You have won!");

     }

     @Test public void gameNotWonWhenTooMuchIsMarked() {

         Mines m = new Mines();

         m.init(10, 10, 10);

         Square additional = null;

         for (Row row : m.getRows()) {

             for (Square sq : row.getColumns()) {

                 if (sq.isMine()) {

                     MinesModel.markMine(m);

                     MinesModel.click(m, sq);

                 } else if (additional == null) {

                     MinesModel.markMine(m);

                     MinesModel.click(m, additional = sq);

                 }

             } 

         }

         assertEquals(m.getState(), MinesModel.GameState.IN_PROGRESS, "One additional mine is marked!");

         // remove the mark

         MinesModel.click(m, additional);

         assertEquals(m.getState(), MinesModel.GameState.WON, "All mines found. You have won!");

     }

     @Test public void bombsSet() {

         Mines m = new Mines();

         m.init(10, 10, 0);

         set(m, 1, 1, SquareType.UNKNOWN, true);

         set(m, 0, 0, SquareType.N_0, false);

         set(m, 0, 1, SquareType.N_2, false);

         set(m, 0, 2, SquareType.N_3, false);

         set(m, 1, 0, SquareType.N_5, false);

         set(m, 2, 0, SquareType.N_8, false);

         set(m, 3, 0, SquareType.N_2, false);

         m.computeMines();

         assertSquare(m, 0, 0, SquareType.N_1);

         assertSquare(m, 1, 0, SquareType.N_1);

         assertSquare(m, 2, 0, SquareType.N_1);

         assertSquare(m, 3, 0, SquareType.N_0);

         assertSquare(m, 0, 1, SquareType.N_1);

         assertSquare(m, 0, 2, SquareType.N_1);

         assertSquare(m, 2, 2, SquareType.UNKNOWN);

     }

     @Test public void gameIsWonIfAllMinesDiscovered() {

         Mines m = new Mines();

         m.init(2, 1, 0);

         set(m, 0, 0, SquareType.UNKNOWN, true);

         m.computeMines();

         assertEquals(m.getState(), MinesModel.GameState.IN_PROGRESS);

         set(m, 1, 0, SquareType.N_0, false);

         m.computeMines();

         assertEquals(m.getState(), MinesModel.GameState.WON, "All non-bomb squares discovered");

     }

     @Test public void unhideNeibourghsOfEmptyPieces() {

         Mines m = new Mines();

         m.init(3, 3, 0);

         set(m, 0, 0, SquareType.UNKNOWN, true);

         MinesModel.click(m, m.getRows().get(2).getColumns().get(2));

         assertSquare(m, 0, 0, SquareType.DISCOVERED);

         assertSquare(m, 0, 1, SquareType.N_1);

         assertSquare(m, 1, 1, SquareType.N_1);

         assertSquare(m, 1, 0, SquareType.N_1);

         assertSquare(m, 2, 0, SquareType.N_0);

         assertSquare(m, 2, 1, SquareType.N_0);

         assertSquare(m, 2, 2, SquareType.N_0);

         assertSquare(m, 1, 2, SquareType.N_0);

     }

     private static void set(Mines m, int x, int y, SquareType squareType, boolean mine) {

         Square sq = m.getRows().get(y).getColumns().get(x);

         sq.setState(squareType);

         sq.setMine(mine);

     }

     private static void assertSquare(Mines m, int x, int y, SquareType t) {

         Square sq = m.getRows().get(y).getColumns().get(x);

         assertEquals(sq.getState(), t, "Expecting at " + x + ":" + y);

     }

 }