first commit

2023-05-19 00:42:48 +08:00
commit 53de9c6c51
243 changed files with 39485 additions and 0 deletions
@@ -0,0 +1,99 @@
+'use strict';
+
+require('../../helper');
+var assert = require('../../support/assert');
+
+var BatchTestClient = require('../../support/batch-test-client');
+var JobStatus = require('../../../batch/job_status');
+
+describe('job query timeout', function() {
+
+    before(function() {
+        this.batchQueryTimeout = global.settings.batch_query_timeout;
+        this.batchTestClient = new BatchTestClient();
+    });
+
+    after(function (done) {
+        global.settings.batch_query_timeout = this.batchQueryTimeout;
+        return this.batchTestClient.drain(done);
+    });
+
+    function createTimeoutQuery(query, timeout) {
+        return {
+            query: {
+                query: [
+                    {
+                        timeout: timeout,
+                        query: query
+                    }
+                ]
+            }
+        };
+    }
+
+    it('should run query with higher user timeout', function (done) {
+        var jobRequest = createTimeoutQuery("select pg_sleep(0.1)", 200);
+        this.batchTestClient.createJob(jobRequest, function(err, jobResult) {
+            if (err) {
+                return done(err);
+            }
+            jobResult.getStatus(function(err, job) {
+                if (err) {
+                    return done(err);
+                }
+                assert.equal(job.status, JobStatus.DONE);
+                done();
+            });
+        });
+    });
+
+    it('should fail to run query with lower user timeout', function (done) {
+        var jobRequest = createTimeoutQuery("select pg_sleep(0.1)", 50);
+        this.batchTestClient.createJob(jobRequest, function(err, jobResult) {
+            if (err) {
+                return done(err);
+            }
+            jobResult.getStatus(function(err, job) {
+                if (err) {
+                    return done(err);
+                }
+                assert.equal(job.status, JobStatus.FAILED);
+                done();
+            });
+        });
+    });
+
+    it('should fail to run query with user timeout if it is higher than config', function (done) {
+        global.settings.batch_query_timeout = 100;
+        var jobRequest = createTimeoutQuery("select pg_sleep(1)", 2000);
+        this.batchTestClient.createJob(jobRequest, function(err, jobResult) {
+            if (err) {
+                return done(err);
+            }
+            jobResult.getStatus(function(err, job) {
+                if (err) {
+                    return done(err);
+                }
+                assert.equal(job.status, JobStatus.FAILED);
+                done();
+            });
+        });
+    });
+
+    it('should fail to run query with user timeout if set to 0 (ignored timeout)', function (done) {
+        global.settings.batch_query_timeout = 100;
+        var jobRequest = createTimeoutQuery("select pg_sleep(1)", 0);
+        this.batchTestClient.createJob(jobRequest, function(err, jobResult) {
+            if (err) {
+                return done(err);
+            }
+            jobResult.getStatus(function(err, job) {
+                if (err) {
+                    return done(err);
+                }
+                assert.equal(job.status, JobStatus.FAILED);
+                done();
+            });
+        });
+    });
+});