allow case with no comment stating where to put the code

pattern-finder · Jul 28, 2021 · eb60478 · eb60478
1 parent 6ef831b
commit eb60478
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/exec-server/godbox.repository.ts b/src/exec-server/godbox.repository.ts
@@ -89,7 +89,9 @@ export class GodBoxRepository {
   ): Promise<GodboxPhaseOutputDTO> {
     const language = await this.languagesService.findByName(bootstrap.language);
 
-    const completeCode = bootstrap.tests.replace('// USER_CODE', code);
+    const completeCode = bootstrap.tests.includes('// USER_CODE')
+      ? bootstrap.tests.replace('// USER_CODE', code)
+      : `${bootstrap.tests}\n${code}`;
 
     const payload = {
       phases: language.phases,