Project test stub updates

bcgov · Nov 22, 2024 · 5b50ef2 · 5b50ef2
1 parent 2611135
commit 5b50ef2
Showing 1 changed file with 28 additions and 1 deletion.
diff --git a/server/wfprev-api/src/test/java/ca/bc/gov/nrs/wfprev/ProjectControllerTest.java b/server/wfprev-api/src/test/java/ca/bc/gov/nrs/wfprev/ProjectControllerTest.java
@@ -16,6 +16,8 @@
 import org.springframework.test.web.servlet.MockMvc;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.put;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.delete;
 import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
 
 import ca.bc.gov.nrs.wfprev.controllers.ProjectController;
@@ -57,12 +59,37 @@ void testGetProject() throws Exception {
 
   @Test
   @WithMockUser
-  void testCreateProject() throws Exception {
+  void testCreateUpdateProject() throws Exception {
     ProjectModel project = new ProjectModel();
     when(projectService.createOrUpdateProject(project)).thenReturn(project);
 
     mockMvc.perform(post("/projects", project)
            .contentType(MediaType.APPLICATION_JSON))
            .andExpect(status().isCreated());
+
+
+    project.setClosestCommunityName("Test");
+    when(projectService.createOrUpdateProject(project)).thenReturn(project);
+
+    mockMvc.perform(put("/projects", project)
+           .contentType(MediaType.APPLICATION_JSON))
+           .andExpect(status().isCreated());
+  }
+
+  @Test
+  @WithMockUser
+  void testDeleteProject() throws Exception {
+    ProjectModel project = new ProjectModel();
+    when(projectService.createOrUpdateProject(project)).thenReturn(project);
+
+    mockMvc.perform(post("/projects", project)
+           .contentType(MediaType.APPLICATION_JSON))
+           .andExpect(status().isCreated());
+
+    when(projectService.deleteProject(project.getProjectGuid())).thenReturn(null);
+
+    mockMvc.perform(delete("/projects/{id}", project.getProjectGuid())
+           .contentType(MediaType.APPLICATION_JSON))
+           .andExpect(status().isOk());
   }
 }