using vercel KV for last stats cache + mocking settimeout in tests

Author: RedHeadphone

package-lock.json

@@ -11,6 +11,7 @@
   },
   "dependencies": {
     "@next/font": "13.1.6",
+    "@vercel/kv": "^2.0.0",
     "antd": "^5.3.1",
     "axios": "^1.0.0",
     "axios-cache-interceptor": "^1.5.3",

package.json

@@ -1,5 +1,6 @@
 import Axios from "axios";
 import { setupCache } from "axios-cache-interceptor";
+import { kv } from "@vercel/kv";
 
 const instance = Axios.create({
   baseURL: "https://codeforces.com/api",
@@ -9,5 +10,20 @@ const instance = Axios.create({
 });
 
 export const api = setupCache(instance);
-export const last_rating_cache = new Map();
-export const last_stats_cache = new Map();
+
+class KVCache {
+  constructor(type) {
+    this.type = type;
+  }
+
+  get(key) {
+    return kv.get(key+"/"+this.type);
+  }
+
+  set(key, value) {
+    kv.set(key+"/"+this.type, value);
+  }
+}
+
+export const last_rating_cache = new KVCache("rating");
+export const last_stats_cache = new KVCache("stats");

src/fetcher-utils.js

@@ -4,13 +4,18 @@ import { api, last_rating_cache, last_stats_cache } from "@/fetcher-utils.js";
 function fetch_error_handler(fetch, username, last_cache) {
   return new Promise((resolve, reject) => {
     const timeoutID = setTimeout(function () {
-      const res = last_cache.get(username);
-      if (res != null) {
-        console.log("Using cached data for", username);
-        resolve(res);
-      }
-      else reject({ status: 500, error: "Codeforces Server Error" });
-    }, 4000);
+      last_cache.get(username).then((res) => {
+        if (res != null) {
+          console.log("Using cached data for", username);
+          resolve(res);
+        }
+        else reject({ status: 500, error: "Codeforces Server Error" });
+      }).catch((error) => {
+        console.error(error);
+        reject({ status: 500, error: "Codeforces Server Error" });
+      });
+      
+    }, 5000);
     fetch()
       .then((result) => {
         clearTimeout(timeoutID);
@@ -51,12 +56,16 @@ export function get_rating(username, cache_seconds) {
           })
           .then((response) => {
             const res = response.data.result[0].rating || 0;
-            last_rating_cache.set(username, res);
+            try { 
+              last_rating_cache.set(username, res);
+            } catch (error) {
+              console.error(error);
+            }
             resolve(res);
           })
           .catch((error) => {
             console.error(error);
-            if (error.response.status === 400)
+            if (error.response && error.response.status === 400)
               reject({ status: 400, error: "Codeforces Handle Not Found" });
             else reject({ status: 500, error: "Codeforces Server Error" });
           });
@@ -118,13 +127,17 @@ export function get_stats(username, cache_seconds) {
               friendOfCount,
               contribution,
             };
-
-            last_stats_cache.set(username, res);
+            
+            try {
+              last_stats_cache.set(username, res);
+            } catch (error) {
+              console.error(error);
+            }
             resolve(res);
           })
           .catch((error) => {
             console.error(error);
-            if (error.response.status === 400)
+            if (error.response && error.response.status === 400)
               reject({ status: 400, error: "Codeforces Handle Not Found" });
             else reject({ status: 500, error: "Codeforces Server Error" });
           });

src/fetcher.js

@@ -30,10 +30,13 @@ describe("badge handler", () => {
       send: jest.fn(),
       status: jest.fn().mockReturnThis(),
     };
+
+    jest.useFakeTimers();
   });
 
   afterEach(() => {
     jest.clearAllMocks();
+    jest.useRealTimers();
   });
 
   it("should return a valid SVG response with proper headers", async () => {
@@ -122,7 +125,17 @@ describe("badge handler", () => {
       },
     });
 
-    await handler(req, res);
+    last_rating_cache.get = () => {
+      return new Promise((resolve) => {
+        resolve(undefined);
+      })
+    };
+
+    const promise = handler(req, res);
+
+    jest.advanceTimersByTime(5000);
+
+    await promise;
 
     expect(res.setHeader).toHaveBeenCalledWith("Content-Type", "text/plain");
     expect(res.setHeader).toHaveBeenCalledWith(

test/badge.test.js

@@ -32,10 +32,13 @@ describe("card handler", () => {
       send: jest.fn(),
       status: jest.fn().mockReturnThis(),
     };
+
+    jest.useFakeTimers();
   });
 
   afterEach(() => {
     jest.clearAllMocks();
+    jest.useRealTimers();
   });
 
   it("should return a valid SVG response with proper headers", async () => {
@@ -95,8 +98,17 @@ describe("card handler", () => {
         status: 403,
       },
     });
+    last_stats_cache.get = () => {
+      return new Promise((resolve) => {
+        resolve(undefined);
+      })
+    };
 
-    await handler(req, res);
+    const promise = handler(req, res);
+
+    jest.advanceTimersByTime(5000);
+
+    await promise;
 
     expect(res.setHeader).toHaveBeenCalledWith("Content-Type", "text/plain");
     expect(res.setHeader).toHaveBeenCalledWith(