Update tests, move maxNeighborsPerNode to parameter (#269)

streamr-dev · May 2, 2019 · 88c84ed · 88c84ed
1 parent b65e7e0
commit 88c84ed
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 31 deletions.
diff --git a/bin/tracker.js b/bin/tracker.js
@@ -4,8 +4,8 @@ const { startTracker } = require('../src/composition')
 
 const port = process.argv[2] || 30300
 const ip = process.argv[3] || '127.0.0.1'
+const maxNeighborsPerNode = process.argv[4] || 4
 const id = `tracker-${port}`
-const maxNeighborsPerNode = 4
 
 startTracker(ip, port, id, maxNeighborsPerNode)
     .then(() => {})

diff --git a/test/unit/OverlayTopology.test.js b/test/unit/OverlayTopology.test.js
@@ -1,5 +1,7 @@
 const OverlayTopology = require('../../src/logic/OverlayTopology')
 
+const maxNeighborsPerNodeArray = [4, 8, 12, 16]
+
 test('forming overlay topology', () => {
     const topology = new OverlayTopology(3, (arr) => arr, (arr) => arr[0])
 
@@ -226,41 +228,45 @@ test('forming overlay topology', () => {
 })
 
 test('unknown nodes are discarded', () => {
-    const topology = new OverlayTopology(3)
+    maxNeighborsPerNodeArray.forEach((maxNeighborsPerNode) => {
+        const topology = new OverlayTopology(maxNeighborsPerNode)
 
-    topology.update('node-1', [])
-    topology.update('node-2', [])
-    topology.update('node-3', [])
+        topology.update('node-1', [])
+        topology.update('node-2', [])
+        topology.update('node-3', [])
 
-    topology.update('node-1', ['node-2', 'node-3', 'node-4'])
-    expect(topology.state()).toEqual({
-        'node-1': [
-            'node-2',
-            'node-3'
-        ],
-        'node-2': [
-            'node-1'
-        ],
-        'node-3': [
-            'node-1'
-        ]
+        topology.update('node-1', ['node-2', 'node-3', 'node-4'])
+        expect(topology.state()).toEqual({
+            'node-1': [
+                'node-2',
+                'node-3'
+            ],
+            'node-2': [
+                'node-1'
+            ],
+            'node-3': [
+                'node-1'
+            ]
+        })
     })
 })
 
 // TODO: remove or write better, since not the best way to test randomness
 test('100 rounds of typical operation does not lead to invariant exception', () => {
-    for (let i = 0; i < 100; ++i) {
-        const topology = new OverlayTopology(4)
-        topology.update('node-1', [])
-        topology.update('node-2', [])
-        topology.formInstructions('node-2')
-        topology.update('node-3', [])
-        topology.formInstructions('node-3')
-        topology.update('node-4', [])
-        topology.formInstructions('node-4')
-        topology.update('node-5', [])
-        topology.formInstructions('node-5')
-        topology.update('node-0', [])
-        topology.formInstructions('node-0')
-    }
+    maxNeighborsPerNodeArray.forEach((maxNeighborsPerNode) => {
+        for (let i = 0; i < 100; ++i) {
+            const topology = new OverlayTopology(maxNeighborsPerNode)
+            topology.update('node-1', [])
+            topology.update('node-2', [])
+            topology.formInstructions('node-2')
+            topology.update('node-3', [])
+            topology.formInstructions('node-3')
+            topology.update('node-4', [])
+            topology.formInstructions('node-4')
+            topology.update('node-5', [])
+            topology.formInstructions('node-5')
+            topology.update('node-0', [])
+            topology.formInstructions('node-0')
+        }
+    })
 })