feat: add form data as optional param

PinataCloud · Mar 24, 2022 · 104bb7c · 104bb7c
1 parent 3de2131
commit 104bb7c
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 5 deletions.
diff --git a/src/commands/pinning/pinFileToIPFS.js b/src/commands/pinning/pinFileToIPFS.js
@@ -24,7 +24,7 @@ export default function pinFileToIPFS(pinataApiKey, pinataSecretApiKey, readStre
 
         const endpoint = `${baseUrl}/pinning/pinFileToIPFS`;
 
-        if (!(readStream instanceof stream.Readable)) {
+        if (!(readStream instanceof stream.Readable || readStream instanceof NodeFormData)) {
             reject(new Error('readStream is not a readable stream'));
         }
 
@@ -41,7 +41,7 @@ export default function pinFileToIPFS(pinataApiKey, pinataSecretApiKey, readStre
 
         axios.post(
             endpoint,
-            data,
+            readStream instanceof NodeFormData ? readStream : data,
             {
                 withCredentials: true,
                 maxContentLength: 'Infinity', //this is needed to prevent axios from erroring out with large files

diff --git a/test/commands/pinning/pinFileToIPFS.test.js b/test/commands/pinning/pinFileToIPFS.test.js
@@ -1,20 +1,23 @@
 import axios from 'axios';
 import pinFileToIPFS from'../../../src/commands/pinning/pinFileToIPFS';
 import fs from 'fs';
+import NodeFormData from 'form-data';
 jest.mock('axios');
 
 //common values
 const nonStream = 'test';
 const validStream = fs.createReadStream('./pinata.png');
+const validFormData = new NodeFormData();
+validFormData.append('file', validStream, {filepath: 'test/filepath'});
 
 
-test('non-readableStream is passed in', () => {
-    expect(pinFileToIPFS('test', 'test', nonStream)).rejects.toEqual(Error('readStream is not a readable stream'));
+test('non-readableStream and non-formData is passed in', () => {
+    expect(pinFileToIPFS('test', 'test', nonStream)).rejects.toEqual(Error('readStream is not a readable stream or form data'));
 
     return undefined;
 });
 
-test('200 status is returned', () => {
+test('200 status is returned with valid stream', () => {
     const goodStatus = {
         status: 200,
         data: 'testData'
@@ -24,6 +27,16 @@ test('200 status is returned', () => {
     expect(pinFileToIPFS('test', 'test', validStream)).resolves.toEqual(goodStatus.data);
 });
 
+test('200 status is returned with valid form data', () => {
+    const goodStatus = {
+        status: 200,
+        data: 'testData'
+    };
+    axios.post.mockResolvedValue(goodStatus);
+    expect.assertions(1);
+    expect(pinFileToIPFS('test', 'test', validFormData)).resolves.toEqual(goodStatus.data);
+});
+
 test('Result other than 200 status is returned', () => {
     const badStatus = {
         status: 700