Merge pull request #1461 from qmonmert/fix/1452-v2

Download: only JHipster Lite project
jhipster · Apr 24, 2022 · 3584edc · 3584edc
2 parents 926d1d5 + 09b600d
commit 3584edc
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/src/main/java/tech/jhipster/lite/generator/init/domain/InitDomainService.java b/src/main/java/tech/jhipster/lite/generator/init/domain/InitDomainService.java
@@ -129,6 +129,9 @@ public void gitInit(Project project) {
 
   @Override
   public byte[] download(Project project) {
-    return projectRepository.download(project);
+    if (projectRepository.isJHipsterLiteProject(project.getFolder())) {
+      return projectRepository.download(project);
+    }
+    throw new GeneratorException("This project is not a JHipster Lite project");
   }
 }
diff --git a/src/test/java/tech/jhipster/lite/generator/init/domain/InitDomainServiceTest.java b/src/test/java/tech/jhipster/lite/generator/init/domain/InitDomainServiceTest.java
@@ -113,6 +113,17 @@ void shouldAddPrettier() {
   void shouldDownload() {
     Project project = tmpProjectWithPomXml();
 
+    when(projectRepository.isJHipsterLiteProject(anyString())).thenReturn(true);
+
     assertThatCode(() -> initDomainService.download(project)).doesNotThrowAnyException();
   }
+
+  @Test
+  void shouldNotDownload() {
+    Project project = tmpProjectWithPomXml();
+
+    when(projectRepository.isJHipsterLiteProject(anyString())).thenReturn(false);
+
+    assertThatThrownBy(() -> initDomainService.download(project)).isExactlyInstanceOf(GeneratorException.class);
+  }
 }