|
29 | 29 | import com.azure.autorest.fluent.template.ResourceManagerUtilsTemplate; |
30 | 30 | import com.azure.autorest.model.javamodel.JavaFile; |
31 | 31 | import com.azure.autorest.model.javamodel.JavaPackage; |
| 32 | +import com.azure.autorest.util.ClassNameUtil; |
32 | 33 | import com.azure.autorest.util.CodeNamer; |
33 | 34 |
|
34 | 35 | import java.util.List; |
@@ -111,11 +112,22 @@ public final JavaFile addSample(FluentExample example) { |
111 | 112 | } |
112 | 113 |
|
113 | 114 | public void addOperationUnitTest(FluentMethodMockUnitTest unitTest) { |
114 | | - |
| 115 | + final String packageName = JavaSettings.getInstance().getPackage("generated"); |
115 | 116 | String className = unitTest.getResourceCollection().getInterfaceType().getName() |
116 | | - + CodeNamer.toPascalCase(unitTest.getCollectionMethod().getMethodName()) |
117 | | - + "MockTests"; |
118 | | - JavaFile javaFile = getJavaFileFactory().createTestFile(JavaSettings.getInstance().getPackage("generated"), className); |
| 117 | + + CodeNamer.toPascalCase(unitTest.getCollectionMethod().getMethodName()); |
| 118 | + |
| 119 | + final String classNameSuffix = "MockTests"; |
| 120 | + |
| 121 | + className = ClassNameUtil.truncateClassName( |
| 122 | + JavaSettings.getInstance().getPackage(), |
| 123 | + "src/tests/java" |
| 124 | + // a hack to count "MockTests" suffix into the length of the full path |
| 125 | + + classNameSuffix, |
| 126 | + packageName, className); |
| 127 | + |
| 128 | + className += classNameSuffix; |
| 129 | + |
| 130 | + JavaFile javaFile = getJavaFileFactory().createTestFile(packageName, className); |
119 | 131 | FluentMethodMockTestTemplate.ClientMethodInfo info = new FluentMethodMockTestTemplate.ClientMethodInfo( |
120 | 132 | className, unitTest); |
121 | 133 | FluentMethodMockTestTemplate.getInstance().write(info, javaFile); |
|
0 commit comments