test: add test for updateDevice

PatrIoT-Framework · Sep 2, 2022 · f72ea54 · f72ea54
1 parent 2e59a11
commit f72ea54
Showing 1 changed file with 27 additions and 3 deletions.
diff --git a/src/test/java/io/patriot_framework/virtual_smart_home/house/HouseTest.java b/src/test/java/io/patriot_framework/virtual_smart_home/house/HouseTest.java
@@ -98,9 +98,33 @@ public void updateDeviceWrongClass() {
         Exception exception = assertThrows(DifferentDeviceException.class, () -> {
             house.updateDevice("fireplace", tv);
         });
+        assertTrue(exception.getMessage().contains("Updating with different device"));
+    }
+
+    @Test
+    public void updateDeviceLabelNull() {
+        Exception exception = assertThrows(IllegalArgumentException.class, () -> {
+            house.updateDevice(null, fireplace);
+        });
+        assertTrue(exception.getMessage().contains("Label of the device can't be null"));
+    }
+
+    @Test
+    public void updateDeviceNull() {
+        devices.put("fireplace", fireplace);
+        house.setDevices(devices);
+        Exception exception = assertThrows(IllegalArgumentException.class, () -> {
+            house.updateDevice("fireplace", null);
+        });
+        assertTrue(exception.getMessage().contains("Device parameter can't be null"));
+    }
+
+    @Test
+    public void updateDeviceWhichIsNotInTheHouse() {
+        Exception exception = assertThrows(NoSuchElementException.class, () -> {
+            house.updateDevice("fireplace", fireplace);
+        });
+        assertTrue(exception.getMessage().contains("Device with label: fireplace is not present in the house"));
 
-        String expectedMessage = "Updating with different device";
-        String actualMessage = exception.getMessage();
-        assertTrue(actualMessage.contains(expectedMessage));
     }
 }