From 0fb70eec89e0bea4ed08f58bc92ec187f3133616 Mon Sep 17 00:00:00 2001
From: Billie Hilton <587740+billiegoose@users.noreply.github.com>
Date: Fri, 21 Jun 2024 13:09:57 -0400
Subject: [PATCH] add test/manual/measure-reconnect.js (#175)

---
 test/manual/measure-reconnect.js | 53 ++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 test/manual/measure-reconnect.js

diff --git a/test/manual/measure-reconnect.js b/test/manual/measure-reconnect.js
new file mode 100644
index 0000000..15af75a
--- /dev/null
+++ b/test/manual/measure-reconnect.js
@@ -0,0 +1,53 @@
+/**
+ * The goal of this test is to measure how quickly a client reconnects
+ * after manually switching networks / e.g. from wifi to mobile data.
+ *
+ * It requires some extra modules to get the relays:
+ * npm install --no-save hypertrace hypercore-id-encoding @holepunchto/keet-default-config
+ */
+
+function customLogger (data) {
+  console.log(`   ... ${data.id} ${Object.keys(data.caller.props || []).join(',')} ${data.caller.filename}:${data.caller.line}:${data.caller.column}`)
+}
+require('hypertrace').setTraceFunction(customLogger)
+
+const { DEV_BLIND_RELAY_KEYS } = require('@holepunchto/keet-default-config')
+const HypercoreId = require('hypercore-id-encoding')
+const DEV_RELAY_KEYS = DEV_BLIND_RELAY_KEYS.map(HypercoreId.decode)
+const relayThrough = (force) => force ? DEV_RELAY_KEYS : null
+
+const Hyperswarm = require('../..')
+
+const topic = Buffer.alloc(32).fill('measure-reconnect')
+const seed = Buffer.alloc(32).fill('measure-reconnect' + require('os').hostname())
+
+const swarm = new Hyperswarm({ seed, relayThrough })
+
+swarm.dht.on('network-change', () => {
+  console.log('NETWORK CHANGE')
+  console.time('RECONNECTION TIME')
+})
+
+let connected = false
+
+swarm.on('connection', async (conn) => {
+  console.log(conn.rawStream.remoteHost)
+  conn.on('error', console.log.bind(console))
+  conn.on('close', console.log.bind(console))
+  conn.on('data', (data) => console.log(data.toString('utf8')))
+  conn.setKeepAlive(5000)
+  conn.write('hello')
+  if (!connected) {
+    connected = true
+    console.timeEnd('INITIAL CONNECTION TIME')
+    return
+  }
+  console.timeEnd('RECONNECTION TIME')
+})
+
+console.time('INITIAL CONNECTION TIME')
+swarm.join(topic)
+
+// process.on('SIGINT', () => {
+//   swarm.leave(topic).then(() => process.exit())
+// })