fix: support localhost urls (#802)

nearform · Mar 2, 2023 · a0bc973 · a0bc973
1 parent a68be05
commit a0bc973
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/packages/autocannon-ui-backend/routes/execute.js b/packages/autocannon-ui-backend/routes/execute.js
@@ -4,7 +4,7 @@ import execute from '../services/autocannon.js'
 
 const schema = {
   body: S.object()
-    .prop('url', S.string().format('url').required())
+    .prop('url', S.string().format('uri').required())
     .prop('connections', S.integer().default(10))
     .prop('duration', S.anyOf([S.number(), S.string()]).default(10))
     .prop('pipelining', S.integer().default(1))

diff --git a/packages/autocannon-ui-backend/test/unit/routes/execute.test.js b/packages/autocannon-ui-backend/test/unit/routes/execute.test.js
@@ -4,7 +4,7 @@ import Fastify from 'fastify'
 import index from '../../../index.js'
 
 t.test('execute', async t => {
-  t.test('returns 404 if url is not valid', async t => {
+  t.test('returns 400 if url is not valid', async t => {
     const fastify = Fastify().register(index)
 
     const response = await fastify.inject({
@@ -17,4 +17,18 @@ t.test('execute', async t => {
 
     t.same(response.statusCode, 400)
   })
+
+  t.test('supports localhost urls with port', async t => {
+    const fastify = Fastify().register(index)
+
+    const response = await fastify.inject({
+      url: '/api/execute',
+      method: 'POST',
+      body: {
+        url: 'http://localhost:3000'
+      }
+    })
+
+    t.same(response.statusCode, 200)
+  })
 })