mirror of
https://github.com/NodeBB/NodeBB.git
synced 2025-10-30 18:46:01 +01:00
354 lines
8.5 KiB
JavaScript
354 lines
8.5 KiB
JavaScript
'use strict';
|
|
|
|
|
|
const async = require('async');
|
|
const assert = require('assert');
|
|
const db = require('../mocks/databasemock');
|
|
|
|
describe('Key methods', () => {
|
|
beforeEach((done) => {
|
|
db.set('testKey', 'testValue', done);
|
|
});
|
|
|
|
it('should set a key without error', (done) => {
|
|
db.set('testKey', 'testValue', function (err) {
|
|
assert.ifError(err);
|
|
assert(arguments.length < 2);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should get a key without error', (done) => {
|
|
db.get('testKey', function (err, value) {
|
|
assert.ifError(err);
|
|
assert.equal(arguments.length, 2);
|
|
assert.strictEqual(value, 'testValue');
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should return null if key does not exist', (done) => {
|
|
db.get('doesnotexist', (err, value) => {
|
|
assert.ifError(err);
|
|
assert.equal(value, null);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should return true if key exist', (done) => {
|
|
db.exists('testKey', function (err, exists) {
|
|
assert.ifError(err);
|
|
assert.equal(arguments.length, 2);
|
|
assert.strictEqual(exists, true);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should return false if key does not exist', (done) => {
|
|
db.exists('doesnotexist', function (err, exists) {
|
|
assert.ifError(err);
|
|
assert.equal(arguments.length, 2);
|
|
assert.strictEqual(exists, false);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should work for an array of keys', (done) => {
|
|
db.exists(['testKey', 'doesnotexist'], (err, exists) => {
|
|
assert.ifError(err);
|
|
assert.deepStrictEqual(exists, [true, false]);
|
|
done();
|
|
});
|
|
});
|
|
|
|
describe('scan', () => {
|
|
it('should scan keys for pattern', async () => {
|
|
await db.sortedSetAdd('ip:123:uid', 1, 'a');
|
|
await db.sortedSetAdd('ip:123:uid', 2, 'b');
|
|
await db.sortedSetAdd('ip:124:uid', 2, 'b');
|
|
await db.sortedSetAdd('ip:1:uid', 1, 'a');
|
|
await db.sortedSetAdd('ip:23:uid', 1, 'a');
|
|
const data = await db.scan({ match: 'ip:1*' });
|
|
assert.equal(data.length, 3);
|
|
assert(data.includes('ip:123:uid'));
|
|
assert(data.includes('ip:124:uid'));
|
|
assert(data.includes('ip:1:uid'));
|
|
});
|
|
});
|
|
|
|
it('should delete a key without error', (done) => {
|
|
db.delete('testKey', function (err) {
|
|
assert.ifError(err);
|
|
assert(arguments.length < 2);
|
|
|
|
db.get('testKey', (err, value) => {
|
|
assert.ifError(err);
|
|
assert.equal(false, !!value);
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
it('should return false if key was deleted', (done) => {
|
|
db.delete('testKey', function (err) {
|
|
assert.ifError(err);
|
|
assert(arguments.length < 2);
|
|
db.exists('testKey', (err, exists) => {
|
|
assert.ifError(err);
|
|
assert.strictEqual(exists, false);
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
it('should delete all keys passed in', (done) => {
|
|
async.parallel([
|
|
function (next) {
|
|
db.set('key1', 'value1', next);
|
|
},
|
|
function (next) {
|
|
db.set('key2', 'value2', next);
|
|
},
|
|
], (err) => {
|
|
if (err) {
|
|
return done(err);
|
|
}
|
|
db.deleteAll(['key1', 'key2'], function (err) {
|
|
assert.ifError(err);
|
|
assert.equal(arguments.length, 1);
|
|
async.parallel({
|
|
key1exists: function (next) {
|
|
db.exists('key1', next);
|
|
},
|
|
key2exists: function (next) {
|
|
db.exists('key2', next);
|
|
},
|
|
}, (err, results) => {
|
|
assert.ifError(err);
|
|
assert.equal(results.key1exists, false);
|
|
assert.equal(results.key2exists, false);
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
});
|
|
|
|
it('should delete all sorted set elements', (done) => {
|
|
async.parallel([
|
|
function (next) {
|
|
db.sortedSetAdd('deletezset', 1, 'value1', next);
|
|
},
|
|
function (next) {
|
|
db.sortedSetAdd('deletezset', 2, 'value2', next);
|
|
},
|
|
], (err) => {
|
|
if (err) {
|
|
return done(err);
|
|
}
|
|
db.delete('deletezset', (err) => {
|
|
assert.ifError(err);
|
|
async.parallel({
|
|
key1exists: function (next) {
|
|
db.isSortedSetMember('deletezset', 'value1', next);
|
|
},
|
|
key2exists: function (next) {
|
|
db.isSortedSetMember('deletezset', 'value2', next);
|
|
},
|
|
}, (err, results) => {
|
|
assert.ifError(err);
|
|
assert.equal(results.key1exists, false);
|
|
assert.equal(results.key2exists, false);
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('increment', () => {
|
|
it('should initialize key to 1', (done) => {
|
|
db.increment('keyToIncrement', (err, value) => {
|
|
assert.ifError(err);
|
|
assert.strictEqual(parseInt(value, 10), 1);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should increment key to 2', (done) => {
|
|
db.increment('keyToIncrement', (err, value) => {
|
|
assert.ifError(err);
|
|
assert.strictEqual(parseInt(value, 10), 2);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should set then increment a key', (done) => {
|
|
db.set('myIncrement', 1, (err) => {
|
|
assert.ifError(err);
|
|
db.increment('myIncrement', (err, value) => {
|
|
assert.ifError(err);
|
|
assert.equal(value, 2);
|
|
db.get('myIncrement', (err, value) => {
|
|
assert.ifError(err);
|
|
assert.equal(value, 2);
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
});
|
|
|
|
it('should return the correct value', (done) => {
|
|
db.increment('testingCache', (err) => {
|
|
assert.ifError(err);
|
|
db.get('testingCache', (err, value) => {
|
|
assert.ifError(err);
|
|
assert.equal(value, 1);
|
|
db.increment('testingCache', (err) => {
|
|
assert.ifError(err);
|
|
db.get('testingCache', (err, value) => {
|
|
assert.ifError(err);
|
|
assert.equal(value, 2);
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('rename', () => {
|
|
it('should rename key to new name', (done) => {
|
|
db.set('keyOldName', 'renamedKeyValue', (err) => {
|
|
if (err) {
|
|
return done(err);
|
|
}
|
|
db.rename('keyOldName', 'keyNewName', function (err) {
|
|
assert.ifError(err);
|
|
assert(arguments.length < 2);
|
|
|
|
db.get('keyNewName', (err, value) => {
|
|
assert.ifError(err);
|
|
assert.equal(value, 'renamedKeyValue');
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
});
|
|
|
|
it('should rename multiple keys', (done) => {
|
|
db.sortedSetAdd('zsettorename', [1, 2, 3], ['value1', 'value2', 'value3'], (err) => {
|
|
assert.ifError(err);
|
|
db.rename('zsettorename', 'newzsetname', (err) => {
|
|
assert.ifError(err);
|
|
db.exists('zsettorename', (err, exists) => {
|
|
assert.ifError(err);
|
|
assert(!exists);
|
|
db.getSortedSetRange('newzsetname', 0, -1, (err, values) => {
|
|
assert.ifError(err);
|
|
assert.deepEqual(['value1', 'value2', 'value3'], values);
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
});
|
|
});
|
|
|
|
it('should not error if old key does not exist', (done) => {
|
|
db.rename('doesnotexist', 'anotherdoesnotexist', (err) => {
|
|
assert.ifError(err);
|
|
db.exists('anotherdoesnotexist', (err, exists) => {
|
|
assert.ifError(err);
|
|
assert(!exists);
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('type', () => {
|
|
it('should return null if key does not exist', (done) => {
|
|
db.type('doesnotexist', (err, type) => {
|
|
assert.ifError(err);
|
|
assert.strictEqual(type, null);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should return hash as type', (done) => {
|
|
db.setObject('typeHash', { foo: 1 }, (err) => {
|
|
assert.ifError(err);
|
|
db.type('typeHash', (err, type) => {
|
|
assert.ifError(err);
|
|
assert.equal(type, 'hash');
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
it('should return zset as type', (done) => {
|
|
db.sortedSetAdd('typeZset', 123, 'value1', (err) => {
|
|
assert.ifError(err);
|
|
db.type('typeZset', (err, type) => {
|
|
assert.ifError(err);
|
|
assert.equal(type, 'zset');
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
it('should return set as type', (done) => {
|
|
db.setAdd('typeSet', 'value1', (err) => {
|
|
assert.ifError(err);
|
|
db.type('typeSet', (err, type) => {
|
|
assert.ifError(err);
|
|
assert.equal(type, 'set');
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
it('should return list as type', (done) => {
|
|
db.listAppend('typeList', 'value1', (err) => {
|
|
assert.ifError(err);
|
|
db.type('typeList', (err, type) => {
|
|
assert.ifError(err);
|
|
assert.equal(type, 'list');
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
it('should return string as type', (done) => {
|
|
db.set('typeString', 'value1', (err) => {
|
|
assert.ifError(err);
|
|
db.type('typeString', (err, type) => {
|
|
assert.ifError(err);
|
|
assert.equal(type, 'string');
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
it('should expire a key using seconds', (done) => {
|
|
db.expire('testKey', 86400, (err) => {
|
|
assert.ifError(err);
|
|
db.ttl('testKey', (err, ttl) => {
|
|
assert.ifError(err);
|
|
assert.equal(Math.round(86400 / 1000), Math.round(ttl / 1000));
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
it('should expire a key using milliseconds', (done) => {
|
|
db.pexpire('testKey', 86400000, (err) => {
|
|
assert.ifError(err);
|
|
db.pttl('testKey', (err, pttl) => {
|
|
assert.ifError(err);
|
|
assert.equal(Math.round(86400000 / 1000000), Math.round(pttl / 1000000));
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
});
|
|
});
|