Remove dangling commas from generated tests

2013-02-11 16:08:05 +04:00
parent 9dedc6d2a9
commit 9f1e808255
3 changed files with 12 additions and 7 deletions
@@ -213,13 +213,18 @@ public class TestGenerator {
        Collection<TestClassModel> innerTestClasses = testClassModel.getInnerTestClasses();
        if (innerTestClasses.isEmpty()) return;
        p.print("@InnerTestClasses({");
-        for (Iterator<TestClassModel> iterator = innerTestClasses.iterator(); iterator.hasNext(); ) {
-            TestClassModel innerTestClass = iterator.next();
+
+        boolean isFirst = true;
+        for (TestClassModel innerTestClass : innerTestClasses) {
            if (!innerTestClass.isEmpty()) {
-                p.printWithNoIndent(testClassModel.getName(), ".", innerTestClass.getName(), ".class");
-                if (iterator.hasNext()) {
+                if (!isFirst) {
                    p.printWithNoIndent(", ");
                }
+                else {
+                    isFirst = false;
+                }
+
+                p.printWithNoIndent(testClassModel.getName(), ".", innerTestClass.getName(), ".class");
            }
        }
        p.printlnWithNoIndent("})");