Add knex AliasesDB

test/db/aliases.js

@@ -0,0 +1,108 @@
+const assert = require('assert');
+const sinon = require('sinon');
+const TestUtilDB = require('../testutil/db');
+const AliasesDB = require('../../lib/db/aliases').AliasesDB;
+
+describe('AliasesDB', () => {
+    let mockTx, mockDB, aliasesDB;
+
+    beforeEach(() => {
+        mockTx = new TestUtilDB.MockTx();
+        mockDB = new TestUtilDB.MockDB(mockTx);
+        aliasesDB = new AliasesDB(mockDB);
+    });
+
+    describe('#addAlias', () => {
+        it('adds a new alias', () => {
+            const ip = '1.2.3.4';
+            const name = 'foo';
+            sinon.stub(mockTx, 'table').withArgs('aliases').returns(mockTx);
+            sinon.stub(mockTx, 'where').withArgs({ ip, name }).returns(mockTx);
+            const del = sinon.stub(mockTx, 'del').resolves();
+            const insert = sinon.stub(mockTx, 'insert').resolves();
+            return aliasesDB.addAlias(ip, name).then(() => {
+                assert(del.called, 'Expected old alias to be purged');
+                assert(insert.called, 'Expected new alias to be inserted');
+                const record = insert.getCall(0).args[0];
+                assert.strictEqual(record.ip, ip);
+                assert.strictEqual(record.name, name);
+                assert(typeof record.time === 'number', 'Expected time field to be a number');
+            });
+        });
+    });
+
+    describe('#getAliasesByIP', () => {
+        it('retrieves aliases by full IP', () => {
+            const ip = '1.2.3.4';
+            const rows = [
+                { ip, name: 'foo' },
+                { ip, name: 'bar' }
+            ];
+
+            sinon.stub(mockTx, 'table').withArgs('aliases').returns(mockTx);
+            sinon.stub(mockTx, 'where').withArgs({ ip }).returns(mockTx);
+            sinon.stub(mockTx, 'select').returns(mockTx);
+            sinon.stub(mockTx, 'distinct').withArgs('name').returns(mockTx);
+            sinon.stub(mockTx, 'orderBy').withArgs('time', 'desc').returns(mockTx);
+            sinon.stub(mockTx, 'limit').withArgs(5).resolves(rows);
+            return aliasesDB.getAliasesByIP(ip).then(names => {
+                assert.deepStrictEqual(names.sort(), ['bar', 'foo']);
+            });
+        });
+
+        it('retrieves aliases by IPv4 range', () => {
+            const ip = '1.2.3';
+            const rows = [
+                { ip: ip + '.4', name: 'foo' },
+                { ip: ip + '.5', name: 'bar' }
+            ];
+
+            sinon.stub(mockTx, 'table').withArgs('aliases').returns(mockTx);
+            sinon.stub(mockTx, 'where').withArgs('ip', 'LIKE', `${ip}.%`).returns(mockTx);
+            sinon.stub(mockTx, 'select').returns(mockTx);
+            sinon.stub(mockTx, 'distinct').withArgs('name').returns(mockTx);
+            sinon.stub(mockTx, 'orderBy').withArgs('time', 'desc').returns(mockTx);
+            sinon.stub(mockTx, 'limit').withArgs(5).resolves(rows);
+            return aliasesDB.getAliasesByIP(ip).then(names => {
+                assert.deepStrictEqual(names.sort(), ['bar', 'foo']);
+            });
+        });
+
+        it('retrieves aliases by IPv6 range', () => {
+            const ip = '1:2:3';
+            const rows = [
+                { ip: ip + '::4', name: 'foo' },
+                { ip: ip + '::5', name: 'bar' }
+            ];
+
+            sinon.stub(mockTx, 'table').withArgs('aliases').returns(mockTx);
+            const where = sinon.stub(mockTx, 'where')
+                    .withArgs('ip', 'LIKE', `${ip}:%`).returns(mockTx);
+            sinon.stub(mockTx, 'select').returns(mockTx);
+            sinon.stub(mockTx, 'distinct').withArgs('name').returns(mockTx);
+            sinon.stub(mockTx, 'orderBy').withArgs('time', 'desc').returns(mockTx);
+            sinon.stub(mockTx, 'limit').withArgs(5).resolves(rows);
+            return aliasesDB.getAliasesByIP(ip).then(names => {
+                assert(where.called, 'Expected WHERE LIKE clause');
+                assert.deepStrictEqual(names.sort(), ['bar', 'foo']);
+            });
+        });
+    });
+
+    describe('#getIPsByName', () => {
+        it('retrieves IPs by name', () => {
+            const name = 'foo';
+            const rows = [
+                { name, ip: '1.2.3.4' },
+                { name, ip: '5.6.7.8' }
+            ];
+
+            sinon.stub(mockTx, 'table').withArgs('aliases').returns(mockTx);
+            sinon.stub(mockTx, 'select').withArgs('ip').returns(mockTx);
+            sinon.stub(mockTx, 'where').withArgs({ name }).resolves(rows);
+            return aliasesDB.getIPsByName(name).then(ips => {
+                assert.deepStrictEqual(ips.sort(), ['1.2.3.4', '5.6.7.8']);
+            });
+        });
+    });
+});

test/testutil/db.js

@@ -14,7 +14,17 @@ function MockTx() {
 
 }
 
-['insert', 'update', 'select', 'del', 'where', 'table'].forEach(method => {
+[
+    'del',
+    'distinct',
+    'insert',
+    'limit',
+    'orderBy',
+    'select',
+    'table',
+    'update',
+    'where',
+].forEach(method => {
     MockTx.prototype[method] = function () {
         return Promise.reject(new Error(`No stub defined for method "${method}"`));
     };