2020/04/03, 02:34, v1.2.0

1 files changed, 202 insertions, 0 deletions

diff --git a/node_modules/needle/test/request_stream_spec.js b/node_modules/needle/test/request_stream_spec.js
new file mode 100644
index 0000000..1f02c8b
--- /dev/null
+++ b/node_modules/needle/test/request_stream_spec.js
@@ -0,0 +1,202 @@
+var fs     = require('fs'),
+    needle = require('..'),
+    stream = require('stream'),
+    http   = require('http'),
+    should = require('should'),
+    sinon  = require('sinon');
+
+var port   = 2233;
+
+describe('request stream length', function() {
+
+  var server, writable;
+
+  function createServer() {
+    return http.createServer(function(req, res) {
+
+      req.on('data', function(chunk) {
+        // console.log(chunk.length);
+      })
+
+      req.on('end', function() {
+        res.writeHeader(200, { 'Content-Type': 'application/json'})
+        res.end(JSON.stringify({ headers: req.headers }))
+      })
+
+    })
+  }
+
+  before(function(done) {
+    server = createServer();
+    server.listen(port, done)
+  })
+
+  beforeEach(function() {
+    writable = new stream.Readable();
+    writable._read = function() {
+      this.push('hello world');
+      this.push(null);
+    }
+  })
+
+  after(function(done) {
+    server.close(done)
+  })
+
+  function send_request(opts, cb) {
+    needle.post('http://localhost:' + port, writable, opts, function(err, resp) {
+      cb(err, resp)
+    })
+  }
+
+  describe('no stream_length set', function() {
+
+    it('doesnt set Content-Length header', function(done) {
+      send_request({}, function(err, resp) {
+        should.not.exist(resp.body.headers['content-length']);
+        done()
+      })
+    })
+
+    it('doesnt work if Transfer-Encoding is set to a blank string', function(done) {
+      send_request({ headers: { 'Transfer-Encoding': '' }}, function(err, resp) {
+        err.code.should.eql('ECONNRESET');
+        done()
+      })
+    })
+
+    it('works if Transfer-Encoding is not set', function(done) {
+      send_request({}, function(err, resp) {
+        should.not.exist(err);
+        resp.statusCode.should.eql(200);
+        done()
+      })
+    })
+
+  })
+
+  describe('stream_length set to invalid value', function() {
+
+    it('sets Content-Length header to that value', function(done) {
+      send_request({ stream_length: 5 }, function(err, resp) {
+        should.exist(err);
+        err.code.should.eql('ECONNRESET');
+        done()
+      })
+    })
+
+    it('doesnt work if Transfer-Encoding is set to a blank string', function(done) {
+      send_request({ stream_length: 5, headers: { 'Transfer-Encoding': '' }}, function(err, resp) {
+        err.code.should.eql('ECONNRESET');
+        done()
+      })
+    })
+
+    it('doesnt work if Transfer-Encoding is not set', function(done) {
+      send_request({ stream_length: 5 }, function(err, resp) {
+        err.code.should.eql('ECONNRESET');
+        done()
+      })
+    })
+
+  })
+
+  describe('stream_length is set to valid value', function() {
+
+    it('sets Content-Length header to that value', function(done) {
+      send_request({ stream_length: 11 }, function(err, resp) {
+        resp.body.headers['content-length'].should.eql('11');
+        done()
+      })
+    })
+
+    it('works if Transfer-Encoding is set to a blank string', function(done) {
+      send_request({ stream_length: 11, headers: { 'Transfer-Encoding': '' }}, function(err, resp) {
+        should.not.exist(err);
+        resp.statusCode.should.eql(200);
+        done()
+      })
+    })
+
+    it('works if Transfer-Encoding is not set', function(done) {
+      send_request({ stream_length: 11 }, function(err, resp) {
+        should.not.exist(err);
+        resp.statusCode.should.eql(200);
+        done()
+      })
+    })
+
+  })
+
+
+  describe('stream_length set to 0', function() {
+
+    describe('stream with path', function() {
+
+      var stub;
+
+      beforeEach(function() {
+        writable.path = '/foo/bar';
+        stub = sinon.stub(fs, 'stat', function(path, cb) {
+          cb(null, { size: 11 })
+        })
+      })
+
+      afterEach(function() {
+        stub.restore();
+      })
+
+      it('sets Content-Length header to streams length', function(done) {
+        send_request({ stream_length: 0 }, function(err, resp) {
+          resp.body.headers['content-length'].should.eql('11');
+          done()
+        })
+      })
+
+      it('works if Transfer-Encoding is set to a blank string', function(done) {
+        send_request({ stream_length: 0, headers: { 'Transfer-Encoding': '' }}, function(err, resp) {
+          should.not.exist(err);
+          resp.statusCode.should.eql(200);
+          done()
+        })
+      })
+
+      it('works if Transfer-Encoding is not set', function(done) {
+        send_request({ stream_length: 0 }, function(err, resp) {
+          should.not.exist(err);
+          resp.statusCode.should.eql(200);
+          done()
+        })
+      })
+
+    })
+
+    describe('stream without path', function() {
+      it('does not set Content-Length header', function(done) {
+        send_request({ stream_length: 0 }, function(err, resp) {
+          should.not.exist(resp.body.headers['content-length']);
+          done()
+        })
+      })
+
+      it('doesnt work if Transfer-Encoding is set to a blank string', function(done) {
+        send_request({ stream_length: 0, headers: { 'Transfer-Encoding': '' }}, function(err, resp) {
+          err.code.should.eql('ECONNRESET');
+          done()
+        })
+      })
+
+      it('works if Transfer-Encoding is not set', function(done) {
+        send_request({ stream_length: 0 }, function(err, resp) {
+          should.not.exist(err);
+          resp.statusCode.should.eql(200);
+          done()
+        })
+      })
+
+    })
+
+
+  })
+
+})