DoorControllerTest

src/main/java/application/controllers/DoorController.java

@@ -20,6 +20,10 @@ import org.springframework.web.bind.annotation.ResponseBody;
 @Controller
 @RequestMapping(value = "/doors")
 public class DoorController {
+    public void setDoorRepository(DoorRepository doorRepository) {
+        this.doorRepository = doorRepository;
+    }
+
     @Autowired
     DoorRepository doorRepository;
 
@@ -46,11 +50,12 @@ public class DoorController {
         loginResponse.put("success", true);
         loginResponse.put("doorId", door.getDoorId());
         JSONObject jsonmessage  = new JSONObject();
-        jsonmessage.put("dateTime",message.getDateTime() );
-        jsonmessage.put("message",message.getMessage() );
-        jsonmessage.put("serverMessageId",message.getServerMessageId() );
-        loginResponse.put("serverMessage", jsonmessage);
-
+        if(message!=null) {
+            jsonmessage.put("dateTime", message.getDateTime());
+            jsonmessage.put("message", message.getMessage());
+            jsonmessage.put("serverMessageId", message.getServerMessageId());
+            loginResponse.put("serverMessage", jsonmessage);
+        }
         return loginResponse.toString();
     }
 }

src/main/java/test/DoorControllerTest.java

@@ -0,0 +1,60 @@
+package test;
+
+/**
+ * Created by lukasgol on 19.01.15.
+ */
+import application.controllers.DoorController;
+import application.model.Door;
+import application.repositories.DoorRepository;
+import org.json.JSONArray;
+import org.json.JSONException;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.springframework.ui.ModelMap;
+
+import static org.junit.Assert.assertEquals;
+import static org.mockito.Mockito.*;
+import static java.util.Arrays.*;
+
+import java.util.List;
+
+public class DoorControllerTest {
+    @Test
+    public void listDoorsJsonTest(){
+        DoorController  doorController=new DoorController();
+        ModelMap model = new ModelMap();
+        String expectedDoorList="[{},{},{}]";
+        List<Door> doorList = asList(new Door(), new Door(), new Door());
+
+        DoorRepository doorRepository= mock(DoorRepository.class);
+        when(doorRepository.findAll()).thenReturn(doorList);
+        String result=null;
+        doorController.setDoorRepository(doorRepository);
+        try {
+            result= doorController.listDoorsJson(model);
+        } catch (JSONException e) {
+            e.printStackTrace();
+        }
+        assertEquals(expectedDoorList,result);
+    }
+
+    @Test
+    public void getUserMessageTest(){
+        long doorId=0;
+        DoorController  doorController=new DoorController();
+        String expectedMessage="{\"success\":true}";
+        Door door = new Door();
+
+        DoorRepository doorRepository= mock(DoorRepository.class);
+        when(doorRepository.findOne(doorId)).thenReturn(door);
+        String result=null;
+        doorController.setDoorRepository(doorRepository);
+        try {
+            result= doorController.getUser(doorId);
+        } catch (JSONException e) {
+            e.printStackTrace();
+        }
+        assertEquals( expectedMessage, result);
+    }
+}

src/main/java/test/RoleControllerTest.java

@@ -0,0 +1,40 @@
+package test;
+
+import application.controllers.DoorController;
+import application.model.Door;
+import application.repositories.DoorRepository;
+import org.json.JSONException;
+import org.junit.Test;
+import org.springframework.ui.ModelMap;
+
+import java.util.List;
+
+import static java.util.Arrays.asList;
+import static org.junit.Assert.assertEquals;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+/**
+ * Created by lukasgol on 19.01.15.
+ */
+public class RoleControllerTest {
+
+    @Test
+    public void listDoorsJsonTest(){
+        DoorController doorController=new DoorController();
+        ModelMap model = new ModelMap();
+        String expectedDoorList="[{},{},{}]";
+        List<Door> doorList = asList(new Door(), new Door(), new Door());
+
+        DoorRepository doorRepository= mock(DoorRepository.class);
+        when(doorRepository.findAll()).thenReturn(doorList);
+        String result=null;
+        doorController.setDoorRepository(doorRepository);
+        try {
+            result= doorController.listDoorsJson(model);
+        } catch (JSONException e) {
+            e.printStackTrace();
+        }
+        assertEquals(expectedDoorList,result);
+    }
+}