aboutsummaryrefslogtreecommitdiff
path: root/test/org/traccar/database/GroupTreeTest.java
diff options
context:
space:
mode:
Diffstat (limited to 'test/org/traccar/database/GroupTreeTest.java')
-rw-r--r--test/org/traccar/database/GroupTreeTest.java55
1 files changed, 55 insertions, 0 deletions
diff --git a/test/org/traccar/database/GroupTreeTest.java b/test/org/traccar/database/GroupTreeTest.java
new file mode 100644
index 000000000..f73cb75b6
--- /dev/null
+++ b/test/org/traccar/database/GroupTreeTest.java
@@ -0,0 +1,55 @@
+package org.traccar.database;
+
+import org.junit.Assert;
+import org.junit.Test;
+import org.traccar.model.Device;
+import org.traccar.model.Group;
+
+import java.util.ArrayList;
+import java.util.Collection;
+
+public class GroupTreeTest {
+
+ private static Group createGroup(long id, String name, long parent) {
+ Group group = new Group();
+ group.setId(id);
+ group.setName(name);
+ group.setGroupId(parent);
+ return group;
+ }
+
+ private static Device createDevice(long id, String name, long parent) {
+ Device device = new Device();
+ device.setId(id);
+ device.setName(name);
+ device.setGroupId(parent);
+ return device;
+ }
+
+ @Test
+ public void testGetDescendants() {
+ Collection<Group> groups = new ArrayList<>();
+ groups.add(createGroup(1, "First", 0));
+ groups.add(createGroup(2, "Second", 1));
+ groups.add(createGroup(3, "Third", 2));
+ groups.add(createGroup(4, "Fourth", 2));
+ groups.add(createGroup(5, "Fifth", 4));
+
+ Collection<Device> devices = new ArrayList<>();
+ devices.add(createDevice(1, "One", 3));
+ devices.add(createDevice(2, "Two", 5));
+ devices.add(createDevice(3, "One", 5));
+
+ GroupTree groupTree = new GroupTree(groups, devices);
+
+ Assert.assertEquals(4, groupTree.getGroups(1).size());
+ Assert.assertEquals(3, groupTree.getGroups(2).size());
+ Assert.assertEquals(0, groupTree.getGroups(3).size());
+ Assert.assertEquals(1, groupTree.getGroups(4).size());
+
+ Assert.assertEquals(3, groupTree.getDevices(1).size());
+ Assert.assertEquals(1, groupTree.getDevices(3).size());
+ Assert.assertEquals(2, groupTree.getDevices(4).size());
+ }
+
+}