improve test cases

1 files changed, 26 insertions, 0 deletions

diff --git a/tests/index.test.ts b/tests/index.test.ts
index 9e1d4ff..896d150 100644
--- a/tests/index.test.ts
+++ b/tests/index.test.ts
@@ -336,6 +336,17 @@ describe('HTTP Client', () => {
         expect(middleware).toHaveBeenCalled()
       })
 
+      it('should be applied to the upload request', () => {
+        const middleware = jest.fn((request, next) => next(request))
+        const factory = new Factory()
+        factory.new().use(middleware).upload({
+          url: 'http://example.com',
+          filePath: 'foo.jpg',
+          name: 'file'
+        })
+        expect(middleware).toHaveBeenCalled()
+      })
+
       it('can modify HTTP request', () => {
         const middleware = (request: any, next: any) =>
           next(request.withHeader('X-Foo', 'bar'))
@@ -348,6 +359,21 @@ describe('HTTP Client', () => {
         )
       })
 
+      it('can modify upload request', () => {
+        const middleware = (request: any, next: any) => next(request.withHeader('X-Foo', 'bar'))
+        const factory = new Factory()
+        factory.new().use(middleware).upload({
+          url: 'http://example.com',
+          filePath: 'foo.jpg',
+          name: 'file'
+        })
+        expect(wx.uploadFile).toHaveBeenCalledWith(
+          expect.objectContaining({
+            header: expect.objectContaining({ 'X-Foo': 'bar' })
+          })
+        )
+      })
+
       it('can retrieve HTTP response', () => {
         ;(wx.request as jest.Mock).mockImplementationOnce(({ success }) => {
           success({