Properly patch package name in case of file annotations on default package

2016-04-22 18:37:31 +03:00
parent 797465a06b
commit 6228b11048
1 changed files with 8 additions and 1 deletions
@@ -244,7 +244,14 @@ public class CodegenTestsOnAndroidGenerator extends UsefulTestCase {
            return matcher.replaceAll("package " + testName);
        }
        else {
-            return "package " + testName + ";\n" + text;
+            String packageDirective = "package " + testName + ";\n";
+            if (text.contains("@file:")) {
+                int index = text.lastIndexOf("@file:");
+                int packageDirectiveIndex = text.indexOf("\n", index);
+                return text.substring(0, packageDirectiveIndex + 1) + packageDirective + text.substring(packageDirectiveIndex + 1);
+            } else {
+                return packageDirective + text;
+            }
        }
    }