init commit
This commit is contained in:
parent
ae639426d0
commit
7a491681cc
257 changed files with 95524 additions and 80 deletions
584
integration_test/channel/kickban.js
Normal file
584
integration_test/channel/kickban.js
Normal file
|
|
@ -0,0 +1,584 @@
|
|||
const assert = require('assert');
|
||||
const KickbanModule = require('../../lib/channel/kickban');
|
||||
const database = require('../../lib/database');
|
||||
const Promise = require('bluebird');
|
||||
const testDB = require('../testutil/db').testDB;
|
||||
|
||||
database.init(testDB);
|
||||
|
||||
describe('KickbanModule', () => {
|
||||
const channelName = `test_${Math.random().toString(31).substring(2)}`;
|
||||
|
||||
let mockChannel;
|
||||
let mockUser;
|
||||
let kickban;
|
||||
|
||||
beforeEach(() => {
|
||||
mockChannel = {
|
||||
name: channelName,
|
||||
refCounter: {
|
||||
ref() { },
|
||||
unref() { }
|
||||
},
|
||||
logger: {
|
||||
log() { }
|
||||
},
|
||||
modules: {
|
||||
permissions: {
|
||||
canBan() {
|
||||
return true;
|
||||
}
|
||||
}
|
||||
},
|
||||
users: []
|
||||
};
|
||||
|
||||
mockUser = {
|
||||
getName() {
|
||||
return 'The_Admin';
|
||||
},
|
||||
|
||||
getLowerName() {
|
||||
return 'the_admin';
|
||||
},
|
||||
|
||||
socket: {
|
||||
emit(frame) {
|
||||
if (frame === 'errorMsg') {
|
||||
throw new Error(arguments[1].msg);
|
||||
}
|
||||
}
|
||||
},
|
||||
|
||||
account: {
|
||||
effectiveRank: 3
|
||||
}
|
||||
};
|
||||
|
||||
kickban = new KickbanModule(mockChannel);
|
||||
});
|
||||
|
||||
afterEach(async () => {
|
||||
await database.getDB().runTransaction(async tx => {
|
||||
await tx.table('channel_bans')
|
||||
.where({ channel: channelName })
|
||||
.del();
|
||||
await tx.table('channel_ranks')
|
||||
.where({ channel: channelName })
|
||||
.del();
|
||||
});
|
||||
});
|
||||
|
||||
describe('#handleCmdBan', () => {
|
||||
it('inserts a valid ban', done => {
|
||||
let kicked = false;
|
||||
|
||||
mockChannel.refCounter.unref = () => {
|
||||
assert(kicked, 'Expected user to be kicked');
|
||||
|
||||
database.getDB().runTransaction(async tx => {
|
||||
const ban = await tx.table('channel_bans')
|
||||
.where({
|
||||
channel: channelName,
|
||||
name: 'test_user'
|
||||
})
|
||||
.first();
|
||||
|
||||
assert.strictEqual(ban.ip, '*');
|
||||
assert.strictEqual(ban.reason, 'because reasons');
|
||||
assert.strictEqual(ban.bannedby, mockUser.getName());
|
||||
|
||||
done();
|
||||
});
|
||||
};
|
||||
|
||||
mockChannel.users = [{
|
||||
getLowerName() {
|
||||
return 'test_user';
|
||||
},
|
||||
|
||||
kick(reason) {
|
||||
assert.strictEqual(reason, "You're banned!");
|
||||
kicked = true;
|
||||
}
|
||||
}];
|
||||
|
||||
kickban.handleCmdBan(
|
||||
mockUser,
|
||||
'/ban test_user because reasons',
|
||||
{}
|
||||
);
|
||||
});
|
||||
|
||||
it('rejects if the user does not have ban permission', done => {
|
||||
mockUser.socket.emit = (frame, obj) => {
|
||||
if (frame === 'errorMsg') {
|
||||
assert.strictEqual(
|
||||
obj.msg,
|
||||
'You do not have ban permissions on this channel'
|
||||
);
|
||||
|
||||
done();
|
||||
}
|
||||
};
|
||||
|
||||
mockChannel.modules.permissions.canBan = () => false;
|
||||
|
||||
kickban.handleCmdBan(
|
||||
mockUser,
|
||||
'/ban test_user because reasons',
|
||||
{}
|
||||
);
|
||||
});
|
||||
|
||||
it('rejects if the user tries to ban themselves', done => {
|
||||
let costanza = false;
|
||||
|
||||
mockUser.socket.emit = (frame, obj) => {
|
||||
if (frame === 'errorMsg') {
|
||||
assert.strictEqual(
|
||||
obj.msg,
|
||||
'You cannot ban yourself'
|
||||
);
|
||||
|
||||
if (!costanza) {
|
||||
throw new Error('Expected costanza for banning self');
|
||||
}
|
||||
|
||||
done();
|
||||
} else if (frame === 'costanza') {
|
||||
assert.strictEqual(
|
||||
obj.msg,
|
||||
"You can't ban yourself"
|
||||
);
|
||||
|
||||
costanza = true;
|
||||
}
|
||||
};
|
||||
|
||||
kickban.handleCmdBan(
|
||||
mockUser,
|
||||
'/ban the_Admin because reasons',
|
||||
{}
|
||||
);
|
||||
});
|
||||
|
||||
it('rejects if the user is ranked below the ban recipient', done => {
|
||||
database.getDB().runTransaction(tx => {
|
||||
return tx.table('channel_ranks')
|
||||
.insert({
|
||||
channel: channelName,
|
||||
name: 'test_user',
|
||||
rank: 5
|
||||
});
|
||||
}).then(() => {
|
||||
mockUser.socket.emit = (frame, obj) => {
|
||||
if (frame === 'errorMsg') {
|
||||
assert.strictEqual(
|
||||
obj.msg,
|
||||
"You don't have permission to ban test_user"
|
||||
);
|
||||
|
||||
done();
|
||||
}
|
||||
};
|
||||
|
||||
kickban.handleCmdBan(
|
||||
mockUser,
|
||||
'/ban test_user because reasons',
|
||||
{}
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
it('rejects if the the ban recipient is already banned', done => {
|
||||
database.getDB().runTransaction(tx => {
|
||||
return tx.table('channel_bans')
|
||||
.insert({
|
||||
channel: channelName,
|
||||
name: 'test_user',
|
||||
ip: '*',
|
||||
bannedby: 'somebody',
|
||||
reason: 'I dunno'
|
||||
});
|
||||
}).then(() => {
|
||||
mockUser.socket.emit = (frame, obj) => {
|
||||
if (frame === 'errorMsg') {
|
||||
assert.strictEqual(
|
||||
obj.msg,
|
||||
'test_user is already banned'
|
||||
);
|
||||
|
||||
done();
|
||||
}
|
||||
};
|
||||
|
||||
kickban.handleCmdBan(
|
||||
mockUser,
|
||||
'/ban test_user because reasons',
|
||||
{}
|
||||
);
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('#handleCmdIPBan', () => {
|
||||
beforeEach(async () => {
|
||||
await database.getDB().runTransaction(async tx => {
|
||||
await tx.table('aliases')
|
||||
.insert([{
|
||||
name: 'test_user',
|
||||
ip: '1.2.3.4',
|
||||
time: Date.now()
|
||||
}]);
|
||||
});
|
||||
});
|
||||
|
||||
afterEach(async () => {
|
||||
await database.getDB().runTransaction(async tx => {
|
||||
await tx.table('aliases')
|
||||
.where({ name: 'test_user' })
|
||||
.orWhere({ ip: '1.2.3.4' })
|
||||
.del();
|
||||
});
|
||||
});
|
||||
|
||||
it('inserts a valid ban', done => {
|
||||
let firstUserKicked = false;
|
||||
let secondUserKicked = false;
|
||||
|
||||
mockChannel.refCounter.unref = () => {
|
||||
assert(firstUserKicked, 'Expected banned user to be kicked');
|
||||
assert(
|
||||
secondUserKicked,
|
||||
'Expected user with banned IP to be kicked'
|
||||
);
|
||||
|
||||
database.getDB().runTransaction(async tx => {
|
||||
const nameBan = await tx.table('channel_bans')
|
||||
.where({
|
||||
channel: channelName,
|
||||
name: 'test_user',
|
||||
ip: '*'
|
||||
})
|
||||
.first();
|
||||
|
||||
assert.strictEqual(nameBan.reason, 'because reasons');
|
||||
assert.strictEqual(nameBan.bannedby, mockUser.getName());
|
||||
|
||||
const ipBan = await tx.table('channel_bans')
|
||||
.where({
|
||||
channel: channelName,
|
||||
ip: '1.2.3.4'
|
||||
})
|
||||
.first();
|
||||
|
||||
assert.strictEqual(ipBan.name, 'test_user');
|
||||
assert.strictEqual(ipBan.reason, 'because reasons');
|
||||
assert.strictEqual(ipBan.bannedby, mockUser.getName());
|
||||
|
||||
done();
|
||||
});
|
||||
};
|
||||
|
||||
mockChannel.users = [{
|
||||
getLowerName() {
|
||||
return 'test_user';
|
||||
},
|
||||
|
||||
realip: '1.2.3.4',
|
||||
|
||||
kick(reason) {
|
||||
assert.strictEqual(reason, "You're banned!");
|
||||
firstUserKicked = true;
|
||||
}
|
||||
}, {
|
||||
getLowerName() {
|
||||
return 'second_user_same_ip';
|
||||
},
|
||||
|
||||
realip: '1.2.3.4',
|
||||
|
||||
kick(reason) {
|
||||
assert.strictEqual(reason, "You're banned!");
|
||||
secondUserKicked = true;
|
||||
}
|
||||
}];
|
||||
|
||||
kickban.handleCmdIPBan(
|
||||
mockUser,
|
||||
'/ipban test_user because reasons',
|
||||
{}
|
||||
);
|
||||
});
|
||||
|
||||
it('inserts a valid range ban', done => {
|
||||
mockChannel.refCounter.unref = () => {
|
||||
database.getDB().runTransaction(async tx => {
|
||||
const ipBan = await tx.table('channel_bans')
|
||||
.where({
|
||||
channel: channelName,
|
||||
ip: '1.2.3'
|
||||
})
|
||||
.first();
|
||||
|
||||
assert.strictEqual(ipBan.name, 'test_user');
|
||||
assert.strictEqual(ipBan.reason, 'because reasons');
|
||||
assert.strictEqual(ipBan.bannedby, mockUser.getName());
|
||||
|
||||
done();
|
||||
});
|
||||
};
|
||||
|
||||
kickban.handleCmdIPBan(
|
||||
mockUser,
|
||||
'/ipban test_user range because reasons',
|
||||
{}
|
||||
);
|
||||
});
|
||||
|
||||
it('inserts a valid wide-range ban', done => {
|
||||
mockChannel.refCounter.unref = () => {
|
||||
database.getDB().runTransaction(async tx => {
|
||||
const ipBan = await tx.table('channel_bans')
|
||||
.where({
|
||||
channel: channelName,
|
||||
ip: '1.2'
|
||||
})
|
||||
.first();
|
||||
|
||||
assert.strictEqual(ipBan.name, 'test_user');
|
||||
assert.strictEqual(ipBan.reason, 'because reasons');
|
||||
assert.strictEqual(ipBan.bannedby, mockUser.getName());
|
||||
|
||||
done();
|
||||
});
|
||||
};
|
||||
|
||||
kickban.handleCmdIPBan(
|
||||
mockUser,
|
||||
'/ipban test_user wrange because reasons',
|
||||
{}
|
||||
);
|
||||
});
|
||||
|
||||
it('inserts a valid IPv6 ban', done => {
|
||||
const longIP = require('../../lib/utilities').expandIPv6('::abcd');
|
||||
|
||||
mockChannel.refCounter.unref = () => {
|
||||
database.getDB().runTransaction(async tx => {
|
||||
const ipBan = await tx.table('channel_bans')
|
||||
.where({
|
||||
channel: channelName,
|
||||
ip: longIP
|
||||
})
|
||||
.first();
|
||||
|
||||
assert.strictEqual(ipBan.name, 'test_user');
|
||||
assert.strictEqual(ipBan.reason, 'because reasons');
|
||||
assert.strictEqual(ipBan.bannedby, mockUser.getName());
|
||||
|
||||
done();
|
||||
});
|
||||
};
|
||||
|
||||
database.getDB().runTransaction(async tx => {
|
||||
await tx.table('aliases')
|
||||
.insert({
|
||||
name: 'test_user',
|
||||
ip: longIP,
|
||||
time: Date.now()
|
||||
});
|
||||
}).then(() => {
|
||||
kickban.handleCmdIPBan(
|
||||
mockUser,
|
||||
'/ipban test_user because reasons',
|
||||
{}
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
it('rejects if the user does not have ban permission', done => {
|
||||
mockUser.socket.emit = (frame, obj) => {
|
||||
if (frame === 'errorMsg') {
|
||||
assert.strictEqual(
|
||||
obj.msg,
|
||||
'You do not have ban permissions on this channel'
|
||||
);
|
||||
|
||||
done();
|
||||
}
|
||||
};
|
||||
|
||||
mockChannel.modules.permissions.canBan = () => false;
|
||||
|
||||
kickban.handleCmdIPBan(
|
||||
mockUser,
|
||||
'/ipban test_user because reasons',
|
||||
{}
|
||||
);
|
||||
});
|
||||
|
||||
it('rejects if the user tries to ban themselves', done => {
|
||||
let costanza = false;
|
||||
|
||||
mockUser.socket.emit = (frame, obj) => {
|
||||
if (frame === 'errorMsg') {
|
||||
assert.strictEqual(
|
||||
obj.msg,
|
||||
'You cannot ban yourself'
|
||||
);
|
||||
|
||||
if (!costanza) {
|
||||
throw new Error('Expected costanza for banning self');
|
||||
}
|
||||
|
||||
done();
|
||||
} else if (frame === 'costanza') {
|
||||
assert.strictEqual(
|
||||
obj.msg,
|
||||
"You can't ban yourself"
|
||||
);
|
||||
|
||||
costanza = true;
|
||||
}
|
||||
};
|
||||
|
||||
kickban.handleCmdIPBan(
|
||||
mockUser,
|
||||
'/ipban the_Admin because reasons',
|
||||
{}
|
||||
);
|
||||
});
|
||||
|
||||
it('rejects if the user is ranked below the ban recipient', done => {
|
||||
database.getDB().runTransaction(tx => {
|
||||
return tx.table('channel_ranks')
|
||||
.insert({
|
||||
channel: channelName,
|
||||
name: 'test_user',
|
||||
rank: 5
|
||||
});
|
||||
}).then(() => {
|
||||
mockUser.socket.emit = (frame, obj) => {
|
||||
if (frame === 'errorMsg') {
|
||||
assert.strictEqual(
|
||||
obj.msg,
|
||||
"You don't have permission to ban IP " +
|
||||
"09l.TFb.5To.HBB"
|
||||
);
|
||||
|
||||
done();
|
||||
}
|
||||
};
|
||||
|
||||
kickban.handleCmdIPBan(
|
||||
mockUser,
|
||||
'/ipban test_user because reasons',
|
||||
{}
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
it('rejects if the user is ranked below an alias of the ban recipient', done => {
|
||||
database.getDB().runTransaction(async tx => {
|
||||
await tx.table('channel_ranks')
|
||||
.insert({
|
||||
channel: channelName,
|
||||
name: 'another_user',
|
||||
rank: 5
|
||||
});
|
||||
await tx.table('aliases')
|
||||
.insert({
|
||||
name: 'another_user',
|
||||
ip: '1.2.3.3', // different IP, same /24 range
|
||||
time: Date.now()
|
||||
});
|
||||
}).then(() => {
|
||||
mockUser.socket.emit = (frame, obj) => {
|
||||
if (frame === 'errorMsg') {
|
||||
assert.strictEqual(
|
||||
obj.msg,
|
||||
"You don't have permission to ban IP " +
|
||||
"09l.TFb.5To.*"
|
||||
);
|
||||
|
||||
done();
|
||||
}
|
||||
};
|
||||
|
||||
kickban.handleCmdIPBan(
|
||||
mockUser,
|
||||
'/ipban test_user range because reasons',
|
||||
{}
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
it('rejects if the the ban recipient IP is already banned', done => {
|
||||
database.getDB().runTransaction(tx => {
|
||||
return tx.table('channel_bans')
|
||||
.insert({
|
||||
channel: channelName,
|
||||
name: 'another_user',
|
||||
ip: '1.2.3.4',
|
||||
bannedby: 'somebody',
|
||||
reason: 'I dunno'
|
||||
});
|
||||
}).then(() => {
|
||||
mockUser.socket.emit = (frame, obj) => {
|
||||
if (frame === 'errorMsg') {
|
||||
assert.strictEqual(
|
||||
obj.msg,
|
||||
'09l.TFb.5To.HBB is already banned'
|
||||
);
|
||||
|
||||
done();
|
||||
}
|
||||
};
|
||||
|
||||
kickban.handleCmdIPBan(
|
||||
mockUser,
|
||||
'/ipban test_user because reasons',
|
||||
{}
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
it('still adds the IP ban even if the name is already banned', done => {
|
||||
mockChannel.refCounter.unref = () => {
|
||||
database.getDB().runTransaction(async tx => {
|
||||
const ipBan = await tx.table('channel_bans')
|
||||
.where({
|
||||
channel: channelName,
|
||||
ip: '1.2.3.4'
|
||||
})
|
||||
.first();
|
||||
|
||||
assert.strictEqual(ipBan.name, 'test_user');
|
||||
assert.strictEqual(ipBan.reason, 'because reasons');
|
||||
assert.strictEqual(ipBan.bannedby, mockUser.getName());
|
||||
|
||||
done();
|
||||
});
|
||||
};
|
||||
|
||||
database.getDB().runTransaction(tx => {
|
||||
return tx.table('channel_bans')
|
||||
.insert({
|
||||
channel: channelName,
|
||||
name: 'test_user',
|
||||
ip: '*',
|
||||
bannedby: 'somebody',
|
||||
reason: 'I dunno'
|
||||
});
|
||||
}).then(() => {
|
||||
kickban.handleCmdIPBan(
|
||||
mockUser,
|
||||
'/ipban test_user because reasons',
|
||||
{}
|
||||
);
|
||||
});
|
||||
});
|
||||
});
|
||||
});
|
||||
88
integration_test/database/accounts.js
Normal file
88
integration_test/database/accounts.js
Normal file
|
|
@ -0,0 +1,88 @@
|
|||
const assert = require('assert');
|
||||
const { testDB } = require('../testutil/db');
|
||||
const accounts = require('../../lib/database/accounts');
|
||||
|
||||
require('../../lib/database').init(testDB);
|
||||
|
||||
describe('AccountsDatabase', () => {
|
||||
describe('#verifyLogin', () => {
|
||||
let ip = '169.254.111.111';
|
||||
let user;
|
||||
let password;
|
||||
|
||||
beforeEach(async () => {
|
||||
return testDB.knex.table('users')
|
||||
.where({ ip })
|
||||
.delete();
|
||||
});
|
||||
|
||||
beforeEach(done => {
|
||||
user = `u${Math.random().toString(31).substring(2)}`;
|
||||
password = 'int!gration_Test';
|
||||
|
||||
accounts.register(
|
||||
user,
|
||||
password,
|
||||
'',
|
||||
ip,
|
||||
(error, res) => {
|
||||
if (error) {
|
||||
throw error;
|
||||
}
|
||||
|
||||
console.log(`Created test user ${user}`);
|
||||
done();
|
||||
}
|
||||
)
|
||||
});
|
||||
|
||||
it('verifies a correct login', done => {
|
||||
accounts.verifyLogin(
|
||||
user,
|
||||
password,
|
||||
(error, res) => {
|
||||
if (error) {
|
||||
throw error;
|
||||
}
|
||||
|
||||
assert.strictEqual(res.name, user);
|
||||
done();
|
||||
}
|
||||
);
|
||||
});
|
||||
|
||||
it('verifies a correct login with an older hash', done => {
|
||||
testDB.knex.table('users')
|
||||
.where({ name: user })
|
||||
.update({
|
||||
// 'test' hashed with old version of bcrypt module
|
||||
password: '$2b$10$2oCG7O9FFqie7T8O33yQDugFPS0NqkgbQjtThTs7Jr8E1QOzdRruK'
|
||||
})
|
||||
.then(() => {
|
||||
accounts.verifyLogin(
|
||||
user,
|
||||
'test',
|
||||
(error, res) => {
|
||||
if (error) {
|
||||
throw error;
|
||||
}
|
||||
|
||||
assert.strictEqual(res.name, user);
|
||||
done();
|
||||
}
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
it('rejects an incorrect login', done => {
|
||||
accounts.verifyLogin(
|
||||
user,
|
||||
'not the right password',
|
||||
(error, res) => {
|
||||
assert.strictEqual(error, 'Invalid username/password combination');
|
||||
done();
|
||||
}
|
||||
);
|
||||
});
|
||||
});
|
||||
});
|
||||
76
integration_test/db/aliases.js
Normal file
76
integration_test/db/aliases.js
Normal file
|
|
@ -0,0 +1,76 @@
|
|||
const assert = require('assert');
|
||||
const AliasesDB = require('../../lib/db/aliases').AliasesDB;
|
||||
const testDB = require('../testutil/db').testDB;
|
||||
|
||||
const aliasesDB = new AliasesDB(testDB);
|
||||
const testIPs = ['111.111.111.111', '111.111.111.222'];
|
||||
const testNames = ['itest1', 'itest2'];
|
||||
|
||||
function cleanup() {
|
||||
return testDB.knex.table('aliases')
|
||||
.where('ip', 'in', testIPs)
|
||||
.del()
|
||||
.then(() => {
|
||||
return testDB.knex.table('aliases')
|
||||
.where('name', 'in', testNames)
|
||||
.del();
|
||||
});
|
||||
}
|
||||
|
||||
function addSomeAliases() {
|
||||
return cleanup().then(() => {
|
||||
return testDB.knex.table('aliases')
|
||||
.insert([
|
||||
{ ip: testIPs[0], name: testNames[0], time: Date.now() },
|
||||
{ ip: testIPs[0], name: testNames[1], time: Date.now() },
|
||||
{ ip: testIPs[1], name: testNames[1], time: Date.now() }
|
||||
]);
|
||||
});
|
||||
}
|
||||
|
||||
describe('AliasesDB', () => {
|
||||
describe('#addAlias', () => {
|
||||
beforeEach(cleanup);
|
||||
afterEach(cleanup);
|
||||
|
||||
it('adds a new alias', () => {
|
||||
return aliasesDB.addAlias(testIPs[0], testNames[0])
|
||||
.then(() => {
|
||||
return testDB.knex.table('aliases')
|
||||
.where({ ip: testIPs[0], name: testNames[0] })
|
||||
.select()
|
||||
.then(rows => {
|
||||
assert.strictEqual(rows.length, 1, 'expected 1 row');
|
||||
});
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('#getAliasesByIP', () => {
|
||||
beforeEach(addSomeAliases);
|
||||
afterEach(cleanup);
|
||||
|
||||
it('retrieves aliases by IP', () => {
|
||||
return aliasesDB.getAliasesByIP(testIPs[0])
|
||||
.then(names => assert.deepStrictEqual(
|
||||
names.sort(), testNames.sort()));
|
||||
});
|
||||
|
||||
it('retrieves aliases by partial IP', () => {
|
||||
return aliasesDB.getAliasesByIP(testIPs[0].substring(4))
|
||||
.then(names => assert.deepStrictEqual(
|
||||
names.sort(), testNames.sort()));
|
||||
});
|
||||
});
|
||||
|
||||
describe('#getIPsByName', () => {
|
||||
beforeEach(addSomeAliases);
|
||||
afterEach(cleanup);
|
||||
|
||||
it('retrieves IPs by name', () => {
|
||||
return aliasesDB.getIPsByName(testNames[1])
|
||||
.then(ips => assert.deepStrictEqual(
|
||||
ips.sort(), testIPs.sort()));
|
||||
});
|
||||
});
|
||||
});
|
||||
92
integration_test/db/globalban.js
Normal file
92
integration_test/db/globalban.js
Normal file
|
|
@ -0,0 +1,92 @@
|
|||
const assert = require('assert');
|
||||
const GlobalBanDB = require('../../lib/db/globalban').GlobalBanDB;
|
||||
const testDB = require('../testutil/db').testDB;
|
||||
const { o } = require('../testutil/o');
|
||||
|
||||
const globalBanDB = new GlobalBanDB(testDB);
|
||||
const testBan = { ip: '8.8.8.8', reason: 'test' };
|
||||
|
||||
function cleanupTestBan() {
|
||||
return testDB.knex.table('global_bans')
|
||||
.where({ ip: testBan.ip })
|
||||
.del();
|
||||
}
|
||||
|
||||
function setupTestBan() {
|
||||
return testDB.knex.table('global_bans')
|
||||
.insert(testBan)
|
||||
.catch(error => {
|
||||
if (error.code === 'ER_DUP_ENTRY') {
|
||||
return testDB.knex.table('global_bans')
|
||||
.where({ ip: testBan.ip })
|
||||
.update({ reason: testBan.reason });
|
||||
}
|
||||
|
||||
throw error;
|
||||
});
|
||||
}
|
||||
|
||||
describe('GlobalBanDB', () => {
|
||||
describe('#listGlobalBans', () => {
|
||||
beforeEach(setupTestBan);
|
||||
afterEach(cleanupTestBan);
|
||||
|
||||
it('lists existing IP bans', () => {
|
||||
return globalBanDB.listGlobalBans().then(bans => {
|
||||
assert.deepStrictEqual([{
|
||||
ip: '8.8.8.8',
|
||||
reason: 'test'
|
||||
}], bans.map(o));
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('#addGlobalIPBan', () => {
|
||||
beforeEach(cleanupTestBan);
|
||||
afterEach(cleanupTestBan);
|
||||
|
||||
it('adds a new ban', () => {
|
||||
return globalBanDB.addGlobalIPBan('8.8.8.8', 'test').then(() => {
|
||||
return testDB.knex.table('global_bans')
|
||||
.where({ ip: '8.8.8.8' })
|
||||
.select()
|
||||
.then(rows => {
|
||||
assert.strictEqual(rows.length, 1, 'Expected 1 row');
|
||||
assert.strictEqual(rows[0].ip, '8.8.8.8');
|
||||
assert.strictEqual(rows[0].reason, 'test');
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
it('updates the reason on an existing ban', () => {
|
||||
return globalBanDB.addGlobalIPBan('8.8.8.8', 'test').then(() => {
|
||||
return globalBanDB.addGlobalIPBan('8.8.8.8', 'different').then(() => {
|
||||
return testDB.knex.table('global_bans')
|
||||
.where({ ip: '8.8.8.8' })
|
||||
.select()
|
||||
.then(rows => {
|
||||
assert.strictEqual(rows.length, 1, 'Expected 1 row');
|
||||
assert.strictEqual(rows[0].ip, '8.8.8.8');
|
||||
assert.strictEqual(rows[0].reason, 'different');
|
||||
});
|
||||
});
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('#removeGlobalIPBan', () => {
|
||||
beforeEach(setupTestBan);
|
||||
afterEach(cleanupTestBan);
|
||||
|
||||
it('removes a ban', () => {
|
||||
return globalBanDB.removeGlobalIPBan('8.8.8.8').then(() => {
|
||||
return testDB.knex.table('global_bans')
|
||||
.where({ ip: '8.8.8.8' })
|
||||
.select()
|
||||
.then(rows => {
|
||||
assert.strictEqual(rows.length, 0, 'Expected 0 rows');
|
||||
});
|
||||
});
|
||||
});
|
||||
});
|
||||
});
|
||||
144
integration_test/db/password-reset.js
Normal file
144
integration_test/db/password-reset.js
Normal file
|
|
@ -0,0 +1,144 @@
|
|||
const assert = require('assert');
|
||||
const PasswordResetDB = require('../../lib/db/password-reset').PasswordResetDB;
|
||||
const testDB = require('../testutil/db').testDB;
|
||||
const { o } = require('../testutil/o');
|
||||
|
||||
const passwordResetDB = new PasswordResetDB(testDB);
|
||||
|
||||
function cleanup() {
|
||||
return testDB.knex.table('password_reset').del();
|
||||
}
|
||||
|
||||
describe('PasswordResetDB', () => {
|
||||
describe('#insert', () => {
|
||||
beforeEach(cleanup);
|
||||
|
||||
const params = {
|
||||
ip: '1.2.3.4',
|
||||
name: 'testing',
|
||||
email: 'test@example.com',
|
||||
hash: 'abcdef',
|
||||
expire: 5678
|
||||
};
|
||||
|
||||
it('adds a new password reset', () => {
|
||||
return passwordResetDB.insert(params).then(() => {
|
||||
return testDB.knex.table('password_reset')
|
||||
.where({ name: 'testing' })
|
||||
.select();
|
||||
}).then(rows => {
|
||||
assert.strictEqual(rows.length, 1);
|
||||
assert.deepStrictEqual(o(rows[0]), params);
|
||||
});
|
||||
});
|
||||
|
||||
it('overwrites an existing reset for the same name', () => {
|
||||
return passwordResetDB.insert(params).then(() => {
|
||||
params.ip = '5.6.7.8';
|
||||
params.email = 'somethingelse@example.com';
|
||||
params.hash = 'qwertyuiop';
|
||||
params.expire = 9999;
|
||||
|
||||
return passwordResetDB.insert(params);
|
||||
}).then(() => {
|
||||
return testDB.knex.table('password_reset')
|
||||
.where({ name: 'testing' })
|
||||
.select();
|
||||
}).then(rows => {
|
||||
assert.strictEqual(rows.length, 1);
|
||||
assert.deepStrictEqual(o(rows[0]), params);
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('#get', () => {
|
||||
const reset = {
|
||||
ip: '1.2.3.4',
|
||||
name: 'testing',
|
||||
email: 'test@example.com',
|
||||
hash: 'abcdef',
|
||||
expire: 5678
|
||||
};
|
||||
|
||||
beforeEach(() => cleanup().then(() => {
|
||||
return testDB.knex.table('password_reset').insert(reset);
|
||||
}));
|
||||
|
||||
it('gets a password reset by hash', () => {
|
||||
return passwordResetDB.get(reset.hash).then(result => {
|
||||
assert.deepStrictEqual(o(result), reset);
|
||||
});
|
||||
});
|
||||
|
||||
it('throws when no reset exists for the input', () => {
|
||||
return passwordResetDB.get('lalala').then(() => {
|
||||
assert.fail('Expected not found error');
|
||||
}).catch(error => {
|
||||
assert.strictEqual(
|
||||
error.message,
|
||||
'No password reset found for hash lalala'
|
||||
);
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('#delete', () => {
|
||||
const reset = {
|
||||
ip: '1.2.3.4',
|
||||
name: 'testing',
|
||||
email: 'test@example.com',
|
||||
hash: 'abcdef',
|
||||
expire: 5678
|
||||
};
|
||||
|
||||
beforeEach(() => cleanup().then(() => {
|
||||
return testDB.knex.table('password_reset').insert(reset);
|
||||
}));
|
||||
|
||||
it('deletes a password reset by hash', () => {
|
||||
return passwordResetDB.delete(reset.hash).then(() => {
|
||||
return testDB.knex.table('password_reset')
|
||||
.where({ name: 'testing' })
|
||||
.select();
|
||||
}).then(rows => {
|
||||
assert.strictEqual(rows.length, 0);
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('#cleanup', () => {
|
||||
const now = Date.now();
|
||||
|
||||
const reset1 = {
|
||||
ip: '1.2.3.4',
|
||||
name: 'testing',
|
||||
email: 'test@example.com',
|
||||
hash: 'abcdef',
|
||||
expire: now - 25 * 60 * 60 * 1000
|
||||
};
|
||||
|
||||
const reset2 = {
|
||||
ip: '5.6.7.8',
|
||||
name: 'testing2',
|
||||
email: 'test@example.com',
|
||||
hash: 'abcdef',
|
||||
expire: now
|
||||
};
|
||||
|
||||
beforeEach(() => cleanup().then(() => {
|
||||
return testDB.knex.table('password_reset')
|
||||
.insert([reset1, reset2]);
|
||||
}));
|
||||
|
||||
it('cleans up old password resets', () => {
|
||||
return passwordResetDB.cleanup().then(() => {
|
||||
return testDB.knex.table('password_reset')
|
||||
.whereIn('name', ['testing1', 'testing2'])
|
||||
.select();
|
||||
}).then(rows => {
|
||||
assert.strictEqual(rows.length, 1);
|
||||
assert.deepStrictEqual(o(rows[0]), reset2);
|
||||
});
|
||||
});
|
||||
});
|
||||
});
|
||||
136
integration_test/regressions/checkban-blank-name.js
Normal file
136
integration_test/regressions/checkban-blank-name.js
Normal file
|
|
@ -0,0 +1,136 @@
|
|||
const assert = require('assert');
|
||||
const KickbanModule = require('../../lib/channel/kickban');
|
||||
const database = require('../../lib/database');
|
||||
const dbChannels = require('../../lib/database/channels');
|
||||
const Promise = require('bluebird');
|
||||
const ChannelModule = require('../../lib/channel/module');
|
||||
const Flags = require('../../lib/flags');
|
||||
const testDB = require('../testutil/db').testDB;
|
||||
|
||||
function randomString(length) {
|
||||
const chars = 'abcdefgihkmnpqrstuvwxyz0123456789';
|
||||
let str = '';
|
||||
for (let i = 0; i < length; i++) {
|
||||
str += chars[Math.floor(Math.random() * chars.length)];
|
||||
}
|
||||
return str;
|
||||
}
|
||||
|
||||
database.init(testDB);
|
||||
|
||||
describe('onPreUserJoin Ban Check', () => {
|
||||
const channelName = `test_${randomString(20)}`;
|
||||
const bannedIP = '1.1.1.1';
|
||||
const bannedName = 'troll';
|
||||
const mockChannel = {
|
||||
name: channelName,
|
||||
modules: {},
|
||||
is(flag) {
|
||||
return flag === Flags.C_REGISTERED;
|
||||
}
|
||||
};
|
||||
const module = new KickbanModule(mockChannel);
|
||||
before(done => {
|
||||
dbChannels.ban(channelName, bannedIP, bannedName, '', '', () => {
|
||||
dbChannels.ban(channelName, bannedIP, '', '', '', () => {
|
||||
dbChannels.ban(channelName, '*', bannedName, '', '', () => {
|
||||
done();
|
||||
});
|
||||
});
|
||||
});
|
||||
});
|
||||
after(done => {
|
||||
dbChannels.deleteBans(channelName, null, () => {
|
||||
done();
|
||||
});
|
||||
});
|
||||
|
||||
it('handles a banned IP with a different name', done => {
|
||||
const user = {
|
||||
getName() {
|
||||
return 'anotherTroll';
|
||||
},
|
||||
|
||||
realip: bannedIP,
|
||||
|
||||
kick() {
|
||||
}
|
||||
};
|
||||
|
||||
module.onUserPreJoin(user, null, (error, res) => {
|
||||
assert.equal(error, null, `Unexpected error: ${error}`);
|
||||
assert.equal(res, ChannelModule.DENY, 'Expected user to be banned');
|
||||
done();
|
||||
});
|
||||
});
|
||||
|
||||
it('handles a banned name with a different IP', done => {
|
||||
const user = {
|
||||
getName() {
|
||||
return 'troll';
|
||||
},
|
||||
|
||||
realip: '5.5.5.5',
|
||||
|
||||
kick() {
|
||||
}
|
||||
};
|
||||
|
||||
module.onUserPreJoin(user, null, (error, res) => {
|
||||
assert.equal(error, null, `Unexpected error: ${error}`);
|
||||
assert.equal(res, ChannelModule.DENY, 'Expected user to be banned');
|
||||
done();
|
||||
});
|
||||
});
|
||||
|
||||
it('handles a banned IP with a blank name', done => {
|
||||
const user = {
|
||||
getName() {
|
||||
return '';
|
||||
},
|
||||
|
||||
realip: bannedIP,
|
||||
|
||||
kick() {
|
||||
}
|
||||
};
|
||||
|
||||
module.onUserPreJoin(user, null, (error, res) => {
|
||||
assert.equal(error, null, `Unexpected error: ${error}`);
|
||||
assert.equal(res, ChannelModule.DENY, 'Expected user to be banned');
|
||||
done();
|
||||
});
|
||||
});
|
||||
|
||||
it('handles a non-banned IP with a blank name', done => {
|
||||
const user = {
|
||||
getName() {
|
||||
return '';
|
||||
},
|
||||
|
||||
realip: '5.5.5.5'
|
||||
};
|
||||
|
||||
module.onUserPreJoin(user, null, (error, res) => {
|
||||
assert.equal(error, null, `Unexpected error: ${error}`);
|
||||
assert.equal(res, ChannelModule.PASSTHROUGH, 'Expected user not to be banned');
|
||||
done();
|
||||
});
|
||||
});
|
||||
|
||||
it('handles a non-banned IP with a non-banned name', done => {
|
||||
const user = {
|
||||
getName() {
|
||||
return 'some_user';
|
||||
},
|
||||
|
||||
realip: '5.5.5.5'
|
||||
};
|
||||
|
||||
module.onUserPreJoin(user, null, (error, res) => {
|
||||
assert.equal(error, null, `Unexpected error: ${error}`);
|
||||
assert.equal(res, ChannelModule.PASSTHROUGH, 'Expected user not to be banned');
|
||||
done();
|
||||
});
|
||||
});
|
||||
});
|
||||
14
integration_test/testutil/config.js
Normal file
14
integration_test/testutil/config.js
Normal file
|
|
@ -0,0 +1,14 @@
|
|||
const loadFromToml = require('../../lib/configuration/configloader').loadFromToml;
|
||||
const path = require('path');
|
||||
|
||||
class IntegrationTestConfig {
|
||||
constructor(config) {
|
||||
this.config = config;
|
||||
}
|
||||
|
||||
get knexConfig() {
|
||||
return this.config.database;
|
||||
}
|
||||
}
|
||||
|
||||
exports.testConfig = loadFromToml(IntegrationTestConfig, path.resolve(__dirname, '..', '..', 'conf', 'integration-test.toml'));
|
||||
4
integration_test/testutil/db.js
Normal file
4
integration_test/testutil/db.js
Normal file
|
|
@ -0,0 +1,4 @@
|
|||
const testConfig = require('./config').testConfig;
|
||||
const Database = require('../../lib/database').Database;
|
||||
|
||||
exports.testDB = new Database(testConfig.knexConfig);
|
||||
4
integration_test/testutil/o.js
Normal file
4
integration_test/testutil/o.js
Normal file
|
|
@ -0,0 +1,4 @@
|
|||
exports.o = function o(obj) {
|
||||
// Workaround for knex returning RowDataPacket and failing assertions
|
||||
return Object.assign({}, obj);
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue