Adding tests for the functionality to adjust partSize dynamically, and fixing that feature.

Author: nathanpeck

README.md

@@ -147,3 +147,9 @@ var UploadStreamObject = new Uploader(
 ```
 npm install s3-upload-stream
 ```
+
+### Running Tests
+
+```
+npm test
+```

lib/s3-upload-stream.js

@@ -43,15 +43,15 @@ module.exports = {
     self.receivedSize = 0;
     self.uploadedSize = 0;
     self.currentPart = Buffer(0);
-    self.maxPartSize = 5242880;
+    self.partSizeThreshold = 5242880;
 
     // Set the maximum amount of data that we will keep in memory before flushing it to S3 as a part
     // of the multipart upload
     self.maxPartSize = function (partSize) {
       if (partSize < 5242880)
         partSize = 5242880;
 
-      self.maxPartSize = partSize;
+      self.partSizeThreshold = partSize;
     };
 
     // Handler to receive data and upload it to S3.
@@ -60,7 +60,7 @@ module.exports = {
 
       // If the current Part buffer is getting too large, or the stream piped in has ended then flush
       // the Part buffer downstream to S3 via the multipart upload API.
-      if (self.currentPart.length > self.maxPartSize)
+      if (self.currentPart.length > self.partSizeThreshold)
         self.flushPart(next);
       else
         next();

tests/test.js

@@ -162,6 +162,41 @@ describe('Creating upload stream', function () {
   });
 });
 
+describe('Stream Methods', function () {
+  var uploadStream, uploadObject;
+
+  before(function (done) {
+    uploadObject = new UploadStream(
+      {
+        s3Client: new AWSstub.S3()
+      },
+      {
+        "Bucket": "test-bucket-name",
+        "Key": "test-file-name"
+      },
+      function (err, data) {
+        expect(err).to.equal(null);
+        uploadStream = data;
+        done();
+      }
+    );
+  });
+
+  describe('Setting max part size to a value greater than 5 MB', function () {
+    it('max part size should be set to that value', function () {
+      uploadObject.maxPartSize(20971520);
+      expect(uploadObject.partSizeThreshold).to.equal(20971520);
+    });
+  });
+
+  describe('Setting max part size to a value less than 5 MB', function () {
+    it('max part size should be set to 5 MB exactly', function () {
+      uploadObject.maxPartSize(4242880);
+      expect(uploadObject.partSizeThreshold).to.equal(5242880);
+    });
+  });
+});
+
 describe('Piping data into the upload stream', function () {
   var uploadStream, uploadObject;