test: increase dgram ref()/unref() coverage

This commit completes code coverage for dgram's Socket#ref() and Socket#unref() methods. PR-URL: nodejs#11240 Reviewed-By: Santiago Gimeno <[email protected]> Reviewed-By: James M Snell <[email protected]>
italoacasas · Feb 14, 2017 · 2333cd3 · 2333cd3
1 parent d9ed965
commit 2333cd3
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 4 deletions.
diff --git a/test/parallel/test-dgram-ref.js b/test/parallel/test-dgram-ref.js
@@ -1,7 +1,14 @@
 'use strict';
-require('../common');
+const common = require('../common');
 const dgram = require('dgram');
 
 // should not hang, see #1282
 dgram.createSocket('udp4');
 dgram.createSocket('udp6');
+
+{
+  // Test the case of ref()'ing a socket with no handle.
+  const s = dgram.createSocket('udp4');
+
+  s.close(common.mustCall(() => s.ref()));
+}
diff --git a/test/parallel/test-dgram-unref.js b/test/parallel/test-dgram-unref.js
@@ -2,8 +2,18 @@
 const common = require('../common');
 const dgram = require('dgram');
 
-const s = dgram.createSocket('udp4');
-s.bind();
-s.unref();
+{
+  // Test the case of unref()'ing a socket with a handle.
+  const s = dgram.createSocket('udp4');
+  s.bind();
+  s.unref();
+}
+
+{
+  // Test the case of unref()'ing a socket with no handle.
+  const s = dgram.createSocket('udp4');
+
+  s.close(common.mustCall(() => s.unref()));
+}
 
 setTimeout(common.mustNotCall(), 1000).unref();