jest integration tests configs

Author: fforbeck

jest.integration.config.json

@@ -0,0 +1,19 @@
+{
+    "testMatch": [
+        "**/test/integration/**/*.spec.js"
+    ],
+        "transform": {
+        "^.+\\.js$": "babel-jest"
+    },
+    "transformIgnorePatterns": [
+        "/node_modules/(?!chai|sinon)"
+    ],
+        "moduleFileExtensions": [
+            "js"
+        ],
+            "globals": {
+        "ts-jest": {
+            "useESM": true
+        }
+    }
+}

jest.unit.config.json

@@ -0,0 +1,19 @@
+{
+    "testMatch": [
+        "**/test/unit/**/*.spec.js"
+    ],
+        "transform": {
+        "^.+\\.js$": "babel-jest"
+    },
+    "transformIgnorePatterns": [
+        "/node_modules/(?!chai|sinon)"
+    ],
+        "moduleFileExtensions": [
+            "js"
+        ],
+            "globals": {
+        "ts-jest": {
+            "useESM": true
+        }
+    }
+}

package.json

@@ -20,7 +20,8 @@
     "build:debug": "export $(npm run -s set-version) && esbuild --bundle src/index.js --format=esm --outfile=dist/worker.mjs",
     "build:tsc": "tsc --build",
     "test": "npm run build:debug && node --test --experimental-vm-modules",
-    "test:unit": "export $(npm run -s set-version) && node --experimental-vm-modules ./node_modules/.bin/jest",
+    "test:unit": "export $(npm run -s set-version) && node --experimental-vm-modules ./node_modules/.bin/jest --config jest.unit.config.json",
+    "test:integration": "export $(npm run -s set-version) && node --experimental-vm-modules ./node_modules/.bin/jest --config jest.integration.config.json",
     "lint": "standard"
   },
   "keywords": [
@@ -66,24 +67,5 @@
     "ignore": [
       "*.ts"
     ]
-  },
-  "jest": {
-    "testMatch": [
-      "**/test/middlewares/**/*.spec.js"
-    ],
-    "transform": {
-      "^.+\\.js$": "babel-jest"
-    },
-    "transformIgnorePatterns": [
-      "/node_modules/(?!chai|sinon)"
-    ],
-    "moduleFileExtensions": [
-      "js"
-    ],
-    "globals": {
-      "ts-jest": {
-        "useESM": true
-      }
-    }
   }
 }

test/integration/index.it.spec.js

@@ -0,0 +1,71 @@
+import fetchHandler from '../../src/index.js'
+import { jest } from '@jest/globals'
+import { RATE_LIMIT_EXCEEDED } from '../../src/constants.js'
+import { HttpError } from '@web3-storage/gateway-lib/util'
+
+describe('Integration Test for fetch handler', () => {
+    let env
+    let rateLimiter
+    let ctx
+
+    beforeEach(() => {
+        rateLimiter = {
+            limit: jest.fn()
+        }
+        env = {
+            RATE_LIMITER: rateLimiter,
+            ACCOUNTING_SERVICE_URL: 'http://example.com',
+            AUTH_TOKEN_METADATA: {
+                get: jest.fn(),
+                put: jest.fn()
+            }
+        }
+        ctx = {
+            dataCid: 'test-cid'
+        }
+    })
+
+    afterEach(() => {
+        jest.restoreAllMocks()
+    })
+
+    it('should call the handler if rate limit is not exceeded', async () => {
+        rateLimiter.limit.mockResolvedValue({ success: true })
+
+        const request = new Request('https://example.com', {
+            method: 'GET',
+            headers: {
+                'Authorization': 'test-token'
+            }
+        })
+
+        const response = await fetchHandler.fetch(request, env, ctx)
+
+        expect(rateLimiter.limit).toHaveBeenCalledTimes(1)
+        expect(rateLimiter.limit).toHaveBeenCalledWith({ key: ctx.dataCid.toString() })
+        expect(response).toBeDefined()
+        expect(response.status).toBe(200)
+    })
+
+    it('should throw an error if rate limit is exceeded', async () => {
+        rateLimiter.limit.mockResolvedValue({ success: false })
+
+        const request = new Request('https://example.com', {
+            method: 'GET',
+            headers: {
+                'Authorization': 'test-token'
+            }
+        })
+
+        try {
+            await fetchHandler.fetch(request, env, ctx)
+            throw new Error('Expected error was not thrown')
+        } catch (err) {
+            expect(rateLimiter.limit).toHaveBeenCalledTimes(1)
+            expect(rateLimiter.limit).toHaveBeenCalledWith({ key: ctx.dataCid.toString() })
+            expect(err).toBeInstanceOf(HttpError)
+            expect(err.message).toBe('Too Many Requests')
+        }
+    })
+
+})

test/middlewares/rate-limiter.spec.js renamed to test/unit/middlewares/rate-limiter.spec.js

@@ -1,5 +1,5 @@
 import { jest } from '@jest/globals'
-import { withRateLimits } from '../../src/middleware.js'
+import { withRateLimits } from '../../../src/middleware.js'
 import { HttpError } from '@web3-storage/gateway-lib/util'
 
 describe('withRateLimits', () => {