first commit

2023-05-19 00:42:48 +08:00
commit 53de9c6c51
243 changed files with 39485 additions and 0 deletions
@@ -0,0 +1,190 @@
+'use strict';
+
+require('../../helper');
+
+var assert = require('../../support/assert');
+var JobStatus = require('../../../batch/job_status');
+var BatchTestClient = require('../../support/batch-test-client');
+
+describe('Use cases', function () {
+    before(function() {
+        this.batchTestClient = new BatchTestClient();
+    });
+
+    after(function(done) {
+        this.batchTestClient.drain(done);
+    });
+
+    it('cancel a done job should return an error', function (done) {
+        var payload = {
+            query: "SELECT * FROM untitle_table_4"
+        };
+
+        this.batchTestClient.createJob(payload, function(err, jobResult) {
+            if (err) {
+                return done(err);
+            }
+
+            jobResult.getStatus(JobStatus.DONE, function (err, job) {
+                if (err) {
+                    return done(err);
+                }
+
+                assert.equal(job.status, JobStatus.DONE);
+
+                jobResult.tryCancel(function (err, body) {
+                    assert.equal(body.error[0], "Cannot set status from done to cancelled");
+                    done();
+                });
+            });
+        });
+    });
+
+    it('cancel a running job', function (done) {
+        var payload = {
+            query: "SELECT * FROM untitle_table_4; select pg_sleep(3)"
+        };
+
+        this.batchTestClient.createJob(payload, function(err, jobResult) {
+            if (err) {
+                return done(err);
+            }
+
+            jobResult.getStatus(JobStatus.RUNNING, function (err, job) {
+                if (err) {
+                    return done(err);
+                }
+
+                assert.equal(job.status, JobStatus.RUNNING);
+
+                jobResult.cancel(function (err, job) {
+                    if (err) {
+                        return done(err);
+                    }
+
+                    assert.equal(job.status, JobStatus.CANCELLED);
+
+                    jobResult.tryCancel(function (err, body) {
+                        assert.equal(body.error[0], "Cannot set status from cancelled to cancelled");
+                        done();
+                    });
+                });
+            });
+        });
+    });
+
+    it('cancel a pending job', function (done) {
+        var self = this;
+        var payload1 = {
+            query: "SELECT * FROM untitle_table_4; select pg_sleep(3)"
+        };
+
+        this.batchTestClient.createJob(payload1, function(err, jobResult1) {
+            if (err) {
+                return done(err);
+            }
+
+            var payload2 = {
+                query: "SELECT * FROM untitle_table_4"
+            };
+
+            self.batchTestClient.createJob(payload2, function(err, jobResult2) {
+                if (err) {
+                    return done(err);
+                }
+
+                jobResult2.getStatus(JobStatus.PENDING, function (err, job) {
+                    if (err) {
+                        return done(err);
+                    }
+
+                    assert.equal(job.status, JobStatus.PENDING);
+
+                    jobResult2.cancel(function (err, job) {
+                        if (err) {
+                            return done(err);
+                        }
+
+                        assert.equal(job.status, JobStatus.CANCELLED);
+
+                        jobResult1.cancel(function (err, job) {
+                            if (err) {
+                                return done(err);
+                            }
+
+                            assert.equal(job.status, JobStatus.CANCELLED);
+                            done();
+                        });
+
+                    });
+                });
+            });
+        });
+    });
+
+    it('cancel a job with quotes', function (done) {
+        var payload = {
+            query: "SELECT name FROM untitle_table_4 WHERE name = 'Hawai'; select pg_sleep(3)"
+        };
+
+        this.batchTestClient.createJob(payload, function(err, jobResult) {
+            if (err) {
+                return done(err);
+            }
+
+            jobResult.getStatus(JobStatus.RUNNING, function (err, job) {
+                if (err) {
+                    return done(err);
+                }
+
+                assert.equal(job.status, JobStatus.RUNNING);
+
+                jobResult.cancel(function (err, job) {
+                    if (err) {
+                        return done(err);
+                    }
+
+                    assert.equal(job.status, JobStatus.CANCELLED);
+                    done();
+                });
+            });
+        });
+    });
+
+    it('cancel a running multiquery job', function (done) {
+        var payload = {
+            query: [
+                "select pg_sleep(1)",
+                "select pg_sleep(1)",
+                "select pg_sleep(1)"
+            ]
+        };
+
+        this.batchTestClient.createJob(payload, function(err, jobResult) {
+            if (err) {
+                return done(err);
+            }
+
+            jobResult.getStatus(JobStatus.RUNNING, function (err, job) {
+                if (err) {
+                    return done(err);
+                }
+
+                assert.equal(job.status, JobStatus.RUNNING);
+
+                jobResult.cancel(function (err, job) {
+                    if (err) {
+                        return done(err);
+                    }
+
+                    assert.equal(job.status, JobStatus.CANCELLED);
+
+                    jobResult.tryCancel(function (err, body) {
+                        assert.equal(body.error[0], "Cannot set status from cancelled to cancelled");
+                        done();
+                    });
+                });
+            });
+        });
+    });
+});