|
18 | 18 | package org.openapitools.codegen.scalaakka;
|
19 | 19 |
|
20 | 20 | import com.google.common.collect.Sets;
|
| 21 | +import com.google.common.io.Resources; |
21 | 22 | import io.swagger.v3.oas.models.OpenAPI;
|
22 | 23 | import io.swagger.v3.oas.models.media.*;
|
23 | 24 | import io.swagger.v3.parser.util.SchemaTypeUtil;
|
|
28 | 29 | import org.testng.annotations.Test;
|
29 | 30 |
|
30 | 31 | import java.io.File;
|
| 32 | +import java.nio.charset.StandardCharsets; |
31 | 33 | import java.nio.file.Files;
|
32 | 34 | import java.util.HashMap;
|
33 | 35 | import java.util.Map;
|
@@ -303,28 +305,8 @@ public void codeGenerationTest() throws Exception {
|
303 | 305 | Assert.assertEquals(generatedFiles.size(), 13);
|
304 | 306 |
|
305 | 307 | final String someObjFilename = new File(output, "src/main/scala/hello/world/model/SomeObj.scala").getAbsolutePath().replace("\\", "/");
|
306 |
| - final String someObjFileContents = generatedFiles.get(someObjFilename); |
307 |
| - Assert.assertTrue(someObjFileContents.contains("package hello.world.model")); |
308 |
| - Assert.assertTrue(someObjFileContents.contains("case class SomeObj")); |
309 |
| - Assert.assertTrue(someObjFileContents.contains("id: Long,")); |
310 |
| - Assert.assertTrue(someObjFileContents.contains("name: Option[String] = None,")); |
311 |
| - Assert.assertTrue(someObjFileContents.contains("`val`: Option[String] = None,")); |
312 |
| - Assert.assertTrue(someObjFileContents.contains("`var`: Option[String] = None,")); |
313 |
| - Assert.assertTrue(someObjFileContents.contains("`class`: Option[String] = None,")); |
314 |
| - Assert.assertTrue(someObjFileContents.contains("`trait`: Option[String] = None,")); |
315 |
| - Assert.assertTrue(someObjFileContents.contains("`object`: Option[String] = None,")); |
316 |
| - Assert.assertTrue(someObjFileContents.contains("`try`: String,")); |
317 |
| - Assert.assertTrue(someObjFileContents.contains("`catch`: String,")); |
318 |
| - Assert.assertTrue(someObjFileContents.contains("`finally`: String,")); |
319 |
| - Assert.assertTrue(someObjFileContents.contains("`def`: Option[String] = None,")); |
320 |
| - Assert.assertTrue(someObjFileContents.contains("`for`: Option[String] = None,")); |
321 |
| - Assert.assertTrue(someObjFileContents.contains("`implicit`: Option[String] = None,")); |
322 |
| - Assert.assertTrue(someObjFileContents.contains("`match`: Option[String] = None,")); |
323 |
| - Assert.assertTrue(someObjFileContents.contains("`case`: Option[String] = None,")); |
324 |
| - Assert.assertTrue(someObjFileContents.contains("`import`: Option[String] = None,")); |
325 |
| - Assert.assertTrue(someObjFileContents.contains("`lazy`: String,")); |
326 |
| - Assert.assertTrue(someObjFileContents.contains("`private`: Option[String] = None,")); |
327 |
| - Assert.assertTrue(someObjFileContents.contains("`type`: Option[String] = None,")); |
328 |
| - Assert.assertTrue(someObjFileContents.contains("foobar: Boolean")); |
| 308 | + Assert.assertEquals( |
| 309 | + generatedFiles.get(someObjFilename), |
| 310 | + Resources.toString(Resources.getResource("codegen/scala/SomeObj.scala.txt"), StandardCharsets.UTF_8)); |
329 | 311 | }
|
330 | 312 | }
|
0 commit comments