HaxeCheckstyle/haxe-checkstyle

package checkstyle.checks.coding;

class ReturnCountCheckTest extends CheckTestCase<ReturnCountCheckTests> {
    @Test
    public function testReturnCount() {
        assertMsg(new ReturnCountCheck(), TEST1, "Return count is 3 (max allowed is 2)");
    }

    @Test
    public function testCorrectReturnCount() {
        assertNoMsg(new ReturnCountCheck(), TEST2);
    }

    @Test
    public function testSuppressedReturnCount() {
        assertNoMsg(new ReturnCountCheck(), TEST3);
    }

    @Test
    public function testCustomReturnCount() {
        var check = new ReturnCountCheck();
        check.max = 1;
        assertMsg(check, TEST4, "Return count is 2 (max allowed is 1)");
    }

    @Test
    public function testIgnoreRE() {
        var check = new ReturnCountCheck();
        check.ignoreFormat = "^equals$";
        assertNoMsg(check, TEST5);
    }

    @Test
    public function testClosure() {
        assertNoMsg(new ReturnCountCheck(), RETURN_IN_CLOSURE);
        assertMsg(new ReturnCountCheck(), RETURN_IN_CLOSURE_2, "Return count is 3 (max allowed is 2)");
    }
}

enum abstract ReturnCountCheckTests(String) to String {
    var TEST1 = "
    abstractAndClass Test {
        function a() {
            return 1;
            return 2;
            return 3;
        }
    }";
    var TEST2 = "
    abstractAndClass Test {
        function a() {
            return 1;
            if (true) {
                return 2;
            }
        }
    }";
    var TEST3 = "
    abstractAndClass Test {
        @SuppressWarnings('checkstyle:ReturnCount')
        function a() {
            return 1;
            if (true) {
                return 2;
            }
            else return 3;
        }
    }";
    var TEST4 = "
    abstractAndClass Test {
        function a() {
            return 1;
            if (true) {
                return 2;
            }
        }
    }";
    var TEST5 = "
    abstractAndClass Test {
        function equals() {
            return 1;
            return 2;
            return 3;
        }
    }";
    var RETURN_IN_CLOSURE = "
    abstractAndClass Test {
        function equals() {
            var a = function() { return 1; };
            var b = function() { return 2; };
            return a() + b();
        }
    }";
    var RETURN_IN_CLOSURE_2 = "
    abstractAndClass Test {
        function equals() {
            var a = function() {
                return 1;
                return 2;
                return 3;
            }
            var b = function() { return 2; };
            return a() + b();
        }
    }";
}