1
0
Fork 0
arangodb/js/common/tests/shell/shell-query-timecritical-sp...

253 lines
7.5 KiB
JavaScript

/* jshint expr: true */
/* eslint no-unused-expressions: 0 */
/* global describe, it, beforeEach, afterEach */
'use strict';
const internal = require('internal');
const tasks = require('@arangodb/tasks');
const expect = require('chai').expect;
const isServer = typeof arango === 'undefined';
const query = 'FOR x IN 1..5 LET y = SLEEP(1) RETURN x';
const taskInfo = {
offset: 0,
command: function () {
var taskQuery = 'FOR x IN 1..5 LET y = SLEEP(1) RETURN x';
require('internal').db._query(taskQuery);
}
};
function filterQueries (q) {
return (q.query === query);
}
function sendQuery (count, async) {
count = count || 1;
for (let i = 0; i < count; ++i) {
if (async === false) {
internal.db._query(query);
} else {
tasks.register(taskInfo);
}
}
if (async === true) {
internal.wait(1);
}
}
function restoreDefaults (testee) {
testee.properties({
enabled: true,
trackSlowQueries: true,
maxSlowQueries: 64,
slowQueryThreshold: 10,
maxQueryStringLength: 8192
});
}
describe('AQL query analyzer', function () {
let testee;
beforeEach(function () {
testee = require('@arangodb/aql/queries');
restoreDefaults(testee);
});
afterEach(function () {
restoreDefaults(testee);
});
it('should be able to activate the tracking', function () {
testee.properties({
enabled: true
});
expect(testee.properties().enabled).to.be.ok;
});
it('should be able to deactivate the tracking', function () {
testee.properties({
enabled: false
});
expect(testee.properties().enabled).not.to.be.ok;
});
it('should be able to set tracking properties', function () {
testee.properties({
enabled: true,
trackSlowQueries: false,
maxSlowQueries: 42,
slowQueryThreshold: 2.5,
maxQueryStringLength: 117
});
expect(testee.properties().enabled).to.be.ok;
expect(testee.properties().trackSlowQueries).not.to.be.ok;
expect(testee.properties().maxSlowQueries).to.equal(42);
expect(testee.properties().slowQueryThreshold).to.equal(2.5);
expect(testee.properties().maxQueryStringLength).to.equal(117);
});
describe('with active tracking', function () {
beforeEach(function () {
if (isServer && internal.debugCanUseFailAt()) {
internal.debugClearFailAt();
}
testee.properties({
enabled: true,
slowQueryThreshold: 20
});
testee.clearSlow();
});
afterEach(function () {
if (isServer && internal.debugCanUseFailAt()) {
internal.debugClearFailAt();
}
const list = testee.current().filter(filterQueries);
for (let item of list) {
try {
testee.kill(item.id);
} catch (e) {
// noop
}
}
});
if (isServer && internal.debugCanUseFailAt()) {
it('should not crash when inserting a query into the current list fails', function () {
internal.debugSetFailAt('QueryList::insert');
// inserting the query will fail
sendQuery(1, true);
expect(testee.current().filter(filterQueries).length).to.equal(0);
});
}
it('should be able to get currently running queries', function () {
sendQuery(1, true);
expect(testee.current().filter(filterQueries).length).to.equal(1);
});
it('should have proper running query descriptions', function () {
sendQuery(1, true);
let queries = testee.current().filter(filterQueries);
expect(queries.length).to.equal(1);
expect(queries[0]).to.have.property('id');
expect(queries[0]).to.have.property('query', query);
expect(queries[0]).to.have.property('bindVars');
expect(queries[0].bindVars).to.eql({});
expect(queries[0]).to.have.property('started');
expect(queries[0]).to.have.property('runTime');
expect(queries[0]).to.have.property('state', 'executing');
});
it('should not track queries if turned off', function () {
testee.properties({
enabled: false
});
sendQuery(1, false);
expect(testee.current().filter(filterQueries).length).to.equal(0);
});
it('should work when tracking is turned off in the middle', function () {
sendQuery(1, true);
expect(testee.current().filter(filterQueries).length).to.equal(1);
// turn it off now
testee.properties({
enabled: false
});
// wait a while, and expect the list of queries to be empty in the end
for (let tries = 0; tries < 20; tries++) {
if (testee.current().filter(filterQueries).length === 0) {
break;
}
internal.wait(1);
}
expect(testee.current().filter(filterQueries).length).to.equal(0);
});
it('should track slow queries by threshold', function () {
sendQuery(1, false);
expect(testee.current().filter(filterQueries).length).to.equal(0);
expect(testee.slow().filter(filterQueries).length).to.equal(0);
testee.properties({
slowQueryThreshold: 2
});
sendQuery(1, false);
expect(testee.current().filter(filterQueries).length).to.equal(0);
let queries = testee.slow().filter(filterQueries);
expect(queries.length).to.equal(1);
expect(queries[0]).to.have.property('id');
expect(queries[0]).to.have.property('query', query);
expect(queries[0]).to.have.property('bindVars');
expect(queries[0].bindVars).to.eql({});
expect(queries[0]).to.have.property('started');
expect(queries[0]).to.have.property('runTime');
expect(queries[0]).to.have.property('state', 'finished');
});
it('should be able to clear the list of slow queries', function () {
testee.properties({
slowQueryThreshold: 2
});
sendQuery(1, false);
expect(testee.slow().filter(filterQueries).length).to.equal(1);
testee.clearSlow();
expect(testee.slow().filter(filterQueries).length).to.equal(0);
});
it('should track at most n slow queries', function () {
const max = 2;
testee.properties({
slowQueryThreshold: 2,
maxSlowQueries: max
});
sendQuery(3, false);
expect(testee.current().filter(filterQueries).length).to.equal(0);
expect(testee.slow().filter(filterQueries).length).to.equal(max);
});
if (isServer && internal.debugCanUseFailAt()) {
it('should not crash when trying to move a query into the slow list', function () {
internal.debugSetFailAt('QueryList::remove');
testee.properties({
slowQueryThreshold: 2
});
sendQuery(3, false);
// no slow queries should have been logged
expect(testee.slow().filter(filterQueries).length).to.equal(0);
});
}
it('should not track slow queries if turned off', function () {
testee.properties({
slowQueryThreshold: 2,
trackSlowQueries: false
});
sendQuery(1, false);
expect(testee.current().filter(filterQueries).length).to.equal(0);
expect(testee.slow().filter(filterQueries).length).to.equal(0);
});
it('should be able to kill a query', function () {
sendQuery(1, true);
let list = testee.current().filter(filterQueries);
expect(list.length).to.equal(1);
expect(function () {
testee.kill(list[0].id);
list = testee.current().filter(filterQueries);
for (let i = 0; i < 10 && list.length > 0; ++i) {
internal.wait(0.1);
list = testee.current().filter(filterQueries);
}
expect(list.length).to.equal(0);
}).not.to.throw();
});
});
});