updates tests and works towards #17

sgnl · Apr 25, 2018 · a9e7aff · a9e7aff
1 parent ff338bd
commit a9e7aff
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 2 deletions.
diff --git a/index.js b/index.js
@@ -1,4 +1,14 @@
 
 'use strict';
 
-module.exports = require('./lib/discovery');
+const Discovery = require('./lib/discovery');
+const Device = require('./lib/device');
+
+function Nodeku(timeout) {
+  return Discovery(timeout);
+}
+
+Nodeku.Discovery = Discovery;
+Nodeku.Device = Device;
+
+module.exports = Nodeku;
diff --git a/test/nodeku.test.js b/test/nodeku.test.js
@@ -16,7 +16,7 @@ require('superagent-mock')(req, ReqMockConfig, utils.logger);
 const device = proxyquire('../lib/device', {
   got: req
 });
-const Nodeku = proxyquire('../lib/discovery', {
+const Nodeku = proxyquire('../index', {
   'node-ssdp': ssdpMock,
   './device': device
 });
@@ -39,6 +39,20 @@ test('Nodeku', t => {
   t.is(typeof Nodeku, 'function', 'is ready');
 });
 
+test('Nodeku.Device exists', t => {
+  t.true(Nodeku.hasOwnProperty('Device'), 'exists');
+});
+
+test('Nodeku.Discovery exists', t => {
+  t.true(Nodeku.hasOwnProperty('Discovery'), 'exists');
+});
+
+wrapper('-method: .ip()', (t, device) => {
+  t.true(device.hasOwnProperty('ip'), 'exists');
+  t.is(typeof device.ip, 'function', 'is a function');
+  t.deepEqual(device.ip(), '192.168.1.17:8060', 'ip address retreived');
+});
+
 wrapper('-method: .ip()', (t, device) => {
   t.true(device.hasOwnProperty('ip'), 'exists');
   t.is(typeof device.ip, 'function', 'is a function');