#39 Add channelIds to generated devices

magx2 · Sep 10, 2019 · 3b17de3 · 3b17de3
1 parent c8fd8fd
commit 3b17de3
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/src/main/kotlin/pl/grzeslowski/jsuplaservermock/InitDb.kt b/src/main/kotlin/pl/grzeslowski/jsuplaservermock/InitDb.kt
@@ -59,11 +59,15 @@ open class InitDb(private val deviceService: DeviceService) : CommandLineRunner
                 // multiple channels
                 buildDeviceWithMultipleChannels("Multiple channels #1"),
                 buildDeviceWithMultipleChannels("Multiple channels #2")
-        ).forEach {
-            deviceService.addDevice(it)
-            updateEnabled(it)
-            updateName(it)
-            updateComment(it)
+        ).forEach { device ->
+            device.channelsIds = device.channels
+                    .stream()
+                    .map { it.id }
+                    .collect(Collectors.toList())
+            deviceService.addDevice(device)
+            updateEnabled(device)
+            updateName(device)
+            updateComment(device)
         }
     }