Add tests for submitting measurements

Author: pyropy

lib/submit-measurement.js

@@ -0,0 +1,19 @@
+/**
+ *
+ * @param {boolean} measurement
+ * @param {typeof globalThis.fetch} fetch
+ */
+export const submit = async (measurement, fetch = globalThis.fetch) => {
+  const res = await fetch('https://api.checker.network/arweave/measurement', {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json'
+    },
+    body: JSON.stringify({ retrievalSucceeded: measurement })
+  })
+  if (!res.ok) {
+    throw new Error(`Failed to submit measurement (status=${res.status})`, {
+      cause: new Error(await res.text().catch(() => null))
+    })
+  }
+}

main.js

@@ -1,8 +1,10 @@
 /* global Arweave */
+/* global Zinnia */
 
 import './vendor/arweave.js'
 import pTimeout from './vendor/p-timeout.js'
 import { getNodes } from './lib/nodes.js'
+import { submit } from './lib/submit-measurement.js'
 
 const ONE_MINUTE = 60_000
 const MEASUREMENT_DELAY = ONE_MINUTE
@@ -31,21 +33,6 @@ const measure = async node => {
   return true
 }
 
-const submit = async measurement => {
-  const res = await fetch('https://api.checker.network/arweave/measurement', {
-    method: 'POST',
-    headers: {
-      'Content-Type': 'application/json'
-    },
-    body: JSON.stringify(measurement)
-  })
-  if (!res.ok) {
-    throw new Error(`Failed to submit measurement (status=${res.status})`, {
-      cause: new Error(await res.text().catch(() => null))
-    })
-  }
-}
-
 let nodes = await getNodes()
 
 ;(async () => {
@@ -71,6 +58,7 @@ while (true) {
     console.error('Error submitting measurement')
     console.error(err)
   }
+  Zinnia.jobComplete()
   console.log(`Waiting ${MEASUREMENT_DELAY / 1_000} seconds...`)
   await new Promise(resolve => setTimeout(resolve, MEASUREMENT_DELAY))
 }

tests/submit-measurement.test.js

@@ -0,0 +1,59 @@
+import { test } from 'zinnia:test'
+import { assertEquals, assertRejects } from 'zinnia:assert'
+import { submit } from '../lib/submit-measurement.js'
+
+test('submit measurement succeeds', async () => {
+  const requests = []
+  const fetch = async (url, allOpts) => {
+    const { signal, ...opts } = allOpts
+    requests.push({ url, opts })
+
+    return {
+      status: 200,
+      ok: true
+
+    }
+  }
+
+  const measurement = true
+  await submit(measurement, fetch)
+  assertEquals(requests.length, 1)
+  assertEquals(requests, [
+    {
+      url: 'https://api.checker.network/arweave/measurement',
+      opts: {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({ retrievalSucceeded: measurement })
+      }
+    }
+  ])
+})
+
+test('submit measurements fails', async () => {
+  const requests = []
+  const fetch = async (url, allOpts) => {
+    const { signal, ...opts } = allOpts
+    requests.push({ url, opts })
+
+    return {
+      status: 500,
+      ok: false,
+      text: async () => 'Internal Server Error'
+    }
+  }
+
+  const measurement = true
+  await assertRejects(async () => await submit(measurement, fetch), 'Failed to submit measurement (status=500)')
+  assertEquals(requests.length, 1)
+  assertEquals(requests, [
+    {
+      url: 'https://api.checker.network/arweave/measurement',
+      opts: {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({ retrievalSucceeded: measurement })
+      }
+    }
+  ])
+})