first commit

2023-05-19 00:42:48 +08:00
commit 53de9c6c51
243 changed files with 39485 additions and 0 deletions
@@ -0,0 +1,126 @@
+'use strict';
+
+require('../../helper');
+
+var assert = require('../../support/assert');
+var TestClient = require('../../support/test-client');
+var JobStatus = require('../../../batch/job_status');
+var BatchTestClient = require('../../support/batch-test-client');
+
+describe('Batch API callback templates', function () {
+    before(function () {
+        this.batchTestClient = new BatchTestClient();
+        this.testClient = new TestClient();
+    });
+
+    after(function (done) {
+        this.batchTestClient.drain(done);
+    });
+
+    it('should use templates for error_message and job_id onerror callback' +
+        ' and keep the original templated query but use the error message', function (done) {
+        var self = this;
+        var payload = {
+            "query": {
+                "query": [
+                    {
+                        "query": "SELECT * FROM invalid_table",
+                        "onerror": "INSERT INTO test_batch_errors " +
+                        "values ('<%= job_id %>', '<%= error_message %>')"
+                    }
+                ]
+            }
+        };
+        var expectedQuery = {
+            query: [
+                {
+                    "query": "SELECT * FROM invalid_table",
+                    "onerror": "INSERT INTO test_batch_errors values ('<%= job_id %>', '<%= error_message %>')",
+                    status: 'failed',
+                    fallback_status: 'done'
+                }
+            ]
+        };
+
+        self.testClient.getResult(
+            'create table test_batch_errors (job_id text, error_message text)', function (err) {
+            if (err) {
+                return done(err);
+            }
+
+            self.batchTestClient.createJob(payload, function(err, jobResult) {
+                if (err) {
+                    return done(err);
+                }
+
+                jobResult.getStatus(JobStatus.FAILED, function (err, job) {
+                    if (err) {
+                        return done(err);
+                    }
+                    jobResult.validateExpectedResponse(expectedQuery);
+                    self.testClient.getResult('select * from test_batch_errors', function(err, rows) {
+                        if (err) {
+                            return done(err);
+                        }
+                        assert.equal(rows[0].job_id, job.job_id);
+                        assert.equal(rows[0].error_message, 'relation "invalid_table" does not exist');
+                        self.testClient.getResult('drop table test_batch_errors', done);
+                    });
+                });
+            });
+        });
+    });
+
+    it('should use template for job_id onsuccess callback ' +
+        'and keep the original templated query but use the job_id', function (done) {
+        var self = this;
+        var payload = {
+            "query": {
+                "query": [
+                    {
+                        query: "create table batch_jobs (job_id text)"
+                    },
+                    {
+                        "query": "SELECT 1",
+                        "onsuccess": "INSERT INTO batch_jobs values ('<%= job_id %>')"
+                    }
+                ]
+            }
+        };
+        var expectedQuery = {
+            query: [
+                {
+                    query: "create table batch_jobs (job_id text)",
+                    status: 'done'
+                },
+                {
+                    query: "SELECT 1",
+                    onsuccess: "INSERT INTO batch_jobs values ('<%= job_id %>')",
+                    status: 'done',
+                    fallback_status: 'done'
+                }
+            ]
+        };
+
+        self.batchTestClient.createJob(payload, function(err, jobResult) {
+            if (err) {
+                return done(err);
+            }
+            jobResult.getStatus(function (err, job) {
+                if (err) {
+                    return done(err);
+                }
+
+                jobResult.validateExpectedResponse(expectedQuery);
+                self.testClient.getResult('select * from batch_jobs', function(err, rows) {
+                    if (err) {
+                        return done(err);
+                    }
+                    assert.equal(rows[0].job_id, job.job_id);
+
+                    self.testClient.getResult('drop table batch_jobs', done);
+                });
+            });
+        });
+    });
+});