From 59416923dcb3a756eaf532cc4259f2f6625c0762 Mon Sep 17 00:00:00 2001
From: Anton Tananaev <anton.tananaev@gmail.com>
Date: Sun, 31 Mar 2019 22:35:39 -0700
Subject: Convert project to gradle

---
 .../java/org/traccar/database/GroupTreeTest.java   | 56 ++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 src/test/java/org/traccar/database/GroupTreeTest.java

(limited to 'src/test/java/org/traccar/database/GroupTreeTest.java')

diff --git a/src/test/java/org/traccar/database/GroupTreeTest.java b/src/test/java/org/traccar/database/GroupTreeTest.java
new file mode 100644
index 000000000..b547aab60
--- /dev/null
+++ b/src/test/java/org/traccar/database/GroupTreeTest.java
@@ -0,0 +1,56 @@
+package org.traccar.database;
+
+import org.junit.Test;
+import org.traccar.model.Device;
+import org.traccar.model.Group;
+
+import java.util.ArrayList;
+import java.util.Collection;
+
+import static org.junit.Assert.assertEquals;
+
+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);
+
+        assertEquals(4, groupTree.getGroups(1).size());
+        assertEquals(3, groupTree.getGroups(2).size());
+        assertEquals(0, groupTree.getGroups(3).size());
+        assertEquals(1, groupTree.getGroups(4).size());
+
+        assertEquals(3, groupTree.getDevices(1).size());
+        assertEquals(1, groupTree.getDevices(3).size());
+        assertEquals(2, groupTree.getDevices(4).size());
+    }
+
+}
-- 
cgit v1.2.3