JUnit_TestCases_for_Controllers

src/test/java/com/nisum/mytime/controllertest/EmployeeBillingControllerTest.java

+package com.nisum.mytime.controllertest;
+
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
+
+import java.util.ArrayList;
+import java.util.Date;
+import java.util.List;
+
+import org.bson.types.ObjectId;
+import org.junit.Test;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.springframework.http.MediaType;
+import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.result.MockMvcResultMatchers;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.nisum.mytime.controller.ProjectTeamController;
+import com.nisum.mytime.model.BillingDetails;
+import com.nisum.mytime.model.EmployeeDashboardVO; 
+import com.nisum.mytime.repository.EmployeeVisaRepo;
+import com.nisum.mytime.service.ProjectService;
+import com.nisum.mytime.service.UserService;
+
+public class EmployeeBillingControllerTest {
+	
+    @Mock
+    UserService userService;
+
+    @Mock
+    ProjectService projectService;
+    
+    @Mock
+    EmployeeVisaRepo employeeVisaRepo;
+    
+    @InjectMocks
+    ProjectTeamController projectTeamController;
+
+    private MockMvc mockMvc;
+    
+    @Test
+    public void testgetEmployeeBillingDetails() throws Exception {
+        List<BillingDetails> billings = CreateTeamMateBilling();
+        when(projectService.getEmployeeBillingDetailsAll("16167")).thenReturn(billings);
+        mockMvc.perform(get("/projectTeam/getProjectAllocations").param("employeeId", "16167"))
+                .andExpect(MockMvcResultMatchers.status().isOk());
+        verify(projectService).getEmployeeBillingDetailsAll("16167");
+    }
+
+    @Test
+    public void testaddEmployeeBilling() throws Exception {
+              BillingDetails billingDetails = new BillingDetails(
+                new ObjectId("9976ef15874c902c98b8a05d"), "16389","Mahesh Mudrakola", "Acc001","Pro001","Macys",
+                "Y", new Date(2017 - 11 - 29),
+                new Date(2017 - 12 - 20),"Comments",true,new Date(2017 - 11 - 29));
+        String jsonvalue = (new ObjectMapper())
+                .writeValueAsString(billingDetails).toString();
+        when(projectService.addEmployeeBillingDetails(billingDetails))
+                .thenReturn(billingDetails);
+        mockMvc.perform(post("/projectTeam/addEmployeeBilling")
+                .contentType(MediaType.APPLICATION_JSON_VALUE)
+                .content(jsonvalue))
+                .andExpect(MockMvcResultMatchers.status().isOk());
+    }
+    
+    @Test
+    public void testupdateEmployeeBilling() throws Exception {
+              BillingDetails billingDetails = new BillingDetails(
+                new ObjectId("9976ef15874c902c98b8a05d"), "16389","Mahesh Mudrakola", "Acc001","Pro001","Macys",
+                "Y", new Date(2017 - 11 - 29),
+                new Date(2017 - 12 - 20),"Comments",true,new Date(2017 - 11 - 29));
+        String jsonvalue = (new ObjectMapper())
+                .writeValueAsString(billingDetails).toString();
+        when(projectService.updateEmployeeBilling(billingDetails))
+                .thenReturn(billingDetails);
+        mockMvc.perform(post("/projectTeam/updateEmployeeBilling")
+                .contentType(MediaType.APPLICATION_JSON_VALUE)
+                .content(jsonvalue))
+                .andExpect(MockMvcResultMatchers.status().isOk());
+    }
+    
+    @Test
+    public void testdeleteTeammate() throws Exception {
+        BillingDetails billingDetails = new BillingDetails(
+                new ObjectId("9976ef15874c902c98b8a05d"), "16389","Mahesh Mudrakola", "Acc001","Pro001","Macys",
+                "Y", new Date(2017 - 11 - 29),
+                new Date(2017 - 12 - 20),"Comments",true,new Date(2017 - 11 - 29));
+        String jsonvalue = (new ObjectMapper())
+                .writeValueAsString(billingDetails).toString();
+        mockMvc.perform(post("/projectTeam/deleteEmployeeBilling")
+                .contentType(MediaType.APPLICATION_JSON_VALUE)
+                .content(jsonvalue))
+                .andExpect(MockMvcResultMatchers.status().isOk());
+        verify(projectService).deleteEmployeeBilling(billingDetails);
+    }
+
+    @Test
+    public void testgetEmployeesDashBoard() throws Exception {
+    	List<EmployeeDashboardVO> dashboard = null;
+        when(projectService.getEmployeesDashBoard()).thenReturn(dashboard);
+        mockMvc.perform(get("/projectTeam/getEmployeesDashBoard"))
+                .andExpect(MockMvcResultMatchers.status().isOk());
+        verify(projectService).getEmployeesDashBoard();
+    }    
+    
+    private List<BillingDetails> CreateTeamMateBilling() {
+        List<BillingDetails> data = new ArrayList<>();
+
+        BillingDetails team1 = new BillingDetails();
+        team1.setId(new ObjectId("5b307d7d708ef705c4ca59cc"));
+        team1.setEmployeeId("16167");
+        team1.setEmployeeName("Mogana Kurmaran");
+        team1.setProjectId("Nisum0000");
+        team1.setProjectName("Free Pool");
+        team1.setBillingStartDate(new Date(2017 - 01 - 10));
+        team1.setBillingEndDate(new Date(2017 - 02 - 10));
+        team1.setActive(true);
+        data.add(team1);
+
+        BillingDetails team2 = new BillingDetails();
+        team2.setId(new ObjectId("5b2f4a969ed316fef01adcce"));
+        team2.setEmployeeId("16050");
+        team2.setEmployeeName("Rajeshekar Sayannagari");
+        team2.setProjectId("Nisum0000");
+        team2.setProjectName("Free Pool");
+        team2.setBillingStartDate(new Date(2017 - 01 - 15));
+        team2.setBillingEndDate(new Date(2017 - 02 - 15));
+        team2.setActive(false);
+        data.add(team2);
+
+        return data;
+
+    }
+}

src/test/java/com/nisum/mytime/controllertest/ProjectControllerTest.java

@@ -61,7 +61,7 @@ public class ProjectControllerTest {
                  null, null, null, null, null, null, null, null, null, null,
                  null, null, new Date(2017 - 11 - 12), new Date(2017 - 12 - 12),
                  null,
-                 null, null,null,new Date(2020 - 01 - 01),new Date(2018 - 01 - 01),new Date(2018 - 02 - 15));
+                 null, null,null,new Date(2020 - 01 - 01),new Date(2018 - 01 - 01),new Date(2018 - 02 - 15),new Date(2018 - 02 - 15),"Mahesh","Mahesh");
     	 when(userService.getEmployeesRole("user@nisum.com"))
          .thenReturn(employeesRole);
 		 mockMvc.perform(
@@ -77,8 +77,9 @@ public class ProjectControllerTest {
         //list.add("16632");
         
         Project employeeRole1 = new Project(
-                new ObjectId("9976ef15874c902c98b8a05d"), "102", "OMS", "Marketing",
-                "Active",list,list, "Acc002", "DOM002", list);
+        		 new ObjectId("9976ef15874c902c98b8a05d"), "102","Macys", "OMS","Active",list,
+                 list, "Gap", "Billable",new Date(2017 - 11 - 29),
+                 new Date(2017 - 12 - 20), list);
 
         Account account = new Account(new ObjectId("5b62cca250e71a6eecc8c682"),
                 "Acc002", "Macys", 2, "Y","Macys","Retail",list);
@@ -110,7 +111,10 @@ public class ProjectControllerTest {
         deliveryLeadIds.add("16654");
         
         
-        Project project = new Project(new ObjectId("5976ef15874c902c98b8a05d"),"Gap0026", "Mosaic", "Move", "Active", employeeIds, managerIds,"Acc002","DOM002",deliveryLeadIds);
+        Project project = new Project(new ObjectId("5976ef15874c902c98b8a05d"),"Gap0026", 
+        		"Mosaic", "Move", "Active", employeeIds, managerIds,"Acc002","DOM002"
+        		,new Date(2017 - 11 - 29),
+                new Date(2017 - 12 - 20),deliveryLeadIds);
         ObjectMapper mapper = new ObjectMapper();
         String jsonString = mapper.writeValueAsString(project);
         when(projectService.updateProject(any())).thenReturn(project);
@@ -143,9 +147,12 @@ public class ProjectControllerTest {
 
     @Test
     public void testgetEmployeeRoleData() throws Exception {
-    	 EmployeeRoles employeesRole = new EmployeeRoles("5b307d7e708ef705c4ca6429", "16999", "B.V.S Satyanarayana", "bsatyanarayana@nisum.com", "Employee", "Employee", "", 
-          		"Java J2EE", null, "Hyderabad", "java, Spring, SpringBoot", "9848012345", "9848012345", 
-          		"test@test.com", "ES", "Active", "Full Time", new Date(), new Date(), "Male", null, null, null, null, new Date(), new Date());      
+    	 EmployeeRoles employeesRole = new EmployeeRoles(
+                 "5976ef15874c902c98b8a05d", null, null, null, null, null, null,
+                 null, null, null, null, null, null, "user@nisum.com", null,
+                 null, null, new Date(2017 - 11 - 20), new Date(2107 - 12 - 23),
+                 null,
+                 null, null,null,new Date(2020 - 01 - 01),new Date(2018 - 01 - 01),new Date(2018 - 02 - 15),new Date(2018 - 02 - 15),"Mahesh","Mahesh");
     	 when(userService.getEmployeesRoleData("16127")) .thenReturn(employeesRole);
         mockMvc.perform(
                 get("/project/getEmployeeRoleData").param("empId", "16127")

src/test/java/com/nisum/mytime/controllertest/ProjectTeamControllerTest.java

@@ -58,7 +58,7 @@ public class ProjectTeamControllerTest {
                 null, null, null, null, null, null, "user@nisum.com", null,
                 null, null, new Date(2017 - 11 - 20), new Date(2107 - 12 - 23),
                 null,
-                null, null,null,new Date(2020 - 01 - 01),new Date(2018 - 01 - 01),new Date(2018 - 02 - 15));
+                null, null,null,new Date(2020 - 01 - 01),new Date(2018 - 01 - 01),new Date(2018 - 02 - 15),new Date(2018 - 02 - 15),"Mahesh","Mahesh");
 
         when(userService.getEmployeesRole("user@nisum.com"))
                 .thenReturn(employeesRole);
@@ -76,7 +76,8 @@ public class ProjectTeamControllerTest {
         list.add("16103");
         Project employeeRole1 = new Project(
                 new ObjectId("9976ef15874c902c98b8a05d"), "102","Macys", "OMS","Active",list,
-                list, "Gap", "Billable", list);
+                list, "Gap", "Billable",new Date(2017 - 11 - 29),
+                new Date(2017 - 12 - 20), list);
         String jsonvalue = (new ObjectMapper())
                 .writeValueAsString(employeeRole1).toString();
         when(projectService.addProject(employeeRole1))
@@ -95,7 +96,7 @@ public class ProjectTeamControllerTest {
                 "Testing", "8755672341", "8800543678", "vsingh@gmail.com", null,
                 null, null, null, null, new Date(2017 - 11 - 29),
                 new Date(2017 - 12 - 20),null,
-                null, null,null,new Date(2020 - 01 - 01),new Date(2018 - 01 - 01),new Date(2018 - 02 - 15));
+                null, null,null,new Date(2020 - 01 - 01),new Date(2018 - 01 - 01),new Date(2018 - 02 - 15),new Date(2018 - 02 - 15),"Mahesh","Mahesh");
         ObjectMapper mapper = new ObjectMapper();
         String jsonString = mapper.writeValueAsString(employeesRoles2);
         when(userService.updateEmployeeRole(anyObject(),anyObject())).thenReturn(employeesRoles2);
@@ -120,7 +121,7 @@ public class ProjectTeamControllerTest {
                 "5976ef15874c902c98b8a05d", "16127", null, null, null, null,
                 null, null, null, null, null, null, null, null, null, null,
                 null, new Date(2017 - 11 - 20), new Date(2107 - 12 - 23), null,
-                null, null,null,new Date(2020 - 01 - 01),new Date(2018 - 01 - 01),new Date(2018 - 02 - 15));
+                null, null,null,new Date(2020 - 01 - 01),new Date(2018 - 01 - 01),new Date(2018 - 02 - 15),new Date(2018 - 02 - 15),"Mahesh","Mahesh");
         when(userService.getEmployeesRoleData("16127"))
                 .thenReturn(employeesRole);
         mockMvc.perform(
@@ -176,10 +177,10 @@ public class ProjectTeamControllerTest {
         String jsonvalue = (new ObjectMapper())
                 .writeValueAsString(updatedTeammate).toString();
         when(projectService.updateTeammate(updatedTeammate))
-                .thenReturn(updatedTeammate);
+                .thenReturn(jsonvalue);
         mockMvc.perform(post("/projectTeam/updateTeammate")
                 .contentType(MediaType.APPLICATION_JSON_VALUE)
-                .content(jsonvalue))
+                .content(jsonvalue))   
                 .andExpect(MockMvcResultMatchers.status().isOk());
     }