From cdb5157958ebf182e44f1679479e9a3dea4b8e12 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Karol=20Wypch=C5=82o?= <kwypchlo@gmail.com>
Date: Mon, 28 Jun 2021 11:38:22 +0200
Subject: [PATCH] add health check for registry write and read (#906)

---
 packages/health-check/src/checks/critical.js | 27 +++++++++++++++++++-
 1 file changed, 26 insertions(+), 1 deletion(-)

diff --git a/packages/health-check/src/checks/critical.js b/packages/health-check/src/checks/critical.js
index 8fa62fdf..c95c9e8f 100644
--- a/packages/health-check/src/checks/critical.js
+++ b/packages/health-check/src/checks/critical.js
@@ -1,7 +1,8 @@
 const got = require("got");
 const FormData = require("form-data");
+const { isEqual } = require("lodash");
 const { calculateElapsedTime, getResponseContent } = require("../utils");
-const { SkynetClient } = require("skynet-js");
+const { SkynetClient, genKeyPairAndSeed } = require("skynet-js");
 
 const skynetClient = new SkynetClient(process.env.SKYNET_PORTAL_API);
 const exampleSkylink = "AACogzrAimYPG42tDOKhS3lXZD8YvlF8Q8R17afe95iV2Q";
@@ -64,6 +65,29 @@ async function accountWebsiteCheck(done) {
   return done(await genericAccessCheck("account_website", url));
 }
 
+// registryWriteAndReadCheck writes to registry and immediately reads and compares the data
+async function registryWriteAndReadCheck(done) {
+  const time = process.hrtime();
+  const data = { name: "registry_write_and_read", up: false };
+  const { privateKey, publicKey } = genKeyPairAndSeed();
+  const expected = { datakey: "foo-key", data: "foo-data", revision: BigInt(0) };
+
+  try {
+    await skynetClient.registry.setEntry(privateKey, expected);
+    const { entry } = await skynetClient.registry.getEntry(publicKey, expected.datakey);
+
+    if (isEqual(expected, entry)) {
+      data.up = true;
+    } else {
+      data.errors = [{ message: "Data mismatch in registry (read after write)", entry, expected }];
+    }
+  } catch (error) {
+    data.errors = [{ message: error.message }];
+  }
+
+  return done({ ...data, time: calculateElapsedTime(time) });
+}
+
 // directServerApiAccessCheck returns the basic server api check on direct server address
 async function directServerApiAccessCheck(done) {
   if (!process.env.SKYNET_SERVER_API) {
@@ -138,6 +162,7 @@ const checks = [
   downloadCheck,
   skylinkSubdomainCheck,
   handshakeSubdomainCheck,
+  registryWriteAndReadCheck,
   directServerApiAccessCheck,
 ];