import XCTest
@testable import Minesweeper

class MinesweeperTests: XCTestCase {
    func testTransform1() {
        let input = ["+------+", "| *  * |", "|  *   |", "|    * |", "|   * *|",
                     "| *  * |", "|      |", "+------+"]
        let output = ["+------+", "|1*22*1|", "|12*322|", "| 123*2|", "|112*4*|",
                      "|1*22*2|", "|111111|", "+------+"]
        XCTAssertEqual(output, try! Board(input).transform())
    }

    func testTransform2() {
        let input = ["+-----+", "| * * |", "|     |", "|   * |", "|  * *|",
                     "| * * |", "+-----+"]
        let output = ["+-----+", "|1*2*1|", "|11322|", "| 12*2|", "|12*4*|",
                      "|1*3*2|", "+-----+"]
        XCTAssertEqual(output, try! Board(input).transform())
    }

    func testTransform3() {
        let input = ["+-----+", "| * * |", "+-----+"]
        let output = ["+-----+", "|1*2*1|", "+-----+"]
        XCTAssertEqual(output, try! Board(input).transform())
    }

    func test_transform4() {
        let input = ["+-+", "|*|", "| |", "|*|", "| |", "| |", "+-+"]
        let output = ["+-+", "|*|", "|2|", "|*|", "|1|", "| |", "+-+"]
        XCTAssertEqual(output, try! Board(input).transform())
    }

    func testTransform5() {
        let input = ["+-+", "|*|", "+-+"]
        let output = ["+-+", "|*|", "+-+"]
        XCTAssertEqual(output, try! Board(input).transform())
    }

    func testTransform6() {
        let input = ["+--+", "|**|", "|**|", "+--+"]
        let output = ["+--+", "|**|", "|**|", "+--+"]
        XCTAssertEqual(output, try! Board(input).transform())
    }

    func testTransform7() {
        let input = ["+--+", "|**|", "|**|", "+--+"]
        let output = ["+--+", "|**|", "|**|", "+--+"]
        XCTAssertEqual(output, try! Board(input).transform())
    }

    func testTransform8() {
        let input = ["+---+", "|***|", "|* *|", "|***|", "+---+"]
        let output = ["+---+", "|***|", "|*8*|", "|***|", "+---+"]
        XCTAssertEqual(output, try! Board(input).transform())
    }

    func testTransform9() {
        let input = ["+-----+", "|     |", "|   * |", "|     |", "|     |",
                     "| *   |", "+-----+"]
        let output = ["+-----+", "|  111|", "|  1*1|", "|  111|", "|111  |",
                      "|1*1  |", "+-----+"]
        XCTAssertEqual(output, try! Board(input).transform())
    }

    func testDifferentLength() {
        let input = ["+-+", "| |", "|*  |", "|  |", "+-+"]

        XCTAssertThrowsError(_ = try Board(input)) { error in
            XCTAssertEqual(error as? Board.BoardError, .differentLength)
        }
    }

    func testFaultyBorder() {
        let input = ["+-----+", "*   * |", "+-- --+"]

        XCTAssertThrowsError(_ = try Board(input)) { error in
            XCTAssertEqual(error as? Board.BoardError, .faultyBorder)
        }
    }

    func testInvalidCharacter() {
        let input = ["+-----+", "|X  * |", "+-----+"]

        XCTAssertThrowsError(_ = try Board(input)) { error in
            XCTAssertEqual(error as? Board.BoardError, .invalidCharacter)
        }
    }

    static var allTests: [(String, (MinesweeperTests) -> () throws -> Void)] {
        return [
            ("testTransform1", testTransform1),
            ("testTransform2", testTransform2),
            ("testTransform3", testTransform3),
            ("test_transform4", test_transform4),
            ("testTransform5", testTransform5),
            ("testTransform6", testTransform6),
            ("testTransform7", testTransform7),
            ("testTransform8", testTransform8),
            ("testTransform9", testTransform9),
            ("testDifferentLength", testDifferentLength),
            ("testFaultyBorder", testFaultyBorder),
            ("testInvalidCharacter", testInvalidCharacter),
        ]
    }
}