zkldi_Tachi/server/src/utils/db.test.ts

import { DecrementCounterValue, GetNextCounterValue } from "./db";
import db from "external/mongo/db";
import t from "tap";
import ResetDBState from "test-utils/resets";

t.test("#GetNextCounterValue", (t) => {
	t.beforeEach(ResetDBState);

	t.test("Increments on valid counter hit", async (t) => {
		const response = await GetNextCounterValue("real-counter");

		// database starts with this at one
		t.equal(response, 2, "Counter should return the current number stored");

		const dbData = await db.counters.findOne({
			counterName: "real-counter",
		});

		t.equal(dbData!.value, 3, "Counter should increment after being hit");
	});

	t.rejects(
		() => GetNextCounterValue("fake-counter"),
		"Could not find sequence document for fake-counter."
	);

	t.end();
});

t.test("#DecrementCounterValue", (t) => {
	t.beforeEach(ResetDBState);

	t.test("Should decrease a counter.", async (t) => {
		await db.counters.insert({ counterName: "foo", value: 3 });

		const res = await DecrementCounterValue("foo");

		t.equal(res, 2);

		const doc = await db.counters.findOne({ counterName: "foo" });

		t.equal(doc?.value, 2);

		t.end();
	});

	t.rejects(
		() => GetNextCounterValue("fake-counter"),
		"Could not find sequence document for fake-counter."
	);

	t.end();
});